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

import { ObjectMetadataItem } from '@/object-metadata/types/ObjectMetadataItem';
import { MainIdentifierMapper } from '@/ui/object/field/types/MainIdentifierMapper';

export const useObjectMainIdentifier = (
  objectMetadataItem?: ObjectMetadataItem,
) => {
  if (!objectMetadataItem) {
    return {
      mainIdentifierMapper: undefined,
      mainIdentifierFieldMetadataId: undefined,
      basePathToShowPage: undefined,
    };
  }

  const mainIdentifierMapper: MainIdentifierMapper = (record: any) => {
    if (objectMetadataItem.nameSingular === 'company') {
      return {
        id: record.id,
        name: record.name,
        avatarUrl: record.avatarUrl,
        avatarType: 'squared',
        record: record,
      };
    }

    if (objectMetadataItem.nameSingular === 'workspaceMember') {
      return {
        id: record.id,
        name: record.name.firstName + ' ' + record.name.lastName,
        avatarUrl: record.avatarUrl,
        avatarType: 'rounded',
        record: record,
      };
    }

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

  const mainIdentifierFieldMetadataId = objectMetadataItem.fields.find(
    ({ name }) => name === 'name',
  )?.id;

  const basePathToShowPage = `/object/${objectMetadataItem.nameSingular}/`;

  return {
    mainIdentifierMapper,
    mainIdentifierFieldMetadataId,
    basePathToShowPage,
  };
};