@axiom-lattice/pg-stores 1.0.67 → 1.0.69

package/dist/index.mjs

@@ -1,5 +1,5 @@
 // src/index.ts
-import { Pool as Pool19 } from "pg";
+import { Pool as Pool20 } from "pg";
 
 // src/stores/PostgreSQLThreadStore.ts
 import { Pool } from "pg";
@@ -2065,6 +2065,29 @@ var createProjectsTable = {
   }
 };
 
+// src/migrations/add_project_config_column.ts
+var addProjectConfigColumn = {
+  version: 24,
+  name: "add_project_config_column",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_projects
+      ADD COLUMN IF NOT EXISTS config JSONB
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_projects_config 
+      ON lattice_projects USING GIN (config)
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP INDEX IF EXISTS idx_lattice_projects_config");
+    await client.query(`
+      ALTER TABLE lattice_projects
+      DROP COLUMN IF EXISTS config
+    `);
+  }
+};
+
 // src/stores/PostgreSQLProjectStore.ts
 var PostgreSQLProjectStore = class {
   constructor(options) {
@@ -2077,6 +2100,7 @@ var PostgreSQLProjectStore = class {
     }
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createProjectsTable);
+    this.migrationManager.register(addProjectConfigColumn);
     if (options.autoMigrate !== false) {
       this.initialize().catch((error) => {
         console.error("Failed to initialize PostgreSQLProjectStore:", error);
@@ -2121,6 +2145,7 @@ var PostgreSQLProjectStore = class {
       workspaceId: row.workspace_id,
       name: row.name,
       description: row.description || void 0,
+      config: row.config || void 0,
       createdAt: row.created_at,
       updatedAt: row.updated_at
     };
@@ -2132,7 +2157,7 @@ var PostgreSQLProjectStore = class {
     await this.ensureInitialized();
     const result = await this.pool.query(
       `
-      SELECT id, tenant_id, workspace_id, name, description, created_at, updated_at
+      SELECT id, tenant_id, workspace_id, name, description, config, created_at, updated_at
       FROM lattice_projects
       WHERE tenant_id = $1 AND workspace_id = $2
       ORDER BY created_at DESC
@@ -2148,7 +2173,7 @@ var PostgreSQLProjectStore = class {
     await this.ensureInitialized();
     const result = await this.pool.query(
       `
-      SELECT id, tenant_id, workspace_id, name, description, created_at, updated_at
+      SELECT id, tenant_id, workspace_id, name, description, config, created_at, updated_at
       FROM lattice_projects
       WHERE id = $1 AND tenant_id = $2
     `,
@@ -2167,15 +2192,16 @@ var PostgreSQLProjectStore = class {
     const now = /* @__PURE__ */ new Date();
     await this.pool.query(
       `
-      INSERT INTO lattice_projects (id, tenant_id, workspace_id, name, description, created_at, updated_at)
-      VALUES ($1, $2, $3, $4, $5, $6, $7)
+      INSERT INTO lattice_projects (id, tenant_id, workspace_id, name, description, config, created_at, updated_at)
+      VALUES ($1, $2, $3, $4, $5, $6, $7, $8)
       ON CONFLICT (id, tenant_id) DO UPDATE SET
         workspace_id = EXCLUDED.workspace_id,
         name = EXCLUDED.name,
         description = EXCLUDED.description,
+        config = EXCLUDED.config,
         updated_at = EXCLUDED.updated_at
     `,
-      [id, tenantId, workspaceId, data.name, data.description || null, now, now]
+      [id, tenantId, workspaceId, data.name, data.description || null, data.config || null, now, now]
     );
     return {
       id,
@@ -2183,12 +2209,17 @@ var PostgreSQLProjectStore = class {
       workspaceId,
       name: data.name,
       description: data.description,
+      config: data.config,
       createdAt: now,
       updatedAt: now
     };
   }
   /**
    * Update an existing project
+   *
+   * @remarks
+   * - The `config` field uses **replace** semantics: if provided, it completely
+   *   overwrites the existing config. To preserve the current config, omit this field.
    */
   async updateProject(tenantId, id, updates) {
     await this.ensureInitialized();
@@ -2207,6 +2238,10 @@ var PostgreSQLProjectStore = class {
       updateFields.push(`description = $${paramIndex++}`);
       updateValues.push(updates.description || null);
     }
+    if (updates.config !== void 0) {
+      updateFields.push(`config = $${paramIndex++}`);
+      updateValues.push(updates.config || null);
+    }
     if (updateFields.length === 0) {
       return existing;
     }
@@ -4980,8 +5015,185 @@ var PostgreSQLChannelInstallationStore = class {
   }
 };
 
-// src/stores/PostgreSQLScheduleStorage.ts
+// src/stores/PostgreSQLA2AApiKeyStore.ts
 import { Pool as Pool16 } from "pg";
+
+// src/migrations/a2a_api_key_migration.ts
+var createA2AApiKeysTable = {
+  version: 130,
+  name: "create_a2a_api_keys_table",
+  up: async (client) => {
+    await client.query(`
+      CREATE TABLE IF NOT EXISTS lattice_a2a_api_keys (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        key_value TEXT NOT NULL UNIQUE,
+        tenant_id VARCHAR(255) NOT NULL,
+        project_id VARCHAR(255),
+        workspace_id VARCHAR(255),
+        label VARCHAR(255),
+        enabled BOOLEAN NOT NULL DEFAULT true,
+        created_at TIMESTAMP NOT NULL DEFAULT NOW(),
+        updated_at TIMESTAMP NOT NULL DEFAULT NOW()
+      )
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_a2a_keys_tenant
+      ON lattice_a2a_api_keys(tenant_id)
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_a2a_keys_value
+      ON lattice_a2a_api_keys(key_value)
+      WHERE enabled = true
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP TABLE IF EXISTS lattice_a2a_api_keys");
+  }
+};
+
+// src/stores/PostgreSQLA2AApiKeyStore.ts
+import { encrypt as encrypt5, decrypt as decrypt5 } from "@axiom-lattice/core";
+import { randomUUID } from "crypto";
+function generateApiKey() {
+  return `a2a_${randomUUID().replace(/-/g, "")}`;
+}
+function mapRowToRecord(row) {
+  return {
+    id: row.id,
+    key: decrypt5(row.key_value),
+    tenantId: row.tenant_id,
+    projectId: row.project_id || void 0,
+    workspaceId: row.workspace_id || void 0,
+    label: row.label || void 0,
+    enabled: row.enabled,
+    createdAt: row.created_at,
+    updatedAt: row.updated_at
+  };
+}
+var PostgreSQLA2AApiKeyStore = class {
+  constructor(options) {
+    this.initialized = false;
+    this.initPromise = null;
+    this.pool = typeof options.poolConfig === "string" ? new Pool16({ connectionString: options.poolConfig }) : new Pool16(options.poolConfig);
+    this.migrationManager = new MigrationManager(this.pool);
+    this.migrationManager.register(createA2AApiKeysTable);
+    if (options.autoMigrate !== false) {
+      this.initialize().catch((error) => {
+        console.error("Failed to initialize PostgreSQLA2AApiKeyStore:", error);
+        throw error;
+      });
+    }
+  }
+  async initialize() {
+    if (this.initialized) return;
+    if (this.initPromise) return this.initPromise;
+    this.initPromise = (async () => {
+      try {
+        await this.migrationManager.migrate();
+        this.initialized = true;
+      } finally {
+        this.initPromise = null;
+      }
+    })();
+    return this.initPromise;
+  }
+  async ensureInitialized() {
+    if (!this.initialized) await this.initialize();
+  }
+  async findByKey(key) {
+    await this.ensureInitialized();
+    const all = await this.pool.query(
+      `SELECT * FROM lattice_a2a_api_keys WHERE enabled = true`
+    );
+    for (const row of all.rows) {
+      if (decrypt5(row.key_value) === key) return mapRowToRecord(row);
+    }
+    return null;
+  }
+  async list(params) {
+    await this.ensureInitialized();
+    const limit = params.limit || 100;
+    const offset = params.offset || 0;
+    if (params.tenantId) {
+      const result2 = await this.pool.query(
+        `SELECT * FROM lattice_a2a_api_keys WHERE tenant_id = $1 ORDER BY created_at DESC LIMIT $2 OFFSET $3`,
+        [params.tenantId, limit, offset]
+      );
+      return result2.rows.map(mapRowToRecord);
+    }
+    const result = await this.pool.query(
+      `SELECT * FROM lattice_a2a_api_keys ORDER BY created_at DESC LIMIT $1 OFFSET $2`,
+      [limit, offset]
+    );
+    return result.rows.map(mapRowToRecord);
+  }
+  async create(input) {
+    await this.ensureInitialized();
+    const key = generateApiKey();
+    const result = await this.pool.query(
+      `INSERT INTO lattice_a2a_api_keys (key_value, tenant_id, project_id, workspace_id, label)
+       VALUES ($1, $2, $3, $4, $5) RETURNING *`,
+      [encrypt5(key), input.tenantId, input.projectId || null, input.workspaceId || null, input.label || null]
+    );
+    const record = mapRowToRecord(result.rows[0]);
+    record.key = key;
+    return record;
+  }
+  async disable(id) {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `UPDATE lattice_a2a_api_keys SET enabled = false, updated_at = NOW() WHERE id = $1 RETURNING *`,
+      [id]
+    );
+    if (result.rows.length === 0) throw new Error(`A2A API key not found: ${id}`);
+    return mapRowToRecord(result.rows[0]);
+  }
+  async enable(id) {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `UPDATE lattice_a2a_api_keys SET enabled = true, updated_at = NOW() WHERE id = $1 RETURNING *`,
+      [id]
+    );
+    if (result.rows.length === 0) throw new Error(`A2A API key not found: ${id}`);
+    return mapRowToRecord(result.rows[0]);
+  }
+  async rotate(id) {
+    await this.ensureInitialized();
+    const key = generateApiKey();
+    const result = await this.pool.query(
+      `UPDATE lattice_a2a_api_keys SET key_value = $1, updated_at = NOW() WHERE id = $2 RETURNING *`,
+      [encrypt5(key), id]
+    );
+    if (result.rows.length === 0) throw new Error(`A2A API key not found: ${id}`);
+    const record = mapRowToRecord(result.rows[0]);
+    record.key = key;
+    return record;
+  }
+  async delete(id) {
+    await this.ensureInitialized();
+    await this.pool.query(`DELETE FROM lattice_a2a_api_keys WHERE id = $1`, [id]);
+  }
+  async loadIntoMap() {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `SELECT * FROM lattice_a2a_api_keys WHERE enabled = true`
+    );
+    const map = /* @__PURE__ */ new Map();
+    for (const row of result.rows) {
+      const key = decrypt5(row.key_value);
+      map.set(key, {
+        key,
+        tenantId: row.tenant_id,
+        projectId: row.project_id || void 0,
+        workspaceId: row.workspace_id || void 0
+      });
+    }
+    return map;
+  }
+};
+
+// src/stores/PostgreSQLScheduleStorage.ts
+import { Pool as Pool17 } from "pg";
 import {
   ScheduledTaskStatus
 } from "@axiom-lattice/protocols";
@@ -5151,9 +5363,9 @@ var PostgreSQLScheduleStorage = class {
   constructor(options) {
     this.initialized = false;
     if (typeof options.poolConfig === "string") {
-      this.pool = new Pool16({ connectionString: options.poolConfig });
+      this.pool = new Pool17({ connectionString: options.poolConfig });
     } else {
-      this.pool = new Pool16(options.poolConfig);
+      this.pool = new Pool17(options.poolConfig);
     }
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createScheduledTasksTable);
@@ -5608,12 +5820,13 @@ function createPgStoreConfig(connectionString) {
     assistant: new PostgreSQLAssistantStore(opts),
     workflowTracking: new PostgreSQLWorkflowTrackingStore(optsAuto),
     threadMessageQueue: new ThreadMessageQueueStore(optsAuto),
+    a2aApiKey: new PostgreSQLA2AApiKeyStore(optsAuto),
     schedule: new PostgreSQLScheduleStorage(opts)
   };
 }
 
 // src/stores/PostgreSQLSkillStore.ts
-import { Pool as Pool17 } from "pg";
+import { Pool as Pool18 } from "pg";
 
 // src/migrations/skill_migrations.ts
 var createSkillsTable = {
@@ -5775,9 +5988,9 @@ var PostgreSQLSkillStore = class {
     this.ownsPool = true;
     this.initPromise = null;
     if (typeof options.poolConfig === "string") {
-      this.pool = new Pool17({ connectionString: options.poolConfig });
+      this.pool = new Pool18({ connectionString: options.poolConfig });
     } else {
-      this.pool = new Pool17(options.poolConfig);
+      this.pool = new Pool18(options.poolConfig);
     }
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createSkillsTable);
@@ -6131,12 +6344,12 @@ var createChannelIdentityMappingTables = {
 };
 
 // src/stores/ChannelIdentityMappingStore.ts
-import { Pool as Pool18 } from "pg";
+import { Pool as Pool19 } from "pg";
 var ChannelIdentityMappingStore = class {
   constructor(options) {
     this.initialized = false;
     this.initPromise = null;
-    this.pool = typeof options.poolConfig === "string" ? new Pool18({ connectionString: options.poolConfig }) : new Pool18(options.poolConfig);
+    this.pool = typeof options.poolConfig === "string" ? new Pool19({ connectionString: options.poolConfig }) : new Pool19(options.poolConfig);
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createChannelIdentityMappingTables);
     if (options.autoMigrate !== false) {
@@ -6351,7 +6564,8 @@ export {
   ChannelBindingStore,
   ChannelIdentityMappingStore,
   MigrationManager,
-  Pool19 as Pool,
+  Pool20 as Pool,
+  PostgreSQLA2AApiKeyStore,
   PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore,
   PostgreSQLDatabaseConfigStore,
@@ -6369,6 +6583,7 @@ export {
   PostgreSQLWorkspaceStore,
   ThreadMessageQueueStore,
   addAssistantTenantId,
+  addProjectConfigColumn,
   addScheduleTenantId,
   addSkillTenantId,
   addThreadTenantId,
@@ -6376,6 +6591,7 @@ export {
   changeAssistantPrimaryKey,
   changeSkillPrimaryKey,
   changeThreadPrimaryKey,
+  createA2AApiKeysTable,
   createAssistantsTable,
   createChannelBindingsTable,
   createChannelIdentityMappingTables,