@axiom-lattice/pg-stores 1.0.55 → 1.0.57

package/dist/index.js

@@ -30,9 +30,10 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  ChannelBindingStore: () => ChannelBindingStore,
   ChannelIdentityMappingStore: () => ChannelIdentityMappingStore,
   MigrationManager: () => MigrationManager,
-  Pool: () => import_pg17.Pool,
+  Pool: () => import_pg18.Pool,
   PostgreSQLAssistantStore: () => PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore: () => PostgreSQLChannelInstallationStore,
   PostgreSQLDatabaseConfigStore: () => PostgreSQLDatabaseConfigStore,
@@ -53,10 +54,12 @@ __export(index_exports, {
   addScheduleTenantId: () => addScheduleTenantId,
   addSkillTenantId: () => addSkillTenantId,
   addThreadTenantId: () => addThreadTenantId,
+  alterChannelInstallationsTable: () => alterChannelInstallationsTable,
   changeAssistantPrimaryKey: () => changeAssistantPrimaryKey,
   changeSkillPrimaryKey: () => changeSkillPrimaryKey,
   changeThreadPrimaryKey: () => changeThreadPrimaryKey,
   createAssistantsTable: () => createAssistantsTable,
+  createChannelBindingsTable: () => createChannelBindingsTable,
   createChannelIdentityMappingTables: () => createChannelIdentityMappingTables,
   createChannelInstallationsTable: () => createChannelInstallationsTable,
   createDatabaseConfigsTable: () => createDatabaseConfigsTable,
@@ -80,7 +83,7 @@ __export(index_exports, {
   getThreadMessageQueueStore: () => getThreadMessageQueueStore
 });
 module.exports = __toCommonJS(index_exports);
-var import_pg17 = require("pg");
+var import_pg18 = require("pg");
 
 // src/stores/PostgreSQLThreadStore.ts
 var import_pg = require("pg");
@@ -4177,6 +4180,82 @@ var createChannelInstallationsTable = {
   }
 };
 
+// src/migrations/channel_installations_alter_migration.ts
+var alterChannelInstallationsTable = {
+  version: 109,
+  name: "alter_channel_installations_add_columns",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_channel_installations
+        ADD COLUMN IF NOT EXISTS enabled BOOLEAN NOT NULL DEFAULT true
+    `);
+    await client.query(`
+      ALTER TABLE lattice_channel_installations
+        ADD COLUMN IF NOT EXISTS fallback_agent_id VARCHAR(255)
+    `);
+    await client.query(`
+      ALTER TABLE lattice_channel_installations
+        ADD COLUMN IF NOT EXISTS reject_when_no_binding BOOLEAN NOT NULL DEFAULT true
+    `);
+  },
+  down: async (client) => {
+    await client.query(
+      "ALTER TABLE lattice_channel_installations DROP COLUMN IF EXISTS reject_when_no_binding"
+    );
+    await client.query(
+      "ALTER TABLE lattice_channel_installations DROP COLUMN IF EXISTS fallback_agent_id"
+    );
+    await client.query(
+      "ALTER TABLE lattice_channel_installations DROP COLUMN IF EXISTS enabled"
+    );
+  }
+};
+
+// src/migrations/channel_bindings_migration.ts
+var createChannelBindingsTable = {
+  version: 110,
+  name: "create_channel_bindings_table",
+  up: async (client) => {
+    await client.query(`
+      CREATE TABLE IF NOT EXISTS lattice_channel_bindings (
+        id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
+        channel VARCHAR(50) NOT NULL,
+        channel_installation_id VARCHAR(255) NOT NULL,
+        tenant_id VARCHAR(255) NOT NULL,
+        sender_id VARCHAR(255) NOT NULL,
+        agent_id VARCHAR(255) NOT NULL,
+        thread_id VARCHAR(255),
+        workspace_id VARCHAR(255),
+        project_id VARCHAR(255),
+        thread_mode VARCHAR(20) NOT NULL DEFAULT 'fixed'
+          CHECK (thread_mode IN ('fixed', 'per_conversation')),
+        sender_display_name VARCHAR(255),
+        sender_metadata JSONB,
+        enabled BOOLEAN NOT NULL DEFAULT true,
+        created_at TIMESTAMP NOT NULL DEFAULT NOW(),
+        updated_at TIMESTAMP NOT NULL DEFAULT NOW(),
+        UNIQUE (channel, channel_installation_id, tenant_id, sender_id)
+      )
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_channel_bindings_lookup
+      ON lattice_channel_bindings(channel, channel_installation_id, tenant_id, sender_id)
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_channel_bindings_agent
+      ON lattice_channel_bindings(agent_id, tenant_id)
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_channel_bindings_enabled
+      ON lattice_channel_bindings(enabled)
+      WHERE enabled = true
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP TABLE IF EXISTS lattice_channel_bindings");
+  }
+};
+
 // src/migrations/workflow_tracking_migrations.ts
 var createWorkflowTrackingTables = {
   name: "create_workflow_tracking_tables",
@@ -4250,6 +4329,190 @@ var createWorkflowTrackingTables = {
   }
 };
 
