@axiom-lattice/local-stores 3.1.5 → 3.1.8

package/dist/index.mjs

@@ -808,12 +808,20 @@ 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 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,
+  PRIMARY KEY (tenant_id, room_id, user_id)
+);
+CREATE INDEX IF NOT EXISTS idx_lt_project_room_read_states_user
+  ON lt_project_room_read_states (tenant_id, user_id);
 `;
 var tableColumns = {
   lt_project_rooms: ["id", "tenant_id", "workspace_id", "project_id", "type", "name", "created_at", "updated_at"],
   lt_project_memberships: ["id", "tenant_id", "project_id", "user_id", "role", "status", "joined_at", "updated_at"],
   lt_project_bot_memberships: ["id", "tenant_id", "workspace_id", "project_id", "room_id", "assistant_id", "role", "title", "responsibility", "mention_name", "status", "room_thread_id", "joined_at", "updated_at"],
-  lt_project_room_messages: ["id", "tenant_id", "workspace_id", "project_id", "room_id", "author", "content", "mentions", "reply_to_message_id", "source", "source_id", "idempotency_key", "created_at"]
+  lt_project_room_messages: ["id", "tenant_id", "workspace_id", "project_id", "room_id", "author", "content", "mentions", "reply_to_message_id", "source", "source_id", "idempotency_key", "created_at"],
+  lt_project_room_read_states: ["tenant_id", "room_id", "user_id", "last_read_at", "updated_at"]
 };
 var indexes = {
   idx_lt_project_memberships_project: { table: "lt_project_memberships", columns: ["tenant_id", "project_id"], unique: false, partial: false },
@@ -822,13 +830,15 @@ var indexes = {
   idx_lt_project_bot_memberships_project: { table: "lt_project_bot_memberships", columns: ["tenant_id", "project_id"], unique: false, partial: false },
   idx_lt_project_bot_memberships_room: { table: "lt_project_bot_memberships", columns: ["tenant_id", "room_id"], unique: false, partial: false },
   uq_lt_project_room_messages_idempotency: { table: "lt_project_room_messages", columns: ["tenant_id", "room_id", "idempotency_key"], unique: true, partial: true, sql: ["idempotency_key is not null"] },
-  idx_lt_project_room_messages_room_created: { table: "lt_project_room_messages", columns: ["tenant_id", "room_id", "created_at", "id"], unique: false, partial: false, sql: ["created_at desc", "id desc"] }
+  idx_lt_project_room_messages_room_created: { table: "lt_project_room_messages", columns: ["tenant_id", "room_id", "created_at", "id"], unique: false, partial: false, sql: ["created_at desc", "id desc"] },
+  idx_lt_project_room_read_states_user: { table: "lt_project_room_read_states", columns: ["tenant_id", "user_id"], unique: false, partial: false }
 };
 var uniqueConstraints = {
   lt_project_rooms: [["tenant_id", "id"], ["tenant_id", "project_id", "type"]],
   lt_project_memberships: [["tenant_id", "id"], ["tenant_id", "project_id", "user_id"]],
   lt_project_bot_memberships: [["tenant_id", "id"], ["tenant_id", "project_id", "assistant_id"]],
-  lt_project_room_messages: [["tenant_id", "id"]]
+  lt_project_room_messages: [["tenant_id", "id"]],
+  lt_project_room_read_states: [["tenant_id", "room_id", "user_id"]]
 };
 var checkConstraints = {
   lt_project_rooms: [{ name: "type", fragment: "type='main'" }],
@@ -840,7 +850,8 @@ var checkConstraints = {
     { name: "role", fragment: "role in ('coordinator','specialist')" },
     { name: "status", fragment: "status in ('active','paused','removed')" }
   ],
-  lt_project_room_messages: [{ name: "source", fragment: "source in ('user','agent','task','routine','system')" }]
+  lt_project_room_messages: [{ name: "source", fragment: "source in ('user','agent','task','routine','system')" }],
+  lt_project_room_read_states: []
 };
 function rows(db, sql) {
   const result = db.getRawDb().exec(sql);
@@ -912,10 +923,10 @@ function validateProjectRoomSchema(db) {
   }
   for (const [name, expected] of Object.entries(indexes)) {
     const listed = rows(db, `PRAGMA index_list(${expected.table})`).find((index) => index.name === name);
-    const columns5 = rows(db, `PRAGMA index_info(${name})`).map((column) => column.name);
+    const columns6 = rows(db, `PRAGMA index_info(${name})`).map((column) => column.name);
     const definition = rows(db, `SELECT sql FROM sqlite_master WHERE type = 'index' AND name = '${name}'`)[0]?.sql;
     const normalized = typeof definition === "string" ? definition.toLowerCase().replace(/\s+/g, " ") : "";
-    if (!listed || listed.unique !== Number(expected.unique) || listed.partial !== Number(expected.partial) || columns5.join(",") !== expected.columns.join(",") || expected.sql?.some((fragment) => !normalized.includes(fragment))) {
+    if (!listed || listed.unique !== Number(expected.unique) || listed.partial !== Number(expected.partial) || columns6.join(",") !== expected.columns.join(",") || expected.sql?.some((fragment) => !normalized.includes(fragment))) {
       throw new Error(`Invalid project room schema: index '${name}' is incompatible`);
     }
   }
@@ -928,6 +939,30 @@ function ensureProjectRoomSchema(db) {
   const applied = hasMigrationTable ? rows(db, `SELECT name, version FROM lattice_schema_migrations WHERE store_name = 'project_room' AND name = '${migrationName}'`)[0] : void 0;
   if (applied) {
     if (applied.version !== 1) throw new Error(`Invalid project room schema migration version: ${String(applied.version)}`);
+    const hasReadStates = rows(db, "SELECT name FROM sqlite_master WHERE type = 'table' AND name = 'lt_project_room_read_states'").length > 0;
+    if (!hasReadStates) {
+      raw.exec("BEGIN IMMEDIATE");
+      try {
+        raw.exec(`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,
+          PRIMARY KEY (tenant_id, room_id, user_id)
+        ); CREATE INDEX IF NOT EXISTS idx_lt_project_room_read_states_user
+          ON lt_project_room_read_states (tenant_id, user_id);`);
+        raw.run(
+          "INSERT INTO lattice_schema_migrations (store_name, name, version, applied_at) VALUES (?, ?, ?, ?)",
+          ["project_room", "create_project_room_read_states", 2, (/* @__PURE__ */ new Date()).toISOString()]
+        );
+        raw.exec("COMMIT");
+        db.save();
+      } catch (error) {
+        try {
+          raw.exec("ROLLBACK");
+        } catch {
+        }
+        throw error;
+      }
+    }
     validateProjectRoomSchema(db);
     marked[schemaMarker] = true;
     return;
@@ -1367,6 +1402,18 @@ var LocalProjectRoomMessageStore = class {
     const row = this.byId(tenantId, id);
     return row ? mapRow4(row) : null;
   }
+  async countAfter(input) {
+    if (!validDate(input.after)) throw new RangeError("Project room message count horizon date is invalid");
+    const rows2 = this.db.prepare(
+      `SELECT author FROM lt_project_room_messages
+       WHERE tenant_id = ? AND room_id = ? AND created_at > ?`
+    ).all(input.tenantId, input.roomId, input.after.toISOString());
+    return rows2.filter((row) => {
+      if (input.excludeAuthorUserId === void 0) return true;
+      const parsed = parseJson(row.author);
+      return !(record(parsed) && parsed.type === "human" && parsed.userId === input.excludeAuthorUserId);
+    }).length;
+  }
   insert(input, idempotent) {
     return transaction(this.db, () => {
       if (idempotent && input.idempotencyKey) {
@@ -1397,6 +1444,50 @@ var LocalProjectRoomMessageStore = class {
   }
 };
 
+// src/stores/LocalProjectRoomReadStateStore.ts
+var columns5 = "tenant_id, room_id, user_id, last_read_at, updated_at";
+function mapRow5(row) {
+  const lastReadAt = new Date(row.last_read_at);
+  const updatedAt = new Date(row.updated_at);
+  if (Number.isNaN(lastReadAt.getTime()) || Number.isNaN(updatedAt.getTime())) throw new Error("Invalid project room read state row");
+  return { tenantId: row.tenant_id, roomId: row.room_id, userId: row.user_id, lastReadAt, updatedAt };
+}
+var LocalProjectRoomReadStateStore = class {
+  /** Creates a store over a managed local database wrapper. */
+  constructor(db) {
+    this.db = db;
+    ensureProjectRoomSchema(db);
+  }
+  /** Returns the user's read marker for a room, or null when never reported. */
+  async get(tenantId, roomId, userId) {
+    const row = this.byKey(tenantId, roomId, userId);
+    return row ? mapRow5(row) : null;
+  }
+  /** Upserts the marker monotonically (an older report never regresses the horizon). */
+  async markRead(input) {
+    if (!validDate(input.lastReadAt)) throw new RangeError("Project room read state lastReadAt is invalid");
+    return transaction(this.db, () => {
+      const existing = this.byKey(input.tenantId, input.roomId, input.userId);
+      const lastReadAt = existing !== void 0 && new Date(existing.last_read_at).getTime() > input.lastReadAt.getTime() ? new Date(existing.last_read_at) : input.lastReadAt;
+      write(
+        this.db,
+        `INSERT INTO lt_project_room_read_states (${columns5})
+         VALUES (?, ?, ?, ?, ?)
+         ON CONFLICT (tenant_id, room_id, user_id) DO UPDATE
+           SET last_read_at = excluded.last_read_at, updated_at = excluded.updated_at`,
+        [input.tenantId, input.roomId, input.userId, lastReadAt.toISOString(), timestamp()]
+      );
+      const row = this.byKey(input.tenantId, input.roomId, input.userId);
+      return mapRow5(row);
+    });
+  }
+  byKey(tenantId, roomId, userId) {
+    return this.db.prepare(
+      `SELECT ${columns5} FROM lt_project_room_read_states WHERE tenant_id = ? AND room_id = ? AND user_id = ?`
+    ).get(tenantId, roomId, userId);
+  }
+};
+
 // src/stores/LocalUserStore.ts
 var DDL5 = `
 CREATE TABLE IF NOT EXISTS lt_users (
@@ -2224,6 +2315,18 @@ function mapRowToEntry3(row) {
       throw new Error("Failed to decrypt MCP server configuration");
     }
   }
