@mastra/mssql 1.5.1-alpha.0 → 1.6.0-alpha.0

package/dist/index.js

@@ -1,5 +1,5 @@
 import { ErrorCategory, ErrorDomain, MastraError } from "@mastra/core/error";
-import { AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, AgentsStorage, BackgroundTasksStorage, MastraCompositeStore, MemoryStorage, ObservabilityStorage, SPAN_SCHEMA, ScoresStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_SNAPSHOT, TraceStatus, WorkflowsStorage, calculatePagination, createStorageErrorId, getDefaultValue, listTracesArgsSchema, normalizePerPage, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
+import { AGENTS_SCHEMA, AGENT_VERSIONS_SCHEMA, AgentsStorage, BackgroundTasksStorage, MastraCompositeStore, MemoryStorage, ObservabilityStorage, SPAN_SCHEMA, ScoresStorage, TABLE_AGENTS, TABLE_AGENT_VERSIONS, TABLE_BACKGROUND_TASKS, TABLE_MESSAGES, TABLE_RESOURCES, TABLE_SCHEMAS, TABLE_SCORERS, TABLE_SPANS, TABLE_THREADS, TABLE_WORKFLOW_DEFINITIONS, TABLE_WORKFLOW_SNAPSHOT, TraceStatus, WORKFLOW_DEFINITIONS_SCHEMA, WorkflowDefinitionsStorage, WorkflowsStorage, calculatePagination, createStorageErrorId, getDefaultValue, listTracesArgsSchema, normalizePerPage, toTraceSpans, transformScoreRow, validateStorageMetadataFilter } from "@mastra/core/storage";
 import sql from "mssql";
 import { randomUUID } from "crypto";
 import { parseSqlIdentifier } from "@mastra/core/utils";
@@ -4087,6 +4087,197 @@ var ScoresMSSQL = class ScoresMSSQL extends ScoresStorage {
 	}
 };
 //#endregion
