openpencil/apps/web/server/api/ai/icon.ts

import { defineEventHandler, getQuery, setResponseHeaders } from 'h3';
import simpleIconsData from '@iconify-json/simple-icons/icons.json';
import lucideData from '@iconify-json/lucide/icons.json';

interface IconResult {
  d: string;
  style: 'stroke' | 'fill';
  width: number;
  height: number;
  iconId: string;
}

type IconifySet = {
  width?: number;
  height?: number;
  icons: Record<string, { body: string; width?: number; height?: number }>;
};

const simpleIcons = simpleIconsData as unknown as IconifySet;
const lucideIcons = lucideData as unknown as IconifySet;

// In-memory cache: normalized name → result (null = confirmed miss)
const iconCache = new Map<string, IconResult | null>();

/**
 * GET /api/ai/icon?name=google
 *
 * Resolves icon names to SVG path data using locally bundled icon sets.
 * Search order: lucide → simple-icons (brand icons)
 * No external network requests — instant, offline-capable.
 */
export default defineEventHandler(async (event) => {
  setResponseHeaders(event, { 'Content-Type': 'application/json' });

  const { name } = getQuery(event) as { name?: string };
  if (!name || typeof name !== 'string') {
    return { icon: null, error: 'Missing required query parameter: name' };
  }

  const normalizedName = name.trim().toLowerCase();
  if (!normalizedName) {
    return { icon: null, error: 'Empty icon name' };
  }

  if (iconCache.has(normalizedName)) {
    return { icon: iconCache.get(normalizedName) ?? null };
  }

  const result = resolveIcon(normalizedName);
  iconCache.set(normalizedName, result);
  return { icon: result };
});

// Common name aliases for icons AI models frequently request.
// Keep in sync with commonAliases in src/services/ai/icon-resolver.ts
const NAME_ALIASES: Record<string, string> = {
  burger: 'hamburger',
  burgers: 'hamburger',
  sushi: 'fish',
  ramen: 'soup',
  noodle: 'soup',
  noodles: 'soup',
  steak: 'beef',
  meat: 'beef',
  dessert: 'cake',
  desserts: 'cake',
  healthy: 'salad',
  salads: 'salad',
  drinks: 'cup-soda',
  beverage: 'cup-soda',
  beverages: 'cup-soda',
  pizzas: 'pizza',
  pasta: 'utensils',
  italian: 'utensils',
  cuisines: 'utensils-crossed',
  dollar: 'dollar-sign',
  currency: 'dollar-sign',
  price: 'dollar-sign',
  profile: 'user',
  time: 'clock',
  deliverytime: 'clock',
  rider: 'bike',
  dismiss: 'x',
  closebutton: 'x',
  cancel: 'x',
  remove: 'x',
  expand: 'maximize-2',
  collapse: 'minimize-2',
  icecream: 'ice-cream-cone',
  donut: 'donut',
  bread: 'croissant',
  fruit: 'apple',
  food: 'utensils',
  drink: 'cup-soda',
  coffee: 'coffee',
  tea: 'cup-soda',
  restaurant: 'utensils-crossed',
  delivery: 'truck',
  order: 'clipboard-list',
  recipe: 'book-open',
  grocery: 'shopping-basket',
  cart: 'shopping-cart',
  bag: 'shopping-bag',
  pay: 'credit-card',
  payment: 'credit-card',
  wallet: 'wallet',
  money: 'banknote',
  coupon: 'ticket',
  discount: 'percent',
  rating: 'star',
  review: 'message-square',
  favorite: 'heart',
  favourites: 'heart',
  favorites: 'heart',
  notification: 'bell',
  address: 'map-pin',
  navigate: 'navigation',
  directions: 'map',
  logout: 'log-out',
  login: 'log-in',
  signup: 'user-plus',
  account: 'user',
  password: 'key',
  security: 'shield',
  privacy: 'eye-off',
  about: 'info',
  faq: 'help-circle',
  support: 'headphones',
  contact: 'phone',
  feedback: 'message-circle',
  language: 'globe',
  theme: 'palette',
  darkmode: 'moon',
  lightmode: 'sun',
  sound: 'volume-2',
  mute: 'volume-x',
  wifi: 'wifi',
  bluetooth: 'bluetooth',
  battery: 'battery',
  location: 'map-pin',
  gps: 'locate',
  scan: 'scan',
  qrcode: 'qr-code',
  barcode: 'barcode',
};

