@axiom-lattice/pg-stores 3.3.3 → 3.3.5

package/dist/index.mjs

@@ -7845,11 +7845,11 @@ var PostgreSQLTaskStore = class {
       params.push(filter.ownerId);
     }
     if (filter.status) {
-      const statuses = filter.status.split(",").map((s) => s.trim()).filter(Boolean);
-      const placeholders = statuses.map((_, i) => `$${paramIndex + i}`).join(", ");
+      const statuses2 = filter.status.split(",").map((s) => s.trim()).filter(Boolean);
+      const placeholders = statuses2.map((_, i) => `$${paramIndex + i}`).join(", ");
       conditions.push(`status IN (${placeholders})`);
-      params.push(...statuses);
-      paramIndex += statuses.length;
+      params.push(...statuses2);
+      paramIndex += statuses2.length;
     }
     if (filter.priority) {
       conditions.push(`priority = $${paramIndex++}`);
@@ -10152,6 +10152,60 @@ var createOutboxTable = {
   }
 };
 
+// src/migrations/plugin_deployment_migration.ts
+var createPluginDeployments = {
+  version: 189,
+  name: "create_plugin_deployments",
+  up: async (client) => {
+    await client.query(`
+      CREATE TABLE IF NOT EXISTS lattice_plugin_deployments (
+        id VARCHAR(255) PRIMARY KEY,
+        tenant_id VARCHAR(255) NOT NULL,
+        plugin_type VARCHAR(255) NOT NULL,
+        plugin_version VARCHAR(255) NOT NULL,
+        mode VARCHAR(16) NOT NULL,
+        status VARCHAR(32) NOT NULL,
+        provider VARCHAR(64) NOT NULL,
+        sandbox_name VARCHAR(255) NOT NULL,
+        port INTEGER NOT NULL,
+        endpoint TEXT,
+        image_ref TEXT,
+        auth_token TEXT NOT NULL,
+        idle_timeout_ms INTEGER NOT NULL,
+        last_health_at TIMESTAMPTZ,
+        last_error TEXT,
+        created_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+        updated_at TIMESTAMPTZ NOT NULL DEFAULT NOW(),
+        UNIQUE (tenant_id, plugin_type)
+      )
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_lattice_plugin_deployments_tenant
+      ON lattice_plugin_deployments (tenant_id)
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP INDEX IF EXISTS idx_lattice_plugin_deployments_tenant");
+    await client.query("DROP TABLE IF EXISTS lattice_plugin_deployments");
+  }
+};
+var dropPluginDeploymentAuthToken = {
+  version: 190,
+  name: "drop_plugin_deployment_auth_token",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_plugin_deployments
+      DROP COLUMN IF EXISTS auth_token
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_plugin_deployments
+      ADD COLUMN IF NOT EXISTS auth_token TEXT NOT NULL DEFAULT ''
+    `);
+  }
+};
+
 // src/migrations/model_provider_migrations.ts
 var createModelProvidersTable = {
   version: 181,
@@ -11935,6 +11989,145 @@ var PostgreSQLOutboxStore = class {
   }
 };
 
