@xopcai/xopc 0.0.103 → 0.0.104

package/dist/src/tui/backends/gateway-sse-backend.js

@@ -1,11 +1,23 @@
 import { createLogger } from "../../utils/logger/index.js";
 import { init_logger } from "../../utils/logger.js";
+import { BUILTIN_AGENT_IDS } from "../../agent/builtin-agent-ids.js";
 import { prependEnvelopeTimestamp } from "../../channels/envelope-timestamp.js";
+import { transcriptRowsToClientHistory } from "../../session/client-history.js";
 import { parseModelRef } from "../../agent/models/selection.js";
 import { consumeSSEStream, parseSSEData } from "../sse-consumer.js";
+import { buildTuiTranscriptTree, transcriptTreeEntryIdToRowNumber } from "../tui-transcript-tree.js";
+import { computeTuiSessionStats } from "../tui-session-stats.js";
 //#region src/tui/backends/gateway-sse-backend.ts
 init_logger();
 const log = createLogger("TUI:GatewaySSE");
+function normalizeGatewayModelChoice(model) {
+	const providerPrefix = `${model.provider}/`;
+	const id = model.id.startsWith(providerPrefix) ? model.id.slice(providerPrefix.length) : model.id;
+	return {
+		...model,
+		id
+	};
+}
 /** Fetch wrapper that adds auth headers. */
 async function gatewayFetch(baseUrl, path, token, init) {
 	const headers = {
@@ -50,18 +62,15 @@ var GatewaySseBackend = class {
 		this.chatAbort?.abort();
 		this.chatAbort = null;
 	}
+	getActiveSignal() {
+		const signal = this.chatAbort?.signal;
+		return signal && !signal.aborted ? signal : void 0;
+	}
 	async sendChat(opts) {
 		this.chatAbort?.abort();
 		this.chatAbort = new AbortController();
 		const signal = this.chatAbort.signal;
 		const runId = crypto.randomUUID();
-		this.onEvent?.({
-			event: "status",
-			data: {
-				status: "started",
-				runId
-			}
-		});
 		(async () => {
 			try {
 				const res = await gatewayFetch(this.baseUrl, "/api/agent", this.token, {
@@ -71,6 +80,7 @@ var GatewaySseBackend = class {
 						message: opts.message.trimStart().startsWith("/") ? opts.message : prependEnvelopeTimestamp(opts.message),
 						channel: "webchat",
 						sessionKey: opts.sessionKey,
+						attachments: opts.attachments,
 						thinking: opts.thinking
 					}),
 					signal
@@ -79,7 +89,17 @@ var GatewaySseBackend = class {
 					const body = await res.json().catch(() => ({}));
 					this.onEvent?.({
 						event: "error",
-						data: { content: body.error?.message ?? `Gateway error: ${res.status}` }
+						data: {
+							type: "error",
+							runId,
+							sessionKey: opts.sessionKey,
+							timestamp: Date.now(),
+							payload: {
+								code: "GATEWAY_ERROR",
+								message: body.error?.message ?? `Gateway error: ${res.status}`
+							}
+						},
+						source: "agent-response"
 					});
 					return;
 				}
@@ -89,19 +109,66 @@ var GatewaySseBackend = class {
 					if (!data) return;
 					this.onEvent?.({
 						event: sseEvent.event,
-						data
+						data,
+						source: "agent-response"
 					});
 				}, signal);
 				else {
 					const json = await res.json();
 					if (json.ok && json.payload?.content) {
+						const messageId = `msg_${runId}_1`;
+						const base = {
+							runId,
+							sessionKey: opts.sessionKey,
+							timestamp: Date.now()
+						};
 						this.onEvent?.({
-							event: "token",
-							data: { content: json.payload.content }
+							event: "run_start",
+							data: {
+								...base,
+								type: "run_start",
+								payload: { channel: "webchat" }
+							},
+							source: "agent-response"
 						});
 						this.onEvent?.({
-							event: "result",
-							data: { ok: true }
+							event: "assistant_message_start",
+							data: {
+								...base,
+								type: "assistant_message_start",
+								payload: { messageId }
+							},
+							source: "agent-response"
+						});
+						this.onEvent?.({
+							event: "assistant_delta",
+							data: {
+								...base,
+								type: "assistant_delta",
+								payload: {
+									messageId,
+									delta: json.payload.content
+								}
+							},
+							source: "agent-response"
+						});
+						this.onEvent?.({
+							event: "assistant_message_end",
+							data: {
+								...base,
+								type: "assistant_message_end",
+								payload: { messageId }
+							},
+							source: "agent-response"
+						});
+						this.onEvent?.({
+							event: "run_end",
+							data: {
+								...base,
+								type: "run_end",
+								payload: { status: "success" }
+							},
+							source: "agent-response"
 						});
 					}
 				}
@@ -110,12 +177,99 @@ var GatewaySseBackend = class {
 				const errorMessage = error instanceof Error ? error.message : String(error);
 				this.onEvent?.({
 					event: "error",
-					data: { content: errorMessage }
+					data: {
+						type: "error",
+						runId,
+						sessionKey: opts.sessionKey,
+						timestamp: Date.now(),
+						payload: {
+							code: "NETWORK_ERROR",
+							message: errorMessage
+						}
+					},
+					source: "agent-response"
 				});
 			}
 		})();
 		return { runId };
 	}
+	async resumeChat(opts) {
+		this.chatAbort?.abort();
+		this.chatAbort = new AbortController();
+		const signal = this.chatAbort.signal;
+		this.onEvent?.({
+			event: "run_start",
+			data: {
+				type: "run_start",
+				runId: opts.runId,
+				sessionKey: opts.sessionKey,
+				timestamp: Date.now(),
+				payload: { channel: "webchat" }
+			},
+			source: "agent-resume"
+		});
+		let res;
+		try {
+			res = await gatewayFetch(this.baseUrl, "/api/agent/resume", this.token, {
+				method: "POST",
+				headers: { Accept: "text/event-stream" },
+				body: JSON.stringify({
+					runId: opts.runId,
+					chatId: opts.sessionKey
+				}),
+				signal
+			});
+		} catch (error) {
+			if (signal.aborted) return {
+				ok: false,
+				reason: "resume aborted"
+			};
+			return {
+				ok: false,
+				reason: error instanceof Error ? error.message : String(error)
+			};
+		}
+		if (!res.ok) return {
+			ok: false,
+			reason: (await res.json().catch(() => ({}))).error?.message ?? `Resume failed: ${res.status}`
+		};
+		if (!(res.headers.get("Content-Type") ?? "").includes("text/event-stream") || !res.body) return {
+			ok: false,
+			reason: "Resume endpoint did not return an SSE stream"
+		};
+		(async () => {
+			try {
+				await consumeSSEStream(res.body, (sseEvent) => {
+					if (signal.aborted) return;
+					const data = parseSSEData(sseEvent.data);
+					if (!data) return;
+					this.onEvent?.({
+						event: sseEvent.event,
+						data,
+						source: "agent-resume"
+					});
+				}, signal);
+			} catch (error) {
+				if (signal.aborted) return;
+				const errorMessage = error instanceof Error ? error.message : String(error);
+				this.onEvent?.({
+					event: "error",
+					data: {
+						type: "error",
+						runId: opts.runId,
+						sessionKey: opts.sessionKey,
+						timestamp: Date.now(),
+						payload: {
+							code: "NETWORK_ERROR",
+							message: errorMessage
+						}
+					},
+					source: "agent-resume"
+				});
+			}
+		})();
+		return { ok: true };
+	}
 	async abortChat(opts) {
 		this.chatAbort?.abort();
 		this.chatAbort = null;
@@ -143,14 +297,39 @@ var GatewaySseBackend = class {
 			return { ok: false };
 		}
 	}
+	async getStartupResources(sessionKey) {
+		const empty = {
+			context: [],
+			skills: [],
+			workflows: [],
+			connectors: []
+		};
+		try {
+			const res = await gatewayFetch(this.baseUrl, `/api/tui/startup-resources?sessionKey=${encodeURIComponent(sessionKey)}`, this.token);
+			if (!res.ok) return empty;
+			const json = await res.json();
+			return {
+				context: Array.isArray(json.payload?.context) ? json.payload.context : [],
+				skills: Array.isArray(json.payload?.skills) ? json.payload.skills : [],
+				workflows: Array.isArray(json.payload?.workflows) ? json.payload.workflows : [],
+				connectors: Array.isArray(json.payload?.connectors) ? json.payload.connectors : []
+			};
+		} catch (error) {
+			const errorMessage = error instanceof Error ? error.message : String(error);
+			log.warn({
+				err: error,
+				sessionKey,
+				errorMessage
+			}, `Failed to load startup resources: ${errorMessage}`);
+			return empty;
+		}
+	}
 	async loadHistory(opts) {
 		try {
-			const params = new URLSearchParams();
-			if (opts.limit) params.set("limit", String(opts.limit));
-			const qs = params.toString();
-			const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(opts.sessionKey)}/messages${qs ? `?${qs}` : ""}`, this.token);
+			const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(opts.sessionKey)}?include=transcriptRows`, this.token);
 			if (!res.ok) return { messages: [] };
-			return { messages: (await res.json()).payload?.messages ?? [] };
+			const json = await res.json();
+			return { messages: transcriptRowsToClientHistory(Array.isArray(json.session?.transcriptRows) ? json.session.transcriptRows : [], { limit: opts.limit }) };
 		} catch (error) {
 			const errorMessage = error instanceof Error ? error.message : String(error);
 			log.warn({
@@ -160,6 +339,38 @@ var GatewaySseBackend = class {
 			return { messages: [] };
 		}
 	}
+	async loadTranscriptTree(sessionKey) {
+		try {
+			const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}?include=transcriptRows`, this.token);
+			if (!res.ok) return [];
+			const json = await res.json();
+			return buildTuiTranscriptTree(Array.isArray(json.session?.transcriptRows) ? json.session.transcriptRows : []);
+		} catch (error) {
+			const errorMessage = error instanceof Error ? error.message : String(error);
+			log.warn({
+				err: error,
+				sessionKey,
+				errorMessage
+			}, `Failed to load transcript tree: ${errorMessage}`);
+			return [];
+		}
+	}
+	async getSessionStats(sessionKey) {
+		try {
+			const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}?include=transcriptRows`, this.token);
+			if (!res.ok) return computeTuiSessionStats([]);
+			const json = await res.json();
+			return computeTuiSessionStats(Array.isArray(json.session?.transcriptRows) ? json.session.transcriptRows : []);
+		} catch (error) {
+			const errorMessage = error instanceof Error ? error.message : String(error);
+			log.warn({
+				err: error,
+				sessionKey,
+				errorMessage
+			}, `Failed to load session stats: ${errorMessage}`);
+			return computeTuiSessionStats([]);
+		}
+	}
 	async listSessions() {
 		try {
 			const res = await gatewayFetch(this.baseUrl, "/api/sessions", this.token);
@@ -170,12 +381,45 @@ var GatewaySseBackend = class {
 				updatedAt: s.updatedAt ? Date.parse(s.updatedAt) : void 0,
 				totalTokens: s.estimatedTokens ?? null,
 				messageCount: typeof s.messageCount === "number" ? s.messageCount : void 0,
-				model: typeof s.customData?.model === "string" ? s.customData.model : typeof s.customData?.modelRef === "string" ? s.customData.modelRef : null
+				model: typeof s.customData?.model === "string" ? s.customData.model : typeof s.customData?.modelRef === "string" ? s.customData.modelRef : null,
+				forkedFromSessionKey: typeof s.customData?.forkedFromSessionKey === "string" ? s.customData.forkedFromSessionKey : void 0,
+				cwd: typeof s.cwd === "string" ? s.cwd : void 0
 			}));
 		} catch {
 			return [];
 		}
 	}
