twenty/front/src/modules/object-metadata/hooks/useMapToObjectRecordIdentifier.ts

import { CoreObjectNameSingular } from '@/object-metadata/types/CoreObjectNameSingular';
import { ObjectMetadataItem } from '@/object-metadata/types/ObjectMetadataItem';
import { ObjectRecordIdentifier } from '@/object-record/types/ObjectRecordIdentifier';
import { getLogoUrlFromDomainName } from '~/utils';

export const useMapToObjectRecordIdentifier = ({
  objectMetadataItem,
}: {
  objectMetadataItem: ObjectMetadataItem;
}) => {
  return (record: any): ObjectRecordIdentifier => {
    switch (objectMetadataItem.nameSingular) {
      case CoreObjectNameSingular.WorkspaceMember:
      case CoreObjectNameSingular.Person:
        return {
          id: record.id,
          name:
            (record.name?.firstName ?? '') +
            ' ' +
            (record.name?.lastName ?? ''),
          avatarUrl: record.avatarUrl,
          avatarType: 'rounded',
        };
      case CoreObjectNameSingular.Opportunity:
        return {
          id: record.id,
          name: record?.company?.name,
          avatarUrl: record.avatarUrl,
          avatarType: 'rounded',
        };
    }

    const labelIdentifierFieldMetadata = objectMetadataItem.fields.find(
      (field) =>
        field.id === objectMetadataItem.labelIdentifierFieldMetadataId ||
        field.name === 'name',
    );

    const labelIdentifierFieldValue = labelIdentifierFieldMetadata
      ? record[labelIdentifierFieldMetadata.name]
      : null;

    const imageIdentifierFieldMetadata = objectMetadataItem.fields.find(
      (field) => field.id === objectMetadataItem.imageIdentifierFieldMetadataId,
    );

    const imageIdentifierFieldValue = imageIdentifierFieldMetadata
      ? (record[imageIdentifierFieldMetadata.name] as string)
      : null;

    const avatarType =
      objectMetadataItem.nameSingular === CoreObjectNameSingular.Company
        ? 'squared'
        : 'rounded';

    const avatarUrl =
      objectMetadataItem.nameSingular === CoreObjectNameSingular.Company
        ? getLogoUrlFromDomainName(imageIdentifierFieldValue ?? '')
        : imageIdentifierFieldValue ?? null;

    return {
      id: record.id,
      name: labelIdentifierFieldValue,
      avatarUrl,
      avatarType,
    };
  };
};