import {
  IconBook2,
  IconDatabase,
  IconMail,
  IconTool,
  IconWorld,
} from 'twenty-ui-deprecated/display';

const TOOL_ICON_MAPPINGS = [
  {
    keywords: ['learn_tools'],
    icon: IconBook2,
  },
  {
    keywords: ['email'],
    icon: IconMail,
  },
  {
    keywords: ['http_request'],
    icon: IconWorld,
  },
  {
    keywords: ['create_', 'update_', 'find_', 'delete_'],
    icon: IconDatabase,
  },
  {
    keywords: ['workflow', 'handoff'],
    icon: IconTool,
  },
] as const;

export const getToolIcon = (toolName: string) => {
  const mapping = TOOL_ICON_MAPPINGS.find(({ keywords }) =>
    keywords.some((keyword) => toolName.includes(keyword)),
  );

  return mapping?.icon ?? IconTool;
};