@axiom-lattice/pg-stores 1.0.68 → 1.0.70

package/dist/index.js

@@ -33,7 +33,8 @@ __export(index_exports, {
   ChannelBindingStore: () => ChannelBindingStore,
   ChannelIdentityMappingStore: () => ChannelIdentityMappingStore,
   MigrationManager: () => MigrationManager,
-  Pool: () => import_pg19.Pool,
+  Pool: () => import_pg20.Pool,
+  PostgreSQLA2AApiKeyStore: () => PostgreSQLA2AApiKeyStore,
   PostgreSQLAssistantStore: () => PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore: () => PostgreSQLChannelInstallationStore,
   PostgreSQLDatabaseConfigStore: () => PostgreSQLDatabaseConfigStore,
@@ -51,6 +52,7 @@ __export(index_exports, {
   PostgreSQLWorkspaceStore: () => PostgreSQLWorkspaceStore,
   ThreadMessageQueueStore: () => ThreadMessageQueueStore,
   addAssistantTenantId: () => addAssistantTenantId,
+  addProjectConfigColumn: () => addProjectConfigColumn,
   addScheduleTenantId: () => addScheduleTenantId,
   addSkillTenantId: () => addSkillTenantId,
   addThreadTenantId: () => addThreadTenantId,
@@ -58,6 +60,7 @@ __export(index_exports, {
   changeAssistantPrimaryKey: () => changeAssistantPrimaryKey,
   changeSkillPrimaryKey: () => changeSkillPrimaryKey,
   changeThreadPrimaryKey: () => changeThreadPrimaryKey,
+  createA2AApiKeysTable: () => createA2AApiKeysTable,
   createAssistantsTable: () => createAssistantsTable,
   createChannelBindingsTable: () => createChannelBindingsTable,
   createChannelIdentityMappingTables: () => createChannelIdentityMappingTables,
@@ -83,7 +86,7 @@ __export(index_exports, {
   evalMigrations: () => evalMigrations
 });
 module.exports = __toCommonJS(index_exports);
-var import_pg19 = require("pg");
+var import_pg20 = require("pg");
 
 // src/stores/PostgreSQLThreadStore.ts
 var import_pg = require("pg");
@@ -2149,6 +2152,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) {
@@ -2161,6 +2187,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);
@@ -2205,6 +2232,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
     };
@@ -2216,7 +2244,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
@@ -2232,7 +2260,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
     `,
@@ -2251,15 +2279,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,
@@ -2267,12 +2296,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();
@@ -2291,6 +2325,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;
     }
@@ -5064,8 +5102,185 @@ var PostgreSQLChannelInstallationStore = class {
   }
 };
 
-// src/stores/PostgreSQLScheduleStorage.ts
+// src/stores/PostgreSQLA2AApiKeyStore.ts
 var import_pg16 = require("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
+var import_core5 = require("@axiom-lattice/core");
+var import_crypto2 = require("crypto");
+function generateApiKey() {
+  return `a2a_${(0, import_crypto2.randomUUID)().replace(/-/g, "")}`;
+}
+function mapRowToRecord(row) {
+  return {
+    id: row.id,
+    key: (0, import_core5.decrypt)(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 import_pg16.Pool({ connectionString: options.poolConfig }) : new import_pg16.Pool(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 ((0, import_core5.decrypt)(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 *`,
+      [(0, import_core5.encrypt)(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 *`,
+      [(0, import_core5.encrypt)(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 = (0, import_core5.decrypt)(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
+var import_pg17 = require("pg");
 var import_protocols = require("@axiom-lattice/protocols");
 
 // src/migrations/schedule_migrations.ts
@@ -5233,9 +5448,9 @@ var PostgreSQLScheduleStorage = class {
   constructor(options) {
     this.initialized = false;
     if (typeof options.poolConfig === "string") {
-      this.pool = new import_pg16.Pool({ connectionString: options.poolConfig });
+      this.pool = new import_pg17.Pool({ connectionString: options.poolConfig });
     } else {
-      this.pool = new import_pg16.Pool(options.poolConfig);
+      this.pool = new import_pg17.Pool(options.poolConfig);
     }
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createScheduledTasksTable);
@@ -5690,12 +5905,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
-var import_pg17 = require("pg");
+var import_pg18 = require("pg");
 
 // src/migrations/skill_migrations.ts
 var createSkillsTable = {
@@ -5857,9 +6073,9 @@ var PostgreSQLSkillStore = class {
     this.ownsPool = true;
     this.initPromise = null;
     if (typeof options.poolConfig === "string") {
-      this.pool = new import_pg17.Pool({ connectionString: options.poolConfig });
+      this.pool = new import_pg18.Pool({ connectionString: options.poolConfig });
     } else {
-      this.pool = new import_pg17.Pool(options.poolConfig);
+      this.pool = new import_pg18.Pool(options.poolConfig);
     }
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createSkillsTable);
@@ -6213,12 +6429,12 @@ var createChannelIdentityMappingTables = {
 };
 
 // src/stores/ChannelIdentityMappingStore.ts
-var import_pg18 = require("pg");
+var import_pg19 = require("pg");
 var ChannelIdentityMappingStore = class {
   constructor(options) {
     this.initialized = false;
     this.initPromise = null;
-    this.pool = typeof options.poolConfig === "string" ? new import_pg18.Pool({ connectionString: options.poolConfig }) : new import_pg18.Pool(options.poolConfig);
+    this.pool = typeof options.poolConfig === "string" ? new import_pg19.Pool({ connectionString: options.poolConfig }) : new import_pg19.Pool(options.poolConfig);
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createChannelIdentityMappingTables);
     if (options.autoMigrate !== false) {
@@ -6435,6 +6651,7 @@ function mapRowToChannelIdentityMapping(row) {
   ChannelIdentityMappingStore,
   MigrationManager,
   Pool,
+  PostgreSQLA2AApiKeyStore,
   PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore,
   PostgreSQLDatabaseConfigStore,
@@ -6452,6 +6669,7 @@ function mapRowToChannelIdentityMapping(row) {
   PostgreSQLWorkspaceStore,
   ThreadMessageQueueStore,
   addAssistantTenantId,
+  addProjectConfigColumn,
   addScheduleTenantId,
   addSkillTenantId,
   addThreadTenantId,
@@ -6459,6 +6677,7 @@ function mapRowToChannelIdentityMapping(row) {
   changeAssistantPrimaryKey,
   changeSkillPrimaryKey,
   changeThreadPrimaryKey,
+  createA2AApiKeysTable,
   createAssistantsTable,
   createChannelBindingsTable,
   createChannelIdentityMappingTables,