@seqyuan/annovibe 0.8.65 → 0.8.66

package/bin/annovibe-im-gateway.js

@@ -86,27 +86,41 @@ function getBaseUrl() {
   return `http://127.0.0.1:${getAnnovibePort()}`;
 }
 
-function readGatewayToken() {
-  const statePath = path.join(getAgentDir(), "im", "gateway.json");
+function getGatewayStatePath() {
+  return path.join(getAgentDir(), "im", "gateway.json");
+}
+
+function readGatewayState() {
   try {
-    const state = JSON.parse(fs.readFileSync(statePath, "utf8"));
-    return typeof state.token === "string" ? state.token.trim() : "";
+    return JSON.parse(fs.readFileSync(getGatewayStatePath(), "utf8"));
   } catch {
-    return "";
+    return {};
   }
 }
 
+function readGatewayToken() {
+  const state = readGatewayState();
+  return typeof state.token === "string" ? state.token.trim() : "";
+}
+
+function writeGatewayState(patch) {
+  const statePath = getGatewayStatePath();
+  try { fs.mkdirSync(path.dirname(statePath), { recursive: true }); } catch {}
+  const state = readGatewayState();
+  fs.writeFileSync(statePath, JSON.stringify({ ...state, ...patch }, null, 2), "utf8");
+}
+
 // ── HTTP helpers ────────────────────────────────────────────────────
 
 function getToken() {
   return process.env.ANNOVIBE_GATEWAY_TOKEN?.trim() || readGatewayToken();
 }
 
-function requestJson(method, pathname, body) {
+function requestJsonRaw(method, pathname, body, options = {}) {
   const url = new URL(pathname, getBaseUrl());
   const payload = body === undefined ? undefined : JSON.stringify(body);
   const lib = url.protocol === "https:" ? https : http;
-  const token = getToken();
+  const token = options.skipAuth ? "" : getToken();
 
   return new Promise((resolve, reject) => {
     const req = lib.request(
@@ -143,6 +157,10 @@ function requestJson(method, pathname, body) {
   });
 }
 
+function requestJson(method, pathname, body) {
+  return requestJsonRaw(method, pathname, body);
+}
+
 function formatImApiError(statusCode, message) {
   if (statusCode === 401) {
     return `${message}. Ensure annovibe is running in this Linux account and gateway token is synced (restart annovibe-im-gateway after saving IM settings).`;
@@ -153,6 +171,18 @@ function formatImApiError(statusCode, message) {
   return message;
 }
 
+async function syncGatewayTokenFromServer() {
+  try {
+    const payload = await requestJsonRaw("GET", "/api/im/gateway-token", undefined, { skipAuth: true });
+    const serverToken = typeof payload.token === "string" ? payload.token.trim() : "";
+    if (!serverToken || readGatewayToken() === serverToken) return;
+    writeGatewayState({ token: serverToken });
+    console.error("[im] synced gateway token from annovibe server");
+  } catch (error) {
+    console.error(`[im] could not sync gateway token: ${String(error)}`);
+  }
+}
+
 /**
  * Consume SSE turn stream, yielding deltas and status changes to callbacks.
  */
@@ -546,8 +576,42 @@ class ProjectWeComBridge {
     };
 
     try {
+      let lastPushAt = 0;
+      let pendingText = null;
+      let pushTimer = null;
+      let pushChain = Promise.resolve();
+      const pushIntervalMs = 300;
+
+      const pushImmediate = async (text) => {
+        pushChain = pushChain.then(() => pushText(text));
+        await pushChain;
+        lastPushAt = Date.now();
+      };
+
+      const schedulePush = (text) => {
+        pendingText = text;
+        const now = Date.now();
+        if (now - lastPushAt >= pushIntervalMs) {
+          void flushPush();
+          return;
+        }
+        if (pushTimer) return;
+        pushTimer = setTimeout(() => { void flushPush(); }, pushIntervalMs - (now - lastPushAt));
+      };
+
+      const flushPush = async () => {
+        if (pushTimer) {
+          clearTimeout(pushTimer);
+          pushTimer = null;
+        }
+        const text = pendingText;
+        if (text === null) return;
+        pendingText = null;
+        await pushImmediate(text);
+      };
+
       // Start processing
-      await pushText("(working 1s)");
+      await pushImmediate("(working 1s)");
 
       refreshTimer = setInterval(() => {
         if (toolsRunning) {
@@ -565,8 +629,7 @@ class ProjectWeComBridge {
           onDelta(text) {
             toolsRunning = false;
             lastText = text;
-            // Schedule push — debounced by WeCom rate limits via replyStream
-            void pushText(text).catch(() => undefined);
+            schedulePush(text);
           },
         },
       );
@@ -578,6 +641,8 @@ class ProjectWeComBridge {
         : "No reply from annovibe.";
       const cleaned = cleanImReplyText(lastText || rawReply);
       const finishTextStr = cleaned || "No reply from annovibe.";
+      if (pushTimer) clearTimeout(pushTimer);
+      pendingText = null;
       await finishText(finishTextStr);
     } catch (error) {
       if (refreshTimer) clearInterval(refreshTimer);
@@ -597,21 +662,14 @@ class ImGateway {
     console.error("[im] Annovibe IM Gateway starting...");
 
     // Write gateway state
-    const imDir = path.join(getAgentDir(), "im");
-    try { fs.mkdirSync(imDir, { recursive: true }); } catch {}
-    const token = crypto.randomBytes(32).toString("hex");
-    fs.writeFileSync(
-      path.join(imDir, "gateway.json"),
-      JSON.stringify({
-        token,
-        startedAt: new Date().toISOString(),
-        pid: process.pid,
-      }, null, 2),
-    );
+    const token = readGatewayToken() || crypto.randomBytes(32).toString("hex");
+    writeGatewayState({ token, startedAt: new Date().toISOString(), pid: process.pid });
 
     console.error(`[im] Gateway token: ${token.slice(0, 8)}...`);
     console.error(`[im] Annovibe port: ${getAnnovibePort()}`);
 
+    await syncGatewayTokenFromServer();
+
     // Start config poll
     await this.refreshProjects();
     this.pollTimer = setInterval(() => this.refreshProjects(), 30000);
@@ -657,8 +715,7 @@ class ImGateway {
     this.bridges.clear();
 
     try {
-      const statePath = path.join(getAgentDir(), "im", "gateway.json");
-      fs.writeFileSync(statePath, JSON.stringify({ token: "" }, null, 2));
+      writeGatewayState({ token: "" });
     } catch {}
   }
 }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@seqyuan/annovibe",
-  "version": "0.8.65",
+  "version": "0.8.66",
   "description": "AI-native bioinformatics workspace by Annoroad",
   "license": "MIT",
   "bin": {