+  if (config.headers && row.is_env_encrypted) {
+    try {
+      const decryptedHeaders = {};
+      for (const [key, value] of Object.entries(config.headers)) {
+        decryptedHeaders[key] = decrypt2(value);
+      }
+      config.headers = decryptedHeaders;
+    } catch (error) {
+      console.error("Failed to decrypt MCP server headers:", error);
+      throw new Error("Failed to decrypt MCP server configuration");
+    }
+  }
   return {
     id: row.id,
     tenantId: row.tenant_id,
@@ -2249,6 +2352,14 @@ function encryptEnvInConfig(config) {
     configCopy.env = encryptedEnv;
     isEnvEncrypted = true;
   }
+  if (configCopy.headers && Object.keys(configCopy.headers).length > 0) {
+    const encryptedHeaders = {};
+    for (const [key, value] of Object.entries(configCopy.headers)) {
+      encryptedHeaders[key] = encrypt2(value);
+    }
+    configCopy.headers = encryptedHeaders;
+    isEnvEncrypted = true;
+  }
   return { config: configCopy, isEnvEncrypted };
 }
 
@@ -3562,7 +3673,7 @@ var LocalChannelInstallationStore = class {
     const row = this.db.prepare(
       `SELECT * FROM lt_channel_installations WHERE id = ?`
     ).get(installationId);
-    return row ? mapRow5(row) : null;
+    return row ? mapRow6(row) : null;
   }
   async getInstallationsByTenant(tenantId, channel) {
     let rows2;
@@ -3575,7 +3686,7 @@ var LocalChannelInstallationStore = class {
         `SELECT * FROM lt_channel_installations WHERE tenant_id = ? ORDER BY created_at DESC`
       ).all(tenantId);
     }
