albex 0.1.0 → 0.3.0

package/src/worker-runtime.ts

@@ -0,0 +1,96 @@
+/**
+ * Albex worker runtime.
+ *
+ * Loads inside a Web Worker, instantiates an `AlbexEngine`, and serves the
+ * wire protocol from `worker-protocol.ts`. Designed to be referenced as:
+ *
+ *     new Worker(new URL('./worker-runtime.js', import.meta.url),
+ *                { type: 'module' });
+ *
+ * The runtime is intentionally side-effectful (registers `onmessage` at
+ * import time). It is not meant to be imported from the main thread.
+ */
+
+import { AlbexEngine, AlbexError } from './albex.js';
+import type { WorkerRequest, WorkerResponse, WorkerOp } from './worker-protocol.js';
+
+let engine: AlbexEngine | null = null;
+
+function ensureEngine(): AlbexEngine {
+  if (!engine) throw new Error('Worker runtime: init() not called yet');
+  return engine;
+}
+
+async function dispatch(op: WorkerOp): Promise<unknown> {
+  switch (op.kind) {
+    case 'init': {
+      engine = new AlbexEngine(op.opts);
+      await engine.init();
+      return undefined;
+    }
+    case 'indexFile': {
+      // Wrap the transferred buffer in a File-like object so existing
+      // indexers work unchanged.
+      const file = new File([op.buffer], op.name);
+      return ensureEngine().indexFile(file);
+    }
+    case 'search':
+      return ensureEngine().search(op.query, op.options);
+    case 'removeDocument':
+      return ensureEngine().removeDocument(op.id);
+    case 'compact':
+      ensureEngine().compact();
+      return undefined;
+    case 'reset':
+      ensureEngine().reset();
+      return undefined;
+    case 'getStats':
+      return ensureEngine().getStats();
+    case 'getLastSearchStats':
+      return ensureEngine().getLastSearchStats();
+    case 'getDocuments':
+      return ensureEngine().documents.slice();
+    case 'setMaxErrors':
+      ensureEngine().setMaxErrors(op.n);
+      return undefined;
+    case 'setThreshold':
+      ensureEngine().setThreshold(op.n);
+      return undefined;
+    case 'setMaxResults':
+      ensureEngine().setMaxResults(op.n);
+      return undefined;
+    case 'setLanguage':
+      ensureEngine().setLanguage(op.lang);
+      return undefined;
+    case 'save':
+      return ensureEngine().save(op.name);
+    case 'load':
+      return ensureEngine().load(op.name);
+    case 'loadOrInit':
+      return ensureEngine().loadOrInit(op.name);
+    case 'deleteSnapshot':
+      return ensureEngine().deleteSnapshot(op.name);
+    case 'listSnapshots':
+      return ensureEngine().listSnapshots();
+  }
+}
+
+self.onmessage = async (ev: MessageEvent<WorkerRequest>) => {
+  const { id, op } = ev.data;
+  try {
+    const result = await dispatch(op);
+    const res: WorkerResponse = { id, ok: true, result };
+    (self as unknown as Worker).postMessage(res);
+  } catch (err) {
+    const e = err as Error & { kind?: string };
+    const res: WorkerResponse = {
+      id, ok: false,
+      error: {
+        name:    e.name ?? 'Error',
+        kind:    err instanceof AlbexError ? err.kind : undefined,
+        message: e.message ?? String(err),
+      },
+    };
+    (self as unknown as Worker).postMessage(res);
+  }
+};