fix: load plugins

Co-authored-by: Su-Yong <simssy2205@gmail.com>

vite-plugins/plugin-importer.ts

@@ -7,7 +7,7 @@ const snakeToCamel = (text: string) =>
   text.replace(/-(\w)/g, (_, letter: string) => letter.toUpperCase());
 
 export const pluginVirtualModuleGenerator = (
-  mode: 'main' | 'preload' | 'renderer' | 'menu',
+  mode: 'main' | 'preload' | 'renderer',
 ) => {
   const project = new Project({
     tsConfigFilePath: resolve(__dirname, '..', 'tsconfig.json'),
@@ -37,7 +37,7 @@ export const pluginVirtualModuleGenerator = (
     // prettier-ignore
     for (const { name, path } of plugins) {
       const relativePath = relative(resolve(srcPath, '..'), path).replace(/\\/g, '/');
-      writer.writeLine(`import ${snakeToCamel(name)}Plugin from "./${relativePath}";`);
+      writer.writeLine(`import ${snakeToCamel(name)}Plugin, { pluginStub as ${snakeToCamel(name)}PluginStub } from "./${relativePath}";`);
     }
 
     writer.blankLine();
@@ -45,15 +45,17 @@ export const pluginVirtualModuleGenerator = (
     // Context-specific exports
     writer.writeLine(`export const ${mode}Plugins = {`);
     for (const { name } of plugins) {
-      writer.writeLine(`  "${name}": ${snakeToCamel(name)}Plugin,`);
+      const checkMode = mode === 'main' ? 'backend' : mode;
+      // HACK: To avoid situation like importing renderer plugins in main
+      writer.writeLine(`  ...(${snakeToCamel(name)}Plugin['${checkMode}'] ? { "${name}": ${snakeToCamel(name)}Plugin } : {}),`);
     }
     writer.writeLine('};');
     writer.blankLine();
 
-    // All plugins export
+    // All plugins export (stub only) // Omit<Plugin, 'backend' | 'preload' | 'renderer'>
     writer.writeLine('export const allPlugins = {');
     for (const { name } of plugins) {
-      writer.writeLine(`  "${name}": ${snakeToCamel(name)}Plugin,`);
+      writer.writeLine(`  "${name}": ${snakeToCamel(name)}PluginStub,`);
     }
     writer.writeLine('};');
     writer.blankLine();

vite-plugins/plugin-loader.ts

@@ -2,7 +2,7 @@ import { readFile } from 'node:fs/promises';
 import { resolve, basename } from 'node:path';
 
 import { createFilter } from 'vite';
-import { Project, ts, ObjectLiteralExpression } from 'ts-morph';
+import { Project, ts, ObjectLiteralExpression, VariableDeclarationKind } from 'ts-morph';
 
 import type { PluginOption } from 'vite';
 
@@ -78,7 +78,7 @@ export default function (mode: 'backend' | 'preload' | 'renderer' | 'none'): Plu
         }
       });
 
-      const contexts = ['backend', 'preload', 'renderer'];
+      const contexts = ['backend', 'preload', 'renderer', 'menu'];
       for (const ctx of contexts) {
         if (mode === 'none') {
           const index = propertyNames.indexOf(ctx);
@@ -89,6 +89,7 @@ export default function (mode: 'backend' | 'preload' | 'renderer' | 'none'): Plu
         }
 
         if (ctx === mode) continue;
+        if (ctx === 'menu' && mode === 'backend') continue;
 
         const index = propertyNames.indexOf(ctx);
         if (index === -1) continue;
@@ -96,6 +97,45 @@ export default function (mode: 'backend' | 'preload' | 'renderer' | 'none'): Plu
         objExpr.getProperty(propertyNames[index])?.remove();
       }
 
+      const stubObjExpr = src.addVariableStatement({
+        isExported: true,
+        declarationKind: VariableDeclarationKind.Const,
+        declarations: [{
+          name: 'pluginStub',
+          initializer: (writer) => writer.write(objExpr!.getText()),
+        }]
+      })
+        .getDeclarations()[0]
+        .getInitializer() as ObjectLiteralExpression;
+
+      const stubProperties = stubObjExpr.getProperties();
+      const stubPropertyNames = stubProperties.map((prop) => {
+        switch (prop.getKind()) {
+          case ts.SyntaxKind.PropertyAssignment:
+            return prop
+              .asKindOrThrow(ts.SyntaxKind.PropertyAssignment)
+              .getName();
+          case ts.SyntaxKind.ShorthandPropertyAssignment:
+            return prop
+              .asKindOrThrow(ts.SyntaxKind.ShorthandPropertyAssignment)
+              .getName();
+          case ts.SyntaxKind.MethodDeclaration:
+            return prop
+              .asKindOrThrow(ts.SyntaxKind.MethodDeclaration)
+              .getName();
+          default:
+            throw new Error('Not implemented');
+        }
+      });
+
+      if (mode === 'backend') contexts.pop();
+      for (const ctx of contexts) {
+        const index = stubPropertyNames.indexOf(ctx);
+        if (index === -1) continue;
+
+        stubObjExpr.getProperty(stubPropertyNames[index])?.remove();
+      }
+
       return {
         code: src.getText(),
       };