import {
  ArrowUpDown,
  BadgeCheck,
  Boxes,
  Braces,
  Calculator,
  Columns3,
  Combine,
  Database,
  Filter,
  GitMerge,
  ListEnd,
  ListFilter,
  ListChecks,
  ListOrdered,
  PanelTopOpen,
  Replace,
  ReplaceAll,
  ScanSearch,
  Sigma,
  SquareFunction,
  Variable,
  type LucideIcon
} from "lucide-react";

const icons: Record<string, LucideIcon> = {
  "arrow-up-down": ArrowUpDown,
  "badge-check": BadgeCheck,
  boxes: Boxes,
  braces: Braces,
  calculator: Calculator,
  "columns-3": Columns3,
  combine: Combine,
  database: Database,
  filter: Filter,
  "git-merge": GitMerge,
  "list-end": ListEnd,
  "list-filter": ListFilter,
  "list-checks": ListChecks,
  "list-ordered": ListOrdered,
  "panel-top-open": PanelTopOpen,
  replace: Replace,
  "replace-all": ReplaceAll,
  "scan-search": ScanSearch,
  sigma: Sigma,
  "function-square": SquareFunction,
  variable: Variable
};

export function dataflowNodeIcon(icon: string): LucideIcon {
  return icons[icon] ?? Braces;
}