@electron-forge/plugin-vite 8.0.0-alpha.7 → 8.0.0-alpha.9

package/src/subprocess-worker.ts

@@ -1,5 +1,8 @@
+import { styleText } from 'node:util';
+
 import { build } from 'vite';
 
+import { viteDevServerUrls } from './config/vite.base.config.js';
 import ViteConfigGenerator from './ViteConfig.js';
 
 import type {
@@ -7,11 +10,14 @@ import type {
   VitePluginConfig,
   VitePluginRendererConfig,
 } from './Config.js';
+import type { Rollup } from 'vite';
 
 const projectDir = process.env.FORGE_VITE_PROJECT_DIR;
 const kind = process.env.FORGE_VITE_KIND as 'build' | 'renderer';
 const index = Number(process.env.FORGE_VITE_INDEX);
 const rawConfig = process.env.FORGE_VITE_CONFIG;
+const watch = process.env.FORGE_VITE_WATCH === '1';
+const rawDevServerUrls = process.env.FORGE_VITE_DEV_SERVER_URLS;
 
 if (!projectDir || !kind || !rawConfig || !Number.isInteger(index)) {
   console.error(
@@ -25,7 +31,7 @@ if (!projectDir || !kind || !rawConfig || !Number.isInteger(index)) {
 // defines when building main targets.
 const pluginConfig = JSON.parse(rawConfig) as VitePluginConfig;
 
-const generator = new ViteConfigGenerator(pluginConfig, projectDir, true);
+const generator = new ViteConfigGenerator(pluginConfig, projectDir, !watch);
 
 let spec: VitePluginBuildConfig | VitePluginRendererConfig;
 let target: 'main' | 'preload' | 'renderer';
@@ -37,11 +43,107 @@ if (kind === 'build') {
   target = 'renderer';
 }
 
+// Seed the module-level URL map so getBuildDefine() produces the same
+// *_VITE_DEV_SERVER_URL defines the in-process path would have.
+if (rawDevServerUrls) {
+  Object.assign(viteDevServerUrls, JSON.parse(rawDevServerUrls));
+}
+
 const resolved = await generator.resolveConfig(spec, target);
 
-await build({
-  configFile: false,
-  logLevel: 'error',
-  ...resolved,
-  clearScreen: false,
-});
+if (!watch) {
+  await build({
+    configFile: false,
+    logLevel: 'error',
+    ...resolved,
+    clearScreen: false,
+  });
+} else {
+  // Matches the format of the default Vite logger
+  const timeFormatter = new Intl.DateTimeFormat(undefined, {
+    hour: 'numeric',
+    minute: 'numeric',
+    second: 'numeric',
+  });
+
+  // Rollup's `input` can be a string, an array of strings, or an object.
+  // https://rollupjs.org/configuration-options/#input
+  const input =
+    resolved.build?.rollupOptions?.input ??
+    (typeof resolved.build?.lib !== 'boolean'
+      ? resolved.build?.lib?.entry
+      : undefined);
+  const targetDisplay = !input
+    ? ''
+    : typeof input === 'string'
+      ? input
+      : Array.isArray(input)
+        ? input.join(' ')
+        : Object.keys(input).join(' ');
+
+  let firstBuildSent = false;
+  const sendOnce = (msg: { type: string; message?: string }) => {
+    if (firstBuildSent) return;
+    firstBuildSent = true;
+    process.send?.(msg);
+  };
+
+  const result = await build({
+    // Avoid recursive builds caused by users configuring @electron-forge/plugin-vite in Vite config file.
+    configFile: false,
+    // We suppress Vite output and instead log lines using RollupWatcher events
+    logLevel: 'silent',
+    ...resolved,
+    plugins: [
+      // `buildEnd` and `closeBundle` are Rollup output generation hooks.
+      // https://rollupjs.org/plugin-development/#output-generation-hooks
+      {
+        name: '@electron-forge/plugin-vite:build-done',
+        buildEnd(err) {
+          if (err instanceof Error) {
+            sendOnce({ type: 'first-build-error', message: err.message });
+          }
+        },
+        closeBundle() {
+          sendOnce({ type: 'first-build-done' });
+          if (target === 'preload') {
+            // pluginHotRestart('reload') is a no-op here because viteDevServers
+            // lives in the parent; ask the parent to fan out the ws full-reload.
+            process.send?.({ type: 'reload-renderers' });
+          }
+        },
+      },
+      ...(resolved.plugins ?? []),
+    ],
+    clearScreen: false,
+  });
+
+  const isRollupWatcher = (x: unknown): x is Rollup.RollupWatcher =>
+    !!x &&
+    typeof x === 'object' &&
+    'on' in x &&
+    typeof x.on === 'function' &&
+    'close' in x &&
+    typeof x.close === 'function';
+
+  if (isRollupWatcher(result)) {
+    // The Rollup watcher emits events for subsequent builds.
+    result.on('event', (event) => {
+      if (event.code === 'ERROR' && resolved.logLevel !== 'silent') {
+        console.error(
+          `\n${styleText('dim', timeFormatter.format(new Date()))} ${event.error.message}`,
+        );
+      } else if (
+        event.code === 'BUNDLE_END' &&
+        (!resolved.logLevel || resolved.logLevel === 'info')
+      ) {
+        console.log(
+          `${styleText('dim', timeFormatter.format(new Date()))} ${styleText(['cyan', 'bold'], '[@electron-forge/plugin-vite]')} ${styleText(
+            'green',
+            'target built',
+          )} ${styleText('dim', targetDisplay)}`,
+        );
+      }
+    });
+  }
+}