import satori from "satori";
import { initWasm, Resvg } from "@resvg/resvg-wasm";
import { join } from "path";

// Load Inter font files (woff, not woff2 — satori doesn't support woff2)
const fontsDir = join(import.meta.dir, "../fonts");
const [interRegular, interBold] = await Promise.all([
  Bun.file(join(fontsDir, "inter-latin-400-normal.woff")).arrayBuffer(),
  Bun.file(join(fontsDir, "inter-latin-700-normal.woff")).arrayBuffer(),
]);

// Initialize resvg-wasm
// Try bundled WASM first, then fall back to node_modules
const possiblePaths = [
  // Bundled WASM file (most reliable)
  join(import.meta.dir, "../vendor/index_bg.wasm"),
  // Development: node_modules is a sibling of src
  join(import.meta.dir, "../node_modules/@resvg/resvg-wasm/index_bg.wasm"),
  // Installed: in parent node_modules
  join(import.meta.dir, "../../@resvg/resvg-wasm/index_bg.wasm"),
  join(import.meta.dir, "../../../@resvg/resvg-wasm/index_bg.wasm"),
];

let wasmPath: string | undefined;
for (const path of possiblePaths) {
  if (await Bun.file(path).exists()) {
    wasmPath = path;
    break;
  }
}

if (!wasmPath) {
  console.error("Failed to find @resvg/resvg-wasm WASM file. Tried:");
  for (const path of possiblePaths) {
    console.error(`  - ${path}`);
  }
  console.error(`\nCurrent directory: ${import.meta.dir}`);
  throw new Error("Could not find @resvg/resvg-wasm WASM file");
}

await initWasm(Bun.file(wasmPath).arrayBuffer());

// Cache generated images by diagram ID
const cache = new Map<string, Buffer>();

export async function generateIndexOgImage(
  mode: "local" | "server",
  diagramCount: number,
): Promise<Buffer> {
  const cacheKey = `__index_${mode}_${diagramCount}`;
  const cached = cache.get(cacheKey);
  if (cached) return cached;

  const subtitle = mode === "server"
    ? `${diagramCount} diagram${diagramCount !== 1 ? "s" : ""} shared`
    : `${diagramCount} diagram${diagramCount !== 1 ? "s" : ""}`;

  const tagline = mode === "server"
    ? "Interactive code walkthrough diagrams, shareable with anyone."
    : "Interactive code walkthrough diagrams";

  const svg = await satori(
    {
      type: "div",
      props: {
        style: {
          width: "100%",
          height: "100%",
          display: "flex",
          flexDirection: "column",
          justifyContent: "center",
          alignItems: "center",
          padding: "60px",
          backgroundColor: "#0a0a0a",
          color: "#e5e5e5",
          fontFamily: "Inter",
        },
        children: [
          {
            type: "div",
            props: {
              style: {
                display: "flex",
                flexDirection: "column",
                alignItems: "center",
                gap: "20px",
              },
              children: [
                {
                  type: "div",
                  props: {
                    style: {
                      fontSize: "80px",
                      fontWeight: 700,
                      color: "#e5e5e5",
                    },
                    children: "Traverse",
                  },
                },
                {
                  type: "div",
                  props: {
                    style: {
                      fontSize: "32px",
                      color: "#a3a3a3",
                      textAlign: "center",
                    },
                    children: tagline,
                  },
                },
                {
                  type: "div",
                  props: {
                    style: {
                      display: "flex",
                      alignItems: "center",
                      gap: "8px",
                      marginTop: "16px",
                      fontSize: "24px",
                      color: "#a3a3a3",
                      backgroundColor: "#1c1c1e",
                      padding: "10px 24px",
                      borderRadius: "8px",
                      border: "1px solid #262626",
                    },
                    children: subtitle,
                  },
                },
              ],
            },
          },
        ],
      },
    },
    {
      width: 1200,
      height: 630,
      fonts: [
        { name: "Inter", data: interRegular, weight: 400, style: "normal" as const },
        { name: "Inter", data: interBold, weight: 700, style: "normal" as const },
      ],
    },
  );

  const resvg = new Resvg(svg, {
    fitTo: { mode: "width", value: 1200 },
  });
  const png = Buffer.from(resvg.render().asPng());

  cache.set(cacheKey, png);
  return png;
}

export async function generateOgImage(
  id: string,
  summary: string,
  nodeNames: string[],
): Promise<Buffer> {
  const cached = cache.get(id);
  if (cached) return cached;

  const nodeCount = nodeNames.length;
  const displayNodes = nodeNames.slice(0, 8);
  const extra = nodeCount > 8 ? nodeCount - 8 : 0;

  const svg = await satori(
    {
      type: "div",
      props: {
        style: {
          width: "100%",
          height: "100%",
          display: "flex",
          flexDirection: "column",
          justifyContent: "space-between",
          padding: "60px",
          backgroundColor: "#0a0a0a",
          color: "#e5e5e5",
          fontFamily: "Inter",
        },
        children: [
          // Top: Traverse label + node count
          {
            type: "div",
            props: {
              style: {
                display: "flex",
                alignItems: "center",
                justifyContent: "space-between",
              },
              children: [
                {
                  type: "div",
                  props: {
                    style: {
                      fontSize: "20px",
                      fontWeight: 700,
                      color: "#a3a3a3",
                      letterSpacing: "0.05em",
                      textTransform: "uppercase" as const,
                    },
                    children: "Traverse",
                  },
                },
                {
                  type: "div",
                  props: {
                    style: {
                      fontSize: "16px",
                      color: "#666",
                    },
                    children: `${nodeCount} node${nodeCount !== 1 ? "s" : ""}`,
                  },
                },
              ],
            },
          },
          // Middle: Summary headline
          {
            type: "div",
            props: {
              style: {
                display: "flex",
                flexDirection: "column",
                gap: "24px",
                flex: 1,
                justifyContent: "center",
              },
              children: [
                {
                  type: "div",
                  props: {
                    style: {
                      fontSize: summary.length > 60 ? "36px" : "44px",
                      fontWeight: 700,
                      lineHeight: 1.2,
                      color: "#e5e5e5",
                      overflow: "hidden",
                      textOverflow: "ellipsis",
                    },
                    children: summary,
                  },
                },
              ],
            },
          },
          // Bottom: Node pills
          {
            type: "div",
            props: {
              style: {
                display: "flex",
                flexWrap: "wrap",
                gap: "8px",
              },
              children: [
                ...displayNodes.map((name) => ({
                  type: "div",
                  props: {
                    style: {
                      fontSize: "14px",
                      color: "#a3a3a3",
                      backgroundColor: "#1c1c1e",
                      padding: "4px 12px",
                      borderRadius: "6px",
                      border: "1px solid #262626",
                    },
                    children: name,
                  },
                })),
                ...(extra > 0
                  ? [
                      {
                        type: "div",
                        props: {
                          style: {
                            fontSize: "14px",
                            color: "#666",
                            padding: "4px 8px",
                          },
                          children: `+${extra} more`,
                        },
                      },
                    ]
                  : []),
              ],
            },
          },
        ],
      },
    },
    {
      width: 1200,
      height: 630,
      fonts: [
        { name: "Inter", data: interRegular, weight: 400, style: "normal" as const },
        { name: "Inter", data: interBold, weight: 700, style: "normal" as const },
      ],
    },
  );

  const resvg = new Resvg(svg, {
    fitTo: { mode: "width", value: 1200 },
  });
  const png = Buffer.from(resvg.render().asPng());

  cache.set(id, png);
  return png;
}