import { quasar } from "@quasar/vite-plugin";
import vue from "@vitejs/plugin-vue";
import { defineConfig, type Plugin } from "vite";
import { copyFile, readFile, writeFile } from "node:fs/promises";
import { resolve } from "node:path";

// __dirname 是当前文件（vite.config.ts）所在目录，即 ClientApp 目录。
// resolve() 把相对路径拼成绝对路径。以下 4 个路径是本项目自己约定的：
const outputRoot = resolve(__dirname, "../wwwroot"); // wwwroot 根目录
const viteManifestPath = resolve(outputRoot, "auth-assets/.vite/manifest.json"); // vite 自动生成的产物映射表（见 manifest: true）
const targetPath = resolve(outputRoot, "auth-assets-manifest.json"); // 给 ASP.NET（Razor）用的正式清单
const localManifestPath = resolve(__dirname, "auth-assets-manifest.json"); // 复制到 ClientApp 的副本，方便本地查看

// 自定义 Vite 插件：在构建结束后读取 vite 生成的 manifest.json，
// 把它改造成只含 4 个入口的简化清单，供 ASP.NET 视图查找带内容哈希的文件名。
function authManifestPlugin(): Plugin {
  return {
    name: "dpz-auth-manifest", // 插件名称（调试/报错时显示）
    apply: "build", // 只在 vite build 时生效，dev/watch 模式下跳过
    async closeBundle() {
      // closeBundle：Rollup 钩子，所有产物文件写出完成后触发。
      // viteManifest 结构类似：
      //   { "src/entries/public.ts": { file: "assets/public-a1b2c3d4.js", css: [...], imports: [...] } }
      const viteManifest = JSON.parse(await readFile(viteManifestPath, "utf8"));
      // 简化后的目标结构：
      //   { "public.js": "auth-assets/assets/public-a1b2c3d4.js", "public.css": ["auth-assets/assets/xxx.css"] }
      const manifest: Record<string, string | string[]> = {};

      // 递归收集某个入口自身及其所有间接依赖 chunk 的 CSS 文件（用 Set 去重）。
      // 因为入口的 CSS 可能被 vite 拆到它 import 的其他 chunk 里，必须顺着 imports 链找全。
      function collectCss(entry: { css?: string[]; imports?: string[] }): Set<string> {
        const cssFiles = new Set<string>();
        for (const importKey of entry.imports ?? []) {
          const imported = viteManifest[importKey];
          if (!imported) {
            continue;
          }

          for (const cssFile of collectCss(imported)) {
            cssFiles.add(cssFile);
          }
        }

        for (const cssFile of entry.css ?? []) {
          cssFiles.add(cssFile);
        }
        return cssFiles;
      }

      // 只关心 4 个入口（src/entries/*.ts），其余 chunk 全部忽略
      for (const [key, value] of Object.entries(viteManifest)) {
        if (!key.startsWith("src/entries/")) {
          continue;
        }

        // "src/entries/public.ts" -> "entries/public"，作为清单里的键名
        const entry = key.replace("src/", "").replace(".ts", "");
        const bundle = value as { file: string; css?: string[]; imports?: string[] };
        // JS 记录为："entries/public.js" -> "auth-assets/assets/public-a1b2c3d4.js"
        manifest[`${entry}.js`] = `auth-assets/${bundle.file}`;

        // CSS 记录为该入口收集到的全部 CSS 文件列表
        manifest[`${entry}.css`] = [...collectCss(bundle)].map(
          (cssFile) => `auth-assets/${cssFile}`,
        );
      }

      // 写入正式清单到 wwwroot，并复制一份到 ClientApp 目录便于本地查看
      await writeFile(targetPath, `${JSON.stringify(manifest, null, 2)}\n`, "utf8");
      await copyFile(targetPath, localManifestPath);
    },
  };
}

export default defineConfig({
  // 插件：vue() 编译 .vue 单文件组件；quasar() 注入 Quasar UI 框架；authManifestPlugin() 是上面自定义的清单插件
  plugins: [vue(), quasar(), authManifestPlugin()],
  // 构建产物的公开 URL 前缀，页面里所有 JS/CSS 引用都会写成 /auth-assets/xxx
  base: "/auth-assets/",
  build: {
    // 产物输出目录（对应 clean-build.mjs 里先删除的那个目录）
    outDir: "../wwwroot/auth-assets",
    emptyOutDir: true, // 构建前自动清空输出目录
    manifest: true, // 额外生成 .vite/manifest.json，记录"源文件 -> hash 文件名"的映射（供上面的插件读取）
    modulePreload: false, // 不自动插入 <link rel="modulepreload"> 预加载标签，加载顺序交给 ASP.NET 端控制
    rollupOptions: {
      // 多入口构建：4 个页面分组各有独立入口（用户访问不同页面时只加载对应 bundle）
      input: {
        "entries/public": resolve(__dirname, "src/entries/public.ts"),
        "entries/identity": resolve(__dirname, "src/entries/identity.ts"),
        "entries/user": resolve(__dirname, "src/entries/user.ts"),
        "entries/admin": resolve(__dirname, "src/entries/admin.ts"),
      },
      output: {
        // manualChunks：手动分包。id 是模块的绝对路径，按依赖的库归类成独立 chunk。
        // 好处：大库（milkdown/codemirror 等）单独成包，改业务代码时这些 chunk 的文件名
        // （内容哈希）不会变，浏览器可以继续用缓存。
        manualChunks(id) {
          // vite 内部的运行时辅助代码单独一个包
          if (id.includes("vite/preload-helper")) {
            return "vite-runtime";
          }

          // 项目自己的源码（不在 node_modules）不手动分包，交给 rollup 默认策略
          if (!id.includes("node_modules")) {
            return;
          }

          if (id.includes("@milkdown")) {
            return "editor-milkdown";
          }

          if (id.includes("@codemirror") || id.includes("@lezer") || id.includes("codemirror")) {
            return "editor-codemirror";
          }

          if (id.includes("photoswipe")) {
            return "gallery-photoswipe";
          }

          if (id.includes("mermaid")) {
            return "editor-mermaid";
          }

          if (id.includes("vue") || id.includes("quasar") || id.includes("@quasar")) {
            return "vendor-vue-quasar";
          }
        },
      },
    },
  },
});