+	async listAgents() {
+		const agents = /* @__PURE__ */ new Map();
+		for (const id of BUILTIN_AGENT_IDS) agents.set(id, {
+			id,
+			source: "builtin",
+			enabled: true
+		});
+		try {
+			const res = await gatewayFetch(this.baseUrl, "/api/agents", this.token);
+			if (res.ok) {
+				const json = await res.json();
+				for (const row of json.payload?.agents ?? []) {
+					if (typeof row.id !== "string" || !row.id.trim()) continue;
+					const id = row.id.trim().toLowerCase();
+					agents.set(id, {
+						id,
+						source: "configured",
+						enabled: true,
+						...typeof row.name === "string" && row.name.trim() ? { displayName: row.name.trim() } : {}
+					});
+				}
+			}
+		} catch (error) {
+			const errorMessage = error instanceof Error ? error.message : String(error);
+			log.warn({
+				err: error,
+				errorMessage
+			}, `Failed to load agents: ${errorMessage}`);
+		}
+		return [...agents.values()].sort((a, b) => a.id.localeCompare(b.id));
+	}
 	async getSessionInfo(sessionKey) {
 		const out = {};
 		try {
@@ -199,6 +443,10 @@ var GatewaySseBackend = class {
 					} else out.model = p.model;
 				}
 				if (p?.thinkingLevel && typeof p.thinkingLevel === "string") out.thinkingLevel = p.thinkingLevel;
+				if (p?.reasoningLevel && typeof p.reasoningLevel === "string") out.reasoningLevel = p.reasoningLevel;
+				if (p?.verboseLevel && typeof p.verboseLevel === "string") out.verboseLevel = p.verboseLevel;
+				if (p?.effectiveWorkspacePath && typeof p.effectiveWorkspacePath === "string") out.effectiveWorkspacePath = p.effectiveWorkspacePath;
+				if (typeof p?.workingDirectoryLocked === "boolean") out.workingDirectoryLocked = p.workingDirectoryLocked;
 			}
 			if (!out.model && session?.customData) {
 				const cd = session.customData;
@@ -226,7 +474,7 @@ var GatewaySseBackend = class {
 		try {
 			const res = await gatewayFetch(this.baseUrl, "/api/models", this.token);
 			if (!res.ok) return [];
-			return (await res.json()).payload?.models ?? [];
+			return ((await res.json()).payload?.models ?? []).map(normalizeGatewayModelChoice);
 		} catch {
 			return [];
 		}
@@ -238,7 +486,10 @@ var GatewaySseBackend = class {
 		try {
 			const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}/compaction/run`, this.token, {
 				method: "POST",
-				body: JSON.stringify({ force: options?.force ?? true })
+				body: JSON.stringify({
+					force: options?.force ?? true,
+					instructions: options?.instructions
+				})
 			});
 			if (!res.ok) return {
 				compacted: false,
@@ -251,7 +502,10 @@ var GatewaySseBackend = class {
 			};
 			return {
 				compacted: true,
-				summary: `Compacted (${result.tokensBefore ?? "?"} → ${result.tokensAfter ?? "?"} tokens)`
+				summary: `Compacted (${result.tokensBefore ?? "?"} → ${result.tokensAfter ?? "?"} tokens)`,
+				tokensBefore: result.tokensBefore,
+				tokensAfter: result.tokensAfter,
+				transcriptSummary: result.summary
 			};
 		} catch (err) {
 			return {
@@ -260,11 +514,146 @@ var GatewaySseBackend = class {
 			};
 		}
 	}
+	async exportSession(sessionKey, format) {
+		const params = new URLSearchParams({ format });
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}/export?${params.toString()}`, this.token);
+		if (!res.ok) throw new Error(`Export failed (${res.status})`);
+		return (await res.json()).content ?? "";
+	}
+	async importSession(targetSessionKey, jsonContent) {
+		const res = await gatewayFetch(this.baseUrl, "/api/sessions/import", this.token, {
+			method: "POST",
+			body: JSON.stringify({
+				targetKey: targetSessionKey,
+				content: jsonContent
+			})
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false || !json.sessionKey) throw new Error(json.error ?? `Import failed (${res.status})`);
+		return {
+			sessionKey: json.sessionKey,
+			rowCount: json.rowCount ?? 0
+		};
+	}
+	async createShare(sessionKey, request, options) {
+		const res = await gatewayFetch(this.baseUrl, "/api/shares/auto", this.token, {
+			method: "POST",
+			body: JSON.stringify({
+				path: request.path,
+				audience: request.audience,
+				mode: request.mode,
+				title: request.title,
+				description: request.description,
+				sessionKey,
+				agentId: options?.agentId
+			})
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false || !json.payload?.share?.shareUrl) throw new Error(json.error?.message ?? `Share failed (${res.status})`);
+		const share = json.payload.share;
+		return {
+			kind: share.kind ?? "share",
+			shareUrl: share.shareUrl,
+			title: share.title,
+			description: share.description,
+			thumbnailUrl: json.payload.thumbnail?.url,
+			reachability: share.reachability,
+			reachabilityHint: share.reachabilityHint,
+			expiresAt: share.expiresAt,
+			maxViews: share.maxViews,
+			routingReason: json.payload.routing?.reason,
+			routingHint: json.payload.routing?.hint
+		};
+	}
+	async btwQuery(sessionKey, question) {
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}/btw`, this.token, {
+			method: "POST",
+			body: JSON.stringify({ question })
+		});
+		if (!res.ok) return {
+			text: "",
+			error: `BTW failed (${res.status})`
+		};
+		return await res.json();
+	}
+	async forkSession(sourceSessionKey, targetSessionKey) {
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sourceSessionKey)}/fork`, this.token, {
+			method: "POST",
+			body: JSON.stringify({ targetKey: targetSessionKey })
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false || !json.sessionKey) throw new Error(json.error ?? `Fork failed (${res.status})`);
+		return {
+			sessionKey: json.sessionKey,
+			rowCount: json.rowCount ?? 0
+		};
+	}
+	async forkSessionAt(sourceSessionKey, targetSessionKey, entryId) {
+		const throughRow = transcriptTreeEntryIdToRowNumber(entryId);
+		if (throughRow == null) throw new Error(`Invalid transcript entry: ${entryId}`);
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sourceSessionKey)}/fork-row`, this.token, {
+			method: "POST",
+			body: JSON.stringify({
+				targetKey: targetSessionKey,
+				throughRow
+			})
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false || !json.sessionKey) throw new Error(json.error ?? `Fork failed (${res.status})`);
+		return {
+			sessionKey: json.sessionKey,
+			rowCount: json.rowCount ?? 0
+		};
+	}
+	async setTranscriptLabel(sessionKey, entryId, label) {
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}/transcript/label`, this.token, {
+			method: "POST",
+			body: JSON.stringify({
+				targetId: entryId,
+				label
+			})
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false) throw new Error(json.error ?? `Label update failed (${res.status})`);
+		return { ok: true };
+	}
+	async appendCustomEntry(sessionKey, customType, data) {
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}/transcript/custom`, this.token, {
+			method: "POST",
+			body: JSON.stringify({
+				customType,
+				data
+			})
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false) throw new Error(json.error ?? `Custom entry append failed (${res.status})`);
+		return { ok: true };
+	}
+	async appendCustomMessage(sessionKey, message) {
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}/transcript/custom-message`, this.token, {
+			method: "POST",
+			body: JSON.stringify(message)
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false) throw new Error(json.error ?? `Custom message append failed (${res.status})`);
+		return { ok: true };
+	}
+	async appendBashExecution(sessionKey, entry) {
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}/transcript/bash`, this.token, {
+			method: "POST",
+			body: JSON.stringify(entry)
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false) throw new Error(json.error ?? `Bash execution append failed (${res.status})`);
+		return { ok: true };
+	}
 	async patchSession(sessionKey, patch) {
-		await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}`, this.token, {
+		const res = await gatewayFetch(this.baseUrl, `/api/sessions/${encodeURIComponent(sessionKey)}/agent-config`, this.token, {
 			method: "PATCH",
 			body: JSON.stringify(patch)
-		}).catch(() => {});
+		});
+		const json = await res.json().catch(() => ({}));
+		if (!res.ok || json.ok === false) throw new Error(json.error ?? `Session config patch failed (${res.status})`);
 	}
 	async renameSession(sessionKey, name) {
 		try {
@@ -315,7 +704,8 @@ var GatewaySseBackend = class {
 					const data = parseSSEData(sseEvent.data);
 					if (data !== null) this.onEvent?.({
 						event: sseEvent.event,
-						data
+						data,
+						source: "broadcast"
 					});
 				}, this.eventAbort.signal);
 				if (!this.eventAbort?.signal.aborted) {