@mobcode/openclaw-plugin 0.1.15 → 0.1.16

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@mobcode/openclaw-plugin",
-  "version": "0.1.15",
+  "version": "0.1.16",
   "description": "MobCode integration plugin for OpenClaw",
   "type": "module",
   "license": "MIT",

package/src/plugin-definition.js

@@ -67,6 +67,16 @@ export function createMobcodePluginDefinition() {
         api,
         store,
       });
+      api.on("tool_result_persist", (event, ctx) => {
+        store.appendMessageSync({
+          sessionKey: ctx?.sessionKey,
+          message: event?.message,
+          messageId:
+            typeof event?.message?.id === "string" && event.message.id.trim()
+              ? event.message.id.trim()
+              : undefined,
+        });
+      });
     },
   };
 }

package/src/state-store.js

@@ -996,10 +996,129 @@ export class MobcodeStateStore {
     this.logger?.warn?.(
       `[mobcode-debug] appendMessage sessionKey=${sessionKey} messageId=${String(update?.messageId ?? normalizedMessage?.id ?? "").trim() || "-"} role=${String(normalizedMessage?.role ?? "").trim() || "-"} toolCallId=${String(normalizedMessage?.toolCallId ?? "").trim() || "-"} sourceKey=${createMessageSourceKey(sessionKey, String(update?.messageId ?? normalizedMessage?.id ?? "").trim() || null, normalizedMessage)}`,
     );
-    await this.indexSessionMessages(sessionKey, [normalizedMessage]);
-    const row = this._db()
-      .prepare(`SELECT COUNT(*) AS count FROM messages WHERE session_key=?`)
-      .get(sessionKey);
+    return this._appendMessageInternal({
+      sessionKey,
+      normalizedMessage,
+      messageId: String(update?.messageId ?? normalizedMessage?.id ?? "").trim() || null,
+    });
+  }
+
+  appendMessageSync(update) {
+    const sessionKey = String(update?.sessionKey ?? "").trim();
+    const normalizedMessage = normalizeMessageObject(update?.message, update?.messageId);
+    if (!sessionKey) {
+      this.logger?.warn?.(
+        `[mobcode-store] appendMessageSync skipped rawSessionKey=${String(update?.sessionKey ?? "").trim() || "-"} messageId=${String(update?.messageId ?? normalizedMessage?.id ?? "").trim() || "-"} reason=missing_session_key`,
+      );
+      return { sessionKey: null, count: 0, skipped: true };
+    }
+    if (!this.database) {
+      this.logger?.warn?.(
+        `[mobcode-store] appendMessageSync skipped sessionKey=${sessionKey} messageId=${String(update?.messageId ?? normalizedMessage?.id ?? "").trim() || "-"} reason=store_not_initialized`,
+      );
+      return { sessionKey, count: 0, skipped: true };
+    }
+    this.logger?.warn?.(
+      `[mobcode-debug] appendMessageSync sessionKey=${sessionKey} messageId=${String(update?.messageId ?? normalizedMessage?.id ?? "").trim() || "-"} role=${String(normalizedMessage?.role ?? "").trim() || "-"} toolCallId=${String(normalizedMessage?.toolCallId ?? "").trim() || "-"} sourceKey=${createMessageSourceKey(sessionKey, String(update?.messageId ?? normalizedMessage?.id ?? "").trim() || null, normalizedMessage)}`,
+    );
+    return this._appendMessageInternal({
+      sessionKey,
+      normalizedMessage,
+      messageId: String(update?.messageId ?? normalizedMessage?.id ?? "").trim() || null,
+    });
+  }
+
+  _appendMessageInternal({ sessionKey, normalizedMessage, messageId }) {
+    const db = this._db();
+    const insert = db.prepare(
+      `INSERT OR IGNORE INTO messages(
+         session_key,
+         source_key,
+         message_id,
+         role,
+         text,
+         raw_json,
+         created_at_ms,
+         indexed_at
+       ) VALUES(?, ?, ?, ?, ?, ?, ?, ?)`,
+    );
+    const touchSession = db.prepare(
+      `INSERT INTO indexed_sessions(session_key, updated_at)
+       VALUES(?, ?)
+       ON CONFLICT(session_key) DO UPDATE SET updated_at=excluded.updated_at`,
+    );
+    const upsertArtifact = db.prepare(
+      `INSERT INTO artifacts(
+         artifact_id,
+         session_key,
+         run_id,
+         kind,
+         title,
+         summary,
+         document_json,
+         created_at_ms,
+         updated_at_ms
+       ) VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?)
+       ON CONFLICT(artifact_id) DO UPDATE SET
+         session_key=excluded.session_key,
+         run_id=excluded.run_id,
+         kind=excluded.kind,
+         title=excluded.title,
+         summary=excluded.summary,
+         document_json=excluded.document_json,
+         updated_at_ms=excluded.updated_at_ms`,
+    );
+    const now = new Date().toISOString();
+    const sourceKey = createMessageSourceKey(sessionKey, messageId, normalizedMessage);
+    const timestampRaw =
+      normalizedMessage?.timestamp ?? normalizedMessage?.createdAt ?? Date.now();
+    const createdAtMs =
+      typeof timestampRaw === "number"
+        ? timestampRaw
+        : Number.parseInt(String(timestampRaw), 10) || Date.now();
+
+    db.exec("BEGIN");
+    try {
+      insert.run(
+        sessionKey,
+        sourceKey,
+        messageId,
+        String(normalizedMessage?.role ?? "").trim() || null,
+        extractMessageText(normalizedMessage),
+        toJson(normalizedMessage),
+        createdAtMs,
+        now,
+      );
+      const artifactDocument = extractArtifactDocumentFromMessage(normalizedMessage);
+      const artifactId = String(
+        artifactDocument?.artifactId ?? artifactDocument?.artifact_id ?? "",
+      ).trim();
+      const artifactKind = String(artifactDocument?.kind ?? "").trim();
+      const artifactTitle = String(artifactDocument?.title ?? "").trim();
+      if (artifactId && artifactKind && artifactTitle) {
+        upsertArtifact.run(
+          artifactId,
+          sessionKey,
+          String(normalizedMessage?.runId ?? normalizedMessage?.run_id ?? "").trim() || null,
+          artifactKind,
+          artifactTitle,
+          String(artifactDocument?.summary ?? "").trim() || null,
+          toJson(artifactDocument),
+          createdAtMs,
+          createdAtMs,
+        );
+      }
+      touchSession.run(sessionKey, now);
+      db.exec("COMMIT");
+    } catch (error) {
+      try {
+        db.exec("ROLLBACK");
+      } catch {
+        // Ignore rollback failures so the original write error is preserved.
+      }
+      throw error;
+    }
+    const row = db.prepare(`SELECT COUNT(*) AS count FROM messages WHERE session_key=?`).get(sessionKey);
     return { sessionKey, count: Number(row?.count ?? 0), skipped: false };
   }