@zooid/transport-matrix 0.9.0 → 0.9.1

package/dist/index.js

@@ -255,6 +255,19 @@ var MatrixClient = class {
     if (!r.ok) throw new Error(`getJoinedMembers(${roomId}) failed: ${r.status}`);
     return await r.json();
   }
+  async sync(opts) {
+    const params = new URLSearchParams({
+      user_id: opts.asUserId,
+      timeout: String(opts.timeoutMs ?? 3e4)
+    });
+    if (opts.since) params.set("since", opts.since);
+    const url = `${this.homeserver}/_matrix/client/v3/sync?${params.toString()}`;
+    const r = await this.fetch(url, {
+      headers: { Authorization: `Bearer ${this.asToken}` }
+    });
+    if (!r.ok) throw new Error(`sync(${opts.asUserId}) failed: ${r.status}`);
+    return r.json();
+  }
   async fetchRoomName(roomId, asUserId) {
     const url = `${this.homeserver}/_matrix/client/v3/rooms/${encodeURIComponent(roomId)}/state/m.room.name/?user_id=${encodeURIComponent(asUserId)}`;
     const r = await this.fetch(url, {
@@ -284,6 +297,7 @@ var MatrixContextProvider = class {
   constructor(opts) {
     this.opts = opts;
   }
+  opts;
   async getRoomHistory(channelId, hopts) {
     const { chunk, end } = await this.opts.client.fetchRoomMessages({
       roomId: channelId,
@@ -443,6 +457,29 @@ function renderRegistration(c) {
   );
 }
 
+// src/identity.ts
+var SLUG_RE = /^[a-z0-9-]+$/;
+var AGENT_KEY_RE = /^[a-z0-9-]+$/;
+var isValidWorkstation = (s) => SLUG_RE.test(s);
+var isValidAgentKey = (s) => AGENT_KEY_RE.test(s);
+function agentMxid(workstation, agent, serverName) {
+  if (!isValidWorkstation(workstation))
+    throw new Error(`invalid workstation: ${JSON.stringify(workstation)}`);
+  if (!isValidAgentKey(agent)) throw new Error(`invalid agent key: ${JSON.stringify(agent)}`);
+  return `@${workstation}.${agent}:${serverName}`;
+}
+function splitAgentLocalpart(localpart2) {
+  const i = localpart2.indexOf(".");
+  if (i <= 0 || i === localpart2.length - 1)
+    throw new Error(`not a workstation.agent localpart: ${localpart2}`);
+  return { workstation: localpart2.slice(0, i), agent: localpart2.slice(i + 1) };
+}
+function workstationUserNamespace(workstation, serverName) {
+  if (!isValidWorkstation(workstation))
+    throw new Error(`invalid workstation: ${JSON.stringify(workstation)}`);
+  return `@${workstation}\\..*:${serverName}`;
+}
+
 // src/mentions.ts
 var MATRIX_TO_RE = /https:\/\/matrix\.to\/#\/(@[^"<>\s]+)/g;
 var RAW_USER_RE = /(@[A-Za-z0-9._\-=/+]+:[A-Za-z0-9.\-]+)/g;
@@ -573,6 +610,8 @@ var BotPool = class {
     this.client = client;
     this.agents = agents;
   }
+  client;
+  agents;
   async bootstrap(opts = {}) {
     const aliasToId = /* @__PURE__ */ new Map();
     const attachedToSpace = /* @__PURE__ */ new Set();
@@ -618,7 +657,7 @@ var BotPool = class {
               } else {
                 const colon = room.indexOf(":");
                 const aliasLocalpart = colon > 1 ? room.slice(1, colon) : room.slice(1);
-                const sender = opts.adminUserId ?? a.userId;
+                const sender = opts.asUserId ?? opts.adminUserId ?? a.userId;
                 const userPowerLevels = buildUserPowerLevels(
                   opts.asUserId,
                   opts.adminUserIds,
@@ -937,6 +976,44 @@ function writeAttachment(input) {
   return { hostPath, agentPath };
 }
 
+// src/sync-loop.ts
+var SyncLoop = class {
+  opts;
+  running = false;
+  constructor(opts) {
+    this.opts = opts;
+  }
+  async tick() {
+    const since = this.opts.loadSince();
+    const res = await this.opts.client.sync({
+      asUserId: this.opts.asUserId,
+      since,
+      timeoutMs: this.opts.timeoutMs ?? 3e4
+    });
+    for (const [roomId, roomState] of Object.entries(res.rooms?.join ?? {})) {
+      for (const baseEvt of roomState.timeline?.events ?? []) {
+        await this.opts.onEvent({ ...baseEvt, room_id: roomId });
+      }
+    }
+    this.opts.saveSince(res.next_batch);
+  }
+  async run() {
+    this.running = true;
+    while (this.running) {
+      try {
+        await this.tick();
+      } catch (err) {
+        if (!this.running) break;
+        console.warn(`[sync-loop] ${this.opts.asUserId} tick failed, retrying:`, err);
+        await new Promise((r) => setTimeout(r, this.opts.retryDelayMs ?? 5e3));
+      }
+    }
+  }
+  stop() {
+    this.running = false;
+  }
+};
+
 // src/transport.ts
 var STARTUP_GRACE_MS = 5e3;
 async function buildMediaBlocks(items, opts) {
@@ -1030,7 +1107,7 @@ function inboundThreadRoot2(evt) {
   return r?.rel_type === "m.thread" && r.event_id ? r.event_id : void 0;
 }
 function createMatrixTransport(opts) {
-  const { agents, approvals, client, bindings, hsToken, adminUserId, botUserId } = opts;
+  const { agents, approvals, client, bindings, hsToken, adminUserId, botUserId, mode = "appservice" } = opts;
   const drainQuietMs = opts.drainQuietMs ?? DRAIN_QUIET_MS;
   const drainMaxMs = opts.drainMaxMs ?? DRAIN_MAX_MS;
   const mediaClient = opts.media;
@@ -1047,7 +1124,7 @@ function createMatrixTransport(opts) {
   const bufferMessageIds = /* @__PURE__ */ new Map();
   const sendQueue = /* @__PURE__ */ new Map();
   const threadStates = /* @__PURE__ */ new Map();
-  const cutoffTs = Date.now() - STARTUP_GRACE_MS;
+  const cutoffTs = mode === "client" ? Number.NEGATIVE_INFINITY : Date.now() - STARTUP_GRACE_MS;
   const seenEventIds = /* @__PURE__ */ new Set();
   const flushedCounts = /* @__PURE__ */ new Map();
   const pendingCommands = /* @__PURE__ */ new Map();
@@ -1184,194 +1261,197 @@ function createMatrixTransport(opts) {
       content
     });
   });
-  const app = new Hono();
-  function authOk(authHeader) {
-    const h = authHeader ?? "";
-    if (!h.startsWith("Bearer ")) return false;
-    const got = h.slice(7);
-    if (got.length !== hsToken.length) return false;
-    return timingSafeEqual(Buffer.from(got), Buffer.from(hsToken));
-  }
-  app.put("/_matrix/app/v1/transactions/:txnId", async (c) => {
-    if (!authOk(c.req.header("authorization"))) {
-      return c.json({ errcode: "M_FORBIDDEN" }, 403);
-    }
-    const body = await c.req.json().catch(() => ({}));
-    for (const evt of body.events ?? []) {
-      if (evt.event_id) {
-        if (seenEventIds.has(evt.event_id)) {
-          continue;
-        }
-        seenEventIds.add(evt.event_id);
-        if (seenEventIds.size > SEEN_EVENT_CAP) {
-          const first = seenEventIds.values().next().value;
-          if (first !== void 0) seenEventIds.delete(first);
-        }
+  async function handleInboundEvent(evt) {
+    if (evt.event_id) {
+      if (seenEventIds.has(evt.event_id)) {
+        return;
+      }
+      seenEventIds.add(evt.event_id);
+      if (seenEventIds.size > SEEN_EVENT_CAP) {
+        const first = seenEventIds.values().next().value;
+        if (first !== void 0) seenEventIds.delete(first);
       }
-      if (evt.origin_server_ts !== void 0 && evt.origin_server_ts < cutoffTs && evt.type === "m.room.message") {
+    }
+    if (evt.origin_server_ts !== void 0 && evt.origin_server_ts < cutoffTs && evt.type === "m.room.message") {
+      console.log(
+        `[matrix] dropping stale message event ${evt.event_id} (ts=${evt.origin_server_ts}, daemon started at ${cutoffTs + STARTUP_GRACE_MS})`
+      );
+      return;
+    }
+    if (evt.type === "m.room.member" && evt.content?.membership === "invite") {
+      const target = evt.state_key;
+      const inviter = evt.sender;
+      if (target && evt.room_id && ourBotUserIds.has(target) && (!inviter || !ourBotUserIds.has(inviter))) {
         console.log(
-          `[matrix] dropping stale message event ${evt.event_id} (ts=${evt.origin_server_ts}, daemon started at ${cutoffTs + STARTUP_GRACE_MS})`
+          `[matrix] declining ad-hoc invite for ${target} in ${evt.room_id} from ${inviter ?? "unknown"}`
+        );
+        await client.leaveRoom(evt.room_id, target, { reason: DECLINE_REASON }).catch(
+          (err) => console.warn(`[matrix] leaveRoom(${evt.room_id}, ${target}) failed:`, err)
         );
-        continue;
       }
-      if (evt.type === "m.room.member" && evt.content?.membership === "invite") {
-        const target = evt.state_key;
-        const inviter = evt.sender;
-        if (target && evt.room_id && ourBotUserIds.has(target) && (!inviter || !ourBotUserIds.has(inviter))) {
-          console.log(
-            `[matrix] declining ad-hoc invite for ${target} in ${evt.room_id} from ${inviter ?? "unknown"}`
-          );
-          await client.leaveRoom(evt.room_id, target, { reason: DECLINE_REASON }).catch(
-            (err) => console.warn(`[matrix] leaveRoom(${evt.room_id}, ${target}) failed:`, err)
-          );
-        }
-        continue;
+      return;
+    }
+    if (evt.type === "dev.zooid.session_reset") {
+      const relates = evt.content?.["m.relates_to"];
+      const threadRoot = relates?.rel_type === "m.thread" && relates.event_id ? relates.event_id : void 0;
+      if (!threadRoot) {
+        console.log("[matrix] dropping dev.zooid.session_reset without thread relation");
+        return;
       }
-      if (evt.type === "dev.zooid.session_reset") {
-        const relates = evt.content?.["m.relates_to"];
-        const threadRoot = relates?.rel_type === "m.thread" && relates.event_id ? relates.event_id : void 0;
-        if (!threadRoot) {
-          console.log("[matrix] dropping dev.zooid.session_reset without thread relation");
-          continue;
-        }
-        console.log(`[matrix] inbound dev.zooid.session_reset in ${evt.room_id} thread=${threadRoot}`);
-        for (const a of bindings) {
-          agents.endSession(a.name, threadRoot);
-        }
-        continue;
+      console.log(`[matrix] inbound dev.zooid.session_reset in ${evt.room_id} thread=${threadRoot}`);
+      for (const a of bindings) {
+        agents.endSession(a.name, threadRoot);
       }
-      if (evt.type === "dev.zooid.interrupt") {
-        const content = evt.content ?? {};
-        const relates = evt.content?.["m.relates_to"];
-        const threadRoot = relates?.rel_type === "m.thread" && relates.event_id ? relates.event_id : void 0;
-        if (threadRoot) {
-          const targets = [];
-          for (const [sessionId, ctx2] of sessions) {
-            if (ctx2.threadRoot === threadRoot) {
-              targets.push({ sessionId, agent: ctx2.agent.name });
-            }
-          }
-          for (const t of targets) {
-            console.log(
-              `[matrix] interrupt session=${t.sessionId} agent=${t.agent} thread=${threadRoot}` + (content.reason ? ` reason=${content.reason}` : "")
-            );
-            await agents.cancelSession(t.agent, t.sessionId).catch((err) => {
-              console.error(`[matrix] cancelSession(${t.agent}, ${t.sessionId}) failed:`, err);
-            });
+      return;
+    }
+    if (evt.type === "dev.zooid.interrupt") {
+      const content = evt.content ?? {};
+      const relates = evt.content?.["m.relates_to"];
+      const threadRoot = relates?.rel_type === "m.thread" && relates.event_id ? relates.event_id : void 0;
+      if (threadRoot) {
+        const targets = [];
+        for (const [sessionId, ctx2] of sessions) {
+          if (ctx2.threadRoot === threadRoot) {
+            targets.push({ sessionId, agent: ctx2.agent.name });
           }
-          continue;
         }
-        if (!content.session_id) {
-          console.warn(`[matrix] dev.zooid.interrupt missing session_id (event_id=${evt.event_id})`);
-          continue;
-        }
-        const ctx = sessions.get(content.session_id);
-        if (!ctx) {
-          continue;
+        for (const t of targets) {
+          console.log(
+            `[matrix] interrupt session=${t.sessionId} agent=${t.agent} thread=${threadRoot}` + (content.reason ? ` reason=${content.reason}` : "")
+          );
+          await agents.cancelSession(t.agent, t.sessionId).catch((err) => {
+            console.error(`[matrix] cancelSession(${t.agent}, ${t.sessionId}) failed:`, err);
+          });
         }
-        console.log(
-          `[matrix] interrupt session=${content.session_id} agent=${ctx.agent.name}` + (content.reason ? ` reason=${content.reason}` : "")
-        );
-        await agents.cancelSession(ctx.agent.name, content.session_id).catch((err) => {
-          console.error(`[matrix] cancelSession(${ctx.agent.name}, ${content.session_id}) failed:`, err);
-        });
-        continue;
+        return;
+      }
+      if (!content.session_id) {
+        console.warn(`[matrix] dev.zooid.interrupt missing session_id (event_id=${evt.event_id})`);
+        return;
       }
-      if (evt.type === "dev.zooid.approval_response") {
-        const content = evt.content ?? {};
-        if (!content.session_id || !content.approval_id || !content.decision) continue;
-        const decision = content.option_id ? { decision: content.decision, optionId: content.option_id } : { decision: content.decision };
-        const ok = approvals.resolve(
-          content.session_id,
-          content.approval_id,
-          decision
+      const ctx = sessions.get(content.session_id);
+      if (!ctx) {
+        return;
+      }
+      console.log(
+        `[matrix] interrupt session=${content.session_id} agent=${ctx.agent.name}` + (content.reason ? ` reason=${content.reason}` : "")
+      );
+      await agents.cancelSession(ctx.agent.name, content.session_id).catch((err) => {
+        console.error(`[matrix] cancelSession(${ctx.agent.name}, ${content.session_id}) failed:`, err);
+      });
+      return;
+    }
+    if (evt.type === "dev.zooid.approval_response") {
+      const content = evt.content ?? {};
+      if (!content.session_id || !content.approval_id || !content.decision) return;
+      const decision = content.option_id ? { decision: content.decision, optionId: content.option_id } : { decision: content.decision };
+      const ok = approvals.resolve(
+        content.session_id,
+        content.approval_id,
+        decision
+      );
+      if (!ok) console.warn(`[matrix] unknown approval ${content.approval_id}`);
+      return;
+    }
+    logInbound(evt);
+    if (evt.type === "m.room.message" && isMediaMsgtype(evt.content?.msgtype) && evt.room_id && evt.event_id && evt.sender && evt.content?.url && !bindings.some((b) => b.userId === evt.sender)) {
+      pendingMedia.add(evt.room_id, inboundThreadRoot2(evt), {
+        eventId: evt.event_id,
+        sender: evt.sender,
+        msgtype: evt.content.msgtype,
+        body: evt.content.body ?? "",
+        filename: evt.content.filename,
+        url: evt.content.url,
+        info: evt.content.info
+      });
+      return;
+    }
+    const promotedRoot = inboundThreadRoot2(evt) ?? evt.event_id;
+    const inboundRel = inboundThreadRoot2(evt);
+    if (evt.type === "m.room.message" && inboundRel && !threadStates.has(inboundRel) && evt.room_id) {
+      try {
+        const rebuilt = await rebuildThreadState(client, evt.room_id, inboundRel, bindings);
+        threadStates.set(inboundRel, rebuilt);
+        console.log(
+          `[matrix] rebuilt threadState for ${inboundRel}: participants=${rebuilt.participants.join(",")} rootMentions=${rebuilt.rootMentions.join(",")}`
         );
-        if (!ok) console.warn(`[matrix] unknown approval ${content.approval_id}`);
-        continue;
+      } catch (err) {
+        console.warn(`[matrix] failed to rebuild threadState for ${inboundRel}:`, err);
       }
-      logInbound(evt);
-      if (evt.type === "m.room.message" && isMediaMsgtype(evt.content?.msgtype) && evt.room_id && evt.event_id && evt.sender && evt.content?.url && !bindings.some((b) => b.userId === evt.sender)) {
-        pendingMedia.add(evt.room_id, inboundThreadRoot2(evt), {
-          eventId: evt.event_id,
-          sender: evt.sender,
-          msgtype: evt.content.msgtype,
-          body: evt.content.body ?? "",
-          filename: evt.content.filename,
-          url: evt.content.url,
-          info: evt.content.info
-        });
-        continue;
+    }
+    const matches = route(evt, bindings, threadStates);
+    const senderIsBot = bindings.some((b) => b.userId === evt.sender);
+    if (evt.type === "m.room.message" && matches.length === 0 && !senderIsBot) {
+      console.warn(
+        `[matrix] no agent matched message in ${evt.room_id} from ${evt.sender} (bindings: ${bindings.map((b) => `${b.name}@${b.userId}[${b.trigger}]`).join(", ")})`
+      );
+    }
+    if (matches.length > 0 && promotedRoot) {
+      let st = threadStates.get(promotedRoot);
+      if (!st) {
+        st = { participants: [], rootMentions: [] };
+        threadStates.set(promotedRoot, st);
       }
-      const promotedRoot = inboundThreadRoot2(evt) ?? evt.event_id;
-      const inboundRel = inboundThreadRoot2(evt);
-      if (evt.type === "m.room.message" && inboundRel && !threadStates.has(inboundRel) && evt.room_id) {
-        try {
-          const rebuilt = await rebuildThreadState(client, evt.room_id, inboundRel, bindings);
-          threadStates.set(inboundRel, rebuilt);
-          console.log(
-            `[matrix] rebuilt threadState for ${inboundRel}: participants=${rebuilt.participants.join(",")} rootMentions=${rebuilt.rootMentions.join(",")}`
-          );
-        } catch (err) {
-          console.warn(`[matrix] failed to rebuild threadState for ${inboundRel}:`, err);
+      const msgMentions = new Set(extractMentions(evt));
+      for (const a of bindings) {
+        if (msgMentions.has(a.userId) && !st.rootMentions.includes(a.name)) {
+          st.rootMentions.push(a.name);
         }
       }
-      const matches = route(evt, bindings, threadStates);
-      const senderIsBot = bindings.some((b) => b.userId === evt.sender);
-      if (evt.type === "m.room.message" && matches.length === 0 && !senderIsBot) {
-        console.warn(
-          `[matrix] no agent matched message in ${evt.room_id} from ${evt.sender} (bindings: ${bindings.map((b) => `${b.name}@${b.userId}[${b.trigger}]`).join(", ")})`
-        );
-      }
-      if (matches.length > 0 && promotedRoot) {
+    }
+    for (const a of matches) {
+      console.log(`[matrix] \u2192 ${a.name} (${a.userId})`);
+      void runTurn(a, evt).then(() => {
+        if (!promotedRoot) return;
         let st = threadStates.get(promotedRoot);
         if (!st) {
           st = { participants: [], rootMentions: [] };
           threadStates.set(promotedRoot, st);
         }
-        const msgMentions = new Set(extractMentions(evt));
-        for (const a of bindings) {
-          if (msgMentions.has(a.userId) && !st.rootMentions.includes(a.name)) {
-            st.rootMentions.push(a.name);
-          }
-        }
-      }
-      for (const a of matches) {
-        console.log(`[matrix] \u2192 ${a.name} (${a.userId})`);
-        void runTurn(a, evt).then(() => {
-          if (!promotedRoot) return;
-          let st = threadStates.get(promotedRoot);
-          if (!st) {
-            st = { participants: [], rootMentions: [] };
-            threadStates.set(promotedRoot, st);
-          }
-          if (st.participants.at(-1) !== a.name) st.participants.push(a.name);
-        }).catch((err) => {
-          console.error(`[matrix] runTurn failed for ${a.name}:`, err);
-          const c2 = classify(err);
-          const threadRoot = inboundThreadRoot2(evt) ?? evt.event_id;
-          if (!threadRoot || !evt.room_id) return;
-          const body2 = toErrorBody(
-            {
-              kind: "error",
-              agentId: a.name,
-              sessionId: null,
-              turnId: null,
-              code: c2.code,
-              message: err instanceof Error ? err.message : String(err),
-              detail: err instanceof Error && err.stack ? err.stack.slice(0, 2e3) : void 0,
-              transient: c2.transient,
-              acp_error: c2.acp_error
-            },
-            threadRoot
-          );
-          void client.sendCustomEvent({
-            roomId: evt.room_id,
-            asUserId: a.userId,
-            eventType: "dev.zooid.error",
-            content: body2
-          }).catch((e) => console.warn(`[matrix:${a.name}] dev.zooid.error send failed:`, e));
-        });
-      }
+        if (st.participants.at(-1) !== a.name) st.participants.push(a.name);
+      }).catch((err) => {
+        console.error(`[matrix] runTurn failed for ${a.name}:`, err);
+        const c = classify(err);
+        const threadRoot = inboundThreadRoot2(evt) ?? evt.event_id;
+        if (!threadRoot || !evt.room_id) return;
+        const body = toErrorBody(
+          {
+            kind: "error",
+            agentId: a.name,
+            sessionId: null,
+            turnId: null,
+            code: c.code,
+            message: err instanceof Error ? err.message : String(err),
+            detail: err instanceof Error && err.stack ? err.stack.slice(0, 2e3) : void 0,
+            transient: c.transient,
+            acp_error: c.acp_error
+          },
+          threadRoot
+        );
+        void client.sendCustomEvent({
+          roomId: evt.room_id,
+          asUserId: a.userId,
+          eventType: "dev.zooid.error",
+          content: body
+        }).catch((e) => console.warn(`[matrix:${a.name}] dev.zooid.error send failed:`, e));
+      });
+    }
+  }
+  const app = new Hono();
+  function authOk(authHeader) {
+    const h = authHeader ?? "";
+    if (!h.startsWith("Bearer ")) return false;
+    const got = h.slice(7);
+    if (got.length !== hsToken.length) return false;
+    return timingSafeEqual(Buffer.from(got), Buffer.from(hsToken));
+  }
+  app.put("/_matrix/app/v1/transactions/:txnId", async (c) => {
+    if (!authOk(c.req.header("authorization"))) {
+      return c.json({ errcode: "M_FORBIDDEN" }, 403);
+    }
+    const body = await c.req.json().catch(() => ({}));
+    for (const evt of body.events ?? []) {
+      await handleInboundEvent(evt);
     }
     return c.json({});
   });
@@ -1475,8 +1555,18 @@ function createMatrixTransport(opts) {
       sendQueue.delete(sessionId);
     }
   }
+  const syncLoops = mode === "client" ? bindings.map(
+    (b) => new SyncLoop({
+      client,
+      asUserId: b.userId,
+      loadSince: () => opts.loadSince?.(b.userId) ?? null,
+      saveSince: (since) => opts.saveSince?.(b.userId, since),
+      onEvent: (evt) => handleInboundEvent(evt)
+    })
+  ) : void 0;
   return {
     app,
+    syncLoops,
     bootstrap: async (bootstrapOpts = {}) => {
       await pool.bootstrap({ adminUserId, ...bootstrapOpts });
       await Promise.allSettled(
@@ -1574,6 +1664,7 @@ async function startWorkforcePublisher(opts) {
   };
 }
 export {
+  AGENT_KEY_RE,
   BotPool,
   INLINE_IMAGE_MIMES,
   MAX_DOWNLOAD_BYTES,
@@ -1584,18 +1675,25 @@ export {
   MatrixContextProvider,
   MediaClient,
   PendingMediaStore,
+  SLUG_RE,
+  SyncLoop,
+  agentMxid,
   buildWorkforceRoster,
   createMatrixTransport,
   ensureDefaultChannel,
   ensureWorkforceSpace,
   extractMentions,
   isMediaMsgtype,
+  isValidAgentKey,
+  isValidWorkstation,
   parseMxcUri,
   publishWorkforce,
   renderRegistration,
   route,
   serverNameFromMxid,
+  splitAgentLocalpart,
   startWorkforcePublisher,
+  workstationUserNamespace,
   writeAttachment
 };
 //# sourceMappingURL=index.js.map