twenty/front/src/modules/ui/editable-field/hooks/useUpdateGenericEntityField.ts

import { useContext } from 'react';

import { EditableFieldMutationContext } from '../contexts/EditableFieldMutationContext';
import { FieldDefinition } from '../types/FieldDefinition';
import {
  FieldBooleanMetadata,
  FieldBooleanValue,
  FieldChipMetadata,
  FieldChipValue,
  FieldDateMetadata,
  FieldDateValue,
  FieldDoubleTextChipMetadata,
  FieldDoubleTextChipValue,
  FieldDoubleTextMetadata,
  FieldDoubleTextValue,
  FieldMetadata,
  FieldNumberMetadata,
  FieldNumberValue,
  FieldPhoneMetadata,
  FieldPhoneValue,
  FieldProbabilityMetadata,
  FieldProbabilityValue,
  FieldRelationMetadata,
  FieldRelationValue,
  FieldTextMetadata,
  FieldTextValue,
  FieldURLMetadata,
  FieldURLValue,
} from '../types/FieldMetadata';
import { isFieldBoolean } from '../types/guards/isFieldBoolean';
import { isFieldBooleanValue } from '../types/guards/isFieldBooleanValue';
import { isFieldChip } from '../types/guards/isFieldChip';
import { isFieldChipValue } from '../types/guards/isFieldChipValue';
import { isFieldDate } from '../types/guards/isFieldDate';
import { isFieldDateValue } from '../types/guards/isFieldDateValue';
import { isFieldDoubleText } from '../types/guards/isFieldDoubleText';
import { isFieldDoubleTextChip } from '../types/guards/isFieldDoubleTextChip';
import { isFieldDoubleTextChipValue } from '../types/guards/isFieldDoubleTextChipValue';
import { isFieldDoubleTextValue } from '../types/guards/isFieldDoubleTextValue';
import { isFieldNumber } from '../types/guards/isFieldNumber';
import { isFieldNumberValue } from '../types/guards/isFieldNumberValue';
import { isFieldPhone } from '../types/guards/isFieldPhone';
import { isFieldPhoneValue } from '../types/guards/isFieldPhoneValue';
import { isFieldProbability } from '../types/guards/isFieldProbability';
import { isFieldProbabilityValue } from '../types/guards/isFieldProbabilityValue';
import { isFieldRelation } from '../types/guards/isFieldRelation';
import { isFieldRelationValue } from '../types/guards/isFieldRelationValue';
import { isFieldText } from '../types/guards/isFieldText';
import { isFieldTextValue } from '../types/guards/isFieldTextValue';
import { isFieldURL } from '../types/guards/isFieldURL';
import { isFieldURLValue } from '../types/guards/isFieldURLValue';

export function useUpdateGenericEntityField() {
  const useUpdateEntityMutation = useContext(EditableFieldMutationContext);

  const [updateEntity] = useUpdateEntityMutation();

  return function updateEntityField<
    ValueType extends FieldMetadata extends FieldDoubleTextMetadata
      ? FieldDoubleTextValue
      : FieldMetadata extends FieldTextMetadata
      ? FieldTextValue
      : FieldMetadata extends FieldPhoneMetadata
      ? FieldPhoneValue
      : FieldMetadata extends FieldURLMetadata
      ? FieldURLValue
      : FieldMetadata extends FieldNumberMetadata
      ? FieldNumberValue
      : FieldMetadata extends FieldDateMetadata
      ? FieldDateValue
      : FieldMetadata extends FieldChipMetadata
      ? FieldChipValue
      : FieldMetadata extends FieldDoubleTextChipMetadata
      ? FieldDoubleTextChipValue
      : FieldMetadata extends FieldRelationMetadata
      ? FieldRelationValue
      : FieldMetadata extends FieldProbabilityMetadata
      ? FieldProbabilityValue
      : FieldMetadata extends FieldBooleanMetadata
      ? FieldBooleanValue
      : unknown,
  >(
    currentEntityId: string,
    field: FieldDefinition<FieldMetadata>,
    newFieldValue: ValueType,
  ) {
    const newFieldValueUnknown = newFieldValue as unknown;
    // TODO: improve type guards organization, maybe with a common typeguard for all fields
    // taking an object of options as parameter ?
    //
    // The goal would be to check that the field value not only is valid,
    // but also that it is validated against the corresponding field type

    // Relation
    if (isFieldRelation(field) && isFieldRelationValue(newFieldValueUnknown)) {
      const newSelectedEntity = newFieldValueUnknown;

      const fieldName = field.metadata.fieldName;

      if (!newSelectedEntity) {
        updateEntity({
          variables: {
            where: { id: currentEntityId },
            data: {
              [fieldName]: {
                disconnect: true,
              },
            },
          },
        });
      } else {
        updateEntity({
          variables: {
            where: { id: currentEntityId },
            data: {
              [fieldName]: {
                connect: { id: newSelectedEntity.id },
              },
            },
          },
        });
      }
      // Chip
    } else if (isFieldChip(field) && isFieldChipValue(newFieldValueUnknown)) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: { [field.metadata.contentFieldName]: newContent },
        },
      });
      // Text
    } else if (isFieldText(field) && isFieldTextValue(newFieldValueUnknown)) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: { [field.metadata.fieldName]: newContent },
        },
      });
      // Double text
    } else if (
      isFieldDoubleText(field) &&
      isFieldDoubleTextValue(newFieldValueUnknown)
    ) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: {
            [field.metadata.firstValueFieldName]: newContent.firstValue,
            [field.metadata.secondValueFieldName]: newContent.secondValue,
          },
        },
      });
      //  Double Text Chip
    } else if (
      isFieldDoubleTextChip(field) &&
      isFieldDoubleTextChipValue(newFieldValueUnknown)
    ) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: {
            [field.metadata.firstValueFieldName]: newContent.firstValue,
            [field.metadata.secondValueFieldName]: newContent.secondValue,
          },
        },
      });
      // Phone
    } else if (isFieldPhone(field) && isFieldPhoneValue(newFieldValueUnknown)) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: { [field.metadata.fieldName]: newContent },
        },
      });
      // URL
    } else if (isFieldURL(field) && isFieldURLValue(newFieldValueUnknown)) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: { [field.metadata.fieldName]: newContent },
        },
      });
      // Number
    } else if (
      isFieldNumber(field) &&
      isFieldNumberValue(newFieldValueUnknown)
    ) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: { [field.metadata.fieldName]: newContent },
        },
      });
      // Date
    } else if (isFieldDate(field) && isFieldDateValue(newFieldValueUnknown)) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: { [field.metadata.fieldName]: newContent },
        },
      });
    } else if (
      isFieldProbability(field) &&
      isFieldProbabilityValue(newFieldValueUnknown)
    ) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: { [field.metadata.fieldName]: newContent },
        },
      });
    }
    // Boolean
    else if (
      isFieldBoolean(field) &&
      isFieldBooleanValue(newFieldValueUnknown)
    ) {
      const newContent = newFieldValueUnknown;

      updateEntity({
        variables: {
          where: { id: currentEntityId },
          data: { [field.metadata.fieldName]: newContent },
        },
      });
    }
  };
}