@axiom-lattice/pg-stores 3.1.10 → 3.1.11

package/dist/index.mjs

@@ -8736,12 +8736,12 @@ var PostgreSQLTaskWorkItemStore = class {
       tenantId: row.tenant_id,
       action: row.action,
       actor: row.actor,
-      threadId: row.thread_id,
-      summary: row.summary,
+      threadId: row.thread_id == null ? void 0 : row.thread_id,
+      summary: row.summary == null ? void 0 : row.summary,
       detail: isRecord2(row.detail) ? row.detail : void 0,
-      attempt: row.attempt,
-      workspaceId: row.workspace_id,
-      projectId: row.project_id,
+      attempt: row.attempt == null ? void 0 : row.attempt,
+      workspaceId: row.workspace_id == null ? void 0 : row.workspace_id,
+      projectId: row.project_id == null ? void 0 : row.project_id,
       eventKey: row.event_key == null ? void 0 : row.event_key,
       createdAt: new Date(row.created_at)
     };
@@ -9884,6 +9884,35 @@ var createProjectRoomReadStates = {
   }
 };
 
+// src/migrations/user_push_subscription_migration.ts
+var createUserPushSubscriptions = {
+  version: 178,
+  name: "create_user_push_subscriptions",
+  up: async (client) => {
+    await client.query(`
+      CREATE TABLE IF NOT EXISTS lattice_user_push_subscriptions (
+        id VARCHAR(255) PRIMARY KEY,
+        tenant_id VARCHAR(255) NOT NULL,
+        user_id VARCHAR(255) NOT NULL,
+        endpoint TEXT NOT NULL,
+        p256dh TEXT NOT NULL,
+        auth TEXT NOT NULL,
+        user_agent TEXT,
+        created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+        UNIQUE (tenant_id, user_id, endpoint)
+      )
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_user_push_subscriptions_user
+      ON lattice_user_push_subscriptions (tenant_id, user_id)
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP INDEX IF EXISTS idx_lattice_user_push_subscriptions_user");
+    await client.query("DROP TABLE IF EXISTS lattice_user_push_subscriptions");
+  }
+};
+
 // src/stores/PostgreSQLCapabilityBundleStore.ts
 import { randomUUID as randomUUID3 } from "crypto";
 import { Pool as Pool26 } from "pg";
@@ -10756,6 +10785,58 @@ var PostgreSQLProjectRoomReadStateStore = class {
   }
 };
 
+// src/stores/PostgreSQLUserPushSubscriptionStore.ts
+import { randomUUID as randomUUID4 } from "crypto";
+var columns4 = "id, tenant_id, user_id, endpoint, p256dh, auth, user_agent, created_at";
+function mapRow7(row) {
+  if (typeof row.id !== "string" || typeof row.tenant_id !== "string" || typeof row.user_id !== "string" || typeof row.endpoint !== "string" || typeof row.p256dh !== "string" || typeof row.auth !== "string" || !(row.created_at instanceof Date) || Number.isNaN(row.created_at.getTime())) {
+    throw new Error("Invalid user push subscription row");
+  }
+  return {
+    id: row.id,
+    tenantId: row.tenant_id,
+    userId: row.user_id,
+    endpoint: row.endpoint,
+    keys: { p256dh: row.p256dh, auth: row.auth },
+    ...typeof row.user_agent === "string" ? { userAgent: row.user_agent } : {},
+    createdAt: row.created_at.toISOString()
+  };
+}
+var PostgreSQLUserPushSubscriptionStore = class {
+  /** Creates a store using an externally managed shared pool. */
+  constructor(options) {
+    this.pool = options.pool;
+  }
+  /** Lists every subscription for a user. */
+  async list(tenantId, userId) {
+    const result = await this.pool.query(
+      `SELECT ${columns4} FROM lattice_user_push_subscriptions WHERE tenant_id = $1 AND user_id = $2`,
+      [tenantId, userId]
+    );
+    return result.rows.map(mapRow7);
+  }
+  /** Upserts a subscription keyed by tenant + user + endpoint. */
+  async save(input) {
+    const result = await this.pool.query(
+      `INSERT INTO lattice_user_push_subscriptions (${columns4})
+       VALUES ($1, $2, $3, $4, $5, $6, $7, NOW())
+       ON CONFLICT (tenant_id, user_id, endpoint) DO UPDATE
+         SET p256dh = EXCLUDED.p256dh, auth = EXCLUDED.auth, user_agent = EXCLUDED.user_agent
+       RETURNING ${columns4}`,
+      [randomUUID4(), input.tenantId, input.userId, input.endpoint, input.keys.p256dh, input.keys.auth, input.userAgent ?? null]
+    );
+    return mapRow7(result.rows[0]);
+  }
+  /** Deletes one subscription by endpoint. */
+  async delete(tenantId, userId, endpoint) {
+    const result = await this.pool.query(
+      `DELETE FROM lattice_user_push_subscriptions WHERE tenant_id = $1 AND user_id = $2 AND endpoint = $3`,
+      [tenantId, userId, endpoint]
+    );
+    return (result.rowCount ?? 0) > 0;
+  }
+};
+
 // src/createPgStoreConfig.ts
 async function createPgStoreConfig(connectionString) {
   const pool = new Pool27({ connectionString });
@@ -10823,6 +10904,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(addProjectLifecycleEventIndex);
   mm.register(addAgentWebAppsIdentity);
   mm.register(createProjectRoomReadStates);
+  mm.register(createUserPushSubscriptions);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {
@@ -10862,6 +10944,7 @@ async function createPgStoreConfig(connectionString) {
     projectBotMembership: new PostgreSQLProjectBotMembershipStore(opts),
     projectRoomMessage: new PostgreSQLProjectRoomMessageStore(opts),
     projectRoomReadState: new PostgreSQLProjectRoomReadStateStore(opts),
+    userPushSubscription: new PostgreSQLUserPushSubscriptionStore(opts),
     vectorStoreProvider: new PGVectorStoreProvider(pool, connectionString),
     checkpoint
   };