Class ModelDelegate

packages/client-generator-js/src/TSClient/Model.ts:458–617

  }
}
export class ModelDelegate implements Generable {
  constructor(
    protected readonly outputType: DMMF.OutputType,
    protected readonly context: GenerateContext,
  ) {}

  /**
   * Returns all available non-aggregate or group actions
   * Includes both dmmf and client-only actions
   *
   * @param availableActions
   * @returns
   */
  private getNonAggregateActions(availableActions: DMMF.ModelAction[]): DMMF.ModelAction[] {
    const actions = availableActions.filter(
      (key) => key !== DMMF.ModelAction.aggregate && key !== DMMF.ModelAction.groupBy && key !== DMMF.ModelAction.count,
    )

    return actions
  }

  public toTS(): string {
    const { name } = this.outputType
    const { dmmf } = this.context

    const mapping = dmmf.mappingsMap[name] ?? { model: name, plural: `${name}s` }
    const modelOrType = dmmf.typeAndModelMap[name]

    const availableActions = getModelActions(dmmf, name)
    const nonAggregateActions = this.getNonAggregateActions(availableActions)
    const groupByArgsName = getGroupByArgsName(name)
    const countArgsName = getModelArgName(name, DMMF.ModelAction.count)

    const genericDelegateParams = [extArgsParam, ts.genericParameter(class="st">'GlobalOmitOptions').default(ts.objectType())]

    const excludedArgsForCount = [class="st">'select', class="st">'include', class="st">'distinct', class="st">'omit']
    if (this.context.isPreviewFeatureOn(class="st">'relationJoins')) {
      excludedArgsForCount.push(class="st">'relationLoadStrategy')
    }

    const excludedArgsForCountType = excludedArgsForCount.map((name) => `class="st">'${name}'`).join(class="st">' | ')

    return `\
${
  availableActions.includes(DMMF.ModelAction.aggregate)
    ? `type ${countArgsName}<ExtArgs extends $Extensions.InternalArgs = $Extensions.DefaultArgs> =
  Omit<${getModelArgName(name, DMMF.ModelAction.findMany)}, ${excludedArgsForCountType}> & {
    select?: ${getCountAggregateInputName(name)} | true
  }
`
    : class="st">''
}
export interface ${name}Delegate<${genericDelegateParams.map((param) => ts.stringify(param)).join(class="st">', ')}> {
${indent(`[K: symbol]: { types: Prisma.TypeMap<ExtArgs>[class="st">'model'][class="st">'${name}'], meta: { name: class="st">'${name}' } }`, TAB_SIZE)}
${nonAggregateActions
  .map((action) => {
    const method = buildModelDelegateMethod(name, action, this.context)
    return ts.stringify(method, { indentLevel: 1, newLine: class="st">'trailing' })