@keystrokehq/keystroke 0.0.100 → 0.0.102

package/dist/credentials.cjs

@@ -1,5 +1,5 @@
 Object.defineProperty(exports, Symbol.toStringTag, { value: "Module" });
-const require_dist = require("./dist-T7a5DjzV.cjs");
+const require_dist = require("./dist-DnqDVVpK.cjs");
 exports.CREDENTIAL_SCOPE_LEVELS = require_dist.CREDENTIAL_SCOPE_LEVELS;
 exports.DEFAULT_CREDENTIAL_RESOLUTION_CHAIN = require_dist.DEFAULT_CREDENTIAL_RESOLUTION_CHAIN;
 exports.credential = require_dist.credential;

package/dist/credentials.mjs

@@ -1,2 +1,2 @@
-import { _ as credentialInputSchema, b as normalizeCredentialList, g as credential, h as DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, p as CREDENTIAL_SCOPE_LEVELS, v as defineCredential, x as toCredentialRequirement, y as isCredentialInput } from "./dist-B1fqDXYa.mjs";
+import { S as toCredentialRequirement, _ as credential, b as isCredentialInput, g as DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, m as CREDENTIAL_SCOPE_LEVELS, v as credentialInputSchema, x as normalizeCredentialList, y as defineCredential } from "./dist-SGAuX401.mjs";
 export { CREDENTIAL_SCOPE_LEVELS, DEFAULT_CREDENTIAL_RESOLUTION_CHAIN, credential, credentialInputSchema, defineCredential, isCredentialInput, normalizeCredentialList, toCredentialRequirement };

package/dist/{dist-BML_gJSh.mjs → dist-BUfE2se6.mjs}

@@ -1,5 +1,5 @@
 import { Bn as __require, Ln as __commonJSMin, Vn as __toESM } from "./dist-f_JGyGWx.mjs";
-import { b as normalizeCredentialList, m as CREDENTIAL_SCOPE_LEVEL_TO_SCOPE_TYPE } from "./dist-B1fqDXYa.mjs";
+import { h as CREDENTIAL_SCOPE_LEVEL_TO_SCOPE_TYPE, x as normalizeCredentialList } from "./dist-SGAuX401.mjs";
 import { AsyncLocalStorage } from "node:async_hooks";
 import "node:fs";
 import { ZodFirstPartyTypeKind } from "zod/v3";
@@ -13255,7 +13255,7 @@ async function connectMcpServer(name, options) {
 }
 async function createTransport(url, transport, requestInit, fetchImpl) {
 	if (transport === "sse") {
-		const { SSEClientTransport } = await import("./sse-Bxvdniu5.mjs");
+		const { SSEClientTransport } = await import("./sse-DezHFNKg.mjs");
 		return new SSEClientTransport(url, {
 			requestInit,
 			fetch: fetchImpl
@@ -15756,6 +15756,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`;
 }
@@ -16334,6 +16344,41 @@ function index(name) {
 function uniqueIndex(name) {
 	return new IndexBuilderOn(name, true);
 }
+const traceContextStorage = new 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 AsyncLocalStorage();
@@ -16393,6 +16438,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(),
@@ -16402,6 +16449,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(),
@@ -16429,7 +16478,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(),
@@ -16440,7 +16491,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(),
@@ -16512,6 +16565,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(),
@@ -16530,6 +16585,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(),
@@ -16937,6 +16994,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,
@@ -16946,6 +17007,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()
 	});
 }
@@ -16983,16 +17046,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
@@ -17992,42 +18100,7 @@ async function resolveActionCredentials(requirements, options) {
 	});
 	return parseResolvedCredentials(requirements, await resolveCredentials(requirements, options.consumer, options.contextOverride));
 }
-const traceContextStorage = new 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
-export { JSONRPCMessageSchema as A, createParser as C, createFetchWithInit as D, extractWWWAuthenticateParams as E, normalizeHeaders as O, isMcp as S, auth as T, touchSession as _, buildCredentialRunContext as a, connectMcpStdio as b, isCredentialError as c, MESSAGE_EVENT_TYPE as d, appendEvent as f, listMessageEvents as g, getSession as h, withSpan as i, zodToJsonSchema as k, resolveActionCredentials as l, getAgentByRoute as m, getTraceContext as n, captureCredentialToolErrors as o, createSession as p, logSystem as r, createCredentialResolver as s, captureConsole as t, resolveMcpTools as u, connectMcpDefinition as v, UnauthorizedError as w, defineMcp as x, connectMcpServer as y };
+export { zodToJsonSchema as A, isMcp as C, extractWWWAuthenticateParams as D, auth as E, createFetchWithInit as O, defineMcp as S, UnauthorizedError as T, logSystem as _, resolveActionCredentials as a, connectMcpServer as b, appendEvent as c, getSession as d, listMessageEvents as f, getTraceContext as g, captureConsole as h, isCredentialError as i, JSONRPCMessageSchema as j, normalizeHeaders as k, createSession as l, touchSession as m, captureCredentialToolErrors as n, resolveMcpTools as o, resolveRunSourceFromTraceContext as p, createCredentialResolver as r, MESSAGE_EVENT_TYPE as s, buildCredentialRunContext as t, getAgentByRoute as u, withSpan as v, createParser as w, connectMcpStdio as x, connectMcpDefinition as y };
 
-//# sourceMappingURL=dist-BML_gJSh.mjs.map
+//# sourceMappingURL=dist-BUfE2se6.mjs.map