yaak-mountain-loop/plugin-runtime/src/PluginHandle.ts

import { BootRequest, InternalEvent } from '@yaakapp-internal/plugin';
import path from 'node:path';
import { Worker } from 'node:worker_threads';
import { EventChannel } from './EventChannel';
import { PluginWorkerData } from './index.worker';

export class PluginHandle {
  #worker: Worker;

  constructor(
    readonly pluginRefId: string,
    readonly bootRequest: BootRequest,
    readonly events: EventChannel,
  ) {
    this.#worker = this.#createWorker();
  }

  sendToWorker(event: InternalEvent) {
    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 workerData: PluginWorkerData = {
      pluginRefId: this.pluginRefId,
      bootRequest: this.bootRequest,
    };
    const worker = new Worker(workerPath, {
      workerData,
    });

    worker.on('message', (e) => this.events.emit(e));
    worker.on('error', this.#handleError.bind(this));
    worker.on('exit', this.#handleExit.bind(this));

    console.log('Created plugin worker for ', this.bootRequest.dir);

    return worker;
  }

  async #handleError(err: Error) {
    console.error('Plugin errored', this.bootRequest.dir, err);
  }

  async #handleExit(code: number) {
    if (code === 0) {
      console.log('Plugin exited successfully', this.bootRequest.dir);
    } else {
      console.log('Plugin exited with status', code, this.bootRequest.dir);
    }
  }
}