@axiom-lattice/pg-stores 1.0.86 → 1.0.87

package/dist/index.mjs

@@ -1247,6 +1247,101 @@ var PostgreSQLDatabaseConfigStore = class {
   }
 };
 
+// src/stores/PostgreSQLConnectionStore.ts
+import { v4 as uuid } from "uuid";
+var TABLE = "lattice_connection_configs";
+var PostgreSQLConnectionStore = class {
+  constructor(db) {
+    this.db = db;
+  }
+  async listByType(tenantId, type) {
+    const result = await this.db.query(
+      `SELECT * FROM ${TABLE} WHERE tenant_id = $1 AND type = $2 ORDER BY created_at`,
+      [tenantId, type]
+    );
+    return result.rows.map((row) => this.mapRow(row));
+  }
+  async getByKey(tenantId, type, key) {
+    const result = await this.db.query(
+      `SELECT * FROM ${TABLE} WHERE tenant_id = $1 AND type = $2 AND key = $3`,
+      [tenantId, type, key]
+    );
+    if (result.rows.length === 0) return null;
+    return this.mapRow(result.rows[0]);
+  }
+  async create(entry) {
+    const id = uuid();
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    await this.db.query(
+      `INSERT INTO ${TABLE} (id, tenant_id, type, key, name, description, config, status, created_at, updated_at)
+       VALUES ($1, $2, $3, $4, $5, $6, $7, 'active', $8::timestamp, $9::timestamp)`,
+      [
+        id,
+        entry.tenantId,
+        entry.type,
+        entry.key,
+        entry.name,
+        entry.description || null,
+        JSON.stringify(entry.config),
+        now,
+        now
+      ]
+    );
+    return { id, ...entry, createdAt: now, updatedAt: now };
+  }
+  async update(tenantId, type, key, updates) {
+    const existing = await this.getByKey(tenantId, type, key);
+    if (!existing) return null;
+    const setClauses = [];
+    const values = [];
+    let paramIndex = 1;
+    if (updates.name !== void 0) {
+      setClauses.push(`name = $${paramIndex++}`);
+      values.push(updates.name);
+    }
+    if (updates.description !== void 0) {
+      setClauses.push(`description = $${paramIndex++}`);
+      values.push(updates.description || null);
+    }
+    if (updates.config !== void 0) {
+      setClauses.push(`config = $${paramIndex++}`);
+      values.push(JSON.stringify(updates.config));
+    }
+    if (setClauses.length === 0) return existing;
+    setClauses.push(`updated_at = $${paramIndex++}::timestamp`);
+    values.push((/* @__PURE__ */ new Date()).toISOString());
+    values.push(tenantId, type, key);
+    const whereTenant = paramIndex++;
+    const whereType = paramIndex++;
+    const whereKey = paramIndex;
+    await this.db.query(
+      `UPDATE ${TABLE} SET ${setClauses.join(", ")} WHERE tenant_id = $${whereTenant} AND type = $${whereType} AND key = $${whereKey}`,
+      values
+    );
+    return this.getByKey(tenantId, type, key);
+  }
+  async delete(tenantId, type, key) {
+    const result = await this.db.query(
+      `DELETE FROM ${TABLE} WHERE tenant_id = $1 AND type = $2 AND key = $3`,
+      [tenantId, type, key]
+    );
+    return (result.rowCount ?? 0) > 0;
+  }
+  mapRow(row) {
+    return {
+      id: row.id,
+      tenantId: row.tenant_id,
+      type: row.type,
+      key: row.key,
+      name: row.name,
+      description: row.description || void 0,
+      config: typeof row.config === "string" ? JSON.parse(row.config) : row.config,
+      createdAt: row.created_at instanceof Date ? row.created_at.toISOString() : String(row.created_at),
+      updatedAt: row.updated_at instanceof Date ? row.updated_at.toISOString() : String(row.updated_at)
+    };
+  }
+};
+
 // src/stores/PostgreSQLMetricsServerConfigStore.ts
 import { Pool as Pool4 } from "pg";
 
@@ -7309,6 +7404,46 @@ var createChannelIdentityMappingTables = {
   }
 };
 
+// src/migrations/connection_config_migrations.ts
+var createConnectionConfigsTable = {
+  version: 160,
+  name: "create_connection_configs_table",
+  up: async (client) => {
+    await client.query(`
+      CREATE TABLE IF NOT EXISTS lattice_connection_configs (
+        id VARCHAR(255) NOT NULL,
+        tenant_id VARCHAR(255) NOT NULL,
+        type VARCHAR(255) NOT NULL,
+        key VARCHAR(255) NOT NULL,
+        name VARCHAR(255) NOT NULL,
+        description TEXT,
+        config JSONB NOT NULL DEFAULT '{}',
+        status VARCHAR(32) NOT NULL DEFAULT 'active',
+        last_test_result JSONB,
+        metadata JSONB NOT NULL DEFAULT '{}',
+        created_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+        updated_at TIMESTAMP NOT NULL DEFAULT CURRENT_TIMESTAMP,
+
+        PRIMARY KEY (tenant_id, id),
+        CONSTRAINT uk_lattice_connection_configs_tenant_type_key UNIQUE (tenant_id, type, key)
+      )
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_connection_configs_tenant_id
+      ON lattice_connection_configs(tenant_id)
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_connection_configs_tenant_type
+      ON lattice_connection_configs(tenant_id, type)
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP INDEX IF EXISTS idx_connection_configs_tenant_type");
+    await client.query("DROP INDEX IF EXISTS idx_connection_configs_tenant_id");
+    await client.query("DROP TABLE IF EXISTS lattice_connection_configs");
+  }
+};
+
 // src/migrations/collection_migrations.ts
 var createCollectionsTable = {
   version: 137,
@@ -7390,6 +7525,7 @@ async function createPgStoreConfig(connectionString) {
   mm.register(createSharedResourcesTable);
   mm.register(addFileContentType);
   mm.register(createCollectionsTable);
+  mm.register(createConnectionConfigsTable);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {
@@ -7407,6 +7543,7 @@ async function createPgStoreConfig(connectionString) {
     channelInstallation: new PostgreSQLChannelInstallationStore(opts),
     thread: new PostgreSQLThreadStore(opts),
     database: new PostgreSQLDatabaseConfigStore(opts),
+    connection: new PostgreSQLConnectionStore(pool),
     metrics: new PostgreSQLMetricsServerConfigStore(opts),
     mcp: new PostgreSQLMcpServerConfigStore(opts),
     assistant: new PostgreSQLAssistantStore(opts),
@@ -7981,6 +8118,7 @@ export {
   PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore,
   PostgreSQLCollectionStore,
+  PostgreSQLConnectionStore,
   PostgreSQLDatabaseConfigStore,
   PostgreSQLEvalStore,
   PostgreSQLMcpServerConfigStore,