-    return rows2.map(mapRow5);
+    return rows2.map(mapRow6);
   }
   async getAllInstallations(channel) {
     let rows2;
@@ -3588,7 +3699,7 @@ var LocalChannelInstallationStore = class {
         `SELECT * FROM lt_channel_installations ORDER BY created_at DESC`
       ).all();
     }
-    return rows2.map(mapRow5);
+    return rows2.map(mapRow6);
   }
   async createInstallation(tenantId, installationId, data) {
     const now = nowISO();
@@ -3664,7 +3775,7 @@ var LocalChannelInstallationStore = class {
     return result.changes > 0;
   }
 };
-function mapRow5(row) {
+function mapRow6(row) {
   return {
     id: row.id,
     tenantId: row.tenant_id,
@@ -3894,8 +4005,8 @@ var LocalThreadMessageQueueStore = class {
     this.capacityTail = Promise.resolve();
     this.db = db;
     ensureTable(db, DDL17);
-    const columns5 = db.prepare("PRAGMA table_info(lt_thread_message_queue)").all();
-    const names = new Set(columns5.map(({ name }) => name));
+    const columns6 = db.prepare("PRAGMA table_info(lt_thread_message_queue)").all();
+    const names = new Set(columns6.map(({ name }) => name));
     if (!names.has("trusted_run_context")) {
       db.run("ALTER TABLE lt_thread_message_queue ADD COLUMN trusted_run_context TEXT");
     }
@@ -5746,6 +5857,7 @@ async function createLocalStoreConfig(options = {}) {
     projectMembership: new LocalProjectMembershipStore(db),
     projectBotMembership: new LocalProjectBotMembershipStore(db),
     projectRoomMessage: new LocalProjectRoomMessageStore(db),
+    projectRoomReadState: new LocalProjectRoomReadStateStore(db),
     user: new LocalUserStore(db),
     tenant: new LocalTenantStore(db),
     userTenantLink: new LocalUserTenantLinkStore(db),
@@ -5822,6 +5934,7 @@ export {
   LocalProjectBotMembershipStore,
   LocalProjectMembershipStore,
   LocalProjectRoomMessageStore,
+  LocalProjectRoomReadStateStore,
   LocalProjectRoomStore,
   LocalProjectStore,
   LocalScheduleStorage,