@axiom-lattice/pg-stores 1.0.86 → 1.0.87

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@axiom-lattice/pg-stores",
-  "version": "1.0.86",
+  "version": "1.0.87",
   "description": "PG stores implementation for Axiom Lattice framework",
   "main": "dist/index.js",
   "module": "dist/index.mjs",
@@ -25,8 +25,8 @@
     "@langchain/core": "1.1.30",
     "pg": "^8.16.3",
     "uuid": "^9.0.1",
-    "@axiom-lattice/core": "2.1.95",
-    "@axiom-lattice/protocols": "2.1.49"
+    "@axiom-lattice/core": "2.1.96",
+    "@axiom-lattice/protocols": "2.1.50"
   },
   "devDependencies": {
     "@types/node": "^20.11.24",

package/src/createPgStoreConfig.ts

@@ -3,6 +3,7 @@ import { MigrationManager } from "./migrations/migration";
 import { PostgreSQLThreadStore } from "./stores/PostgreSQLThreadStore";
 import { PostgreSQLAssistantStore } from "./stores/PostgreSQLAssistantStore";
 import { PostgreSQLDatabaseConfigStore } from "./stores/PostgreSQLDatabaseConfigStore";
+import { PostgreSQLConnectionStore } from "./stores/PostgreSQLConnectionStore";
 import { PostgreSQLMetricsServerConfigStore } from "./stores/PostgreSQLMetricsServerConfigStore";
 import { PostgreSQLMcpServerConfigStore } from "./stores/PostgreSQLMcpServerConfigStore";
 import { PostgreSQLWorkspaceStore } from "./stores/PostgreSQLWorkspaceStore";
@@ -50,6 +51,7 @@ import { createChannelInstallationsTable } from "./migrations/channel_installati
 import { alterChannelInstallationsTable } from "./migrations/channel_installations_alter_migration";
 import { createChannelBindingsTable } from "./migrations/channel_bindings_migration";
 import { createDatabaseConfigsTable } from "./migrations/database_config_migrations";
+import { createConnectionConfigsTable } from "./migrations/connection_config_migrations";
 import { createMetricsConfigsTable } from "./migrations/metrics_config_migrations";
 import { createMcpServerConfigsTable } from "./migrations/mcp_server_config_migrations";
 import { createThreadMessageQueueTable } from "./migrations/thread_message_queue_migrations";
@@ -116,6 +118,7 @@ export async function createPgStoreConfig(connectionString: string) {
   mm.register(createSharedResourcesTable);       // v135
   mm.register(addFileContentType);               // v136
   mm.register(createCollectionsTable);           // v137
+  mm.register(createConnectionConfigsTable);      // v160
 
   await mm.migrate();
 
@@ -137,6 +140,7 @@ export async function createPgStoreConfig(connectionString: string) {
     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),

package/src/index.ts

@@ -15,6 +15,7 @@ export * from "./stores/PostgreSQLAssistantStore";
 export * from "./stores/PostgreSQLScheduleStorage";
 export * from "./stores/PostgreSQLSkillStore";
 export * from "./stores/PostgreSQLDatabaseConfigStore";
+export * from "./stores/PostgreSQLConnectionStore";
 export * from "./stores/PostgreSQLMetricsServerConfigStore";
 export * from "./stores/PostgreSQLMcpServerConfigStore";
 export * from "./stores/PostgreSQLWorkspaceStore";
@@ -74,6 +75,7 @@ export { PostgreSQLAssistantStore } from "./stores/PostgreSQLAssistantStore";
 export { PostgreSQLScheduleStorage } from "./stores/PostgreSQLScheduleStorage";
 export { PostgreSQLSkillStore } from "./stores/PostgreSQLSkillStore";
 export { PostgreSQLDatabaseConfigStore } from "./stores/PostgreSQLDatabaseConfigStore";
+export { PostgreSQLConnectionStore } from "./stores/PostgreSQLConnectionStore";
 export { PostgreSQLMetricsServerConfigStore } from "./stores/PostgreSQLMetricsServerConfigStore";
 export { PostgreSQLMcpServerConfigStore } from "./stores/PostgreSQLMcpServerConfigStore";
 export { PostgreSQLWorkspaceStore } from "./stores/PostgreSQLWorkspaceStore";
@@ -111,6 +113,8 @@ export type {
   UpdateDatabaseConfigRequest,
   DatabaseConfig,
   DatabaseType,
+  ConnectionStore,
+  ConnectionEntry,
   Binding,
   BindingRegistry,
   CreateBindingInput,

package/src/migrations/connection_config_migrations.ts

@@ -0,0 +1,50 @@
+/**
+ * Connection configs table migration
+ *
+ * Generic connection storage for plugin connections (ERP, CRM, etc.).
+ * Uses JSONB for config/metadata to accommodate any plugin type's shape
+ * without schema changes.
+ */
+import { PoolClient } from "pg";
+import { Migration } from "./migration";
+
+export const createConnectionConfigsTable: Migration = {
+  version: 160,
+  name: "create_connection_configs_table",
+  up: async (client: PoolClient) => {
+    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: PoolClient) => {
+    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");
+  },
+};

package/src/stores/PostgreSQLConnectionStore.ts

@@ -0,0 +1,128 @@
+/**
+ * PostgreSQL implementation of ConnectionStore
+ */
+
+import { Pool } from "pg";
+import type { ConnectionEntry, ConnectionStore } from "@axiom-lattice/protocols";
+import { v4 as uuid } from "uuid";
+
+interface ConnectionRow {
+  id: string;
+  tenant_id: string;
+  type: string;
+  key: string;
+  name: string;
+  description: string | null;
+  config: unknown;
+  status: string;
+  last_test_result: unknown;
+  metadata: unknown;
+  created_at: Date;
+  updated_at: Date;
+}
+
+const TABLE = "lattice_connection_configs";
+
+export class PostgreSQLConnectionStore implements ConnectionStore {
+  private db: Pool;
+
+  constructor(db: Pool) {
+    this.db = db;
+  }
+
+  async listByType(tenantId: string, type: string): Promise<ConnectionEntry[]> {
+    const result = await this.db.query<ConnectionRow>(
+      `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: string, type: string, key: string): Promise<ConnectionEntry | null> {
+    const result = await this.db.query<ConnectionRow>(
+      `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: Omit<ConnectionEntry, "id" | "createdAt" | "updatedAt">): Promise<ConnectionEntry> {
+    const id = uuid();
+    const now = 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: string, type: string, key: string, updates: Partial<ConnectionEntry>,
+  ): Promise<ConnectionEntry | null> {
+    const existing = await this.getByKey(tenantId, type, key);
+    if (!existing) return null;
+
+    const setClauses: string[] = [];
+    const values: unknown[] = [];
+    let paramIndex = 1;
+
+    if (updates.name !== undefined) {
+      setClauses.push(`name = $${paramIndex++}`);
+      values.push(updates.name);
+    }
+    if (updates.description !== undefined) {
+      setClauses.push(`description = $${paramIndex++}`);
+      values.push(updates.description || null);
+    }
+    if (updates.config !== undefined) {
+      setClauses.push(`config = $${paramIndex++}`);
+      values.push(JSON.stringify(updates.config));
+    }
+
+    if (setClauses.length === 0) return existing;
+
+    setClauses.push(`updated_at = $${paramIndex++}::timestamp`);
+    values.push(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: string, type: string, key: string): Promise<boolean> {
+    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;
+  }
+
+  private mapRow(row: ConnectionRow): ConnectionEntry {
+    return {
+      id: row.id,
+      tenantId: row.tenant_id,
+      type: row.type,
+      key: row.key,
+      name: row.name,
+      description: row.description || undefined,
+      config: (typeof row.config === "string" ? JSON.parse(row.config) : row.config) as Record<string, unknown>,
+      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),
+    };
+  }
+}