@axiom-lattice/pg-stores 3.1.6 → 3.1.8

package/dist/index.mjs

@@ -2016,6 +2016,18 @@ var PostgreSQLMcpServerConfigStore = class {
         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] = decrypt3(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,
@@ -2044,6 +2056,14 @@ var PostgreSQLMcpServerConfigStore = class {
       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] = encrypt3(value);
+      }
+      configCopy.headers = encryptedHeaders;
+      isEnvEncrypted = true;
+    }
     return { config: configCopy, isEnvEncrypted };
   }
 };
@@ -9586,6 +9606,30 @@ var createProjectRoomTables = {
     await client.query("DROP TABLE IF EXISTS lattice_project_rooms");
   }
 };
+var createProjectRoomReadStates = {
+  version: 177,
+  name: "create_project_room_read_states",
+  up: async (client) => {
+    await client.query(`
+      CREATE TABLE IF NOT EXISTS lattice_project_room_read_states (
+        tenant_id VARCHAR(255) NOT NULL,
+        room_id VARCHAR(255) NOT NULL,
+        user_id VARCHAR(255) NOT NULL,
+        last_read_at TIMESTAMPTZ NOT NULL,
+        updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+        PRIMARY KEY (tenant_id, room_id, user_id)
+      )
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_project_room_read_states_user
+      ON lattice_project_room_read_states (tenant_id, user_id)
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP INDEX IF EXISTS idx_lattice_project_room_read_states_user");
+    await client.query("DROP TABLE IF EXISTS lattice_project_room_read_states");
+  }
+};
 
 // src/stores/PostgreSQLCapabilityBundleStore.ts
 import { randomUUID as randomUUID3 } from "crypto";
@@ -10373,6 +10417,19 @@ var PostgreSQLProjectRoomMessageStore = class {
     );
     return result.rows[0] ? mapRow5(result.rows[0]) : null;
   }
+  /** Counts messages created strictly after a horizon, optionally excluding one human author. */
+  async countAfter(input) {
+    if (!(input.after instanceof Date) || Number.isNaN(input.after.getTime())) {
+      throw new RangeError("Project room message count horizon date is invalid");
+    }
+    const result = await this.pool.query(
+      `SELECT COUNT(*)::int AS count FROM lattice_project_room_messages
+       WHERE tenant_id = $1 AND room_id = $2 AND created_at > $3
+         AND ($4::text IS NULL OR author->>'type' <> 'human' OR author->>'userId' <> $4)`,
+      [input.tenantId, input.roomId, input.after, input.excludeAuthorUserId ?? null]
+    );
+    return result.rows[0]?.count ?? 0;
+  }
   parameters(input) {
     return [
       input.id,
@@ -10391,6 +10448,45 @@ var PostgreSQLProjectRoomMessageStore = class {
   }
 };
 
+// src/stores/PostgreSQLProjectRoomReadStateStore.ts
+function mapRow6(row) {
+  if (typeof row.tenant_id !== "string" || typeof row.room_id !== "string" || typeof row.user_id !== "string" || !(row.last_read_at instanceof Date) || Number.isNaN(row.last_read_at.getTime()) || !(row.updated_at instanceof Date) || Number.isNaN(row.updated_at.getTime())) {
+    throw new Error("Invalid project room read state row");
+  }
+  return { tenantId: row.tenant_id, roomId: row.room_id, userId: row.user_id, lastReadAt: row.last_read_at, updatedAt: row.updated_at };
+}
+var PostgreSQLProjectRoomReadStateStore = class {
+  /** Creates a store using an externally managed shared pool. */
+  constructor(options) {
+    this.pool = options.pool;
+  }
+  /** Returns the user's read marker for a room, or null when never reported. */
+  async get(tenantId, roomId, userId) {
+    const result = await this.pool.query(
+      `SELECT tenant_id, room_id, user_id, last_read_at, updated_at
+       FROM lattice_project_room_read_states WHERE tenant_id = $1 AND room_id = $2 AND user_id = $3`,
+      [tenantId, roomId, userId]
+    );
+    return result.rows[0] ? mapRow6(result.rows[0]) : null;
+  }
+  /** Upserts the marker monotonically using GREATEST so older reports never regress it. */
+  async markRead(input) {
+    if (!(input.lastReadAt instanceof Date) || Number.isNaN(input.lastReadAt.getTime())) {
+      throw new RangeError("Project room read state lastReadAt is invalid");
+    }
+    const result = await this.pool.query(
+      `INSERT INTO lattice_project_room_read_states (tenant_id, room_id, user_id, last_read_at, updated_at)
+       VALUES ($1, $2, $3, $4, NOW())
+       ON CONFLICT (tenant_id, room_id, user_id) DO UPDATE
+         SET last_read_at = GREATEST(lattice_project_room_read_states.last_read_at, EXCLUDED.last_read_at),
+             updated_at = NOW()
+       RETURNING tenant_id, room_id, user_id, last_read_at, updated_at`,
+      [input.tenantId, input.roomId, input.userId, input.lastReadAt]
+    );
+    return mapRow6(result.rows[0]);
+  }
+};
+
 // src/createPgStoreConfig.ts
 async function createPgStoreConfig(connectionString) {
   const pool = new Pool26({ connectionString });
@@ -10454,6 +10550,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(addTrustedRunContextColumn);
   mm.register(addProjectLifecycleEventIndex);
   mm.register(addAgentWebAppsIdentity);
+  mm.register(createProjectRoomReadStates);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {
@@ -10491,6 +10588,7 @@ async function createPgStoreConfig(connectionString) {
     projectMembership: new PostgreSQLProjectMembershipStore(opts),
     projectBotMembership: new PostgreSQLProjectBotMembershipStore(opts),
     projectRoomMessage: new PostgreSQLProjectRoomMessageStore(opts),
+    projectRoomReadState: new PostgreSQLProjectRoomReadStateStore(opts),
     vectorStoreProvider: new PGVectorStoreProvider(pool, connectionString),
     checkpoint
   };
@@ -11066,6 +11164,7 @@ export {
   PostgreSQLProjectBotMembershipStore,
   PostgreSQLProjectMembershipStore,
   PostgreSQLProjectRoomMessageStore,
+  PostgreSQLProjectRoomReadStateStore,
   PostgreSQLProjectRoomStore,
   PostgreSQLProjectStore,
   PostgreSQLScheduleStorage,
@@ -11118,6 +11217,7 @@ export {
   createMetricsConfigsTable,
   createPGVectorStoreProvider,
   createPgStoreConfig,
+  createProjectRoomReadStates,
   createProjectRoomTables,
   createProjectsTable,
   createScheduledTasksTable,