twenty_crm/front/src/modules/companies/hooks/useSpreadsheetCompanyImport.ts

import { v4 as uuidv4 } from 'uuid';

import { useSpreadsheetImport } from '@/spreadsheet-import/hooks/useSpreadsheetImport';
import { SpreadsheetOptions } from '@/spreadsheet-import/types';
import { useSnackBar } from '@/ui/snack-bar/hooks/useSnackBar';
import { useUpsertEntityTableItems } from '@/ui/table/hooks/useUpsertEntityTableItems';
import { useUpsertTableRowIds } from '@/ui/table/hooks/useUpsertTableRowIds';
import {
  GetPeopleDocument,
  useInsertManyCompanyMutation,
} from '~/generated/graphql';

import { fieldsForCompany } from '../utils/fieldsForCompany';

export type FieldCompanyMapping = (typeof fieldsForCompany)[number]['key'];

export function useSpreadsheetCompanyImport() {
  const { openSpreadsheetImport } = useSpreadsheetImport<FieldCompanyMapping>();
  const upsertEntityTableItems = useUpsertEntityTableItems();
  const upsertTableRowIds = useUpsertTableRowIds();
  const { enqueueSnackBar } = useSnackBar();

  const [createManyCompany] = useInsertManyCompanyMutation();

  const openCompanySpreadsheetImport = (
    options?: Omit<
      SpreadsheetOptions<FieldCompanyMapping>,
      'fields' | 'isOpen' | 'onClose'
    >,
  ) => {
    openSpreadsheetImport({
      ...options,
      async onSubmit(data) {
        // TODO: Add better type checking in spreadsheet import later
        const createInputs = data.validData.map((company) => ({
          id: uuidv4(),
          name: company.name as string,
          domainName: company.domainName as string,
          address: company.address as string,
          employees: parseInt(company.employees as string, 10),
          linkedinUrl: company.linkedinUrl as string | undefined,
        }));

        try {
          const result = await createManyCompany({
            variables: {
              data: createInputs,
            },
            refetchQueries: [GetPeopleDocument],
          });

          if (result.errors) {
            throw result.errors;
          }

          upsertTableRowIds(createInputs.map((company) => company.id));
          upsertEntityTableItems(createInputs);
        } catch (error: any) {
          enqueueSnackBar(error?.message || 'Something went wrong', {
            variant: 'error',
          });
        }
      },
      fields: fieldsForCompany,
    });
  };

  return { openCompanySpreadsheetImport };
}