Class Model

packages/client-generator-js/src/TSClient/Model.ts:49–457

import * as tsx from class="st">'./utils/type-builders'

export class Model implements Generable {
  protected type: DMMF.OutputType
  protected createManyAndReturnType: undefined | DMMF.OutputType
  protected updateManyAndReturnType: undefined | DMMF.OutputType
  protected mapping?: DMMF.ModelMapping
  private dmmf: DMMFHelper
  constructor(
    protected readonly model: DMMF.Model,
    protected readonly context: GenerateContext,
  ) {
    this.dmmf = context.dmmf
    this.type = this.context.dmmf.outputTypeMap.model[model.name]

    this.createManyAndReturnType = this.context.dmmf.outputTypeMap.model[getCreateManyAndReturnOutputType(model.name)]
    this.updateManyAndReturnType = this.context.dmmf.outputTypeMap.model[getUpdateManyAndReturnOutputType(model.name)]
    this.mapping = this.context.dmmf.mappings.modelOperations.find((m) => m.model === model.name)!
  }

  protected get argsTypes(): ts.Export<ts.TypeDeclaration>[] {
    const argsTypes: ts.Export<ts.TypeDeclaration>[] = []
    for (const action of Object.keys(DMMF.ModelAction)) {
      const fieldName = this.rootFieldNameForAction(action as DMMF.ModelAction)
      if (!fieldName) {
        continue
      }
      const field = this.dmmf.rootFieldMap[fieldName]
      if (!field) {
        throw new Error(`Oops this must not happen. Could not find field ${fieldName} on either Query or Mutation`)
      }

      if (
        action === class="st">'updateMany' ||
        action === class="st">'deleteMany' ||
        action === class="st">'createMany' ||
        action === class="st">'findRaw' ||
        action === class="st">'aggregateRaw'
      ) {
        argsTypes.push(
          new ArgsTypeBuilder(this.type, this.context, action as DMMF.ModelAction)
            .addSchemaArgs(field.args)
            .createExport(),
        )
      } else if (action === class="st">'createManyAndReturn') {
        const args = new ArgsTypeBuilder(this.type, this.context, action as DMMF.ModelAction)
          .addSelectArg(getSelectCreateManyAndReturnName(this.type.name))
          .addOmitArg()
          .addSchemaArgs(field.args)

        if (this.createManyAndReturnType) {
          args.addIncludeArgIfHasRelations(
            getIncludeCreateManyAndReturnName(this.model.name),
            this.createManyAndReturnType,
          )
        }
        argsTypes.push(args.createExport())
      } else if (action === class="st">'updateManyAndReturn') {
        const args = new ArgsTypeBuilder(this.type, this.context, action as DMMF.ModelAction)
          .addSelectArg(getSelectUpdateManyAndReturnName(this.type.name))