+//#region src/storage/domains/workflow-definitions/index.ts
+function parseJson(value, column, rowId) {
+	if (value == null) return null;
+	if (typeof value === "string") try {
+		return JSON.parse(value);
+	} catch {
+		throw new Error(`Workflow definition row "${String(rowId)}" has malformed JSON in column "${column}".`);
+	}
+	return value;
+}
+function rowToDefinition(row) {
+	const inputSchema = parseJson(row.inputSchema, "inputSchema", row.id);
+	const outputSchema = parseJson(row.outputSchema, "outputSchema", row.id);
+	const graph = parseJson(row.graph, "graph", row.id);
+	if (inputSchema == null || outputSchema == null || graph == null) throw new Error(`Workflow definition row "${String(row.id)}" is missing required JSON columns.`);
+	const def = {
+		id: String(row.id),
+		inputSchema,
+		outputSchema,
+		graph,
+		status: String(row.status),
+		source: String(row.source),
+		createdAt: row.createdAt instanceof Date ? row.createdAt : new Date(row.createdAt),
+		updatedAt: row.updatedAt instanceof Date ? row.updatedAt : new Date(row.updatedAt)
+	};
+	if (row.description != null) def.description = String(row.description);
+	const metadata = parseJson(row.metadata, "metadata", row.id);
+	if (metadata != null) def.metadata = metadata;
+	const stateSchema = parseJson(row.stateSchema, "stateSchema", row.id);
+	if (stateSchema != null) def.stateSchema = stateSchema;
+	const requestContextSchema = parseJson(row.requestContextSchema, "requestContextSchema", row.id);
+	if (requestContextSchema != null) def.requestContextSchema = requestContextSchema;
+	if (row.authorId != null) def.authorId = String(row.authorId);
+	return def;
+}
+var WorkflowDefinitionsMSSQL = class WorkflowDefinitionsMSSQL extends WorkflowDefinitionsStorage {
+	pool;
+	schema;
+	db;
+	needsConnect;
+	skipDefaultIndexes;
+	indexes;
+	static MANAGED_TABLES = [TABLE_WORKFLOW_DEFINITIONS];
+	constructor(config) {
+		super();
+		const { pool, schemaName, skipDefaultIndexes, indexes, needsConnect } = resolveMssqlConfig(config);
+		this.pool = pool;
+		this.schema = schemaName;
+		this.db = new MssqlDB({
+			pool,
+			schemaName,
+			skipDefaultIndexes
+		});
+		this.needsConnect = needsConnect;
+		this.skipDefaultIndexes = skipDefaultIndexes;
+		this.indexes = indexes?.filter((idx) => WorkflowDefinitionsMSSQL.MANAGED_TABLES.includes(idx.table));
+	}
+	async init() {
+		if (this.needsConnect) {
+			await this.pool.connect();
+			this.needsConnect = false;
+		}
+		await this.db.createTable({
+			tableName: TABLE_WORKFLOW_DEFINITIONS,
+			schema: WORKFLOW_DEFINITIONS_SCHEMA
+		});
+		await this.createDefaultIndexes();
+		await this.createCustomIndexes();
+	}
+	getDefaultIndexDefinitions() {
+		return [{
+			name: `${this.schema && this.schema !== "dbo" ? `${this.schema}_` : ""}mastra_workflow_definitions_status_idx`,
+			table: TABLE_WORKFLOW_DEFINITIONS,
+			columns: ["status"]
+		}];
+	}
+	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_WORKFLOW_DEFINITIONS });
+	}
+	async upsert(input) {
+		const now = /* @__PURE__ */ new Date();
+		if (!await this.get(input.id)) {
+			if (!("inputSchema" in input) || input.inputSchema === void 0) throw new Error(`Cannot create workflow definition "${input.id}": inputSchema is required.`);
+			if (!("outputSchema" in input) || input.outputSchema === void 0) throw new Error(`Cannot create workflow definition "${input.id}": outputSchema is required.`);
+			if (!("graph" in input) || input.graph === void 0) throw new Error(`Cannot create workflow definition "${input.id}": graph is required.`);
+			const record = {
+				id: input.id,
+				description: input.description ?? null,
+				metadata: input.metadata ?? null,
+				inputSchema: input.inputSchema,
+				outputSchema: input.outputSchema,
+				stateSchema: input.stateSchema ?? null,
+				requestContextSchema: input.requestContextSchema ?? null,
+				graph: input.graph,
+				status: "active",
+				source: "storage",
+				authorId: "authorId" in input ? input.authorId ?? null : null,
+				createdAt: now,
+				updatedAt: now
+			};
+			try {
+				await this.db.insert({
+					tableName: TABLE_WORKFLOW_DEFINITIONS,
+					record
+				});
+			} catch (error) {
+				if (!await this.get(input.id)) throw error;
+				return this.applyUpdate(input, now);
+			}
+			const created = await this.get(input.id);
+			if (!created) throw new Error(`Failed to persist workflow definition "${input.id}".`);
+			return created;
+		}
+		return this.applyUpdate(input, now);
+	}
+	async applyUpdate(input, now) {
+		const data = { updatedAt: now };
+		if ("description" in input && input.description !== void 0) data.description = input.description;
+		if ("metadata" in input && input.metadata !== void 0) data.metadata = input.metadata;
+		if ("inputSchema" in input && input.inputSchema !== void 0) data.inputSchema = input.inputSchema;
+		if ("outputSchema" in input && input.outputSchema !== void 0) data.outputSchema = input.outputSchema;
+		if ("stateSchema" in input && input.stateSchema !== void 0) data.stateSchema = input.stateSchema;
+		if ("requestContextSchema" in input && input.requestContextSchema !== void 0) data.requestContextSchema = input.requestContextSchema;
+		if ("graph" in input && input.graph !== void 0) data.graph = input.graph;
+		if ("status" in input && input.status !== void 0) data.status = input.status;
+		if ("authorId" in input && input.authorId !== void 0) data.authorId = input.authorId;
+		await this.db.update({
+			tableName: TABLE_WORKFLOW_DEFINITIONS,
+			keys: { id: input.id },
+			data
+		});
+		const updated = await this.get(input.id);
+		if (!updated) throw new Error(`Failed to update workflow definition "${input.id}".`);
+		return updated;
+	}
+	async get(id) {
+		const row = await this.db.load({
+			tableName: TABLE_WORKFLOW_DEFINITIONS,
+			keys: { id }
+		});
+		return row ? rowToDefinition(row) : null;
+	}
+	async list(args) {
+		const tableName = getTableName({
+			indexName: TABLE_WORKFLOW_DEFINITIONS,
+			schemaName: getSchemaName(this.schema)
+		});
+		const request = this.pool.request();
+		const conditions = [];
+		if (args?.status) {
+			request.input("status", args.status);
+			conditions.push(`[status] = @status`);
+		}
+		if (args?.authorId !== void 0) {
+			request.input("authorId", args.authorId);
+			conditions.push(`[authorId] = @authorId`);
+		}
+		const where = conditions.length ? `WHERE ${conditions.join(" AND ")}` : "";
+		const definitions = (await request.query(`SELECT * FROM ${tableName} ${where} ORDER BY [updatedAt] DESC`)).recordset.map(rowToDefinition);
+		return {
+			definitions,
+			total: definitions.length
+		};
+	}
+	async delete(id) {
+		const tableName = getTableName({
+			indexName: TABLE_WORKFLOW_DEFINITIONS,
+			schemaName: getSchemaName(this.schema)
+		});
+		const request = this.pool.request();
+		request.input("id", id);
+		await request.query(`DELETE FROM ${tableName} WHERE [id] = @id`);
+	}
+};
+//#endregion
 //#region src/storage/domains/workflows/index.ts
 var WorkflowsMSSQL = class WorkflowsMSSQL extends WorkflowsStorage {
 	pool;
@@ -4558,13 +4749,15 @@ var MSSQLStore = class extends MastraCompositeStore {
 			const observability = new ObservabilityMSSQL(domainConfig);
 			const backgroundTasks = new BackgroundTasksMSSQL(domainConfig);
 			const agents = new AgentsMSSQL(domainConfig);
+			const workflowDefinitions = new WorkflowDefinitionsMSSQL(domainConfig);
 			this.stores = {
 				scores,
 				workflows,
 				memory,
 				observability,
 				backgroundTasks,
-				agents
+				agents,
+				workflowDefinitions
 			};
 		} catch (e) {
 			throw new MastraError({
@@ -4607,6 +4800,6 @@ var MSSQLStore = class extends MastraCompositeStore {
 	}
 };
 //#endregion
-export { AgentsMSSQL, BackgroundTasksMSSQL, MSSQLStore, MemoryMSSQL, ObservabilityMSSQL, ScoresMSSQL, WorkflowsMSSQL };
+export { AgentsMSSQL, BackgroundTasksMSSQL, MSSQLStore, MemoryMSSQL, ObservabilityMSSQL, ScoresMSSQL, WorkflowDefinitionsMSSQL, WorkflowsMSSQL };
 
 //# sourceMappingURL=index.js.map