@memtensor/memos-local-openclaw-plugin 1.0.6-beta.1 → 1.0.6-beta.11

package/dist/viewer/server.js

@@ -108,26 +108,6 @@ function applyMigrationItemToState(state, d) {
     state.lastItem = d;
     state.success = computeMigrationSuccess(state);
 }
-/** Epoch ms for Chunk; OpenClaw SQLite may store Unix seconds or ms. */
-function normalizeTimestamp(value) {
-    if (value == null)
-        return Date.now();
-    if (typeof value === "string") {
-        const parsed = Date.parse(value.trim());
-        if (Number.isFinite(parsed))
-            return parsed;
-        const n = Number(value);
-        if (Number.isFinite(n))
-            return normalizeTimestamp(n);
-        return Date.now();
-    }
-    if (typeof value === "number" && Number.isFinite(value)) {
-        if (value > 0 && value < 5_000_000_000)
-            return Math.round(value * 1000);
-        return Math.round(value);
-    }
-    return Date.now();
-}
 class ViewerServer {
     server = null;
     store;
@@ -2090,18 +2070,24 @@ class ViewerServer {
     handleRetryJoin(req, res) {
         this.readBody(req, async (_body) => {
             if (!this.ctx)
-                return this.jsonResponse(res, { ok: false, error: "sharing_unavailable" });
+                return this.jsonResponse(res, { ok: false, error: "sharing_unavailable", errorCode: "sharing_unavailable" });
             const sharing = this.ctx.config.sharing;
             if (!sharing?.enabled || sharing.role !== "client") {
-                return this.jsonResponse(res, { ok: false, error: "not_in_client_mode" });
+                return this.jsonResponse(res, { ok: false, error: "not_in_client_mode", errorCode: "not_in_client_mode" });
             }
             const hubAddress = sharing.client?.hubAddress ?? "";
             const teamToken = sharing.client?.teamToken ?? "";
             if (!hubAddress || !teamToken) {
-                return this.jsonResponse(res, { ok: false, error: "missing_hub_address_or_team_token" });
+                return this.jsonResponse(res, { ok: false, error: "missing_hub_address_or_team_token", errorCode: "missing_config" });
+            }
+            const hubUrl = (0, hub_1.normalizeHubUrl)(hubAddress);
+            try {
+                await (0, hub_1.hubRequestJson)(hubUrl, "", "/api/v1/hub/info", { method: "GET" });
+            }
+            catch {
+                return this.jsonResponse(res, { ok: false, error: "hub_unreachable", errorCode: "hub_unreachable" });
             }
             try {
-                const hubUrl = (0, hub_1.normalizeHubUrl)(hubAddress);
                 const os = await Promise.resolve().then(() => __importStar(require("os")));
                 const nickname = sharing.client?.nickname;
                 const username = nickname || os.userInfo().username || "user";
@@ -2130,10 +2116,20 @@ class ViewerServer {
                     lastKnownStatus: result.status || "",
                     hubInstanceId,
                 });
+                if (result.status === "blocked") {
+                    return this.jsonResponse(res, { ok: false, error: "blocked", errorCode: "blocked" });
+                }
                 this.jsonResponse(res, { ok: true, status: result.status || "pending" });
             }
             catch (err) {
-                this.jsonResponse(res, { ok: false, error: String(err) });
+                const errStr = String(err);
+                if (errStr.includes("(409)") || errStr.includes("username_taken")) {
+                    return this.jsonResponse(res, { ok: false, error: "username_taken", errorCode: "username_taken" });
+                }
+                if (errStr.includes("(403)") || errStr.includes("invalid_team_token")) {
+                    return this.jsonResponse(res, { ok: false, error: "invalid_team_token", errorCode: "invalid_team_token" });
+                }
+                this.jsonResponse(res, { ok: false, error: errStr, errorCode: "unknown" });
             }
         });
     }
@@ -3118,24 +3114,24 @@ class ViewerServer {
                 const nowClient = Boolean(finalSharing?.enabled) && finalSharing?.role === "client";
                 const previouslyClient = oldSharingEnabled && oldSharingRole === "client";
                 let joinStatus;
+                let joinError;
                 if (nowClient && !previouslyClient) {
                     try {
                         joinStatus = await this.autoJoinOnSave(finalSharing);
                     }
                     catch (e) {
-                        this.log.warn(`Auto-join on save failed: ${e}`);
+                        const msg = String(e instanceof Error ? e.message : e);
+                        this.log.warn(`Auto-join on save failed: ${msg}`);
+                        if (msg === "hub_unreachable" || msg === "username_taken" || msg === "invalid_team_token") {
+                            joinError = msg;
+                        }
                     }
                 }
-                this.jsonResponse(res, { ok: true, joinStatus, restart: true });
-                setTimeout(() => {
-                    this.log.info("config-save: triggering gateway restart via SIGUSR1...");
-                    try {
-                        process.kill(process.pid, "SIGUSR1");
-                    }
-                    catch (sig) {
-                        this.log.warn(`SIGUSR1 failed: ${sig}`);
-                    }
-                }, 500);
+                if (joinError) {
+                    this.jsonResponse(res, { ok: true, joinError, restart: false });
+                    return;
+                }
+                this.jsonResponseAndRestart(res, { ok: true, joinStatus, restart: true }, "config-save");
             }
             catch (e) {
                 this.log.warn(`handleSaveConfig error: ${e}`);
@@ -3151,16 +3147,35 @@ class ViewerServer {
         if (!hubAddress || !teamToken)
             return undefined;
         const hubUrl = (0, hub_1.normalizeHubUrl)(hubAddress);
+        try {
+            await (0, hub_1.hubRequestJson)(hubUrl, "", "/api/v1/hub/info", { method: "GET" });
+        }
+        catch {
+            throw new Error("hub_unreachable");
+        }
         const os = await Promise.resolve().then(() => __importStar(require("os")));
         const nickname = String(clientCfg?.nickname || "");
         const username = nickname || os.userInfo().username || "user";
         const hostname = os.hostname() || "unknown";
         const persisted = this.store.getClientHubConnection();
         const existingIdentityKey = persisted?.identityKey || "";
-        const result = await (0, hub_1.hubRequestJson)(hubUrl, "", "/api/v1/hub/join", {
-            method: "POST",
-            body: JSON.stringify({ teamToken, username, deviceName: hostname, identityKey: existingIdentityKey }),
-        });
+        let result;
+        try {
+            result = await (0, hub_1.hubRequestJson)(hubUrl, "", "/api/v1/hub/join", {
+                method: "POST",
+                body: JSON.stringify({ teamToken, username, deviceName: hostname, identityKey: existingIdentityKey }),
+            });
+        }
+        catch (err) {
+            const errStr = String(err);
+            if (errStr.includes("(409)") || errStr.includes("username_taken")) {
+                throw new Error("username_taken");
+            }
+            if (errStr.includes("(403)") || errStr.includes("invalid_team_token")) {
+                throw new Error("invalid_team_token");
+            }
+            throw err;
+        }
         const returnedIdentityKey = String(result.identityKey || existingIdentityKey || "");
         let hubInstanceId = persisted?.hubInstanceId || "";
         try {
@@ -3206,16 +3221,7 @@ class ViewerServer {
                         this.log.info("handleLeaveTeam: config updated, sharing disabled");
                     }
                 }
-                this.jsonResponse(res, { ok: true, restart: true });
-                setTimeout(() => {
-                    this.log.info("handleLeaveTeam: triggering gateway restart via SIGUSR1...");
-                    try {
-                        process.kill(process.pid, "SIGUSR1");
-                    }
-                    catch (sig) {
-                        this.log.warn(`SIGUSR1 failed: ${sig}`);
-                    }
-                }, 500);
+                this.jsonResponseAndRestart(res, { ok: true, restart: true }, "handleLeaveTeam");
             }
             catch (e) {
                 this.log.warn(`handleLeaveTeam error: ${e}`);
@@ -3389,17 +3395,45 @@ class ViewerServer {
                     }
                 }
                 catch { }
-                const url = hubUrl.replace(/\/+$/, "") + "/api/v1/hub/info";
+                const baseUrl = hubUrl.replace(/\/+$/, "");
+                const infoUrl = baseUrl + "/api/v1/hub/info";
                 const ctrl = new AbortController();
                 const timeout = setTimeout(() => ctrl.abort(), 8000);
                 try {
-                    const r = await fetch(url, { signal: ctrl.signal });
+                    const r = await fetch(infoUrl, { signal: ctrl.signal });
                     clearTimeout(timeout);
                     if (!r.ok) {
                         this.jsonResponse(res, { ok: false, error: `HTTP ${r.status}` });
                         return;
                     }
                     const info = await r.json();
+                    const { teamToken, nickname } = JSON.parse(body);
+                    if (teamToken) {
+                        const username = (typeof nickname === "string" && nickname.trim()) || node_os_1.default.userInfo().username || "user";
+                        const persisted = this.store.getClientHubConnection();
+                        const identityKey = persisted?.identityKey || "";
+                        try {
+                            const joinR = await fetch(baseUrl + "/api/v1/hub/join", {
+                                method: "POST",
+                                headers: { "content-type": "application/json" },
+                                body: JSON.stringify({ teamToken, username, identityKey, deviceName: node_os_1.default.hostname(), dryRun: true }),
+                            });
+                            const joinData = await joinR.json();
+                            if (!joinR.ok && joinData.error === "username_taken") {
+                                this.jsonResponse(res, { ok: false, error: "username_taken", teamName: info.teamName || "" });
+                                return;
+                            }
+                            if (!joinR.ok && joinData.error === "invalid_team_token") {
+                                this.jsonResponse(res, { ok: false, error: "invalid_team_token", teamName: info.teamName || "" });
+                                return;
+                            }
+                            if (joinR.ok && joinData.status === "blocked") {
+                                this.jsonResponse(res, { ok: false, error: "blocked", teamName: info.teamName || "" });
+                                return;
+                            }
+                        }
+                        catch { /* join check is best-effort; connection itself is OK */ }
+                    }
                     this.jsonResponse(res, { ok: true, teamName: info.teamName || "", apiVersion: info.apiVersion || "" });
                 }
                 catch (e) {
@@ -3486,26 +3520,35 @@ class ViewerServer {
         return null;
     }
     async handleUpdateCheck(res) {
+        const sendNoStore = (data, statusCode = 200) => {
+            res.writeHead(statusCode, {
+                "Content-Type": "application/json; charset=utf-8",
+                "Cache-Control": "no-store, no-cache, must-revalidate, max-age=0",
+                "Pragma": "no-cache",
+                "Expires": "0",
+            });
+            res.end(JSON.stringify(data));
+        };
         try {
             const pkgPath = this.findPluginPackageJson();
             if (!pkgPath) {
-                this.jsonResponse(res, { updateAvailable: false, error: "package.json not found" });
+                sendNoStore({ updateAvailable: false, error: "package.json not found" });
                 return;
             }
             const pkg = JSON.parse(node_fs_1.default.readFileSync(pkgPath, "utf-8"));
             const current = pkg.version;
             const name = pkg.name;
             if (!current || !name) {
-                this.jsonResponse(res, { updateAvailable: false, current });
+                sendNoStore({ updateAvailable: false, current });
                 return;
             }
             const { computeUpdateCheck } = await Promise.resolve().then(() => __importStar(require("../update-check")));
             const result = await computeUpdateCheck(name, current, fetch, 6_000);
             if (!result) {
-                this.jsonResponse(res, { updateAvailable: false, current, packageName: name });
+                sendNoStore({ updateAvailable: false, current, packageName: name });
                 return;
             }
-            this.jsonResponse(res, {
+            sendNoStore({
                 updateAvailable: result.updateAvailable,
                 current: result.current,
                 latest: result.latest,
@@ -3517,7 +3560,7 @@ class ViewerServer {
         }
         catch (e) {
             this.log.warn(`handleUpdateCheck error: ${e}`);
-            this.jsonResponse(res, { updateAvailable: false, error: String(e) });
+            sendNoStore({ updateAvailable: false, error: String(e) });
         }
     }
     handleUpdateInstall(req, res) {
@@ -3525,13 +3568,14 @@ class ViewerServer {
         req.on("data", (chunk) => { body += chunk.toString(); });
         req.on("end", () => {
             try {
-                const { packageSpec: rawSpec } = JSON.parse(body);
+                const { packageSpec: rawSpec, targetVersion: rawTargetVersion } = JSON.parse(body);
                 if (!rawSpec || typeof rawSpec !== "string") {
                     res.writeHead(400, { "Content-Type": "application/json" });
                     res.end(JSON.stringify({ ok: false, error: "Missing packageSpec" }));
                     return;
                 }
                 const packageSpec = rawSpec.trim().replace(/^(?:npx\s+)?openclaw\s+plugins\s+install\s+/i, "");
+                const targetVersion = typeof rawTargetVersion === "string" ? rawTargetVersion.trim() : "";
                 const allowed = /^@[\w-]+\/[\w.-]+(@[\w.-]+)?$/;
                 this.log.info(`update-install: received packageSpec="${packageSpec}" (len=${packageSpec.length})`);
                 if (!allowed.test(packageSpec)) {
@@ -3552,18 +3596,50 @@ class ViewerServer {
                 const shortName = pluginName?.replace(/^@[\w-]+\//, "") ?? "memos-local-openclaw-plugin";
                 const extDir = node_path_1.default.join(node_os_1.default.homedir(), ".openclaw", "extensions", shortName);
                 const tmpDir = node_path_1.default.join(node_os_1.default.tmpdir(), `openclaw-update-${Date.now()}`);
+                const backupDir = node_path_1.default.join(node_path_1.default.dirname(extDir), `${shortName}.backup-${Date.now()}`);
+                let backupReady = false;
+                const cleanupTmpDir = () => {
+                    try {
+                        node_fs_1.default.rmSync(tmpDir, { recursive: true, force: true });
+                    }
+                    catch { }
+                };
+                const rollbackInstall = () => {
+                    try {
+                        node_fs_1.default.rmSync(extDir, { recursive: true, force: true });
+                    }
+                    catch { }
+                    if (!backupReady)
+                        return;
+                    try {
+                        node_fs_1.default.renameSync(backupDir, extDir);
+                        backupReady = false;
+                        this.log.info(`update-install: restored previous version from ${backupDir}`);
+                    }
+                    catch (restoreErr) {
+                        this.log.warn(`update-install: failed to restore previous version: ${restoreErr?.message ?? restoreErr}`);
+                    }
+                };
+                const discardBackup = () => {
+                    if (!backupReady)
+                        return;
+                    try {
+                        node_fs_1.default.rmSync(backupDir, { recursive: true, force: true });
+                        backupReady = false;
+                    }
+                    catch (cleanupErr) {
+                        this.log.warn(`update-install: failed to remove backup dir ${backupDir}: ${cleanupErr?.message ?? cleanupErr}`);
+                    }
+                };
                 // Download via npm pack, extract, and replace extension dir.
                 // Does NOT touch openclaw.json → no config watcher SIGUSR1.
                 this.log.info(`update-install: downloading ${packageSpec} via npm pack...`);
                 node_fs_1.default.mkdirSync(tmpDir, { recursive: true });
-                (0, node_child_process_1.exec)(`npm pack ${packageSpec} --pack-destination ${tmpDir}`, { timeout: 60_000 }, (packErr, packOut) => {
+                (0, node_child_process_1.exec)(`npm pack ${packageSpec} --pack-destination ${tmpDir} --prefer-online`, { timeout: 60_000 }, (packErr, packOut) => {
                     if (packErr) {
                         this.log.warn(`update-install: npm pack failed: ${packErr.message}`);
                         this.jsonResponse(res, { ok: false, error: `Download failed: ${packErr.message}` });
-                        try {
-                            node_fs_1.default.rmSync(tmpDir, { recursive: true, force: true });
-                        }
-                        catch { }
+                        cleanupTmpDir();
                         return;
                     }
                     const tgzFile = packOut.trim().split("\n").pop();
@@ -3575,40 +3651,45 @@ class ViewerServer {
                         if (tarErr) {
                             this.log.warn(`update-install: tar extract failed: ${tarErr.message}`);
                             this.jsonResponse(res, { ok: false, error: `Extract failed: ${tarErr.message}` });
-                            try {
-                                node_fs_1.default.rmSync(tmpDir, { recursive: true, force: true });
-                            }
-                            catch { }
+                            cleanupTmpDir();
                             return;
                         }
                         // npm pack extracts to a "package" subdirectory
                         const srcDir = node_path_1.default.join(extractDir, "package");
                         if (!node_fs_1.default.existsSync(srcDir)) {
                             this.jsonResponse(res, { ok: false, error: "Extracted package has no 'package' dir" });
-                            try {
-                                node_fs_1.default.rmSync(tmpDir, { recursive: true, force: true });
-                            }
-                            catch { }
+                            cleanupTmpDir();
                             return;
                         }
                         // Replace extension directory
                         this.log.info(`update-install: replacing ${extDir}...`);
                         try {
-                            node_fs_1.default.rmSync(extDir, { recursive: true, force: true });
+                            node_fs_1.default.mkdirSync(node_path_1.default.dirname(extDir), { recursive: true });
+                            try {
+                                node_fs_1.default.rmSync(backupDir, { recursive: true, force: true });
+                            }
+                            catch { }
+                            if (node_fs_1.default.existsSync(extDir)) {
+                                node_fs_1.default.renameSync(extDir, backupDir);
+                                backupReady = true;
+                            }
+                            node_fs_1.default.renameSync(srcDir, extDir);
+                        }
+                        catch (replaceErr) {
+                            this.log.warn(`update-install: replace failed: ${replaceErr?.message ?? replaceErr}`);
+                            cleanupTmpDir();
+                            rollbackInstall();
+                            this.jsonResponse(res, { ok: false, error: `Replace failed: ${replaceErr?.message ?? replaceErr}` });
+                            return;
                         }
-                        catch { }
-                        node_fs_1.default.mkdirSync(node_path_1.default.dirname(extDir), { recursive: true });
-                        node_fs_1.default.renameSync(srcDir, extDir);
                         // Install dependencies
                         this.log.info(`update-install: installing dependencies...`);
                         const npmCmd = process.platform === "win32" ? "npm.cmd" : "npm";
                         (0, node_child_process_1.execFile)(npmCmd, ["install", "--omit=dev", "--ignore-scripts"], { cwd: extDir, timeout: 120_000 }, (npmErr, npmOut, npmStderr) => {
                             if (npmErr) {
-                                try {
-                                    node_fs_1.default.rmSync(tmpDir, { recursive: true, force: true });
-                                }
-                                catch { }
                                 this.log.warn(`update-install: npm install failed: ${npmErr.message}`);
+                                cleanupTmpDir();
+                                rollbackInstall();
                                 this.jsonResponse(res, { ok: false, error: `Dependency install failed: ${npmStderr || npmErr.message}` });
                                 return;
                             }
@@ -3621,34 +3702,34 @@ class ViewerServer {
                                 }
                                 this.log.info(`update-install: running postinstall...`);
                                 (0, node_child_process_1.execFile)(process.execPath, ["scripts/postinstall.cjs"], { cwd: extDir, timeout: 180_000 }, (postErr, postOut, postStderr) => {
-                                    try {
-                                        node_fs_1.default.rmSync(tmpDir, { recursive: true, force: true });
-                                    }
-                                    catch { }
+                                    cleanupTmpDir();
                                     if (postErr) {
                                         this.log.warn(`update-install: postinstall failed: ${postErr.message}`);
                                         const postStderrStr = String(postStderr || "").trim();
                                         if (postStderrStr)
                                             this.log.warn(`update-install: postinstall stderr: ${postStderrStr.slice(0, 500)}`);
+                                        rollbackInstall();
+                                        this.jsonResponse(res, { ok: false, error: `Postinstall failed: ${postStderrStr || postErr.message}` });
+                                        return;
                                     }
-                                    // Read new version
                                     let newVersion = "unknown";
                                     try {
                                         const newPkg = JSON.parse(node_fs_1.default.readFileSync(node_path_1.default.join(extDir, "package.json"), "utf-8"));
                                         newVersion = newPkg.version ?? newVersion;
                                     }
                                     catch { }
+                                    if (targetVersion && newVersion !== targetVersion) {
+                                        this.log.warn(`update-install: version mismatch! expected=${targetVersion}, got=${newVersion} — rolling back`);
+                                        rollbackInstall();
+                                        this.jsonResponse(res, {
+                                            ok: false,
+                                            error: `Version mismatch: expected ${targetVersion} but downloaded ${newVersion}. npm cache may be stale — please try again.`,
+                                        });
+                                        return;
+                                    }
+                                    discardBackup();
                                     this.log.info(`update-install: success! Updated to ${newVersion}`);
-                                    this.jsonResponse(res, { ok: true, version: newVersion });
-                                    setTimeout(() => {
-                                        this.log.info(`update-install: triggering gateway restart via SIGUSR1...`);
-                                        try {
-                                            process.kill(process.pid, "SIGUSR1");
-                                        }
-                                        catch (sig) {
-                                            this.log.warn(`SIGUSR1 failed: ${sig}`);
-                                        }
-                                    }, 500);
+                                    this.jsonResponseAndRestart(res, { ok: true, version: newVersion }, "update-install");
                                 });
                             });
                         });
@@ -4223,8 +4304,8 @@ class ViewerServer {
                                     mergeCount: 0,
                                     lastHitAt: null,
                                     mergeHistory: "[]",
-                                    createdAt: normalizeTimestamp(row.updated_at),
-                                    updatedAt: normalizeTimestamp(row.updated_at),
+                                    createdAt: Number(row.updated_at) < 1e12 ? Number(row.updated_at) * 1000 : Number(row.updated_at),
+                                    updatedAt: Number(row.updated_at) < 1e12 ? Number(row.updated_at) * 1000 : Number(row.updated_at),
                                 };
                                 this.store.insertChunk(chunk);
                                 if (embedding && dedupStatus === "active") {
@@ -4781,6 +4862,20 @@ class ViewerServer {
         req.on("data", (chunk) => { body += chunk.toString(); });
         req.on("end", () => cb(body));
     }
+    jsonResponseAndRestart(res, data, source, delayMs = 1500, statusCode = 200) {
+        res.writeHead(statusCode, { "Content-Type": "application/json; charset=utf-8" });
+        res.end(JSON.stringify(data), () => {
+            setTimeout(() => {
+                this.log.info(`${source}: triggering gateway restart via SIGUSR1...`);
+                try {
+                    process.kill(process.pid, "SIGUSR1");
+                }
+                catch (sig) {
+                    this.log.warn(`SIGUSR1 failed: ${sig}`);
+                }
+            }, delayMs);
+        });
+    }
     jsonResponse(res, data, statusCode = 200) {
         res.writeHead(statusCode, { "Content-Type": "application/json; charset=utf-8" });
         res.end(JSON.stringify(data));