import type { Plugin } from "vite";
import { writeFileSync, readFileSync, existsSync, copyFileSync } from "fs";
import { resolve } from "path";

interface PageConfig {
  /** Page name (used for filename and script name) */
  name: string;
  /** Page title */
  title: string;
  /** Script filename (e.g., 'homepage.min.js') */
  script: string;
  /** Optional CSS filename */
  css?: string;
  /** Target filename (defaults to name + '.html') */
  filename?: string;
}

// Load the favicon SVG from assets
function getFaviconSVG(): string {
  const logoPath = resolve(__dirname, "..", "assets", "logo.svg");
  return readFileSync(logoPath, "utf-8");
}

// Generate HTML content for a page
function generatePageHTML(config: PageConfig): string {
  const cssLink = config.css
    ? `<link rel="stylesheet" href="./${config.css}">`
    : "";

  return `<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <meta name="viewport" content="width=device-width, initial-scale=1.0">
    <title>${config.title}</title>
    <link rel="icon" type="image/x-icon" href="dumb://homepage/favicon.ico">
    ${cssLink}
</head>
<body>
    <script src="./${config.script}"></script>
</body>
</html>`;
}

/**
 * Vite plugin to generate dumb:// protocol pages
 * Supports multiple pages like homepage, config, about, etc.
 */
export function pageGenerator(pages: PageConfig[]): Plugin {
  return {
    name: "dumb-page-generator",
    writeBundle() {
      const assetsDir = resolve(__dirname, "..", "assets", "webui");

      try {
        // Generate HTML files for each page
        for (const page of pages) {
          const htmlContent = generatePageHTML(page);
          const filename = page.filename || `${page.name}.html`;
          writeFileSync(resolve(assetsDir, filename), htmlContent);
          console.log(`✓ Generated ${filename} for dumb://${page.name}`);
        }

        // Generate favicon files for scheme handler lookups
        const faviconContent = getFaviconSVG();
        writeFileSync(resolve(assetsDir, "favicon.svg"), faviconContent);
        console.log("✓ Generated favicon.svg");

        // Copy pre-generated PNG and ICO favicons if they exist
        const sourceDir = resolve(__dirname, "..", "assets", "webui");
        try {
          const pngPath = resolve(sourceDir, "favicon.png");
          const icoPath = resolve(sourceDir, "favicon.ico");

          if (existsSync(pngPath)) {
            copyFileSync(pngPath, resolve(assetsDir, "favicon.png"));
            console.log("✓ Copied favicon.png");
          }

          if (existsSync(icoPath)) {
            copyFileSync(icoPath, resolve(assetsDir, "favicon.ico"));
            console.log("✓ Copied favicon.ico");
          }
        } catch (err) {
          console.warn("⚠ Could not copy favicon PNG/ICO files:", err);
        }
      } catch (error) {
        console.error("✗ Failed to generate page files:", error);
        throw error;
      }
    },
  };
}

/**
 * Convenience function for homepage generation
 * @deprecated Use pageGenerator with homepage config instead
 */
export function homepageGenerator(): Plugin {
  return pageGenerator([
    {
      name: "homepage",
      title: "Dumber Browser",
      script: "homepage.min.js",
      filename: "index.html", // Special case: homepage uses index.html
    },
  ]);
}