+// src/stores/PostgreSQLPluginDeploymentStore.ts
+var columns5 = "id, tenant_id, plugin_type, plugin_version, mode, status, provider, sandbox_name, port, endpoint, image_ref, idle_timeout_ms, last_health_at, last_error, created_at, updated_at";
+var statuses = [
+  "requested",
+  "provisioning",
+  "running",
+  "paused",
+  "stopping",
+  "stopped",
+  "failed"
+];
+function isStatus4(value) {
+  return typeof value === "string" && statuses.includes(value);
+}
+function mapRow8(row) {
+  if (typeof row.id !== "string" || typeof row.tenant_id !== "string" || typeof row.plugin_type !== "string" || typeof row.plugin_version !== "string" || row.mode !== "onDemand" && row.mode !== "alwaysOn" || !isStatus4(row.status) || typeof row.provider !== "string" || typeof row.sandbox_name !== "string" || typeof row.port !== "number" || typeof row.idle_timeout_ms !== "number" || !(row.created_at instanceof Date) || Number.isNaN(row.created_at.getTime()) || !(row.updated_at instanceof Date) || Number.isNaN(row.updated_at.getTime())) {
+    throw new Error("Invalid plugin deployment row");
+  }
+  return {
+    id: row.id,
+    tenantId: row.tenant_id,
+    pluginType: row.plugin_type,
+    pluginVersion: row.plugin_version,
+    mode: row.mode,
+    status: row.status,
+    provider: row.provider,
+    sandboxName: row.sandbox_name,
+    port: row.port,
+    ...typeof row.endpoint === "string" ? { endpoint: row.endpoint } : {},
+    ...typeof row.image_ref === "string" ? { imageRef: row.image_ref } : {},
+    idleTimeoutMs: row.idle_timeout_ms,
+    ...row.last_health_at instanceof Date && !Number.isNaN(row.last_health_at.getTime()) ? { lastHealthAt: row.last_health_at.toISOString() } : {},
+    ...typeof row.last_error === "string" ? { lastError: row.last_error } : {},
+    createdAt: row.created_at.toISOString(),
+    updatedAt: row.updated_at.toISOString()
+  };
+}
+var PostgreSQLPluginDeploymentStore = class {
+  /** Creates a store using an externally managed shared pool. */
+  constructor(options) {
+    this.pool = options.pool;
+  }
+  /** Lists every deployment for a tenant, ordered by plugin type. */
+  async listByTenant(tenantId) {
+    const result = await this.pool.query(
+      `SELECT ${columns5} FROM lattice_plugin_deployments WHERE tenant_id = $1 ORDER BY plugin_type`,
+      [tenantId]
+    );
+    return result.rows.map(mapRow8);
+  }
+  /** Returns the deployment for a tenant + plugin type, or undefined. */
+  async get(tenantId, pluginType) {
+    const result = await this.pool.query(
+      `SELECT ${columns5} FROM lattice_plugin_deployments WHERE tenant_id = $1 AND plugin_type = $2`,
+      [tenantId, pluginType]
+    );
+    return result.rows[0] ? mapRow8(result.rows[0]) : void 0;
+  }
+  /** Upserts a deployment keyed by tenant + plugin type, preserving the original id and createdAt. */
+  async upsert(deployment) {
+    const result = await this.pool.query(
+      `INSERT INTO lattice_plugin_deployments (${columns5})
+       VALUES ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16)
+       ON CONFLICT (tenant_id, plugin_type) DO UPDATE SET
+         plugin_version = EXCLUDED.plugin_version,
+         mode = EXCLUDED.mode,
+         status = EXCLUDED.status,
+         provider = EXCLUDED.provider,
+         sandbox_name = EXCLUDED.sandbox_name,
+         port = EXCLUDED.port,
+         endpoint = EXCLUDED.endpoint,
+         image_ref = EXCLUDED.image_ref,
+         idle_timeout_ms = EXCLUDED.idle_timeout_ms,
+         last_health_at = EXCLUDED.last_health_at,
+         last_error = EXCLUDED.last_error,
+         updated_at = NOW()
+       RETURNING ${columns5}`,
+      [
+        deployment.id,
+        deployment.tenantId,
+        deployment.pluginType,
+        deployment.pluginVersion,
+        deployment.mode,
+        deployment.status,
+        deployment.provider,
+        deployment.sandboxName,
+        deployment.port,
+        deployment.endpoint ?? null,
+        deployment.imageRef ?? null,
+        deployment.idleTimeoutMs,
+        deployment.lastHealthAt ? new Date(deployment.lastHealthAt) : null,
+        deployment.lastError ?? null,
+        new Date(deployment.createdAt),
+        new Date(deployment.updatedAt)
+      ]
+    );
+    return mapRow8(result.rows[0]);
+  }
+  /** Applies a partial update to mutable columns; returns undefined when no row exists. */
+  async updateStatus(tenantId, pluginType, patch) {
+    const assignments = [];
+    const params = [];
+    const assign = (column, value) => {
+      params.push(value);
+      assignments.push(`${column} = $${params.length}`);
+    };
+    if (patch.status !== void 0) assign("status", patch.status);
+    if (patch.mode !== void 0) assign("mode", patch.mode);
+    if (patch.endpoint !== void 0) assign("endpoint", patch.endpoint);
+    if (patch.imageRef !== void 0) assign("image_ref", patch.imageRef);
+    if (patch.idleTimeoutMs !== void 0) assign("idle_timeout_ms", patch.idleTimeoutMs);
+    if (patch.lastHealthAt !== void 0) assign("last_health_at", new Date(patch.lastHealthAt));
+    if (patch.lastError !== void 0) assign("last_error", patch.lastError);
+    if (patch.pluginVersion !== void 0) assign("plugin_version", patch.pluginVersion);
+    if (patch.provider !== void 0) assign("provider", patch.provider);
+    if (patch.sandboxName !== void 0) assign("sandbox_name", patch.sandboxName);
+    if (patch.port !== void 0) assign("port", patch.port);
+    assignments.push("updated_at = NOW()");
+    params.push(tenantId, pluginType);
+    const tenantParam = params.length - 1;
+    const pluginParam = params.length;
+    const result = await this.pool.query(
+      `UPDATE lattice_plugin_deployments SET ${assignments.join(", ")}
+       WHERE tenant_id = $${tenantParam} AND plugin_type = $${pluginParam}
+       RETURNING ${columns5}`,
+      params
+    );
+    return result.rows[0] ? mapRow8(result.rows[0]) : void 0;
+  }
+  /** Deletes a deployment by tenant + plugin type; returns whether a row was removed. */
+  async delete(tenantId, pluginType) {
+    const result = await this.pool.query(
+      `DELETE FROM lattice_plugin_deployments WHERE tenant_id = $1 AND plugin_type = $2`,
+      [tenantId, pluginType]
+    );
+    return (result.rowCount ?? 0) > 0;
+  }
+};
+
 // src/createPgStoreConfig.ts
 async function createPgStoreConfig(connectionString) {
   const pool = new Pool30({ connectionString });
@@ -12007,6 +12200,8 @@ async function createPgStoreConfig(connectionString) {
   mm.register(createDistributedLeases);
   mm.register(createOutboxTable);
   mm.register(addConnectionSecretFields);
+  mm.register(createPluginDeployments);
+  mm.register(dropPluginDeploymentAuthToken);
   await mm.migrate();
   const checkpoint = PostgresSaver.fromConnString(connectionString);
   checkpoint.setup().catch((err) => {
@@ -12050,6 +12245,7 @@ async function createPgStoreConfig(connectionString) {
     userPushSubscription: new PostgreSQLUserPushSubscriptionStore(opts),
     lease: new PostgreSQLLeaseStore(opts),
     outbox: new PostgreSQLOutboxStore(opts),
+    pluginDeployment: new PostgreSQLPluginDeploymentStore(opts),
     vectorStoreProvider: new PGVectorStoreProvider(pool, connectionString),
     checkpoint
   };
@@ -12624,6 +12820,7 @@ export {
   PostgreSQLMetricsServerConfigStore,
   PostgreSQLModelProviderStore,
   PostgreSQLOpenAuditStore,
+  PostgreSQLPluginDeploymentStore,
   PostgreSQLProjectBotMembershipStore,
   PostgreSQLProjectMembershipStore,
   PostgreSQLProjectRoomMessageStore,
@@ -12681,6 +12878,7 @@ export {
   createMetricsConfigsTable,
   createPGVectorStoreProvider,
   createPgStoreConfig,
+  createPluginDeployments,
   createProjectRoomReadStates,
   createProjectRoomTables,
   createProjectsTable,
@@ -12693,6 +12891,7 @@ export {
   createUsersTable,
   createWorkflowTrackingTables,
   createWorkspacesTable,
+  dropPluginDeploymentAuthToken,
   enforceUniqueEvalProjectName,
   evalMigrations,
   mapRowToRunStep,