import {
  ArrowUpDown,
  Braces,
  Columns3,
  Combine,
  Database,
  Filter,
  GitMerge,
  ListEnd,
  ListFilter,
  PanelTopOpen,
  Sigma,
  Variable,
  type LucideIcon
} from "lucide-react";

const icons: Record<string, LucideIcon> = {
  "arrow-up-down": ArrowUpDown,
  braces: Braces,
  "columns-3": Columns3,
  combine: Combine,
  database: Database,
  filter: Filter,
  "git-merge": GitMerge,
  "list-end": ListEnd,
  "list-filter": ListFilter,
  "panel-top-open": PanelTopOpen,
  sigma: Sigma,
  variable: Variable
};

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