@mastra/pg 1.9.4-alpha.0 → 1.9.4

package/dist/index.js

@@ -1,5 +1,5 @@
 import { MastraError, ErrorCategory, ErrorDomain } from '@mastra/core/error';
-import { createVectorErrorId, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_SCHEMAS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, DatasetsStorage, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, DATASETS_SCHEMA, TABLE_CONFIGS, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, ensureDate, safelyParseJSON, ExperimentsStorage, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, ScorerDefinitionsStorage, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, TABLE_SCORERS, SkillsStorage, TABLE_SKILLS, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, TraceStatus, getDefaultValue, transformScoreRow as transformScoreRow$1, getSqlType } from '@mastra/core/storage';
+import { createVectorErrorId, AgentsStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_SCHEMAS, createStorageErrorId, normalizePerPage, calculatePagination, BackgroundTasksStorage, TABLE_BACKGROUND_TASKS, BlobStore, TABLE_SKILL_BLOBS, ChannelsStorage, TABLE_CHANNEL_INSTALLATIONS, TABLE_CHANNEL_CONFIG, DatasetsStorage, TABLE_DATASETS, TABLE_DATASET_ITEMS, TABLE_DATASET_VERSIONS, DATASETS_SCHEMA, TABLE_CONFIGS, DATASET_ITEMS_SCHEMA, DATASET_VERSIONS_SCHEMA, ensureDate, safelyParseJSON, ExperimentsStorage, TABLE_EXPERIMENTS, TABLE_EXPERIMENT_RESULTS, EXPERIMENTS_SCHEMA, EXPERIMENT_RESULTS_SCHEMA, MCPClientsStorage, TABLE_MCP_CLIENTS, TABLE_MCP_CLIENT_VERSIONS, MCPServersStorage, TABLE_MCP_SERVERS, TABLE_MCP_SERVER_VERSIONS, MemoryStorage, TABLE_THREADS, TABLE_MESSAGES, TABLE_RESOURCES, ObservabilityStorage, TABLE_SPANS, listTracesArgsSchema, toTraceSpans, PromptBlocksStorage, TABLE_PROMPT_BLOCKS, TABLE_PROMPT_BLOCK_VERSIONS, ScorerDefinitionsStorage, TABLE_SCORER_DEFINITIONS, TABLE_SCORER_DEFINITION_VERSIONS, ScoresStorage, TABLE_SCORERS, SkillsStorage, TABLE_SKILLS, TABLE_SKILL_VERSIONS, WorkflowsStorage, TABLE_WORKFLOW_SNAPSHOT, WorkspacesStorage, TABLE_WORKSPACES, TABLE_WORKSPACE_VERSIONS, MastraCompositeStore, TraceStatus, getDefaultValue, transformScoreRow as transformScoreRow$1, getSqlType } from '@mastra/core/storage';
 import { parseSqlIdentifier, parseFieldKey } from '@mastra/core/utils';
 import { MastraVector, validateTopK, validateUpsertInput } from '@mastra/core/vector';
 import { Mutex } from 'async-mutex';
@@ -4633,6 +4633,202 @@ var BlobsPG = class extends BlobStore {
     };
   }
 };
