feat(plugin-infra): support worker thread in server side (#3462)

apps/electron/src/main/application-menu/create.ts

@@ -1,8 +1,8 @@
 import { app, Menu } from 'electron';
 
+import { isMacOS } from '../../shared/utils';
 import { revealLogFile } from '../logger';
 import { checkForUpdates } from '../updater';
-import { isMacOS } from '../utils';
 import { applicationMenuSubjects } from './subject';
 
 // Unique id for menuitems

apps/electron/src/main/helper-process.ts

@@ -15,8 +15,8 @@ import {
   type WebContents,
 } from 'electron';
 
+import { MessageEventChannel } from '../shared/utils';
 import { logger } from './logger';
-import { MessageEventChannel } from './utils';
 
 const HELPER_PROCESS_PATH = path.join(__dirname, './helper.js');
 

apps/electron/src/main/logger.ts

@@ -2,6 +2,7 @@ import { shell } from 'electron';
 import log from 'electron-log';
 
 export const logger = log.scope('main');
+export const pluginLogger = log.scope('plugin');
 log.initialize();
 
 export function getLogFilePath() {

apps/electron/src/main/main-window.ts

@@ -4,10 +4,10 @@ import { BrowserWindow, nativeTheme } from 'electron';
 import electronWindowState from 'electron-window-state';
 import { join } from 'path';
 
+import { isMacOS, isWindows } from '../shared/utils';
 import { getExposedMeta } from './exposed';
 import { ensureHelperProcess } from './helper-process';
 import { logger } from './logger';
-import { isMacOS, isWindows } from './utils';
 
 const IS_DEV: boolean =
   process.env.NODE_ENV === 'development' && !process.env.CI;
@@ -114,6 +114,7 @@ async function createWindow() {
 
 // singleton
 let browserWindow: Electron.BrowserWindow | undefined;
+
 /**
  * Restore existing BrowserWindow or Create new BrowserWindow
  */

apps/electron/src/main/plugin.ts

@@ -1,7 +1,14 @@
 import { join, resolve } from 'node:path';
+import { Worker } from 'node:worker_threads';
 
-import { logger } from '@affine/electron/main/logger';
+import { logger, pluginLogger } from '@affine/electron/main/logger';
+import { AsyncCall } from 'async-call-rpc';
 import { ipcMain } from 'electron';
+import { readFile } from 'fs/promises';
+
+import { MessageEventChannel } from '../shared/utils';
+
+const builtInPlugins = ['bookmark'];
 
 declare global {
   // fixme(himself65):
@@ -10,26 +17,41 @@ declare global {
   var asyncCall: Record<string, (...args: any) => PromiseLike<any>>;
 }
 
-export function registerPlugin() {
+export async function registerPlugin() {
   logger.info('import plugin manager');
-  globalThis.asyncCall = {};
-  const bookmarkPluginPath = join(
-    process.env.PLUGIN_DIR ?? resolve(__dirname, './plugins'),
-    './bookmark/index.js'
+  const asyncCall = AsyncCall<
+    Record<string, (...args: any) => PromiseLike<any>>
+  >(
+    {
+      log: (...args: any[]) => {
+        pluginLogger.log(...args);
+      },
+    },
+    {
+      channel: new MessageEventChannel(
+        new Worker(resolve(__dirname, './worker.js'), {})
+      ),
+    }
+  );
+  globalThis.asyncCall = asyncCall;
+  await Promise.all(
+    builtInPlugins.map(async plugin => {
+      const pluginPackageJsonPath = join(
+        process.env.PLUGIN_DIR ?? resolve(__dirname, './plugins'),
+        `./${plugin}/package.json`
+      );
+      logger.info(`${plugin} plugin path:`, pluginPackageJsonPath);
+      const packageJson = JSON.parse(
+        await readFile(pluginPackageJsonPath, 'utf-8')
+      );
+      console.log('packageJson', packageJson);
+      const serverCommand: string[] = packageJson.affinePlugin.serverCommand;
+      serverCommand.forEach(command => {
+        ipcMain.handle(command, async (_, ...args) => {
+          logger.info(`plugin ${plugin} called`);
+          return asyncCall[command](...args);
+        });
+      });
+    })
   );
-  logger.info('bookmark plugin path:', bookmarkPluginPath);
-  // eslint-disable-next-line @typescript-eslint/no-var-requires
-  const { entry } = require(bookmarkPluginPath);
-
-  entry({
-    registerCommand: (command: string, handler: (...args: any[]) => any) => {
-      logger.info('register plugin command', command);
-      ipcMain.handle(command, (event, ...args) => handler(...args));
-      globalThis.asyncCall[command] = handler;
-    },
-    registerCommands: (command: string) => {
-      ipcMain.removeHandler(command);
-      delete globalThis.asyncCall[command];
-    },
-  });
 }

apps/electron/src/main/ui/index.ts

@@ -1,7 +1,7 @@
 import { app, BrowserWindow, nativeTheme } from 'electron';
 
+import { isMacOS } from '../../shared/utils';
 import type { NamespaceHandlers } from '../type';
-import { isMacOS } from '../utils';
 import { getGoogleOauthCode } from './google-auth';
 
 export const uiHandlers = {

apps/electron/src/main/updater/electron-updater.ts

@@ -2,8 +2,8 @@ import { app } from 'electron';
 import { autoUpdater } from 'electron-updater';
 import { z } from 'zod';
 
+import { isMacOS } from '../../shared/utils';
 import { logger } from '../logger';
-import { isMacOS } from '../utils';
 import { updaterSubjects } from './event';
 
 export const ReleaseTypeSchema = z.enum([

apps/electron/src/worker/plugin.ts

@@ -0,0 +1,45 @@
+import { join, resolve } from 'node:path';
+import { parentPort } from 'node:worker_threads';
+
+import type { ServerContext } from '@toeverything/plugin-infra/server';
+import { AsyncCall } from 'async-call-rpc';
+
+import { MessageEventChannel } from '../shared/utils';
+
+if (!parentPort) {
+  throw new Error('parentPort is null');
+}
+const commandProxy: Record<string, (...args: any[]) => Promise<any>> = {};
+
+parentPort.start();
+
+const mainThread = AsyncCall<{
+  log: (...args: any[]) => Promise<void>;
+}>(commandProxy, {
+  channel: new MessageEventChannel(parentPort),
+});
+
+globalThis.console.log = mainThread.log;
+globalThis.console.error = mainThread.log;
+globalThis.console.info = mainThread.log;
+globalThis.console.debug = mainThread.log;
+globalThis.console.warn = mainThread.log;
+
+// eslint-disable-next-line @typescript-eslint/no-var-requires
+const bookmarkPluginModule = require(join(
+  process.env.PLUGIN_DIR ?? resolve(__dirname, './plugins'),
+  './bookmark/index.js'
+));
+
+const serverContext: ServerContext = {
+  registerCommand: (command, fn) => {
+    console.log('register command', command);
+    commandProxy[command] = fn;
+  },
+  unregisterCommand: command => {
+    console.log('unregister command', command);
+    delete commandProxy[command];
+  },
+};
+
+bookmarkPluginModule.entry(serverContext);