function resolveIcon(name: string): IconResult | null {
  const kebab = toKebabCase(name);
  const aliased = NAME_ALIASES[name] ?? NAME_ALIASES[kebab];
  const candidates = new Set([name, kebab]);
  if (aliased) candidates.add(aliased);

  // 1. Try simple-icons first (brand/product icons).
  //    simple-icons only contains brand logos, so a hit here is unambiguously
  //    a brand — no risk of shadowing UI icon names like "search" or "home".
  for (const n of candidates) {
    const result = lookupLocal(simpleIcons, 'simple-icons', n);
    if (result) return result;
  }

  // 2. Try Lucide (UI icons)
  for (const n of candidates) {
    const result = lookupLocal(lucideIcons, 'lucide', n);
    if (result) return result;
  }

  return null;
}

function lookupLocal(set: IconifySet, collection: string, iconName: string): IconResult | null {
  const icon = set.icons[iconName];
  if (!icon) return null;
  const w = icon.width ?? set.width ?? 24;
  const h = icon.height ?? set.height ?? 24;
  return parseIconBody(icon.body, w, h, `${collection}:${iconName}`);
}

/**
 * Parse the SVG `body` field from Iconify into path data.
 * Extracts `d` from `<path>` elements and detects stroke vs fill style.
 */
function parseIconBody(
  body: string,
  width: number,
  height: number,
  iconId: string,
): IconResult | null {
  const pathRegex = /<path\s[^>]*?\bd="([^"]+)"[^>]*?\/?>/gi;
  const paths: string[] = [];
  let hasStroke = false;
  let hasFill = false;
  let match: RegExpExecArray | null;

  while ((match = pathRegex.exec(body)) !== null) {
    paths.push(match[1]);
    const tag = match[0];
    if (/\bstroke=/.test(tag) || /\bstroke-width=/.test(tag) || /\bstroke-linecap=/.test(tag)) {
      hasStroke = true;
    }
    if (/\bfill="(?!none)[^"]*"/.test(tag)) {
      hasFill = true;
    }
    if (/\bfill="none"/.test(tag)) {
      hasStroke = true;
    }
  }

  if (paths.length === 0) return null;

  // Check body-level stroke/fill attributes
  if (/\bstroke="currentColor"/.test(body) || /\bstroke-linecap=/.test(body)) {
    hasStroke = true;
  }
  if (/\bfill="currentColor"/.test(body) && !/\bfill="none"/.test(body)) {
    hasFill = true;
  }

  // When joining multiple <path> d-values, ensure each sub-path starts with
  // absolute M. A standalone <path> treats initial lowercase "m" as absolute,
  // but after concatenation it becomes relative to the previous endpoint.
  for (let i = 1; i < paths.length; i++) {
    if (paths[i].startsWith('m')) {
      paths[i] = 'M' + paths[i].slice(1);
    }
  }
  const d = paths.join(' ');
  const style: 'stroke' | 'fill' = hasStroke && !hasFill ? 'stroke' : 'fill';

  return { d, style, width, height, iconId };
}

/**
 * Convert concatenated lowercase to kebab-case for icon name matching.
 * e.g. "arrowright" → "arrow-right", "chevrondown" → "chevron-down"
 */
function toKebabCase(name: string): string {
  const prefixes = [
    'arrow',
    'chevron',
    'circle',
    'alert',
    'help',
    'external',
    'bar',
    'message',
    'log',
  ];
  for (const prefix of prefixes) {
    if (name.startsWith(prefix) && name.length > prefix.length) {
      return `${prefix}-${name.slice(prefix.length)}`;
    }
  }
  return name;
}