+var ChannelsPG = class _ChannelsPG extends ChannelsStorage {
+  #db;
+  #schema;
+  #skipDefaultIndexes;
+  #indexes;
+  static MANAGED_TABLES = [TABLE_CHANNEL_INSTALLATIONS, TABLE_CHANNEL_CONFIG];
+  constructor(config) {
+    super();
+    const { client, schemaName, skipDefaultIndexes, indexes } = resolvePgConfig(config);
+    this.#db = new PgDB({ client, schemaName, skipDefaultIndexes });
+    this.#schema = schemaName || "public";
+    this.#skipDefaultIndexes = skipDefaultIndexes;
+    this.#indexes = indexes?.filter((idx) => _ChannelsPG.MANAGED_TABLES.includes(idx.table));
+  }
+  async init() {
+    await this.#db.createTable({
+      tableName: TABLE_CHANNEL_INSTALLATIONS,
+      schema: TABLE_SCHEMAS[TABLE_CHANNEL_INSTALLATIONS]
+    });
+    await this.#db.createTable({
+      tableName: TABLE_CHANNEL_CONFIG,
+      schema: TABLE_SCHEMAS[TABLE_CHANNEL_CONFIG]
+    });
+    await this.createDefaultIndexes();
+    await this.createCustomIndexes();
+  }
+  static getDefaultIndexDefs(schemaPrefix) {
+    return [
+      {
+        name: `${schemaPrefix}idx_channel_installations_webhook`,
+        table: TABLE_CHANNEL_INSTALLATIONS,
+        columns: ["webhookId"],
+        unique: true
+      },
+      {
+        name: `${schemaPrefix}idx_channel_installations_platform_agent`,
+        table: TABLE_CHANNEL_INSTALLATIONS,
+        columns: ["platform", "agentId"]
+      }
+    ];
+  }
+  static getExportDDL(schemaName) {
+    const statements = [];
+    const parsedSchema = schemaName ? parseSqlIdentifier(schemaName, "schema name") : "";
+    const schemaPrefix = parsedSchema && parsedSchema !== "public" ? `${parsedSchema}_` : "";
+    for (const tableName of _ChannelsPG.MANAGED_TABLES) {
+      statements.push(
+        generateTableSQL({
+          tableName,
+          schema: TABLE_SCHEMAS[tableName],
+          schemaName,
+          includeAllConstraints: true
+        })
+      );
+    }
+    for (const idx of _ChannelsPG.getDefaultIndexDefs(schemaPrefix)) {
+      statements.push(generateIndexSQL(idx, schemaName));
+    }
+    return statements;
+  }
+  getDefaultIndexDefinitions() {
+    const schemaPrefix = this.#schema !== "public" ? `${this.#schema}_` : "";
+    return _ChannelsPG.getDefaultIndexDefs(schemaPrefix);
+  }
+  async createDefaultIndexes() {
+    if (this.#skipDefaultIndexes) return;
+    for (const indexDef of this.getDefaultIndexDefinitions()) {
+      try {
+        await this.#db.createIndex(indexDef);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create index ${indexDef.name}:`, error);
+      }
+    }
+  }
+  async createCustomIndexes() {
+    if (!this.#indexes || this.#indexes.length === 0) return;
+    for (const indexDef of this.#indexes) {
+      try {
+        await this.#db.createIndex(indexDef);
+      } catch (error) {
+        this.logger?.warn?.(`Failed to create custom index ${indexDef.name}:`, error);
+      }
+    }
+  }
+  async dangerouslyClearAll() {
+    await this.#db.clearTable({ tableName: TABLE_CHANNEL_INSTALLATIONS });
+    await this.#db.clearTable({ tableName: TABLE_CHANNEL_CONFIG });
+  }
+  async saveInstallation(installation) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_INSTALLATIONS, schemaName });
+    const now = (/* @__PURE__ */ new Date()).toISOString();
+    await this.#db.client.none(
+      `INSERT INTO ${tableName} ("id", "platform", "agentId", "status", "webhookId", "data", "configHash", "error", "createdAt", "updatedAt")
+       VALUES ($1, $2, $3, $4, $5, $6::jsonb, $7, $8, $9, $10)
+       ON CONFLICT ("id") DO UPDATE SET
+         "platform" = EXCLUDED."platform",
+         "agentId" = EXCLUDED."agentId",
+         "status" = EXCLUDED."status",
+         "webhookId" = EXCLUDED."webhookId",
+         "data" = EXCLUDED."data",
+         "configHash" = EXCLUDED."configHash",
+         "error" = EXCLUDED."error",
+         "updatedAt" = EXCLUDED."updatedAt"`,
+      [
+        installation.id,
+        installation.platform,
+        installation.agentId,
+        installation.status,
+        installation.webhookId ?? null,
+        JSON.stringify(installation.data),
+        installation.configHash ?? null,
+        installation.error ?? null,
+        installation.createdAt?.toISOString() ?? now,
+        now
+      ]
+    );
+  }
+  async getInstallation(id) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_INSTALLATIONS, schemaName });
+    const row = await this.#db.client.oneOrNone(`SELECT * FROM ${tableName} WHERE "id" = $1`, [id]);
+    return row ? this.#parseInstallationRow(row) : null;
+  }
+  async getInstallationByAgent(platform, agentId) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_INSTALLATIONS, schemaName });
+    const row = await this.#db.client.oneOrNone(
+      `SELECT * FROM ${tableName} WHERE "platform" = $1 AND "agentId" = $2 ORDER BY CASE "status" WHEN 'active' THEN 0 WHEN 'pending' THEN 1 ELSE 2 END, "updatedAt" DESC LIMIT 1`,
+      [platform, agentId]
+    );
+    return row ? this.#parseInstallationRow(row) : null;
+  }
+  async getInstallationByWebhookId(webhookId) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_INSTALLATIONS, schemaName });
+    const row = await this.#db.client.oneOrNone(`SELECT * FROM ${tableName} WHERE "webhookId" = $1`, [webhookId]);
+    return row ? this.#parseInstallationRow(row) : null;
+  }
+  async listInstallations(platform) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_INSTALLATIONS, schemaName });
+    const rows = await this.#db.client.manyOrNone(
+      `SELECT * FROM ${tableName} WHERE "platform" = $1 ORDER BY "createdAt" DESC`,
+      [platform]
+    );
+    return rows.map((row) => this.#parseInstallationRow(row));
+  }
+  async deleteInstallation(id) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_INSTALLATIONS, schemaName });
+    await this.#db.client.none(`DELETE FROM ${tableName} WHERE "id" = $1`, [id]);
+  }
+  async saveConfig(config) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_CONFIG, schemaName });
+    await this.#db.client.none(
+      `INSERT INTO ${tableName} ("platform", "data", "updatedAt")
+       VALUES ($1, $2::jsonb, $3)
+       ON CONFLICT ("platform") DO UPDATE SET
+         "data" = EXCLUDED."data",
+         "updatedAt" = EXCLUDED."updatedAt"`,
+      [config.platform, JSON.stringify(config.data), config.updatedAt.toISOString()]
+    );
+  }
+  async getConfig(platform) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_CONFIG, schemaName });
+    const row = await this.#db.client.oneOrNone(`SELECT * FROM ${tableName} WHERE "platform" = $1`, [platform]);
+    if (!row) return null;
+    return {
+      platform: row.platform,
+      data: typeof row.data === "string" ? JSON.parse(row.data) : row.data,
+      updatedAt: new Date(row.updatedAt)
+    };
+  }
+  async deleteConfig(platform) {
+    const schemaName = getSchemaName2(this.#schema);
+    const tableName = getTableName2({ indexName: TABLE_CHANNEL_CONFIG, schemaName });
+    await this.#db.client.none(`DELETE FROM ${tableName} WHERE "platform" = $1`, [platform]);
+  }
+  #parseInstallationRow(row) {
+    return {
+      id: row.id,
+      platform: row.platform,
+      agentId: row.agentId,
+      status: row.status,
+      webhookId: row.webhookId || void 0,
+      data: typeof row.data === "string" ? JSON.parse(row.data) : row.data,
+      configHash: row.configHash || void 0,
+      error: row.error || void 0,
+      createdAt: row.createdAt instanceof Date ? row.createdAt : new Date(row.createdAt),
+      updatedAt: row.updatedAt instanceof Date ? row.updatedAt : new Date(row.updatedAt)
+    };
+  }
+};
 function jsonbArg(value) {
   return value === void 0 || value === null ? null : JSON.stringify(value);
 }