+// src/stores/ChannelBindingStore.ts
+var import_pg13 = require("pg");
+var ChannelBindingStore = class {
+  constructor(options) {
+    this.initialized = false;
+    this.initPromise = null;
+    this.pool = typeof options.poolConfig === "string" ? new import_pg13.Pool({ connectionString: options.poolConfig }) : new import_pg13.Pool(options.poolConfig);
+    this.migrationManager = new MigrationManager(this.pool);
+    this.migrationManager.register(createChannelBindingsTable);
+    if (options.autoMigrate !== false) {
+      this.initialize().catch((error) => {
+        console.error("Failed to initialize ChannelBindingStore:", 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 resolve(params) {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `SELECT * FROM lattice_channel_bindings
+       WHERE channel = $1
+         AND sender_id = $2
+         AND channel_installation_id = $3
+         AND tenant_id = $4
+         AND enabled = true
+       LIMIT 1`,
+      [params.channel, params.senderId, params.channelInstallationId, params.tenantId]
+    );
+    if (result.rows.length === 0) return null;
+    return this.mapRowToBinding(result.rows[0]);
+  }
+  async create(input) {
+    await this.ensureInitialized();
+    const result = await this.pool.query(
+      `INSERT INTO lattice_channel_bindings
+         (channel, channel_installation_id, tenant_id, sender_id, agent_id,
+          thread_mode, sender_display_name, sender_metadata, workspace_id, project_id)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
+       RETURNING *`,
+      [
+        input.channel,
+        input.channelInstallationId,
+        input.tenantId,
+        input.senderId,
+        input.agentId,
+        input.threadMode || "fixed",
+        input.senderDisplayName || null,
+        input.senderMetadata ? JSON.stringify(input.senderMetadata) : null,
+        input.workspaceId || null,
+        input.projectId || null
+      ]
+    );
+    return this.mapRowToBinding(result.rows[0]);
+  }
+  async update(id, patch) {
+    await this.ensureInitialized();
+    const existing = await this.pool.query(
+      `SELECT * FROM lattice_channel_bindings WHERE id = $1`,
+      [id]
+    );
+    if (existing.rows.length === 0) {
+      throw new Error(`Binding ${id} not found`);
+    }
+    const row = existing.rows[0];
+    const updated = {
+      channel: patch.channel ?? row.channel,
+      channel_installation_id: patch.channelInstallationId ?? row.channel_installation_id,
+      sender_id: patch.senderId ?? row.sender_id,
+      agent_id: patch.agentId ?? row.agent_id,
+      thread_id: patch.threadId !== void 0 ? patch.threadId : row.thread_id,
+      workspace_id: patch.workspaceId !== void 0 ? patch.workspaceId : row.workspace_id,
+      project_id: patch.projectId !== void 0 ? patch.projectId : row.project_id,
+      thread_mode: patch.threadMode ?? row.thread_mode,
+      sender_display_name: patch.senderDisplayName !== void 0 ? patch.senderDisplayName : row.sender_display_name,
+      sender_metadata: patch.senderMetadata !== void 0 ? patch.senderMetadata : row.sender_metadata,
+      enabled: patch.enabled ?? row.enabled
+    };
+    const result = await this.pool.query(
+      `UPDATE lattice_channel_bindings SET
+         channel = $1, channel_installation_id = $2, sender_id = $3,
+         agent_id = $4, thread_id = $5, workspace_id = $6, project_id = $7,
+         thread_mode = $8, sender_display_name = $9, sender_metadata = $10,
+         enabled = $11, updated_at = NOW()
+       WHERE id = $12
+       RETURNING *`,
+      [
+        updated.channel,
+        updated.channel_installation_id,
+        updated.sender_id,
+        updated.agent_id,
+        updated.thread_id,
+        updated.workspace_id,
+        updated.project_id,
+        updated.thread_mode,
+        updated.sender_display_name,
+        updated.sender_metadata ? JSON.stringify(updated.sender_metadata) : null,
+        updated.enabled,
+        id
+      ]
+    );
+    return this.mapRowToBinding(result.rows[0]);
+  }
+  async delete(id) {
+    await this.ensureInitialized();
+    await this.pool.query(`DELETE FROM lattice_channel_bindings WHERE id = $1`, [id]);
+  }
+  async list(params) {
+    await this.ensureInitialized();
+    const conditions = ["tenant_id = $1"];
+    const values = [params.tenantId];
+    let idx = 2;
+    if (params.channel) {
+      conditions.push(`channel = $${idx++}`);
+      values.push(params.channel);
+    }
+    if (params.agentId) {
+      conditions.push(`agent_id = $${idx++}`);
+      values.push(params.agentId);
+    }
+    if (params.channelInstallationId) {
+      conditions.push(`channel_installation_id = $${idx++}`);
+      values.push(params.channelInstallationId);
+    }
+    const limit = params.limit ?? 50;
+    const offset = params.offset ?? 0;
+    values.push(limit, offset);
+    const result = await this.pool.query(
+      `SELECT * FROM lattice_channel_bindings
+       WHERE ${conditions.join(" AND ")}
+       ORDER BY created_at DESC
+       LIMIT $${idx++} OFFSET $${idx}`,
+      values
+    );
+    return result.rows.map((r) => this.mapRowToBinding(r));
+  }
+  async import(bindings) {
+    const result = [];
+    for (const input of bindings) {
+      result.push(await this.create(input));
+    }
+    return result;
+  }
+  async export(params) {
+    return this.list({ tenantId: params.tenantId, limit: 1e4, offset: 0 });
+  }
+  async ensureInitialized() {
+    if (!this.initialized) {
+      await this.initialize();
+    }
+  }
+  mapRowToBinding(row) {
+    return {
+      id: row.id,
+      channel: row.channel,
+      channelInstallationId: row.channel_installation_id,
+      tenantId: row.tenant_id,
+      senderId: row.sender_id,
+      agentId: row.agent_id,
+      threadId: row.thread_id || void 0,
+      workspaceId: row.workspace_id || void 0,
+      projectId: row.project_id || void 0,
+      threadMode: row.thread_mode,
+      senderDisplayName: row.sender_display_name || void 0,
+      senderMetadata: row.sender_metadata || void 0,
+      enabled: row.enabled,
+      createdAt: row.created_at,
+      updatedAt: row.updated_at
+    };
+  }
+};
+
 // src/stores/ThreadMessageQueueStore.ts
 var import_crypto = __toESM(require("crypto"));
 
@@ -4540,12 +4803,12 @@ var ThreadMessageQueueStore = class _ThreadMessageQueueStore {
 var getThreadMessageQueueStore = () => ThreadMessageQueueStore.getInstance();
 
 // src/stores/ChannelIdentityMappingStore.ts
-var import_pg13 = require("pg");
+var import_pg14 = require("pg");
 var ChannelIdentityMappingStore = class {
   constructor(options) {
     this.initialized = false;
     this.initPromise = null;
-    this.pool = typeof options.poolConfig === "string" ? new import_pg13.Pool({ connectionString: options.poolConfig }) : new import_pg13.Pool(options.poolConfig);
+    this.pool = typeof options.poolConfig === "string" ? new import_pg14.Pool({ connectionString: options.poolConfig }) : new import_pg14.Pool(options.poolConfig);
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createChannelIdentityMappingTables);
     if (options.autoMigrate !== false) {
@@ -4758,15 +5021,16 @@ function mapRowToChannelIdentityMapping(row) {
 }
 
 // src/stores/PostgreSQLChannelInstallationStore.ts
-var import_pg14 = require("pg");
+var import_pg15 = require("pg");
 var import_core4 = require("@axiom-lattice/core");
 var PostgreSQLChannelInstallationStore = class {
   constructor(options) {
     this.initialized = false;
     this.initPromise = null;
-    this.pool = typeof options.poolConfig === "string" ? new import_pg14.Pool({ connectionString: options.poolConfig }) : new import_pg14.Pool(options.poolConfig);
+    this.pool = typeof options.poolConfig === "string" ? new import_pg15.Pool({ connectionString: options.poolConfig }) : new import_pg15.Pool(options.poolConfig);
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createChannelInstallationsTable);
+    this.migrationManager.register(alterChannelInstallationsTable);
     if (options.autoMigrate !== false) {
       this.initialize().catch((error) => {
         console.error(
@@ -4798,7 +5062,7 @@ var PostgreSQLChannelInstallationStore = class {
     await this.ensureInitialized();
     const result = await this.pool.query(
       `
-      SELECT id, tenant_id, channel, name, config, created_at, updated_at
+      SELECT id, tenant_id, channel, name, config, enabled, fallback_agent_id, reject_when_no_binding, created_at, updated_at
       FROM lattice_channel_installations
       WHERE id = $1
       LIMIT 1
@@ -4814,7 +5078,7 @@ var PostgreSQLChannelInstallationStore = class {
     await this.ensureInitialized();
     const result = channel ? await this.pool.query(
       `
-          SELECT id, tenant_id, channel, name, config, created_at, updated_at
+          SELECT id, tenant_id, channel, name, config, enabled, fallback_agent_id, reject_when_no_binding, created_at, updated_at
           FROM lattice_channel_installations
           WHERE tenant_id = $1 AND channel = $2
           ORDER BY created_at DESC
@@ -4822,7 +5086,7 @@ var PostgreSQLChannelInstallationStore = class {
       [tenantId, channel]
     ) : await this.pool.query(
       `
-          SELECT id, tenant_id, channel, name, config, created_at, updated_at
+          SELECT id, tenant_id, channel, name, config, enabled, fallback_agent_id, reject_when_no_binding, created_at, updated_at
           FROM lattice_channel_installations
           WHERE tenant_id = $1
           ORDER BY created_at DESC
@@ -4838,8 +5102,8 @@ var PostgreSQLChannelInstallationStore = class {
     await this.pool.query(
       `
       INSERT INTO lattice_channel_installations (
-        id, tenant_id, channel, name, config, created_at, updated_at
-      ) VALUES ($1, $2, $3, $4, $5, $6, $7)
+        id, tenant_id, channel, name, config, enabled, fallback_agent_id, reject_when_no_binding, created_at, updated_at
+      ) VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10)
       `,
       [
         installationId,
@@ -4847,6 +5111,9 @@ var PostgreSQLChannelInstallationStore = class {
         data.channel,
         data.name || null,
         JSON.stringify(encryptedConfig),
+        data.enabled ?? true,
+        data.fallbackAgentId || null,
+        data.rejectWhenNoBinding ?? true,
         now,
         now
       ]
@@ -4857,6 +5124,9 @@ var PostgreSQLChannelInstallationStore = class {
       channel: data.channel,
       name: data.name,
       config: data.config,
+      enabled: data.enabled ?? true,
+      fallbackAgentId: data.fallbackAgentId,
+      rejectWhenNoBinding: data.rejectWhenNoBinding ?? true,
       createdAt: now,
       updatedAt: now
     };
@@ -4867,8 +5137,9 @@ var PostgreSQLChannelInstallationStore = class {
     if (!existing || existing.tenantId !== tenantId) {
       return null;
     }
+    const existingConfig = existing.config ?? {};
     const mergedConfig = {
-      ...existing.config,
+      ...existingConfig,
       ...updates.config || {}
     };
     const now = /* @__PURE__ */ new Date();
@@ -4877,12 +5148,18 @@ var PostgreSQLChannelInstallationStore = class {
       UPDATE lattice_channel_installations
       SET name = $1,
           config = $2,
-          updated_at = $3
-      WHERE tenant_id = $4 AND id = $5
+          enabled = $3,
+          fallback_agent_id = $4,
+          reject_when_no_binding = $5,
+          updated_at = $6
+      WHERE tenant_id = $7 AND id = $8
       `,
       [
         updates.name ?? existing.name ?? null,
         JSON.stringify(this.encryptSecrets(mergedConfig)),
+        updates.enabled ?? existing.enabled,
+        updates.fallbackAgentId !== void 0 ? updates.fallbackAgentId : existing.fallbackAgentId ?? null,
+        updates.rejectWhenNoBinding ?? existing.rejectWhenNoBinding,
         now,
         tenantId,
         installationId
@@ -4892,6 +5169,9 @@ var PostgreSQLChannelInstallationStore = class {
       ...existing,
       name: updates.name ?? existing.name,
       config: mergedConfig,
+      enabled: updates.enabled ?? existing.enabled,
+      fallbackAgentId: updates.fallbackAgentId !== void 0 ? updates.fallbackAgentId : existing.fallbackAgentId,
+      rejectWhenNoBinding: updates.rejectWhenNoBinding ?? existing.rejectWhenNoBinding,
       updatedAt: now
     };
   }
@@ -4920,6 +5200,9 @@ var PostgreSQLChannelInstallationStore = class {
       config: this.decryptSecrets(
         typeof row.config === "string" ? JSON.parse(row.config) : row.config
       ),
+      enabled: row.enabled,
+      fallbackAgentId: row.fallback_agent_id || void 0,
+      rejectWhenNoBinding: row.reject_when_no_binding,
       createdAt: row.created_at,
       updatedAt: row.updated_at
     };
@@ -4934,28 +5217,24 @@ var PostgreSQLChannelInstallationStore = class {
   }
   decryptSecrets(config) {
     return {
-      appId: String(config.appId || ""),
-      appSecret: typeof config.appSecret === "string" ? (0, import_core4.decrypt)(config.appSecret) : "",
-      verificationToken: typeof config.verificationToken === "string" ? (0, import_core4.decrypt)(config.verificationToken) : void 0,
-      encryptKey: typeof config.encryptKey === "string" ? (0, import_core4.decrypt)(config.encryptKey) : void 0,
-      mappingMode: config.mappingMode === "user" || config.mappingMode === "group" || config.mappingMode === "hybrid" ? config.mappingMode : "hybrid",
-      assistantId: String(config.assistantId || ""),
-      workspaceId: typeof config.workspaceId === "string" ? config.workspaceId : void 0,
-      projectId: typeof config.projectId === "string" ? config.projectId : void 0
+      ...config,
+      appSecret: typeof config.appSecret === "string" ? (0, import_core4.decrypt)(config.appSecret) : config.appSecret,
+      verificationToken: typeof config.verificationToken === "string" ? (0, import_core4.decrypt)(config.verificationToken) : config.verificationToken,
+      encryptKey: typeof config.encryptKey === "string" ? (0, import_core4.decrypt)(config.encryptKey) : config.encryptKey
     };
   }
 };
 
 // src/stores/PostgreSQLWorkflowTrackingStore.ts
-var import_pg15 = require("pg");
+var import_pg16 = require("pg");
 var PostgreSQLWorkflowTrackingStore = class {
   constructor(options) {
     this.initialized = false;
     this.initPromise = null;
     if (typeof options.poolConfig === "string") {
-      this.pool = new import_pg15.Pool({ connectionString: options.poolConfig });
+      this.pool = new import_pg16.Pool({ connectionString: options.poolConfig });
     } else {
-      this.pool = new import_pg15.Pool(options.poolConfig);
+      this.pool = new import_pg16.Pool(options.poolConfig);
     }
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createWorkflowTrackingTables);
@@ -5222,7 +5501,7 @@ var PostgreSQLWorkflowTrackingStore = class {
 };
 
 // src/stores/PostgreSQLEvalStore.ts
-var import_pg16 = require("pg");
+var import_pg17 = require("pg");
 
 // src/migrations/eval_migrations.ts
 var createEvalProjectsTable = {
@@ -5422,9 +5701,9 @@ var PostgreSQLEvalStore = class {
     this.ownsPool = true;
     this.initPromise = null;
     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);
     for (const m of evalMigrations) {
@@ -6143,6 +6422,7 @@ var PostgreSQLEvalStore = class {
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  ChannelBindingStore,
   ChannelIdentityMappingStore,
   MigrationManager,
   Pool,
@@ -6166,10 +6446,12 @@ var PostgreSQLEvalStore = class {
   addScheduleTenantId,
   addSkillTenantId,
   addThreadTenantId,
+  alterChannelInstallationsTable,
   changeAssistantPrimaryKey,
   changeSkillPrimaryKey,
   changeThreadPrimaryKey,
   createAssistantsTable,
+  createChannelBindingsTable,
   createChannelIdentityMappingTables,
   createChannelInstallationsTable,
   createDatabaseConfigsTable,