vite.plugins.config.ts

import { createReadStream } from 'node:fs';
import { cp, stat } from 'node:fs/promises';
import { builtinModules } from 'node:module';
import path from 'node:path';
import type { Plugin } from 'vite';

const MONACO_VS = 'node_modules/monaco-editor/min/vs';

const BUILTINS = new Set([
  'electron',
  'electron/main',
  'electron/common',
  'electron/renderer',
  ...builtinModules,
  ...builtinModules.map((name) => `node:${name}`),
]);

// 出力コードに残った生の require("x")。Rollup は元コードの require() を
// 解釈せずそのまま出すため、これらは chunk.imports に載らない。
// 直前が引用符のものは文字列リテラル内(ajv が standalone 用に生成する
// `'require("ajv/dist/runtime/equal").default'` 等)なので除く
const RAW_REQUIRE = /(?<!['"`])require\(["']([^"']+)["']\)/g;

/**
 * Fails the build when the bundle keeps an unexpected bare dependency.
 *
 * バンドルから外れた依存は node_modules から実行時 require される。同梱の
 * 一覧(forge.config.ts の packagedPaths)に無いとパッケージ版だけが
 * MODULE_NOT_FOUND で落ち、しかも Electron が main のロード例外をダイアログに
 * 出すため stderr には何も出ず「窓が開かない」としか見えない。webpack の
 * asset relocator が自動でやっていた解決を人手の一覧で代替する以上、
 * 食い違いはビルド時に落とす。
 *
 * chunk.imports(静的 import 由来)だけでなく出力コードの生 require() も見る。
 * 後者を見落とすと、まさに位置を保つために require のまま残した依存
 * (vite.base.config.ts の positionalRequireDependencies)が素通りする。
 * @param {string[]} allowed - Dependencies that are intentionally external.
 * @returns {Plugin} The Vite plugin.
 */
export function assertExternals(allowed: string[]): Plugin {
  return {
    name: 'apm:assert-externals',
    generateBundle(_options, bundle) {
      const unexpected = new Set<string>();
      const check = (id: string, isInternalChunk: boolean) => {
        if (isInternalChunk) return;
        if (BUILTINS.has(id)) return;
        if (allowed.some((name) => id === name || id.startsWith(`${name}/`)))
          return;
        unexpected.add(id);
      };

      for (const output of Object.values(bundle)) {
        if (output.type !== 'chunk') continue;
        for (const id of [...output.imports, ...output.dynamicImports]) {
          // 内部チャンクは出力ファイル名で現れる
          check(id, id in bundle);
        }
        for (const [, id] of output.code.matchAll(RAW_REQUIRE)) {
          check(id, id.startsWith('.') || id.startsWith('/'));
        }
      }

      if (unexpected.size > 0) {
        throw new Error(
          `バンドルされなかった依存があります: ${[...unexpected].sort().join(', ')}\n` +
            'バンドルできるよう静的 import に直すか、vite.base.config.ts の ' +
            'externalDependencies に足してパッケージへ同梱すること。',
        );
      }
    },
  };
}

// JSON エディタから到達しない大物は同梱しない(27MB -> 6MB 弱)。
// 0.53 で AMD ビルドが Rollup 製に変わりチャンク名にハッシュが付いたため、
// ディレクトリ名の前方一致ではなく正規表現で表す。
//   language/  : 0.52 までの単独エントリの名残。editor/editor.main.js は同じ
//                worker を assets/ 配下のハッシュ付きチャンク経由で読むので、
//                同じ内容が二重に入っている
//   nls/       : vs/nls の availableLanguages を設定したときだけ読まれる。
//                apm は設定しないので Monaco の UI は英語のまま(0.52 で
//                nls.messages.*.js を除外していたのと結果は同じ)
//   assets/{css,html,ts}.worker : その言語のモデルを作った時点で初めて
//                fetch される。apm が開くのは JSON だけ
// basic-languages と各言語のトークナイザは除外しない。前者は editor.main の
// 静的依存に入っており(欠けると AMD の解決が止まる)、後者は合計 0.5MB 程度
const MONACO_EXCLUDED = [
  /^language(?:\/|$)/,
  /^nls(?:\/|$)/,
  /^assets\/(?:css|html|ts)\.worker-[^/]*\.js$/,
];

/**
 * Returns the exclusion pattern covering the given path, if any.
 * @param {string} relative - Path relative to the `vs` directory.
 * @returns {RegExp | undefined} The matching pattern, or undefined to copy.
 */
function monacoExclusionFor(relative: string): RegExp | undefined {
  const normalized = relative.split(path.sep).join('/');
  return MONACO_EXCLUDED.find((pattern) => pattern.test(normalized));
}

/**
 * Bundles the Monaco AMD build as a static asset under `vs/`.
 *
 * 値インポートするとビルドにヒープ拡大が必要になるため、Monaco は Vite の
 * 依存解決に載せず AMD ビルドを丸ごと同梱し、@monaco-editor/loader の
 * 読み込み先をそこへ向ける(CDN 許可を CSP から外すため)。
 * vite-plugin-static-copy を使わないのは、コピー元の相対ディレクトリ構造を
 * 出力先にも保つ仕様で、node_modules/... の階層ごと出力されてしまうため。
 * @param {string} projectRoot - The repository root.
 * @returns {Plugin} The Vite plugin.
 */
export function monacoAssets(projectRoot: string): Plugin {
  const source = path.join(projectRoot, MONACO_VS);
  let outDir = '';

  return {
    name: 'apm:monaco-assets',
    configResolved(config) {
      outDir = config.build.outDir;
    },
    configureServer(server) {
      // 製品版では index.html と同じ階層の vs/ を読むため、dev サーバでも
      // 同じ URL(/vs)で引けるようにする
      server.middlewares.use('/vs', (req, res, next) => {
        const relative = decodeURIComponent((req.url ?? '/').split('?')[0]);
        const filePath = path.join(source, relative);
        if (!filePath.startsWith(source + path.sep)) return next();
        void stat(filePath).then(
          (stats) => {
            if (!stats.isFile()) return next();
            if (filePath.endsWith('.js'))
              res.setHeader('Content-Type', 'text/javascript');
            if (filePath.endsWith('.css'))
              res.setHeader('Content-Type', 'text/css');
            createReadStream(filePath).pipe(res);
          },
          () => next(),
        );
      });
    },
    async writeBundle() {
      const used = new Set<RegExp>();
      await cp(source, path.join(outDir, 'vs'), {
        recursive: true,
        filter: (src) => {
          const relative = path.relative(source, src);
          if (relative === '') return true;
          const pattern = monacoExclusionFor(relative);
          if (!pattern) return true;
          used.add(pattern);
          return false;
        },
      });

      // 1 件も当たらない除外は、Monaco 側の min/vs の構成が変わって意図した
      // 除外が効かなくなった合図。同梱物が増えるだけなら気付けないのに、
      // 逆に必要なものが除外側へ回り込むと「ビルドは通るのに起動時に落ちる」
      // ため、構成が動いたこと自体をここで落とす
      const dead = MONACO_EXCLUDED.filter((pattern) => !used.has(pattern));
      if (dead.length > 0) {
        throw new Error(
          `一致しなかった Monaco の除外パターンがあります: ${dead.join(', ')}\n` +
            'monaco-editor の min/vs の構成が変わっている。同梱すべきものが' +
            '除外されていないか確かめて MONACO_EXCLUDED を見直すこと。',
        );
      }
    },
  };
}

/**
 * Relaxes the index.html CSP for the dev server only.
 *
 * 製品版の CSP は index.html の meta が単一ソースで、`script-src-elem 'self'`
 * を保つ。一方 dev では Vite client と React Fast Refresh のプリアンブルが
 * インライン module script として注入されるため、serve のときだけ meta を
 * 書き換えて通す(webpack 構成の devContentSecurityPolicy と同じ役回り。
 * 出力ファイルには一切影響しない)。
 * @returns {Plugin} The Vite plugin.
 */
export function devContentSecurityPolicy(): Plugin {
  return {
    name: 'apm:dev-csp',
    apply: 'serve',
    transformIndexHtml(html) {
      return html.replace(
        /(<meta\s+http-equiv="Content-Security-Policy"\s+content=")([^"]*)(")/,
        (_match, before: string, policy: string, after: string) =>
          before +
          policy
            .replace(
              "script-src-elem 'self'",
              "script-src-elem 'self' 'unsafe-inline'",
            )
            .replace(
              "default-src 'self'",
              "default-src 'self' ws: http://localhost:*",
            ) +
          after,
      );
    },
  };
}