twenty/packages/twenty-front/src/modules/command-menu-item/engine-command/hooks/useMountEngineCommand.ts

import { useCallback } from 'react';

import { currentWorkspaceMemberState } from '@/auth/states/currentWorkspaceMemberState';
import { mountedEngineCommandsState } from '@/command-menu-item/engine-command/states/mountedEngineCommandsState';
import { contextStoreAnyFieldFilterValueComponentState } from '@/context-store/states/contextStoreAnyFieldFilterValueComponentState';
import { contextStoreCurrentObjectMetadataItemIdComponentState } from '@/context-store/states/contextStoreCurrentObjectMetadataItemIdComponentState';
import { contextStoreCurrentViewIdComponentState } from '@/context-store/states/contextStoreCurrentViewIdComponentState';
import { contextStoreFilterGroupsComponentState } from '@/context-store/states/contextStoreFilterGroupsComponentState';
import { contextStoreFiltersComponentState } from '@/context-store/states/contextStoreFiltersComponentState';
import { contextStoreTargetedRecordsRuleComponentState } from '@/context-store/states/contextStoreTargetedRecordsRuleComponentState';
import { computeContextStoreFilters } from '@/context-store/utils/computeContextStoreFilters';
import { objectMetadataItemsSelector } from '@/object-metadata/states/objectMetadataItemsSelector';
import { recordStoreFamilyState } from '@/object-record/record-store/states/recordStoreFamilyState';
import { getRecordIndexIdFromObjectNamePluralAndViewId } from '@/object-record/utils/getRecordIndexIdFromObjectNamePluralAndViewId';
import { useStore } from 'jotai';
import { isDefined } from 'twenty-shared/utils';
import { type EngineComponentKey } from '~/generated-metadata/graphql';

export const useMountEngineCommand = () => {
  const store = useStore();

  const mountEngineCommand = useCallback(
    (
      engineCommandId: string,
      contextStoreInstanceId: string,
      engineComponentKey: EngineComponentKey,
    ) => {
      const objectMetadataItemId = store.get(
        contextStoreCurrentObjectMetadataItemIdComponentState.atomFamily({
          instanceId: contextStoreInstanceId,
        }),
      );

      const objectMetadataItems = store.get(objectMetadataItemsSelector.atom);
      const objectMetadataItem = objectMetadataItems.find(
        (item) => item.id === objectMetadataItemId,
      );

      const currentViewId = store.get(
        contextStoreCurrentViewIdComponentState.atomFamily({
          instanceId: contextStoreInstanceId,
        }),
      );

      const targetedRecordsRule = store.get(
        contextStoreTargetedRecordsRuleComponentState.atomFamily({
          instanceId: contextStoreInstanceId,
        }),
      );

      const selectedRecords = (
        targetedRecordsRule.mode === 'selection'
          ? targetedRecordsRule.selectedRecordIds
          : []
      )
        .map((id) => store.get(recordStoreFamilyState.atomFamily(id)))
        .filter(isDefined);

      const filters = store.get(
        contextStoreFiltersComponentState.atomFamily({
          instanceId: contextStoreInstanceId,
        }),
      );

      const filterGroups = store.get(
        contextStoreFilterGroupsComponentState.atomFamily({
          instanceId: contextStoreInstanceId,
        }),
      );

      const anyFieldFilterValue = store.get(
        contextStoreAnyFieldFilterValueComponentState.atomFamily({
          instanceId: contextStoreInstanceId,
        }),
      );

      const currentWorkspaceMember = store.get(
        currentWorkspaceMemberState.atom,
      );
      const systemTimeZone = Intl.DateTimeFormat().resolvedOptions().timeZone;
      const userTimezone =
        currentWorkspaceMember?.timeZone !== 'system'
          ? (currentWorkspaceMember?.timeZone ?? systemTimeZone)
          : systemTimeZone;

      const graphqlFilter = isDefined(objectMetadataItem)
        ? computeContextStoreFilters({
            contextStoreTargetedRecordsRule: targetedRecordsRule,
            contextStoreFilters: filters,
            contextStoreFilterGroups: filterGroups,
            objectMetadataItem,
            filterValueDependencies: {
              currentWorkspaceMemberId: currentWorkspaceMember?.id,
              timeZone: userTimezone,
            },
            contextStoreAnyFieldFilterValue: anyFieldFilterValue,
          })
        : null;

      const recordIndexId =
        objectMetadataItem && currentViewId
          ? getRecordIndexIdFromObjectNamePluralAndViewId(
              objectMetadataItem.namePlural,
              currentViewId,
            )
          : null;

      store.set(mountedEngineCommandsState.atom, (previousMap) => {
        const newMap = new Map(previousMap);

        newMap.set(engineCommandId, {
          engineComponentKey,
          contextStoreInstanceId,
          objectMetadataItem: objectMetadataItem ?? null,
          currentViewId,
          recordIndexId,
          targetedRecordsRule,
          selectedRecords,
          graphqlFilter,
        });

        return newMap;
      });
    },
    [store],
  );

  return mountEngineCommand;
};