twenty/server/src/workspace/workspace-query-builder/workspace-query-builder.factory.ts

import { Injectable, Logger } from '@nestjs/common';

import { WorkspaceQueryBuilderOptions } from 'src/workspace/workspace-query-builder/interfaces/workspace-query-builder-options.interface';
import {
  Record as IRecord,
  RecordFilter,
  RecordOrderBy,
} from 'src/workspace/workspace-query-builder/interfaces/record.interface';
import {
  FindManyResolverArgs,
  FindOneResolverArgs,
  CreateManyResolverArgs,
  UpdateOneResolverArgs,
  DeleteOneResolverArgs,
  UpdateManyResolverArgs,
  DeleteManyResolverArgs,
} from 'src/workspace/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';

import { FindManyQueryFactory } from './factories/find-many-query.factory';
import { FindOneQueryFactory } from './factories/find-one-query.factory';
import { CreateManyQueryFactory } from './factories/create-many-query.factory';
import { UpdateOneQueryFactory } from './factories/update-one-query.factory';
import { DeleteOneQueryFactory } from './factories/delete-one-query.factory';
import { UpdateManyQueryFactory } from './factories/update-many-query.factory';
import { DeleteManyQueryFactory } from './factories/delete-many-query.factory';

@Injectable()
export class WorkspaceQueryBuilderFactory {
  private readonly logger = new Logger(WorkspaceQueryBuilderFactory.name);

  constructor(
    private readonly findManyQueryFactory: FindManyQueryFactory,
    private readonly findOneQueryFactory: FindOneQueryFactory,
    private readonly createManyQueryFactory: CreateManyQueryFactory,
    private readonly updateOneQueryFactory: UpdateOneQueryFactory,
    private readonly deleteOneQueryFactory: DeleteOneQueryFactory,
    private readonly updateManyQueryFactory: UpdateManyQueryFactory,
    private readonly deleteManyQueryFactory: DeleteManyQueryFactory,
  ) {}

  findMany<
    Filter extends RecordFilter = RecordFilter,
    OrderBy extends RecordOrderBy = RecordOrderBy,
  >(
    args: FindManyResolverArgs<Filter, OrderBy>,
    options: WorkspaceQueryBuilderOptions,
  ): string {
    return this.findManyQueryFactory.create<Filter, OrderBy>(args, options);
  }

  findOne<Filter extends RecordFilter = RecordFilter>(
    args: FindOneResolverArgs<Filter>,
    options: WorkspaceQueryBuilderOptions,
  ): string {
    return this.findOneQueryFactory.create<Filter>(args, options);
  }

  createMany<Record extends IRecord = IRecord>(
    args: CreateManyResolverArgs<Record>,
    options: WorkspaceQueryBuilderOptions,
  ): string {
    return this.createManyQueryFactory.create<Record>(args, options);
  }

  updateOne<Record extends IRecord = IRecord>(
    initialArgs: UpdateOneResolverArgs<Record>,
    options: WorkspaceQueryBuilderOptions,
  ): string {
    return this.updateOneQueryFactory.create<Record>(initialArgs, options);
  }

  deleteOne(
    args: DeleteOneResolverArgs,
    options: WorkspaceQueryBuilderOptions,
  ): string {
    return this.deleteOneQueryFactory.create(args, options);
  }

  updateMany<
    Record extends IRecord = IRecord,
    Filter extends RecordFilter = RecordFilter,
  >(
    args: UpdateManyResolverArgs<Record, Filter>,
    options: WorkspaceQueryBuilderOptions,
  ): string {
    return this.updateManyQueryFactory.create(args, options);
  }

  deleteMany<Filter extends RecordFilter = RecordFilter>(
    args: DeleteManyResolverArgs<Filter>,
    options: WorkspaceQueryBuilderOptions,
  ): string {
    return this.deleteManyQueryFactory.create(args, options);
  }
}