twenty/packages/twenty-front/src/modules/navigation-menu-item/hooks/useCreateNavigationMenuItem.ts

import { isDefined } from 'twenty-shared/utils';
import { useMutation } from '@apollo/client/react';
import { CreateNavigationMenuItemDocument } from '~/generated-metadata/graphql';

import { usePrefetchedNavigationMenuItemsData } from '@/navigation-menu-item/hooks/usePrefetchedNavigationMenuItemsData';
import { objectMetadataItemsState } from '@/object-metadata/states/objectMetadataItemsState';
import { useAtomStateValue } from '@/ui/utilities/state/jotai/hooks/useAtomStateValue';
import { type ObjectRecord } from '@/object-record/types/ObjectRecord';

export const useCreateNavigationMenuItem = () => {
  const { navigationMenuItems, currentWorkspaceMemberId } =
    usePrefetchedNavigationMenuItemsData();
  const objectMetadataItems = useAtomStateValue(objectMetadataItemsState);

  const [createNavigationMenuItemMutation] = useMutation(
    CreateNavigationMenuItemDocument,
    {
      refetchQueries: ['FindManyNavigationMenuItems'],
    },
  );

  const createNavigationMenuItem = async (
    targetRecord: ObjectRecord,
    targetObjectNameSingular: string,
    folderId?: string,
  ) => {
    const isView = targetObjectNameSingular === 'view';

    if (isView) {
      const relevantItems = folderId
        ? navigationMenuItems.filter((item) => item.folderId === folderId)
        : navigationMenuItems.filter(
            (item) =>
              !isDefined(item.folderId) && isDefined(item.userWorkspaceId),
          );

      const maxPosition = Math.max(
        ...relevantItems.map((item) => item.position),
        0,
      );

      await createNavigationMenuItemMutation({
        variables: {
          input: {
            viewId: targetRecord.id,
            userWorkspaceId: currentWorkspaceMemberId,
            folderId,
            position: maxPosition + 1,
          },
        },
      });
    } else {
      const objectMetadataItem = objectMetadataItems.find(
        (item) => item.nameSingular === targetObjectNameSingular,
      );

      if (!isDefined(objectMetadataItem)) {
        throw new Error(
          `Object metadata item not found for nameSingular: ${targetObjectNameSingular}`,
        );
      }

      const relevantItems = folderId
        ? navigationMenuItems.filter((item) => item.folderId === folderId)
        : navigationMenuItems.filter(
            (item) =>
              !isDefined(item.folderId) && isDefined(item.userWorkspaceId),
          );

      const maxPosition = Math.max(
        ...relevantItems.map((item) => item.position),
        0,
      );

      await createNavigationMenuItemMutation({
        variables: {
          input: {
            targetRecordId: targetRecord.id,
            targetObjectMetadataId: objectMetadataItem.id,
            userWorkspaceId: currentWorkspaceMemberId,
            folderId,
            position: maxPosition + 1,
          },
        },
      });
    }
  };

  return { createNavigationMenuItem };
};