@axiom-lattice/pg-stores 1.0.80 → 1.0.81

package/dist/index.mjs

@@ -1,5 +1,5 @@
 // src/index.ts
-import { Pool as Pool22 } from "pg";
+import { Pool as Pool23 } from "pg";
 
 // src/stores/PostgreSQLThreadStore.ts
 import { Pool } from "pg";
@@ -6231,7 +6231,7 @@ var createMenuItemsTable = {
         name VARCHAR(255) NOT NULL,
         icon VARCHAR(50),
         sort_order INT NOT NULL DEFAULT 0,
-        content_type VARCHAR(20) NOT NULL CHECK (content_type IN ('agent', 'html', 'custom')),
+         content_type VARCHAR(20) NOT NULL CHECK (content_type IN ('agent', 'html', 'custom', 'file')),
         content_config JSONB NOT NULL DEFAULT '{}',
         enabled BOOLEAN NOT NULL DEFAULT true,
         created_at TIMESTAMP NOT NULL DEFAULT NOW(),
@@ -6252,6 +6252,34 @@ var createMenuItemsTable = {
   }
 };
 
+// src/migrations/menu_items_add_file_type.ts
+var addFileContentType = {
+  version: 136,
+  name: "add_file_content_type_to_menu_items",
+  up: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_menu_items
+      DROP CONSTRAINT IF EXISTS lattice_menu_items_content_type_check
+    `);
+    await client.query(`
+      ALTER TABLE lattice_menu_items
+      ADD CONSTRAINT lattice_menu_items_content_type_check
+      CHECK (content_type IN ('agent', 'html', 'custom', 'file'))
+    `);
+  },
+  down: async (client) => {
+    await client.query(`
+      ALTER TABLE lattice_menu_items
+      DROP CONSTRAINT IF EXISTS lattice_menu_items_content_type_check
+    `);
+    await client.query(`
+      ALTER TABLE lattice_menu_items
+      ADD CONSTRAINT lattice_menu_items_content_type_check
+      CHECK (content_type IN ('agent', 'html', 'custom'))
+    `);
+  }
+};
+
 // src/stores/MenuStore.ts
 var MenuStore = class {
   constructor(options) {
@@ -6260,6 +6288,7 @@ var MenuStore = class {
     this.pool = typeof options.poolConfig === "string" ? new Pool19({ connectionString: options.poolConfig }) : new Pool19(options.poolConfig);
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createMenuItemsTable);
+    this.migrationManager.register(addFileContentType);
     if (options.autoMigrate !== false) {
       this.initialize().catch((error) => {
         console.error("Failed to initialize MenuStore:", error);
@@ -6403,6 +6432,215 @@ var MenuStore = class {
   }
 };
 
+// src/stores/PostgresSharedResourceStore.ts
+import { Pool as Pool20 } from "pg";
+
+// src/migrations/shared_resources_migration.ts
+var createSharedResourcesTable = {
+  version: 135,
+  name: "create_shared_resources_table",
+  up: async (client) => {
+    await client.query(`
+      CREATE TABLE IF NOT EXISTS lattice_shared_resources (
+        token         VARCHAR(64) PRIMARY KEY,
+        address       JSONB NOT NULL,
+        title         VARCHAR(256),
+        created_by    VARCHAR(64) NOT NULL,
+        tenant_id     VARCHAR(64) NOT NULL,
+        workspace_id  VARCHAR(64) NOT NULL,
+        project_id    VARCHAR(64) NOT NULL,
+        assistant_id  VARCHAR(64),
+        vm_isolation  VARCHAR(16) NOT NULL DEFAULT 'project',
+        visibility    VARCHAR(16) NOT NULL DEFAULT 'public',
+        password_hash VARCHAR(128),
+        expires_at    TIMESTAMPTZ,
+        max_access    INT,
+        access_count  INT DEFAULT 0,
+        revoked       BOOLEAN DEFAULT FALSE,
+        created_at    TIMESTAMPTZ DEFAULT NOW(),
+        version       INT DEFAULT 1
+      )
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_shared_resources_tenant
+      ON lattice_shared_resources(tenant_id)
+    `);
+    await client.query(`
+      CREATE INDEX IF NOT EXISTS idx_shared_resources_user
+      ON lattice_shared_resources(created_by)
+    `);
+  },
+  down: async (client) => {
+    await client.query("DROP TABLE IF EXISTS lattice_shared_resources");
+  }
+};
+
+// src/stores/PostgresSharedResourceStore.ts
+var PostgresSharedResourceStore = class {
+  constructor(options) {
+    this.initialized = false;
+    this.initPromise = null;
+    this.pool = typeof options.poolConfig === "string" ? new Pool20({ connectionString: options.poolConfig }) : new Pool20(options.poolConfig);
+    this.migrationManager = new MigrationManager(this.pool);
+    this.migrationManager.register(createSharedResourcesTable);
+    if (options.autoMigrate !== false) {
+      this.initialize().catch((error) => {
+        console.error(
+          "Failed to initialize PostgresSharedResourceStore:",
+          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 findByToken(token) {
+    await this.ensureInitialized();
+    const { rows } = await this.pool.query(
+      "SELECT * FROM lattice_shared_resources WHERE token = $1",
+      [token]
+    );
+    if (rows.length === 0) return null;
+    return this._mapRow(rows[0]);
+  }
+  async create(record) {
+    await this.ensureInitialized();
+    const { rows } = await this.pool.query(
+      `INSERT INTO lattice_shared_resources
+       (token, address, title, created_by, tenant_id, workspace_id, project_id,
+        assistant_id, vm_isolation, visibility, password_hash,
+        expires_at, max_access, revoked)
+       VALUES ($1,$2,$3,$4,$5,$6,$7,$8,$9,$10,$11,$12,$13,$14)
+       RETURNING *`,
+      [
+        record.token,
+        JSON.stringify(record.address),
+        record.title,
+        record.createdBy,
+        record.tenantId,
+        record.workspaceId,
+        record.projectId,
+        record.assistantId,
+        record.vmIsolation,
+        record.visibility,
+        record.passwordHash,
+        record.expiresAt ? record.expiresAt.toISOString() : null,
+        record.maxAccess,
+        record.revoked
+      ]
+    );
+    return this._mapRow(rows[0]);
+  }
+  async update(token, patch) {
+    await this.ensureInitialized();
+    const ALLOWED_COLUMNS = /* @__PURE__ */ new Set([
+      "title",
+      "expiresAt",
+      "maxAccess",
+      "revoked",
+      "visibility",
+      "passwordHash"
+    ]);
+    const COLUMN_MAP = {
+      "title": "title",
+      "expiresAt": "expires_at",
+      "maxAccess": "max_access",
+      "revoked": "revoked",
+      "visibility": "visibility",
+      "passwordHash": "password_hash"
+    };
+    const sets = [];
+    const vals = [];
+    let i = 1;
+    for (const [col, val] of Object.entries(patch)) {
+      const dbCol = COLUMN_MAP[col];
+      if (dbCol && val !== void 0) {
+        sets.push(`${dbCol} = $${i++}`);
+        vals.push(val);
+      }
+    }
+    if (sets.length === 0) return;
+    vals.push(token);
+    await this.pool.query(
+      `UPDATE lattice_shared_resources SET ${sets.join(", ")} WHERE token = $${i}`,
+      vals
+    );
+  }
+  async listByUser(tenantId, userId) {
+    await this.ensureInitialized();
+    const { rows } = await this.pool.query(
+      `SELECT * FROM lattice_shared_resources
+       WHERE tenant_id = $1 AND created_by = $2
+       ORDER BY created_at DESC`,
+      [tenantId, userId]
+    );
+    return rows.map((r) => this._mapRow(r));
+  }
+  async incrementAccess(token) {
+    await this.ensureInitialized();
+    await this.pool.query(
+      "UPDATE lattice_shared_resources SET access_count = access_count + 1 WHERE token = $1",
+      [token]
+    );
+  }
+  async atomicIncrementAccess(token) {
+    await this.ensureInitialized();
+    const { rows } = await this.pool.query(
+      `UPDATE lattice_shared_resources SET access_count = access_count + 1
+       WHERE token = $1 AND NOT revoked
+         AND (expires_at IS NULL OR expires_at > NOW())
+         AND (max_access IS NULL OR access_count < max_access)
+       RETURNING 1`,
+      [token]
+    );
+    return rows.length > 0;
+  }
+  async ensureInitialized() {
+    if (!this.initialized) {
+      await this.initialize();
+    }
+  }
+  _mapRow(row) {
+    let address;
+    if (typeof row.address === "string") {
+      address = JSON.parse(row.address);
+    } else {
+      address = row.address;
+    }
+    return {
+      token: row.token,
+      address,
+      title: row.title,
+      createdBy: row.created_by,
+      tenantId: row.tenant_id,
+      workspaceId: row.workspace_id,
+      projectId: row.project_id,
+      assistantId: row.assistant_id,
+      vmIsolation: row.vm_isolation,
+      visibility: row.visibility,
+      passwordHash: row.password_hash,
+      expiresAt: row.expires_at ? new Date(row.expires_at) : null,
+      maxAccess: row.max_access,
+      accessCount: row.access_count ?? 0,
+      revoked: row.revoked ?? false,
+      createdAt: new Date(row.created_at),
+      version: row.version ?? 1
+    };
+  }
+};
+
 // src/createPgStoreConfig.ts
 import { PostgresSaver } from "@langchain/langgraph-checkpoint-postgres";
 function createPgStoreConfig(connectionString) {
@@ -6431,12 +6669,13 @@ function createPgStoreConfig(connectionString) {
     a2aApiKey: new PostgreSQLA2AApiKeyStore(opts),
     schedule: new PostgreSQLScheduleStorage(opts),
     menu: new MenuStore(opts),
+    sharedResource: new PostgresSharedResourceStore(opts),
     checkpoint
   };
 }
 
 // src/stores/PostgreSQLSkillStore.ts
-import { Pool as Pool20 } from "pg";
+import { Pool as Pool21 } from "pg";
 
 // src/migrations/skill_migrations.ts
 var createSkillsTable = {
@@ -6598,9 +6837,9 @@ var PostgreSQLSkillStore = class {
     this.ownsPool = true;
     this.initPromise = null;
     if (typeof options.poolConfig === "string") {
-      this.pool = new Pool20({ connectionString: options.poolConfig });
+      this.pool = new Pool21({ connectionString: options.poolConfig });
     } else {
-      this.pool = new Pool20(options.poolConfig);
+      this.pool = new Pool21(options.poolConfig);
     }
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createSkillsTable);
@@ -6954,12 +7193,12 @@ var createChannelIdentityMappingTables = {
 };
 
 // src/stores/ChannelIdentityMappingStore.ts
-import { Pool as Pool21 } from "pg";
+import { Pool as Pool22 } from "pg";
 var ChannelIdentityMappingStore = class {
   constructor(options) {
     this.initialized = false;
     this.initPromise = null;
-    this.pool = typeof options.poolConfig === "string" ? new Pool21({ connectionString: options.poolConfig }) : new Pool21(options.poolConfig);
+    this.pool = typeof options.poolConfig === "string" ? new Pool22({ connectionString: options.poolConfig }) : new Pool22(options.poolConfig);
     this.migrationManager = new MigrationManager(this.pool);
     this.migrationManager.register(createChannelIdentityMappingTables);
     if (options.autoMigrate !== false) {
@@ -7175,7 +7414,7 @@ export {
   ChannelIdentityMappingStore,
   MenuStore,
   MigrationManager,
-  Pool22 as Pool,
+  Pool23 as Pool,
   PostgreSQLA2AApiKeyStore,
   PostgreSQLAssistantStore,
   PostgreSQLChannelInstallationStore,
@@ -7192,8 +7431,10 @@ export {
   PostgreSQLUserTenantLinkStore,
   PostgreSQLWorkflowTrackingStore,
   PostgreSQLWorkspaceStore,
+  PostgresSharedResourceStore,
   ThreadMessageQueueStore,
   addAssistantTenantId,
+  addFileContentType,
   addProjectConfigColumn,
   addScheduleTenantId,
   addSkillTenantId,
@@ -7220,6 +7461,7 @@ export {
   createPgStoreConfig,
   createProjectsTable,
   createScheduledTasksTable,
+  createSharedResourcesTable,
   createSkillsTable,
   createTenantsTable,
   createThreadMessageQueueTable,