mirror of https://github.com/Chocobozzz/PeerTube
119 lines
3.9 KiB
TypeScript
119 lines
3.9 KiB
TypeScript
import isPlainObject from 'lodash-es/isPlainObject.js'
|
|
import { Model as SequelizeModel, ModelStatic, Sequelize } from 'sequelize'
|
|
import { logger } from '@server/helpers/logger.js'
|
|
|
|
/**
|
|
*
|
|
* Build Sequelize models from sequelize raw query (that must use { nest: true } options)
|
|
*
|
|
* In order to sequelize to correctly build the JSON this class will ingest,
|
|
* the columns selected in the raw query should be in the following form:
|
|
* * All tables must be Pascal Cased (for example "VideoChannel")
|
|
* * Root table must end with `Model` (for example "VideoCommentModel")
|
|
* * Joined tables must contain the origin table name + '->JoinedTable'. For example:
|
|
* * "Actor" is joined to "Account": "Actor" table must be renamed "Account->Actor"
|
|
* * "Account->Actor" is joined to "Server": "Server" table must be renamed to "Account->Actor->Server"
|
|
* * Selected columns must be renamed to contain the JSON path:
|
|
* * "videoComment"."id": "VideoCommentModel"."id"
|
|
* * "Account"."Actor"."Server"."id": "Account.Actor.Server.id"
|
|
* * All tables must contain the row id
|
|
*/
|
|
|
|
export class ModelBuilder <T extends SequelizeModel> {
|
|
private readonly modelRegistry = new Map<string, T>()
|
|
|
|
constructor (private readonly sequelize: Sequelize) {
|
|
|
|
}
|
|
|
|
createModels (jsonArray: any[], baseModelName: string): T[] {
|
|
const result: T[] = []
|
|
|
|
for (const json of jsonArray) {
|
|
const { created, model } = this.createModel(json, baseModelName, json.id + '.' + baseModelName)
|
|
|
|
if (created) result.push(model)
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
private createModel (json: any, modelName: string, keyPath: string) {
|
|
if (!json.id) return { created: false, model: null }
|
|
|
|
const { created, model } = this.createOrFindModel(json, modelName, keyPath)
|
|
|
|
for (const key of Object.keys(json)) {
|
|
const value = json[key]
|
|
if (!value) continue
|
|
|
|
// Child model
|
|
if (isPlainObject(value)) {
|
|
const { created, model: subModel } = this.createModel(value, key, keyPath + '.' + json.id + '.' + key)
|
|
if (!created || !subModel) continue
|
|
|
|
const Model = this.findModelBuilder(modelName)
|
|
const association = Model.associations[key]
|
|
|
|
if (!association) {
|
|
logger.error('Cannot find association %s of model %s', key, modelName, { associations: Object.keys(Model.associations) })
|
|
continue
|
|
}
|
|
|
|
if (association.isMultiAssociation) {
|
|
if (!Array.isArray(model[key])) model[key] = []
|
|
|
|
model[key].push(subModel)
|
|
} else {
|
|
model[key] = subModel
|
|
}
|
|
}
|
|
}
|
|
|
|
return { created, model }
|
|
}
|
|
|
|
private createOrFindModel (json: any, modelName: string, keyPath: string) {
|
|
const registryKey = this.getModelRegistryKey(json, keyPath)
|
|
if (this.modelRegistry.has(registryKey)) {
|
|
return {
|
|
created: false,
|
|
model: this.modelRegistry.get(registryKey)
|
|
}
|
|
}
|
|
|
|
const Model = this.findModelBuilder(modelName)
|
|
|
|
if (!Model) {
|
|
logger.error(
|
|
'Cannot build model %s that does not exist', this.buildSequelizeModelName(modelName),
|
|
{ existing: this.sequelize.modelManager.all.map(m => m.name) }
|
|
)
|
|
return { created: false, model: null }
|
|
}
|
|
|
|
const model = Model.build(json, { raw: true, isNewRecord: false })
|
|
|
|
this.modelRegistry.set(registryKey, model)
|
|
|
|
return { created: true, model }
|
|
}
|
|
|
|
private findModelBuilder (modelName: string) {
|
|
return this.sequelize.modelManager.getModel(this.buildSequelizeModelName(modelName)) as ModelStatic<T>
|
|
}
|
|
|
|
private buildSequelizeModelName (modelName: string) {
|
|
if (modelName === 'Avatars') return 'ActorImageModel'
|
|
if (modelName === 'ActorFollowing') return 'ActorModel'
|
|
if (modelName === 'ActorFollower') return 'ActorModel'
|
|
if (modelName === 'FlaggedAccount') return 'AccountModel'
|
|
|
|
return modelName + 'Model'
|
|
}
|
|
|
|
private getModelRegistryKey (json: any, keyPath: string) {
|
|
return keyPath + json.id
|
|
}
|
|
}
|