@keystrokehq/keystroke 0.0.100 → 0.0.102

package/dist/{dist-ByV6mHKd.cjs → dist-BkC9pJIu.cjs}

@@ -1,5 +1,5 @@
 const require_dist$1 = require("./dist-CLqJza2Y.cjs");
-const require_dist$2 = require("./dist-T7a5DjzV.cjs");
+const require_dist$2 = require("./dist-DnqDVVpK.cjs");
 let node_async_hooks = require("node:async_hooks");
 require("node:fs");
 let zod_v3 = require("zod/v3");
@@ -13262,7 +13262,7 @@ async function connectMcpServer(name, options) {
 }
 async function createTransport(url, transport, requestInit, fetchImpl) {
 	if (transport === "sse") {
-		const { SSEClientTransport } = await Promise.resolve().then(() => require("./sse-DGaTY14k.cjs"));
+		const { SSEClientTransport } = await Promise.resolve().then(() => require("./sse-SwZnScle.cjs"));
 		return new SSEClientTransport(url, {
 			requestInit,
 			fetch: fetchImpl
@@ -15763,6 +15763,16 @@ function and(...unfilteredConditions) {
 		new StringChunk(")")
 	]);
 }
+function or(...unfilteredConditions) {
+	const conditions = unfilteredConditions.filter((c) => c !== void 0);
+	if (conditions.length === 0) return;
+	if (conditions.length === 1) return new SQL(conditions);
+	return new SQL([
+		new StringChunk("("),
+		sql.join(conditions, new StringChunk(" or ")),
+		new StringChunk(")")
+	]);
+}
 function isNull(value) {
 	return sql`${value} is null`;
 }
@@ -16341,6 +16351,41 @@ function index(name) {
 function uniqueIndex(name) {
 	return new IndexBuilderOn(name, true);
 }
+const traceContextStorage = new node_async_hooks.AsyncLocalStorage();
+function getTraceContext() {
+	return traceContextStorage.getStore();
+}
+const CONSOLE_LEVELS = [
+	"log",
+	"debug",
+	"info",
+	"warn",
+	"error"
+];
+async function captureConsole(fn) {
+	const originals = /* @__PURE__ */ new Map();
+	for (const level of CONSOLE_LEVELS) {
+		originals.set(level, console[level].bind(console));
+		console[level] = (...args) => {
+			originals.get(level)?.(...args);
+			getTraceContext();
+		};
+	}
+	try {
+		return await fn();
+	} finally {
+		for (const level of CONSOLE_LEVELS) {
+			const original = originals.get(level);
+			if (original) console[level] = original;
+		}
+	}
+}
+async function logSystem(level, message, data) {
+	getTraceContext();
+}
+async function withSpan(input, fn) {
+	return await fn();
+}
 //#endregion
 //#region ../database/dist/index.mjs
 const tenantScopeStorage = new node_async_hooks.AsyncLocalStorage();
@@ -16400,6 +16445,8 @@ const agentSessions = pgTable("agent_sessions", {
 	id: text$1("id").primaryKey(),
 	projectId: text$1("project_id").notNull(),
 	agentId: text$1("agent_id").notNull().references(() => agents.id),
+	sourceKind: text$1("source_kind").$type().notNull().default("api"),
+	sourceId: text$1("source_id"),
 	parentSpanId: text$1("parent_span_id"),
 	createdAt: timestamp("created_at", { withTimezone: true }).notNull(),
 	updatedAt: timestamp("updated_at", { withTimezone: true }).notNull(),
@@ -16409,6 +16456,8 @@ const agentSessionsSqlite = sqliteTable("agent_sessions", {
 	id: text("id").primaryKey(),
 	projectId: text("project_id").notNull(),
 	agentId: text("agent_id").notNull().references(() => agentsSqlite.id),
+	sourceKind: text("source_kind").$type().notNull().default("api"),
+	sourceId: text("source_id"),
 	parentSpanId: text("parent_span_id"),
 	createdAt: integer("created_at", { mode: "timestamp_ms" }).notNull(),
 	updatedAt: integer("updated_at", { mode: "timestamp_ms" }).notNull(),
@@ -16436,7 +16485,9 @@ const triggerAttachments = pgTable("trigger_attachments", {
 	id: text$1("id").primaryKey(),
 	projectId: text$1("project_id").notNull(),
 	key: text$1("key").notNull(),
-	workflowKey: text$1("workflow_key").notNull(),
+	targetKind: text$1("target_kind").$type().notNull().default("workflow"),
+	workflowKey: text$1("workflow_key"),
+	agentKey: text$1("agent_key"),
 	moduleFile: text$1("module_file").notNull(),
 	source: jsonb("source").$type().notNull(),
 	registeredAt: timestamp("registered_at", { withTimezone: true }).notNull(),
@@ -16447,7 +16498,9 @@ const triggerAttachmentsSqlite = sqliteTable("trigger_attachments", {
 	id: text("id").primaryKey(),
 	projectId: text("project_id").notNull(),
 	key: text("key").notNull(),
-	workflowKey: text("workflow_key").notNull(),
+	targetKind: text("target_kind").$type().notNull().default("workflow"),
+	workflowKey: text("workflow_key"),
+	agentKey: text("agent_key"),
 	moduleFile: text("module_file").notNull(),
 	source: text("source", { mode: "json" }).$type().notNull(),
 	registeredAt: integer("registered_at", { mode: "timestamp_ms" }).notNull(),
@@ -16519,6 +16572,8 @@ const workflowRuns = pgTable("workflow_runs", {
 	triggerRunId: text$1("trigger_run_id").references(() => triggerRuns.id),
 	parentWorkflowRunId: text$1("parent_workflow_run_id"),
 	parentSpanId: text$1("parent_span_id"),
+	sourceKind: text$1("source_kind").$type().notNull().default("api"),
+	sourceId: text$1("source_id"),
 	status: text$1("status").$type().notNull(),
 	trigger: text$1("trigger").$type().notNull(),
 	triggeredAt: timestamp("triggered_at", { withTimezone: true }).notNull(),
@@ -16537,6 +16592,8 @@ const workflowRunsSqlite = sqliteTable("workflow_runs", {
 	triggerRunId: text("trigger_run_id").references(() => triggerRunsSqlite.id),
 	parentWorkflowRunId: text("parent_workflow_run_id"),
 	parentSpanId: text("parent_span_id"),
+	sourceKind: text("source_kind").$type().notNull().default("api"),
+	sourceId: text("source_id"),
 	status: text("status").$type().notNull(),
 	trigger: text("trigger").$type().notNull(),
 	triggeredAt: integer("triggered_at", { mode: "timestamp_ms" }).notNull(),
@@ -16944,6 +17001,10 @@ function scopedAnd$1(table, ...conditions) {
 function projectScopeId() {
 	return getProjectScopeId();
 }
+/** Use with left joins on workflowRuns/triggerRuns/agentSessions keyed by trace or ref id. */
+function traceRootProjectScopeFilter() {
+	return or(eq(workflowRuns.projectId, getProjectScopeId()), eq(triggerRuns.projectId, getProjectScopeId()), eq(agentSessions.projectId, getProjectScopeId()));
+}
 async function selectActiveAgentByRoute(route) {
 	return (await getDb$1().select({
 		id: agents.id,
@@ -16953,6 +17014,8 @@ async function selectActiveAgentByRoute(route) {
 async function insertAgentSession(values) {
 	await getDb$1().insert(agentSessions).values({
 		...values,
+		sourceKind: values.sourceKind ?? "api",
+		sourceId: values.sourceId ?? null,
 		projectId: projectScopeId()
 	});
 }
@@ -16990,16 +17053,61 @@ async function selectCredentialInstancesForScope$1(key, scopeType, scopeId) {
 	const scopeCondition = scopeId === null ? and(eq(credentialInstances.scopeType, scopeType), isNull(credentialInstances.scopeId)) : and(eq(credentialInstances.scopeType, scopeType), eq(credentialInstances.scopeId, scopeId));
 	return db.select().from(credentialInstances).where(credentialScopeAnd(scopeType, eq(credentialInstances.key, key), scopeCondition));
 }
+async function selectTraceSpanById(id) {
+	const db = getDb$1();
+	const rows = await db.select({ span: traceSpans }).from(traceSpans).leftJoin(workflowRuns, eq(workflowRuns.id, traceSpans.traceId)).leftJoin(triggerRuns, eq(triggerRuns.id, traceSpans.traceId)).leftJoin(agentSessions, eq(agentSessions.id, traceSpans.traceId)).where(and(eq(traceSpans.id, id), traceRootProjectScopeFilter())).limit(1);
+	if (rows[0]) return rows[0].span;
+	return (await db.select({ span: traceSpans }).from(traceSpans).leftJoin(workflowRuns, eq(workflowRuns.id, traceSpans.refId)).leftJoin(triggerRuns, eq(triggerRuns.id, traceSpans.refId)).leftJoin(agentSessions, eq(agentSessions.id, traceSpans.refId)).where(and(eq(traceSpans.id, id), traceRootProjectScopeFilter())).limit(1))[0]?.span;
+}
+function runSource(kind, id) {
+	return {
+		kind,
+		id: id ?? null
+	};
+}
+function sourceFromSpan(span) {
+	switch (span.kind) {
+		case "workflow_run": return runSource("workflow", span.refId);
+		case "agent_session": return runSource("agent", span.refId);
+		case "trigger_run": return runSource("trigger", span.refId);
+		case "gateway_event": {
+			const metadata = span.metadata;
+			if (metadata && typeof metadata === "object" && "gatewayAttachmentId" in metadata) {
+				const attachmentId = metadata.gatewayAttachmentId;
+				if (typeof attachmentId === "string") return runSource("gateway", attachmentId);
+			}
+			return;
+		}
+		case "action": {
+			const metadata = span.metadata;
+			if (metadata && typeof metadata === "object" && "runId" in metadata) {
+				const runId = metadata.runId;
+				if (typeof runId === "string") return runSource("workflow", runId);
+			}
+			return;
+		}
+		default: return;
+	}
+}
+async function resolveRunSourceFromTraceContext() {
+	const parent = getTraceContext();
+	if (!parent?.spanId) return runSource("api");
+	const span = await selectTraceSpanById(parent.spanId);
+	if (!span) return runSource("api");
+	return sourceFromSpan(span) ?? runSource("api");
+}
 async function getAgentByRoute(route) {
 	return selectActiveAgentByRoute(route);
 }
 const MESSAGE_EVENT_TYPE = "message";
-async function createSession(agentId, id) {
+async function createSession(agentId, id, source = runSource("api")) {
 	const now = /* @__PURE__ */ new Date();
 	const sessionId = id ?? crypto.randomUUID();
 	await insertAgentSession({
 		id: sessionId,
 		agentId,
+		sourceKind: source.kind,
+		sourceId: source.id,
 		createdAt: now,
 		updatedAt: now,
 		deletedAt: null
@@ -17999,41 +18107,6 @@ async function resolveActionCredentials(requirements, options) {
 	});
 	return parseResolvedCredentials(requirements, await resolveCredentials(requirements, options.consumer, options.contextOverride));
 }
-const traceContextStorage = new node_async_hooks.AsyncLocalStorage();
-function getTraceContext() {
-	return traceContextStorage.getStore();
-}
-const CONSOLE_LEVELS = [
-	"log",
-	"debug",
-	"info",
-	"warn",
-	"error"
-];
-async function captureConsole(fn) {
-	const originals = /* @__PURE__ */ new Map();
-	for (const level of CONSOLE_LEVELS) {
-		originals.set(level, console[level].bind(console));
-		console[level] = (...args) => {
-			originals.get(level)?.(...args);
-			getTraceContext();
-		};
-	}
-	try {
-		return await fn();
-	} finally {
-		for (const level of CONSOLE_LEVELS) {
-			const original = originals.get(level);
-			if (original) console[level] = original;
-		}
-	}
-}
-async function logSystem(level, message, data) {
-	getTraceContext();
-}
-async function withSpan(input, fn) {
-	return await fn();
-}
 //#endregion
 Object.defineProperty(exports, "JSONRPCMessageSchema", {
 	enumerable: true,
@@ -18197,6 +18270,12 @@ Object.defineProperty(exports, "resolveMcpTools", {
 		return resolveMcpTools;
 	}
 });
+Object.defineProperty(exports, "resolveRunSourceFromTraceContext", {
+	enumerable: true,
+	get: function() {
+		return resolveRunSourceFromTraceContext;
+	}
+});
 Object.defineProperty(exports, "touchSession", {
 	enumerable: true,
 	get: function() {
@@ -18216,4 +18295,4 @@ Object.defineProperty(exports, "zodToJsonSchema", {
 	}
 });
 
-//# sourceMappingURL=dist-ByV6mHKd.cjs.map
+//# sourceMappingURL=dist-BkC9pJIu.cjs.map