packages/twenty-server/src/engine/api/graphql/direct-execution/utils/assert-merge-many-args.util.ts

import { isBoolean, isNumber, isObject } from 'class-validator';

import { isDefined } from 'twenty-shared/utils';

import {
  GraphqlDirectExecutionException,
  GraphqlDirectExecutionExceptionCode,
} from 'src/engine/api/graphql/direct-execution/errors/graphql-direct-execution.exception';
import { STANDARD_ERROR_MESSAGE } from 'src/engine/api/common/common-query-runners/errors/standard-error-message.constant';
import { type MergeManyResolverArgs } from 'src/engine/api/graphql/workspace-resolver-builder/interfaces/workspace-resolvers-builder.interface';

export function assertMergeManyArgs(
  args: unknown,
): asserts args is MergeManyResolverArgs {
  if (!isObject(args)) {
    throw new GraphqlDirectExecutionException(
      'Invalid argument: it must be an object',
      GraphqlDirectExecutionExceptionCode.INVALID_QUERY_INPUT,
      { userFriendlyMessage: STANDARD_ERROR_MESSAGE },
    );
  }

  const argKeys = Object.keys(args);

  const allowedKeys = new Set(['ids', 'conflictPriorityIndex', 'dryRun']);

  for (const key of argKeys) {
    if (!allowedKeys.has(key)) {
      throw new GraphqlDirectExecutionException(
        `Argument not allowed: ${key}`,
        GraphqlDirectExecutionExceptionCode.INVALID_QUERY_INPUT,
        { userFriendlyMessage: STANDARD_ERROR_MESSAGE },
      );
    }
  }

  if (!('ids' in args) || !Array.isArray(args.ids)) {
    throw new GraphqlDirectExecutionException(
      'Missing required argument: "ids" (array)',
      GraphqlDirectExecutionExceptionCode.INVALID_QUERY_INPUT,
      { userFriendlyMessage: STANDARD_ERROR_MESSAGE },
    );
  }

  if (
    !('conflictPriorityIndex' in args) ||
    !isNumber(args.conflictPriorityIndex)
  ) {
    throw new GraphqlDirectExecutionException(
      'Missing required argument: "conflictPriorityIndex" (number)',
      GraphqlDirectExecutionExceptionCode.INVALID_QUERY_INPUT,
      { userFriendlyMessage: STANDARD_ERROR_MESSAGE },
    );
  }

  if ('dryRun' in args && isDefined(args.dryRun) && !isBoolean(args.dryRun)) {
    throw new GraphqlDirectExecutionException(
      'Invalid argument: "dryRun" must be a boolean',
      GraphqlDirectExecutionExceptionCode.INVALID_QUERY_INPUT,
      { userFriendlyMessage: STANDARD_ERROR_MESSAGE },
    );
  }
}