import { loader } from "fumadocs-core/source";
import { icons } from "lucide-react";
import { createElement } from "react";
import { i18n } from "@/lib/i18n";
import { docs } from "@/.source/server";

const pageIcons: Record<string, keyof typeof icons> = {
  "what-is-dbx": "BookOpen",
  changelog: "History",
  "getting-started": "Rocket",
  databases: "Database",
  "query-editor": "SquareTerminal",
  "data-grid": "Table2",
  "schema-browser": "Network",
  redis: "KeyRound",
  mongodb: "Braces",
  "object-browser": "Package",
  "schema-diff": "GitCompareArrows",
  "data-transfer": "ArrowRightLeft",
  "table-structure": "TableProperties",
  "field-lineage": "GitBranch",
  "table-import": "Upload",
  "sql-file": "FileCode2",
  "database-export": "Download",
  "sql-snippets": "ScrollText",
  "ai-assistant": "Bot",
  cli: "Terminal",
  mcp: "Cable",
  plugins: "Plug",
  "driver-management": "Package",
  "keyboard-shortcuts": "Keyboard",
  "connection-import": "Import",
  "config-export": "FileUp",
  "ssh-tunnel": "ShieldCheck",
};

export const source = loader({
  baseUrl: "/docs",
  source: docs.toFumadocsSource(),
  i18n,
  plugins: [
    {
      name: "dbx-sidebar-icons",
      transformPageTree: {
        file(node) {
          const slug = node.url.split("/").filter(Boolean).at(-1);
          const iconName = slug ? pageIcons[slug] : undefined;
          const Icon = iconName ? icons[iconName] : undefined;

          if (!Icon) return node;

          return {
            ...node,
            icon: createElement(Icon, {
              "aria-hidden": true,
              className: "size-4",
              strokeWidth: 2,
            }),
          };
        },
      },
    },
  ],
});