@axiom-lattice/pg-stores 3.1.16 → 3.1.18

package/dist/index.mjs

@@ -8817,6 +8817,34 @@ var addFileContentType = {
   }
 };
 
+// src/migrations/menu_items_add_mcp_app_type.ts
+var addMcpAppContentType = {
+  version: 181,
+  name: "add_mcp_app_content_type_to_menu_items",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_menu_items
+      DROP CONSTRAINT IF EXISTS lattice_menu_items_content_type_check
+    `);
+    await client.query(`
+      ALTER TABLE lattice_menu_items
+      ADD CONSTRAINT lattice_menu_items_content_type_check
+      CHECK (content_type IN ('agent', 'html', 'custom', 'file', 'mcpApp'))
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_menu_items
+      DROP CONSTRAINT IF EXISTS lattice_menu_items_content_type_check
+    `);
+    await client.query(`
+      ALTER TABLE lattice_menu_items
+      ADD CONSTRAINT lattice_menu_items_content_type_check
+      CHECK (content_type IN ('agent', 'html', 'custom', 'file'))
+    `);
+  }
+};
+
 // src/stores/MenuStore.ts
 var MenuStore = class {
   constructor(options) {
@@ -8835,6 +8863,7 @@ var MenuStore = class {
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createMenuItemsTable);
     this.migrationManager.register(addFileContentType);
+    this.migrationManager.register(addMcpAppContentType);
     if (options.autoMigrate !== false) {
       this.initialize().catch((error) => {
         console.error("Failed to initialize MenuStore:", error);
@@ -10054,6 +10083,10 @@ var PostgreSQLCapabilityBundleStore = class {
   }
   async update(tenantId, id, input) {
     await this.ready();
+    const expectedUpdatedAt = input.expectedUpdatedAt;
+    if (typeof expectedUpdatedAt !== "string" || expectedUpdatedAt.trim().length === 0) {
+      return this.resolveCasMiss(tenantId, id);
+    }
     const fields = [];
     const values = [];
     const add = (field, value) => {
@@ -10068,37 +10101,46 @@ var PostgreSQLCapabilityBundleStore = class {
       add("capabilities", JSON.stringify(input.capabilities));
     }
     if (fields.length === 0) {
-      const expectedUpdatedAt2 = input.expectedUpdatedAt;
       const result = await this.pool.query(
         `SELECT id, tenant_id, bundle_key, name, description, capabilities, created_at, updated_at::text AS updated_at
          FROM lattice_capability_bundles
-         WHERE tenant_id = $1 AND id = $2${expectedUpdatedAt2 === void 0 ? "" : " AND updated_at::text = $3"}`,
-        expectedUpdatedAt2 === void 0 ? [tenantId, id] : [tenantId, id, expectedUpdatedAt2]
+         WHERE tenant_id = $1 AND id = $2 AND updated_at::text = $3`,
+        [tenantId, id, expectedUpdatedAt]
       );
       if (result.rows[0]) return map(result.rows[0]);
-      return expectedUpdatedAt2 === void 0 ? null : { status: "conflict" };
+      return this.resolveCasMiss(tenantId, id);
     }
-    const expectedUpdatedAt = input.expectedUpdatedAt;
-    values.push(tenantId, id);
-    if (expectedUpdatedAt !== void 0) values.push(expectedUpdatedAt);
-    const tenantParam = values.length - (expectedUpdatedAt === void 0 ? 1 : 2);
+    values.push(tenantId, id, expectedUpdatedAt);
+    const tenantParam = values.length - 2;
     const idParam = tenantParam + 1;
-    const revisionPredicate = expectedUpdatedAt === void 0 ? "" : ` AND updated_at::text = $${values.length}`;
+    const revisionParam = values.length;
     try {
       const result = await this.pool.query(
         `UPDATE lattice_capability_bundles
-          SET ${fields.join(", ")}, updated_at = GREATEST(updated_at + interval '1 microsecond', clock_timestamp())
-          WHERE tenant_id = $${tenantParam} AND id = $${idParam}${revisionPredicate}
+         SET ${fields.join(", ")}, updated_at = GREATEST(updated_at + interval '1 microsecond', clock_timestamp())
+         WHERE tenant_id = $${tenantParam} AND id = $${idParam} AND updated_at::text = $${revisionParam}
          RETURNING id, tenant_id, bundle_key, name, description, capabilities, created_at, updated_at::text AS updated_at`,
         values
       );
       if (result.rows[0]) return map(result.rows[0]);
-      return expectedUpdatedAt === void 0 ? null : { status: "conflict" };
+      return this.resolveCasMiss(tenantId, id);
     } catch (error) {
       if (isDuplicate(error)) throw new Error(duplicateMessage);
       throw error;
     }
   }
+  /**
+   * Distinguishes a stale-revision CAS miss from a missing bundle.
+   *
+   * @returns `null` when no such bundle exists, otherwise a conflict marker.
+   */
+  async resolveCasMiss(tenantId, id) {
+    const result = await this.pool.query(
+      "SELECT id FROM lattice_capability_bundles WHERE tenant_id = $1 AND id = $2",
+      [tenantId, id]
+    );
+    return result.rows[0] ? { status: "conflict" } : null;
+  }
   async deleteIfUnreferenced(tenantId, id) {
     await this.ready();
     const client = await this.pool.connect();
@@ -10928,6 +10970,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(createMenuItemsTable);
   mm.register(createSharedResourcesTable);
   mm.register(addFileContentType);
+  mm.register(addMcpAppContentType);
   mm.register(createCollectionsTable);
   mm.register(createTaskWorkItemsMigration);
   mm.register(addWorkItemProjectFieldsMigration);
@@ -11588,6 +11631,7 @@ export {
   addHoldoutToEvalRuns,
   addInterruptColumnsToEval,
   addInterruptPolicyToEvalCases,
+  addMcpAppContentType,
   addProjectConfigColumn,
   addScheduleTenantId,
   addSkillTenantId,