@axiom-lattice/pg-stores 3.1.10 → 3.1.11

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/pg-stores",
-  "version": "3.1.10",
+  "version": "3.1.11",
   "description": "PG stores implementation for Axiom Lattice framework",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -25,8 +25,8 @@
     "@langchain/core": "1.1.30",
     "pg": "^8.16.3",
     "uuid": "^9.0.1",
-    "@axiom-lattice/core": "4.2.10",
-    "@axiom-lattice/protocols": "4.2.1"
+    "@axiom-lattice/core": "4.2.11",
+    "@axiom-lattice/protocols": "4.2.2"
   },
   "devDependencies": {
     "@types/jest": "^29.5.14",

package/src/__tests__/PostgreSQLTaskWorkItemStore.test.ts

@@ -82,6 +82,19 @@ describe("PostgreSQLTaskWorkItemStore", () => {
     expect(result[0].eventKey).toBeUndefined();
   });
 
+  it("maps null optional columns to undefined for strict projection snapshots", async () => {
+    const query = jest.fn().mockResolvedValue({ rows: [row] });
+    const store = new PostgreSQLTaskWorkItemStore({ query } as unknown as Pool);
+
+    const result = await store.list({ tenantId: "tenant-1", taskId: "task-1" });
+
+    expect(result[0].threadId).toBeUndefined();
+    expect(result[0].attempt).toBeUndefined();
+    expect(result[0].workspaceId).toBeUndefined();
+    expect(result[0].projectId).toBeUndefined();
+    expect(result[0].detail).toBeUndefined();
+  });
+
   it.each([null, [], "legacy", 1] as const)("maps non-object detail %p to undefined", async (detail) => {
     const query = jest.fn().mockResolvedValue({ rows: [{ ...row, detail }] });
     const store = new PostgreSQLTaskWorkItemStore({ query } as unknown as Pool);

package/src/createPgStoreConfig.ts

@@ -87,12 +87,14 @@ import { createAgentWebAppsTable } from "./migrations/agent_web_apps_migration";
 import { addAgentWebAppsIdentity } from "./migrations/add_agent_web_apps_identity_migration";
 import { createCapabilityBundlesTable } from "./migrations/capability_bundle_migration";
 import { createProjectRoomReadStates, createProjectRoomTables } from "./migrations/project_room_migration";
+import { createUserPushSubscriptions } from "./migrations/user_push_subscription_migration";
 import { PostgreSQLCapabilityBundleStore } from "./stores/PostgreSQLCapabilityBundleStore";
 import { PostgreSQLProjectRoomStore } from "./stores/PostgreSQLProjectRoomStore";
 import { PostgreSQLProjectMembershipStore } from "./stores/PostgreSQLProjectMembershipStore";
 import { PostgreSQLProjectBotMembershipStore } from "./stores/PostgreSQLProjectBotMembershipStore";
 import { PostgreSQLProjectRoomMessageStore } from "./stores/PostgreSQLProjectRoomMessageStore";
 import { PostgreSQLProjectRoomReadStateStore } from "./stores/PostgreSQLProjectRoomReadStateStore";
+import { PostgreSQLUserPushSubscriptionStore } from "./stores/PostgreSQLUserPushSubscriptionStore";
 
 export async function createPgStoreConfig(connectionString: string) {
   const pool = new Pool({ connectionString });
@@ -164,6 +166,7 @@ export async function createPgStoreConfig(connectionString: string) {
   mm.register(addProjectLifecycleEventIndex);   // v175
   mm.register(addAgentWebAppsIdentity);         // v176
   mm.register(createProjectRoomReadStates);     // v177
+  mm.register(createUserPushSubscriptions);     // v178
 
   await mm.migrate();
 
@@ -208,6 +211,7 @@ export async function createPgStoreConfig(connectionString: string) {
     projectBotMembership: new PostgreSQLProjectBotMembershipStore(opts),
     projectRoomMessage: new PostgreSQLProjectRoomMessageStore(opts),
     projectRoomReadState: new PostgreSQLProjectRoomReadStateStore(opts),
+    userPushSubscription: new PostgreSQLUserPushSubscriptionStore(opts),
     vectorStoreProvider: new PGVectorStoreProvider(pool, connectionString),
     checkpoint,
   };

package/src/migrations/user_push_subscription_migration.ts

@@ -0,0 +1,30 @@
+import type { Migration } from "./migration";
+
+/** Adds per-user Web Push subscriptions for PWA notifications. */
+export const createUserPushSubscriptions: Migration = {
+  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");
+  },
+};

package/src/stores/PostgreSQLTaskWorkItemStore.ts

@@ -284,12 +284,12 @@ export class PostgreSQLTaskWorkItemStore implements TaskWorkItemStore {
       tenantId: row.tenant_id as string,
       action: row.action as string,
       actor: row.actor as string,
-      threadId: row.thread_id as string | undefined,
-      summary: row.summary as string | undefined,
+      threadId: row.thread_id == null ? undefined : row.thread_id as string,
+      summary: row.summary == null ? undefined : row.summary as string,
       detail: isRecord(row.detail) ? row.detail : undefined,
-      attempt: row.attempt as number | undefined,
-      workspaceId: row.workspace_id as string | undefined,
-      projectId: row.project_id as string | undefined,
+      attempt: row.attempt == null ? undefined : row.attempt as number,
+      workspaceId: row.workspace_id == null ? undefined : row.workspace_id as string,
+      projectId: row.project_id == null ? undefined : row.project_id as string,
       eventKey: row.event_key == null ? undefined : row.event_key as string,
       createdAt: new Date(row.created_at as string),
     };

package/src/stores/PostgreSQLUserPushSubscriptionStore.ts

@@ -0,0 +1,61 @@
+import { randomUUID } from "node:crypto";
+import type { UserPushSubscription, UserPushSubscriptionStore } from "@axiom-lattice/protocols";
+import type { Pool } from "pg";
+
+type Row = {
+  id: unknown; tenant_id: unknown; user_id: unknown; endpoint: unknown;
+  p256dh: unknown; auth: unknown; user_agent: unknown; created_at: unknown;
+};
+const columns = "id, tenant_id, user_id, endpoint, p256dh, auth, user_agent, created_at";
+
+function mapRow(row: Row): UserPushSubscription {
+  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(),
+  };
+}
+
+/** Persists per-user Web Push subscriptions in PostgreSQL. */
+export class PostgreSQLUserPushSubscriptionStore implements UserPushSubscriptionStore {
+  /** Creates a store using an externally managed shared pool. */
+  constructor(options: { pool: Pool }) { this.pool = options.pool; }
+  private readonly pool: Pool;
+
+  /** Lists every subscription for a user. */
+  async list(tenantId: string, userId: string): Promise<UserPushSubscription[]> {
+    const result = await this.pool.query<Row>(
+      `SELECT ${columns} FROM lattice_user_push_subscriptions WHERE tenant_id = $1 AND user_id = $2`,
+      [tenantId, userId],
+    );
+    return result.rows.map(mapRow);
+  }
+
+  /** Upserts a subscription keyed by tenant + user + endpoint. */
+  async save(input: Omit<UserPushSubscription, "id" | "createdAt">): Promise<UserPushSubscription> {
+    const result = await this.pool.query<Row>(
+      `INSERT INTO lattice_user_push_subscriptions (${columns})
+       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 ${columns}`,
+      [randomUUID(), input.tenantId, input.userId, input.endpoint, input.keys.p256dh, input.keys.auth, input.userAgent ?? null],
+    );
+    return mapRow(result.rows[0]);
+  }
+
+  /** Deletes one subscription by endpoint. */
+  async delete(tenantId: string, userId: string, endpoint: string): Promise<boolean> {
+    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;
+  }
+}