mirror of
https://github.com/go-vikunja/vikunja.git
synced 2026-05-05 19:36:46 +00:00
72 lines
1.6 KiB
TypeScript
72 lines
1.6 KiB
TypeScript
import {seed} from './seed'
|
|
|
|
/**
|
|
* A factory makes it easy to seed the database with data.
|
|
*/
|
|
export class Factory {
|
|
static table: string | null = null
|
|
|
|
static factory() {
|
|
return {}
|
|
}
|
|
|
|
/**
|
|
* Seeds a bunch of fake data into the database.
|
|
*
|
|
* Takes an override object as its single argument which will override the data from the factory.
|
|
* If the value of one of the override fields is `{increment}` that value will be replaced with an incrementing
|
|
* number through all created entities.
|
|
*
|
|
* @param override
|
|
* @returns {[]}
|
|
*/
|
|
static create(count = 1, override = {}, truncate = true) {
|
|
const data = []
|
|
|
|
for (let i = 1; i <= count; i++) {
|
|
const entry = {
|
|
...this.factory(),
|
|
...override,
|
|
}
|
|
for (const e in entry) {
|
|
if(typeof entry[e] === 'function') {
|
|
entry[e] = entry[e](i)
|
|
continue
|
|
}
|
|
if (entry[e] === '{increment}') {
|
|
entry[e] = i
|
|
}
|
|
}
|
|
data.push(entry)
|
|
}
|
|
|
|
// Create a flattened copy of the data for seeding
|
|
// This removes nested objects/arrays that the backend can't handle
|
|
const flatData = data.map(item => {
|
|
const flatItem = {}
|
|
for (const key in item) {
|
|
const value = item[key]
|
|
// Only include primitive values (string, number, boolean, null, Date)
|
|
if (value === null || value === undefined ||
|
|
typeof value === 'string' ||
|
|
typeof value === 'number' ||
|
|
typeof value === 'boolean' ||
|
|
value instanceof Date) {
|
|
flatItem[key] = value
|
|
}
|
|
// Skip arrays, objects, and other complex types
|
|
}
|
|
return flatItem
|
|
})
|
|
|
|
seed(this.table, flatData, truncate)
|
|
|
|
return data
|
|
}
|
|
|
|
static truncate() {
|
|
seed(this.table, null)
|
|
}
|
|
}
|
|
|