twenty/front/src/modules/metadata/hooks/useUpdateOneObject.ts

import { gql, useMutation } from '@apollo/client';
import { getOperationName } from '@apollo/client/utilities';

import { MetadataObjectIdentifier } from '../types/MetadataObjectIdentifier';
import { generateUpdateOneObjectMutation } from '../utils/generateUpdateOneObjectMutation';

import { useFindOneMetadataObject } from './useFindOneMetadataObject';

export const useUpdateOneObject = ({
  objectNamePlural,
  objectNameSingular,
}: MetadataObjectIdentifier) => {
  const { foundMetadataObject, objectNotFoundInMetadata, findManyQuery } =
    useFindOneMetadataObject({
      objectNamePlural,
      objectNameSingular,
    });

  const generatedMutation = foundMetadataObject
    ? generateUpdateOneObjectMutation({
        metadataObject: foundMetadataObject,
      })
    : gql`
        mutation EmptyMutation {
          empty
        }
      `;

  // TODO: type this with a minimal type at least with Record<string, any>
  const [mutate] = useMutation(generatedMutation, {
    refetchQueries: [getOperationName(findManyQuery) ?? ''],
  });

  const updateOneObject = foundMetadataObject
    ? ({
        idToUpdate,
        input,
      }: {
        idToUpdate: string;
        input: Record<string, any>;
      }) => {
        return mutate({
          variables: {
            idToUpdate: idToUpdate,
            input: {
              ...input,
            },
          },
        });
      }
    : undefined;

  return {
    updateOneObject,
    objectNotFoundInMetadata,
  };
};