@axiom-lattice/local-stores 3.1.12 → 3.1.13

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/local-stores",
-  "version": "3.1.12",
+  "version": "3.1.13",
   "description": "Local SQLite-based stores for Axiom Lattice framework",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -25,7 +25,7 @@
     "@types/sql.js": "^1.4.11",
     "sql.js": "^1.14.1",
     "uuid": "^14.0.1",
-    "@axiom-lattice/core": "4.2.12",
+    "@axiom-lattice/core": "4.2.13",
     "@axiom-lattice/protocols": "4.2.3"
   },
   "devDependencies": {

package/src/__tests__/LocalProjectRoomStores.test.ts

@@ -316,6 +316,23 @@ describe("local project room stores", () => {
     expect(await reopened.messages.list({ tenantId: "t1", roomId: "r1", before: { createdAt: second.createdAt, id: second.id }, limit: 10 })).toEqual([first]);
   });
 
+  it("deletes old chat messages in bounded batches while retaining task events", async () => {
+    const store = new LocalProjectRoomMessageStore(await initDatabase({ dbPath }));
+    const base = { tenantId: "t1", workspaceId: "w1", projectId: "p1", roomId: "r1", content: { type: "text" as const, text: "x" }, mentions: [] };
+    await store.create({ ...base, id: "user-1", author: { type: "human", userId: "u1" }, source: "user" });
+    await store.create({ ...base, id: "agent-1", author: { type: "bot", membershipId: "bot", assistantId: "a1" }, source: "agent" });
+    await store.create({ ...base, id: "task-1", author: { type: "system" }, source: "task" });
+    const horizon = new Date(Date.now() + 60_000);
+
+    await expect(store.deleteChatOlderThan({ before: horizon, limit: 1 })).resolves.toBe(1);
+    await expect(store.deleteChatOlderThan({ before: horizon, limit: 1 })).resolves.toBe(1);
+    await expect(store.deleteChatOlderThan({ before: horizon, limit: 1 })).resolves.toBe(0);
+    await expect(store.findById("t1", "task-1")).resolves.toMatchObject({ id: "task-1" });
+    await expect(store.findById("t1", "user-1")).resolves.toBeNull();
+    await expect(store.findById("t1", "agent-1")).resolves.toBeNull();
+    await expect(store.deleteChatOlderThan({ before: new Date("invalid"), limit: 1 })).rejects.toBeInstanceOf(RangeError);
+  });
+
   it("round-trips cloned message JSON, maps uniqueness, idempotency, and exact descending cursors", async () => {
     const store = new LocalProjectRoomMessageStore(await initDatabase({ dbPath }));
     const message = { id: "m1", tenantId: "t1", workspaceId: "w1", projectId: "p1", roomId: "r1", author: { type: "human" as const, userId: "u1" }, content: { type: "text" as const, text: "hello" }, mentions: [{ type: "team" as const }], source: "user" as const, idempotencyKey: "key-1" };

package/src/stores/LocalProjectRoomMessageStore.ts

@@ -74,6 +74,25 @@ export class LocalProjectRoomMessageStore implements ProjectRoomMessageStore {
       return !(record(parsed) && parsed.type === "human" && parsed.userId === input.excludeAuthorUserId);
     }).length;
   }
+  /**
+   * Deletes one bounded batch of chat messages older than a horizon.
+   *
+   * @param input Exclusive upper time bound and requested batch size (clamped to 1..1000).
+   * @returns The number of deleted rows.
+   * @remarks Only `source: "user" | "agent"` rows are eligible; task/system events are retained.
+   */
+  async deleteChatOlderThan(input: { before: Date; limit: number }): Promise<number> {
+    if (!validDate(input.before)) throw new RangeError("Project room message retention horizon date is invalid");
+    const limit = Math.min(1_000, Math.max(1, Math.trunc(Number.isFinite(input.limit) ? input.limit : 1)));
+    return transaction(this.db, () => write(this.db,
+      `DELETE FROM lt_project_room_messages WHERE rowid IN (
+         SELECT rowid FROM lt_project_room_messages
+         WHERE created_at < ? AND source IN ('user','agent')
+         ORDER BY created_at ASC LIMIT ?
+       )`,
+      [input.before.toISOString(), limit]));
+  }
+
   private insert(input: Omit<ProjectRoomMessage, "createdAt">, idempotent: boolean): ProjectRoomMessage {
     return transaction(this.db, () => {
       if (idempotent && input.idempotencyKey) { const canonical = this.byKey(input.tenantId, input.roomId, input.idempotencyKey); if (canonical) return mapRow(canonical); }

package/src/stores/project-room-schema.ts

@@ -49,6 +49,8 @@ CREATE UNIQUE INDEX IF NOT EXISTS uq_lt_project_room_messages_idempotency
   ON lt_project_room_messages (tenant_id, room_id, idempotency_key) WHERE idempotency_key IS NOT NULL;
 CREATE INDEX IF NOT EXISTS idx_lt_project_room_messages_room_created
   ON lt_project_room_messages (tenant_id, room_id, created_at DESC, id DESC);
+CREATE INDEX IF NOT EXISTS idx_lt_project_room_messages_chat_created
+  ON lt_project_room_messages (created_at) WHERE source IN ('user','agent');
 CREATE TABLE IF NOT EXISTS lt_project_room_read_states (
   tenant_id TEXT NOT NULL, room_id TEXT NOT NULL, user_id TEXT NOT NULL,
   last_read_at TEXT NOT NULL, updated_at TEXT NOT NULL,
@@ -210,6 +212,20 @@ export function ensureProjectRoomSchema(db: DatabaseWrapper): void {
       }
     }
     validateProjectRoomSchema(db);
+    const hasRetentionIndex = rows(db, "SELECT name FROM sqlite_master WHERE type = 'index' AND name = 'idx_lt_project_room_messages_chat_created'").length > 0;
+    if (!hasRetentionIndex) {
+      raw.exec("BEGIN IMMEDIATE");
+      try {
+        raw.exec("CREATE INDEX IF NOT EXISTS idx_lt_project_room_messages_chat_created ON lt_project_room_messages (created_at) WHERE source IN ('user','agent');");
+        raw.run("INSERT INTO lattice_schema_migrations (store_name, name, version, applied_at) VALUES (?, ?, ?, ?)",
+          ["project_room", "add_project_room_message_retention_index", 3, new Date().toISOString()]);
+        raw.exec("COMMIT");
+        db.save();
+      } catch (error) {
+        try { raw.exec("ROLLBACK"); } catch { /* preserve the schema failure */ }
+        throw error;
+      }
+    }
     marked[schemaMarker] = true;
     return;
   }