@@ -14009,7 +14205,8 @@ var ALL_DOMAINS = [
   WorkflowsPG,
   DatasetsPG,
   ExperimentsPG,
-  BackgroundTasksPG
+  BackgroundTasksPG,
+  ChannelsPG
 ];
 function exportSchemas(schemaName) {
   const statements = [];
@@ -14064,7 +14261,8 @@ var PostgresStore = class extends MastraCompositeStore {
         blobs: new BlobsPG(domainConfig),
         datasets: new DatasetsPG(domainConfig),
         experiments: new ExperimentsPG(domainConfig),
-        backgroundTasks: new BackgroundTasksPG(domainConfig)
+        backgroundTasks: new BackgroundTasksPG(domainConfig),
+        channels: new ChannelsPG(domainConfig)
       };
     } catch (e) {
       throw new MastraError(
@@ -14253,6 +14451,6 @@ Example Complex Query:
   ]
 }`;
 
-export { AgentsPG, BackgroundTasksPG, BlobsPG, DatasetsPG, ExperimentsPG, MCPClientsPG, MCPServersPG, MemoryPG, ObservabilityPG, PGVECTOR_PROMPT, PgVector, PoolAdapter, PostgresStore, PromptBlocksPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
+export { AgentsPG, BackgroundTasksPG, BlobsPG, ChannelsPG, DatasetsPG, ExperimentsPG, MCPClientsPG, MCPServersPG, MemoryPG, ObservabilityPG, PGVECTOR_PROMPT, PgVector, PoolAdapter, PostgresStore, PromptBlocksPG, ScorerDefinitionsPG, ScoresPG, SkillsPG, WorkflowsPG, WorkspacesPG, exportSchemas };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map