@pi-oxide/extension-js 0.2.3 → 0.3.0

package/worker.js

@@ -1,22 +1,23 @@
-import R, { setLogLevel as y, ExtensionSession as _ } from "./extension_js.js";
-const g = {
+import z, { setLogLevel as L, ExtensionSession as G, registerJsCallBatch as H } from "../../pkg/extension_js.js";
+import "zod";
+const w = {
   debug: 0,
   info: 1,
   warn: 2,
   error: 3,
   none: 4
 };
-let u = "error", m = null;
-function E(r) {
-  u = r, m && m(g[r]);
+let E = "error", S = null;
+function J(r) {
+  E = r, S && S(w[r]);
 }
-function x(r) {
-  m = r, u !== "error" && r(g[u]);
+function V(r) {
+  S = r, E !== "error" && r(w[E]);
 }
-function $(r) {
-  return g[r] >= g[u];
+function q(r) {
+  return w[r] >= w[E];
 }
-function k(r, e = "info") {
+function U(r, e = "info") {
   var t;
   if (r === null) return "null";
   if (r === void 0) return "undefined";
@@ -25,65 +26,65 @@ function k(r, e = "info") {
     return String(r);
   if (typeof r == "bigint") return `${r}n`;
   if (r instanceof Error) {
-    const s = e === "debug" ? r.stack : (t = r.stack) == null ? void 0 : t.split(`
+    const n = e === "debug" ? r.stack : (t = r.stack) == null ? void 0 : t.split(`
 `)[0];
-    return JSON.stringify({ message: r.message, name: r.name, stack: s });
+    return JSON.stringify({ message: r.message, name: r.name, stack: n });
   }
   if (typeof r == "function") return "[Function]";
   if (typeof r == "symbol") return String(r);
   if (typeof r == "object")
     try {
       return JSON.stringify(r);
-    } catch (s) {
-      return s instanceof TypeError && s.message.includes("circular") ? "[Circular]" : `[Unserializable: ${s instanceof Error ? s.message : String(s)}]`;
+    } catch (n) {
+      return n instanceof TypeError && n.message.includes("circular") ? "[Circular]" : `[Unserializable: ${n instanceof Error ? n.message : String(n)}]`;
     }
   return String(r);
 }
-function I(r) {
+function K(r) {
   return `[extension-js][${r}]`;
 }
-function A(r, e) {
+function X(r, e) {
   if (!r) return "";
   const t = [];
   try {
-    for (const [s, o] of Object.entries(r))
-      t.push(`${s}=${k(o, e)}`);
+    for (const [n, s] of Object.entries(r))
+      t.push(`${n}=${U(s, e)}`);
   } catch {
     return " metadata=[unreadable]";
   }
   return t.length > 0 ? ` ${t.join(" ")}` : "";
 }
-function l(r, e) {
+function h(r, e) {
   return e.length === 0 ? { event: r } : e.length === 1 && typeof e[0] == "object" && e[0] !== null && !Array.isArray(e[0]) ? { event: r, metadata: e[0] } : {
     event: r,
-    metadata: { _args: e.map((t) => k(t)).join(" ") }
+    metadata: { _args: e.map((t) => U(t)).join(" ") }
   };
 }
-class h {
+class I {
   constructor(e = "root") {
     this.namespace = e;
   }
-  log(e, t, s) {
+  log(e, t, n) {
     try {
-      if (!$(e)) return;
-      const o = I(this.namespace), p = A(s, e), c = `${o} ${t}${p}`;
+      if (!q(e)) return;
+      const s = K(this.namespace), a = X(n, e), o = `${s} ${t}${a}`;
       switch (e) {
         case "debug":
-          console.log(c);
+          console.log(o);
           break;
         case "info":
-          console.log(c);
+          console.log(o);
           break;
         case "warn":
-          console.warn(c);
+          console.warn(o);
           break;
         case "error":
-          console.error(c);
+          console.error(o);
           break;
         case "none":
           break;
         default: {
-          const w = e;
+          const f = e;
           break;
         }
       }
@@ -91,85 +92,368 @@ class h {
     }
   }
   debug(e, ...t) {
-    const { event: s, metadata: o } = l(e, t);
-    this.log("debug", s, o);
+    const { event: n, metadata: s } = h(e, t);
+    this.log("debug", n, s);
   }
   info(e, ...t) {
-    const { event: s, metadata: o } = l(e, t);
-    this.log("info", s, o);
+    const { event: n, metadata: s } = h(e, t);
+    this.log("info", n, s);
   }
   warn(e, ...t) {
-    const { event: s, metadata: o } = l(e, t);
-    this.log("warn", s, o);
+    const { event: n, metadata: s } = h(e, t);
+    this.log("warn", n, s);
   }
   error(e, ...t) {
-    const { event: s, metadata: o } = l(e, t);
-    this.log("error", s, o);
+    const { event: n, metadata: s } = h(e, t);
+    this.log("error", n, s);
   }
   child(e) {
-    return new h(`${this.namespace}.${e}`);
+    return new I(`${this.namespace}.${e}`);
   }
-  timer(e, t, s = "info") {
-    const o = typeof performance < "u" && performance.now, p = o ? performance.now() : Date.now();
-    return (c) => {
+  timer(e, t, n = "info") {
+    const s = typeof performance < "u" && performance.now, a = s ? performance.now() : Date.now();
+    return (o) => {
       try {
-        const w = o ? performance.now() : Date.now(), S = Math.round(w - p), L = {
+        const f = s ? performance.now() : Date.now(), l = Math.round(f - a), d = {
           ...t,
-          ...c,
-          duration_ms: S
+          ...o,
+          duration_ms: l
         };
-        this.log(s, e, L);
+        this.log(n, e, d);
       } catch {
       }
     };
   }
 }
-const f = new h("root");
-let a = null, b = !1;
-const d = /* @__PURE__ */ new Map();
-let i;
-function T() {
-  return Math.random().toString(36).slice(2) + Date.now().toString(36);
+const m = new I("root"), Y = /* @__PURE__ */ new Set([
+  "page_url",
+  "page_title",
+  "page_click",
+  "page_fill",
+  "page_type",
+  "page_append",
+  "page_press",
+  "page_select",
+  "page_check",
+  "page_hover",
+  "page_unhover",
+  "page_scroll",
+  "page_scroll_to",
+  "page_dblclick",
+  "page_back",
+  "tab_click",
+  "tab_fill",
+  "tab_type",
+  "tab_press",
+  "tab_select",
+  "tab_check",
+  "tab_hover",
+  "tab_unhover",
+  "tab_scroll",
+  "tab_scroll_to",
+  "tab_dblclick",
+  "tab_back"
+]);
+function Q(r) {
+  return Y.has(r);
 }
-function j(r) {
+const Z = /* @__PURE__ */ new Map();
+function x(r, e) {
+  Z.set(r, e);
+}
+function ee(r) {
+  return r.startsWith("tab_") ? "required" : "active";
+}
+function re(r) {
+  return r === "content-script" ? "content-script" : r === "main-thread" ? "main-thread" : r === "worker" ? "worker:default" : r.startsWith("worker:") ? r : "main-thread";
+}
+function te(r, e) {
+  return e !== "main-thread" ? e : Q(r) ? "content-script" : e;
+}
+function ne(r, e) {
+  return {
+    endpoint: re(te(r, e)),
+    tabPolicy: ee(r)
+  };
+}
+function se(r) {
+  for (const e of r)
+    x(e.action, ne(e.action, e.owner));
+}
+function k(r) {
+  return r == null ? {} : r instanceof Map ? Object.fromEntries(
+    [...r.entries()].map(([e, t]) => [e, k(t)])
+  ) : Array.isArray(r) ? r.map(k) : r;
+}
+function oe(r) {
+  return {
+    action: r.action,
+    namespace: r.namespace,
+    name: r.name,
+    publicName: r.publicName,
+    description: r.description,
+    fields: r.fields,
+    aliases: (r.aliases ?? []).map((e) => ({
+      namespace: e.namespace,
+      name: e.name,
+      fields: e.fields
+    })),
+    paramsDoc: r.paramsDoc.map((e) => ({
+      name: e.name,
+      type: e.type,
+      required: e.required,
+      description: e.description
+    })),
+    returnsDoc: {
+      type: r.returnsDoc.type,
+      description: r.returnsDoc.description
+    },
+    errorCode: r.errorCode,
+    errorCategory: r.errorCategory ?? null
+  };
+}
+m.child("tool-registry");
+let c = null, T = !1;
+const y = /* @__PURE__ */ new Map();
+function R(r, e) {
+  try {
+    return r.postMessage(e), !0;
+  } catch (t) {
+    const n = t instanceof Error ? t.message : String(t);
+    return m.error("port_post_failed", { error: n }), !1;
+  }
+}
+const B = 1e3;
+function j(r, e, t) {
+  e === "main-thread" || e === "content-script" ? R(t, { type: "relayCancel", id: r, owner: e }) : R(t, { type: "registryCallCancel", id: r });
+}
+function F(r, e) {
+  for (const [t, n] of y)
+    clearTimeout(n.timeoutId), j(t, n.owner, n.port), n.settle({
+      ok: !1,
+      error: { message: e, code: r }
+    }), y.delete(t);
+}
+const A = /* @__PURE__ */ new Map(), p = /* @__PURE__ */ new Map();
+function i(r, e) {
+  A.set(r, e);
+}
+function ae() {
+  const r = new Uint8Array(16);
+  return crypto.getRandomValues(r), Array.from(r, (e) => e.toString(16).padStart(2, "0")).join("");
+}
+function ie(r) {
   return r <= 0 ? "debug" : r === 1 ? "info" : r === 2 ? "warn" : r === 3 ? "error" : "none";
 }
-const W = self;
-W.__extension_js_relay = (r) => (f.debug("relay", {
-  action: r == null ? void 0 : r.action,
-  runId: i
-}), new Promise((e) => {
-  const t = T();
-  d.set(t, e), self.postMessage({
-    type: "asyncRelay",
-    id: t,
-    command: r,
-    runId: i
+const C = /* @__PURE__ */ new Map(), _ = /* @__PURE__ */ new Map();
+function le(r, e) {
+  var t;
+  if (C.has(r))
+    throw new Error(`Worker port already registered for owner: ${r}`);
+  if (typeof e.addEventListener != "function")
+    throw new Error(`Worker port for owner "${r}" cannot receive responses`);
+  C.set(r, e), e.addEventListener("message", async (n) => {
+    var a;
+    const s = n.data;
+    if (s !== null && (s.type === "asyncRelayResult" || s.type === "registryCallResult") && typeof s.id == "string") {
+      $(s.id, s.result);
+      return;
+    }
+    if (s !== null && s.type === "registryCallCancel" && typeof s.id == "string") {
+      (a = _.get(s.id)) == null || a.abort(), _.delete(s.id);
+      return;
+    }
+    if (s !== null && s.type === "registryCall" && typeof s.id == "string" && typeof s.action == "string") {
+      const o = s.id, f = new AbortController();
+      _.set(o, f);
+      const l = A.get(s.action);
+      let d;
+      if (!l)
+        d = {
+          ok: !1,
+          error: { message: `Unknown worker action: ${s.action}`, code: "E_UNKNOWN" }
+        };
+      else
+        try {
+          const u = await l(s.params, {
+            action: s.action,
+            callId: s.callId,
+            runId: s.runId,
+            signal: f.signal
+          });
+          d = f.signal.aborted ? { ok: !1, error: { message: "Relay aborted", code: "E_ABORT" } } : { ok: !0, value: u };
+        } catch (u) {
+          d = {
+            ok: !1,
+            error: {
+              message: u instanceof Error ? u.message : String(u),
+              code: f.signal.aborted ? "E_ABORT" : "E_WORKER_HANDLER"
+            }
+          };
+        }
+      _.delete(o), R(e, { type: "registryCallResult", id: o, result: d }) || $(o, {
+        ok: !1,
+        error: { message: "Failed to deliver worker handler response", code: "E_PORT" }
+      });
+    }
+  }), (t = e.start) == null || t.call(e);
+}
+function $(r, e) {
+  const t = y.get(r);
+  return t ? (t.settle(e), !0) : (m.warn("asyncRelayResult_no_pending_relay", { id: r }), !1);
+}
+function ce(r) {
+  if (r === "main-thread" || r === "content-script")
+    return self;
+  const e = C.get(r);
+  return e || null;
+}
+function fe(r) {
+  const { owner: e, action: t, timeoutMs: n, tabPolicy: s } = r;
+  return (a, o) => new Promise((f, l) => {
+    var D;
+    if ((D = o == null ? void 0 : o.signal) != null && D.aborted) {
+      const g = new Error(`Relay aborted for action: ${t}`);
+      g.code = "E_ABORT", l(g);
+      return;
+    }
+    const d = ce(e);
+    if (!d) {
+      l(new Error(`No port available for action: ${t}`));
+      return;
+    }
+    if (y.size >= B) {
+      l(new Error(
+        `Too many pending calls (${B} limit exceeded). Action: ${t}`
+      ));
+      return;
+    }
+    const u = ae();
+    let O = !1;
+    const b = (g) => {
+      O || (O = !0, clearTimeout(v), o != null && o.signal && o.signal.removeEventListener("abort", M), y.delete(u), g());
+    }, W = () => {
+      j(u, e, d);
+    }, M = () => {
+      W();
+      const g = new Error(`Relay aborted for action: ${t}`);
+      g.code = "E_ABORT", b(() => l(g));
+    };
+    o != null && o.signal && o.signal.addEventListener("abort", M);
+    const v = setTimeout(() => {
+      W(), b(() => l(new Error(`Relay timeout for action: ${t}`)));
+    }, n);
+    y.set(u, {
+      settle: (g) => b(() => f(g)),
+      timeoutId: v,
+      abort: M,
+      owner: e,
+      port: d
+    });
+    const P = o == null ? void 0 : o.runId, N = o == null ? void 0 : o.callId;
+    R(d, {
+      type: e === "main-thread" || e === "content-script" ? "asyncRelay" : "registryCall",
+      id: u,
+      owner: e,
+      action: t,
+      params: a,
+      callId: N,
+      tabPolicy: s,
+      command: { action: t, params: a, runId: P, callId: N },
+      runId: P
+    }) || b(() => l(new Error(`Failed to post relay for action: ${t}`)));
   });
-}));
-const M = /* @__PURE__ */ new Map();
-function n(r, e) {
-  M.set(r, e);
 }
-function B(r) {
-  n("exists", (e) => r.fsExists(e)), n("stat", (e) => r.fsStat(e)), n("read", (e) => r.fsRead(e)), n("readText", (e) => r.fsReadText(e)), n("readBase64", (e) => r.fsReadBase64(e)), n("list", (e) => r.fsList(e)), n("mkdir", (e) => r.fsMkdir(e)), n("delete", (e) => r.fsDelete(e)), n("copy", (e) => r.fsCopy(e)), n("move", (e) => r.fsMove(e)), n("write", (e) => r.fsWrite(e)), n("writeText", (e) => r.fsWriteText(e)), n("writeBase64", (e) => r.fsWriteBase64(e)), n("append", (e) => r.fsAppend(e)), n("appendText", (e) => r.fsAppendText(e)), n("appendBase64", (e) => r.fsAppendBase64(e)), n("readRange", (e) => r.fsReadRange(e)), n("update", (e) => r.fsUpdate(e)), n("hash", (e) => r.fsHash(e));
+const de = "worker", ue = 3e4;
+function ge(r) {
+  if (r.owner === de) {
+    const e = A.get(r.action);
+    if (!e)
+      throw new Error(
+        `No worker-local handler registered for action: ${r.action}`
+      );
+    return async (t, n) => {
+      var a;
+      t = k(t);
+      const s = (n == null ? void 0 : n.signal) ?? (n != null && n.runId ? (a = p.get(n.runId)) == null ? void 0 : a.signal : void 0);
+      try {
+        return { ok: !0, value: await e(t, { ...n, action: r.action, signal: s }) };
+      } catch (o) {
+        const f = o instanceof Error ? o.message : String(o), l = typeof o == "object" && o !== null && "code" in o && typeof o.code == "string" ? o.code : r.errorCode;
+        return { ok: !1, error: { message: f, code: l } };
+      }
+    };
+  } else {
+    const e = fe({
+      owner: r.owner,
+      action: r.action,
+      timeoutMs: ue
+    });
+    return (t, n) => {
+      var s;
+      return e(k(t), {
+        ...n,
+        signal: (n == null ? void 0 : n.signal) ?? (n != null && n.runId ? (s = p.get(n.runId)) == null ? void 0 : s.signal : void 0)
+      });
+    };
+  }
+}
+function pe(r) {
+  i("exists", (e) => r.fsExists(e)), i("stat", (e) => r.fsStat(e)), i("read", (e) => r.fsRead(e)), i("readText", (e) => r.fsReadText(e)), i("readBase64", (e) => r.fsReadBase64(e)), i("list", (e) => r.fsList(e)), i("mkdir", (e) => r.fsMkdir(e)), i("delete", (e) => r.fsDelete(e)), i("copy", (e) => r.fsCopy(e)), i("move", (e) => r.fsMove(e)), i("write", (e) => r.fsWrite(e)), i("writeText", (e) => r.fsWriteText(e)), i("writeBase64", (e) => r.fsWriteBase64(e)), i("append", (e) => r.fsAppend(e)), i("appendText", (e) => r.fsAppendText(e)), i("appendBase64", (e) => r.fsAppendBase64(e)), i("readRange", (e) => r.fsReadRange(e)), i("update", (e) => r.fsUpdate(e)), i("hash", (e) => r.fsHash(e));
 }
-async function C() {
-  b || (await R(), a = new _(), y(3), x(y), B(a), b = !0);
+async function me(r, e) {
+  if (T) return;
+  await z(), c = new G(), L(3), V(L), pe(c), se(r);
+  const n = r.map((a) => ({
+    entry: oe(a),
+    callback: ge(a)
+  }));
+  try {
+    H(n);
+  } catch (a) {
+    const o = a instanceof Error ? a.message : String(a);
+    throw new Error(`Registry registration failed: ${o}`);
+  }
+  const { freezeManifest: s } = await import("../../pkg/extension_js.js");
+  if (s(), c.injectRegistryBindings(), e) {
+    const a = JSON.stringify(e);
+    await c.runCellAsync(
+      `(function(){var r=globalThis.chrome&&globalThis.chrome.runtime;if(!r){r={};if(!globalThis.chrome)globalThis.chrome={};globalThis.chrome.runtime=r;}r.id=${a};})();`,
+      "",
+      "inject-runtime-id"
+    );
+  }
+  T = !0;
 }
-C().then(() => {
-  self.postMessage({ type: "ready" });
-}).catch((r) => {
-  const e = r instanceof Error ? r.message : String(r);
-  self.postMessage({ type: "error", error: `WASM init failed: ${e}` });
-});
 self.onmessage = async (r) => {
   const e = r.data;
+  if (e.type === "asyncRelayResult") {
+    $(e.id, e.result);
+    return;
+  }
+  if (e.type === "registerWorkerPort") {
+    const t = r.ports[0];
+    if (!t) {
+      m.error("register_worker_port_missing", { owner: e.owner });
+      return;
+    }
+    le(e.owner, t);
+    return;
+  }
+  if (e.type === "init") {
+    try {
+      await me(e.manifest, e.extensionId), self.postMessage({ type: "ready" });
+    } catch (t) {
+      const n = t instanceof Error ? t.message : String(t);
+      m.error("worker_init_failed", { error: n }), self.postMessage({ type: "error", error: `WASM init failed: ${n}` });
+    }
+    return;
+  }
   if (e.type === "setLogLevel") {
-    y(e.level), E(j(e.level)), f.debug("set_log_level", { level: e.level });
+    L(e.level), J(ie(e.level)), m.debug("set_log_level", { level: e.level });
     return;
   }
-  if (!b || !a) {
+  if (!T || !c) {
     self.postMessage({
       type: "error",
       id: e.id,
@@ -179,84 +463,85 @@ self.onmessage = async (r) => {
   }
   switch (e.type) {
     case "runCell": {
-      i = e.runId;
+      const t = e.runId, n = new AbortController();
+      t && p.set(t, n);
       try {
-        const t = await a.runCellAsync(
+        const s = await c.runCellAsync(
           e.code,
           e.stdin || "",
-          e.runId || ""
+          t || ""
           // propagate correlation ID to WASM so trace spans can be linked end-to-end
         );
         self.postMessage({
           type: "result",
           id: e.id,
-          data: t,
-          runId: i
+          data: s,
+          runId: t
         });
-      } catch (t) {
-        const s = t instanceof Error ? t.message : String(t);
-        f.error("runCell_error", { runId: i, error: s }), self.postMessage({
+      } catch (s) {
+        const a = s instanceof Error ? s.message : String(s);
+        m.error("runCell_error", { runId: t, error: a }), self.postMessage({
           type: "error",
           id: e.id,
-          error: s,
-          runId: i
+          error: a,
+          runId: t
         });
       } finally {
-        i = void 0;
+        t && p.delete(t);
       }
       break;
     }
     case "reset": {
+      c.setAborted(!0);
+      for (const t of p.values()) t.abort();
+      p.clear(), F("E_RESET", "Worker reset");
       try {
-        a.reset(), self.postMessage({ type: "result", id: e.id, data: { ok: !0 } });
+        c.reset(), self.postMessage({ type: "result", id: e.id, data: { ok: !0 } });
       } catch (t) {
-        const s = t instanceof Error ? t.message : String(t);
-        self.postMessage({ type: "error", id: e.id, error: s });
+        const n = t instanceof Error ? t.message : String(t);
+        self.postMessage({ type: "error", id: e.id, error: n });
       }
       break;
     }
     case "stop": {
-      for (const [t, s] of d)
-        s({
-          ok: !1,
-          error: { message: "Worker stopped", code: "E_STOPPED" }
-        }), d.delete(t);
-      self.postMessage({ type: "result", id: e.id, data: { ok: !0 } });
+      c.setAborted(!0);
+      for (const t of p.values()) t.abort();
+      p.clear(), F("E_STOPPED", "Worker stopped"), self.postMessage({ type: "result", id: e.id, data: { ok: !0 } });
       break;
     }
     case "setFuelLimit": {
       try {
-        a.set_fuel_limit(e.limit), e.id && self.postMessage({ type: "result", id: e.id, data: { ok: !0 } });
+        c.set_fuel_limit(e.limit), e.id && self.postMessage({ type: "result", id: e.id, data: { ok: !0 } });
       } catch (t) {
         if (e.id) {
-          const s = t instanceof Error ? t.message : String(t);
-          self.postMessage({ type: "error", id: e.id, error: s });
+          const n = t instanceof Error ? t.message : String(t);
+          self.postMessage({ type: "error", id: e.id, error: n });
         }
       }
       break;
     }
     case "inspectGlobals": {
       try {
-        const t = a.inspect_globals();
+        const t = c.inspect_globals();
         self.postMessage({ type: "result", id: e.id, data: t });
       } catch (t) {
-        const s = t instanceof Error ? t.message : String(t);
-        self.postMessage({ type: "error", id: e.id, error: s });
+        const n = t instanceof Error ? t.message : String(t);
+        self.postMessage({ type: "error", id: e.id, error: n });
       }
       break;
     }
     case "loadLibrary": {
       try {
-        const t = a.load_library(e.source);
+        const t = c.load_library(e.source);
         self.postMessage({ type: "result", id: e.id, data: t });
       } catch (t) {
-        const s = t instanceof Error ? t.message : String(t);
-        self.postMessage({ type: "error", id: e.id, error: s });
+        const n = t instanceof Error ? t.message : String(t);
+        self.postMessage({ type: "error", id: e.id, error: n });
       }
       break;
     }
     case "fsCall": {
-      const t = M.get(e.action);
+      const t = A.get(e.action);
       if (!t) {
         self.postMessage({
           type: "error",
@@ -266,21 +551,16 @@ self.onmessage = async (r) => {
         break;
       }
       try {
-        const s = await t(e.params);
-        self.postMessage({ type: "result", id: e.id, data: s });
-      } catch (s) {
-        const o = s instanceof Error ? s.message : String(s);
-        self.postMessage({ type: "error", id: e.id, error: o });
+        const n = await t(e.params);
+        self.postMessage({ type: "result", id: e.id, data: n });
+      } catch (n) {
+        const s = n instanceof Error ? n.message : String(n);
+        self.postMessage({ type: "error", id: e.id, error: s });
       }
       break;
     }
-    case "asyncRelayResult": {
-      f.debug("asyncRelayResult", {
-        id: e.id,
-        resultType: typeof e.result
-      });
-      const t = d.get(e.id);
-      t ? (d.delete(e.id), t(e.result)) : f.warn("asyncRelayResult_no_pending_relay", { id: e.id });
+    default: {
+      m.error("unhandled_worker_message", { type: e.type });
       break;
     }
   }