Refactor plugin manager and gRPC server (#96)

plugin-runtime/src/PluginHandle.ts

@@ -18,6 +18,10 @@ export class PluginHandle {
     this.#worker.postMessage(event);
   }
 
+  async terminate() {
+    await this.#worker.terminate();
+  }
+
   #createWorker(): Worker {
     const workerPath = process.env.YAAK_WORKER_PATH ?? path.join(__dirname, 'index.worker.cjs');
     const worker = new Worker(workerPath, {

plugin-runtime/src/index.ts

@@ -22,13 +22,19 @@ const plugins: Record<string, PluginHandle> = {};
         plugins[pluginEvent.pluginRefId] = plugin;
       }
 
-      // Once booted, forward all events to plugin's worker
+      // Once booted, forward all events to the plugin worker
       const plugin = plugins[pluginEvent.pluginRefId];
       if (!plugin) {
         console.warn('Failed to get plugin for event by', pluginEvent.pluginRefId);
         continue;
       }
 
+      if (pluginEvent.payload.type === 'terminate_request') {
+        await plugin.terminate();
+        console.log('Terminated plugin worker', pluginEvent.pluginRefId);
+        delete plugins[pluginEvent.pluginRefId];
+      }
+
       plugin.sendToWorker(pluginEvent);
     }
     console.log('Stream ended');

plugin-runtime/src/index.worker.ts

@@ -155,6 +155,14 @@ async function initialize() {
         return;
       }
 
+      if (payload.type === 'terminate_request') {
+        const payload: InternalEventPayload = {
+          type: 'terminate_response',
+        };
+        sendPayload(payload, replyId);
+        return;
+      }
+
       if (payload.type === 'import_request' && typeof mod.pluginHookImport === 'function') {
         const reply: ImportResponse | null = await mod.pluginHookImport(ctx, payload.content);
         if (reply != null) {