opencode-tbot 0.1.11 → 0.1.12

package/dist/plugin.js

@@ -211,6 +211,14 @@ function buildOpenCodeSdkConfig(options) {
 		...apiKey ? { auth: apiKey } : {}
 	};
 }
+var OpenCodePromptTimeoutError = class extends Error {
+	data;
+	constructor(data) {
+		super(data.message ?? "OpenCode prompt timed out.");
+		this.name = "OpenCodePromptTimeoutError";
+		this.data = data;
+	}
+};
 var EMPTY_RESPONSE_TEXT = "OpenCode returned empty response.";
 var PROMPT_MESSAGE_POLL_INITIAL_DELAYS_MS = [
 	0,
@@ -220,8 +228,11 @@ var PROMPT_MESSAGE_POLL_INITIAL_DELAYS_MS = [
 	1e3
 ];
 var PROMPT_MESSAGE_POLL_INTERVAL_MS = 2e3;
+var PROMPT_POLL_REQUEST_TIMEOUT_MS = 15e3;
+var PROMPT_SEND_TIMEOUT_MS = 3e4;
 var PROMPT_MESSAGE_POLL_TIMEOUT_MS = 6e4;
 var PROMPT_MESSAGE_POLL_LIMIT = 20;
+var PROMPT_LOG_SERVICE = "opencode-tbot";
 var STRUCTURED_REPLY_SCHEMA = {
 	type: "json_schema",
 	retryCount: 2,
@@ -243,6 +254,11 @@ var StructuredReplySchema = z.object({ body_md: z.string() });
 var OpenCodeClient = class {
 	client;
 	fetchFn;
+	promptRequestTimeouts = {
+		pollRequestMs: PROMPT_POLL_REQUEST_TIMEOUT_MS,
+		sendMs: PROMPT_SEND_TIMEOUT_MS,
+		totalPollMs: PROMPT_MESSAGE_POLL_TIMEOUT_MS
+	};
 	modelCache = {
 		expiresAt: 0,
 		promise: null,
@@ -287,8 +303,7 @@ var OpenCodeClient = class {
 		return unwrapSdkData(await this.client.mcp.status(directory ? { directory } : void 0, SDK_OPTIONS));
 	}
 	async getSessionStatuses() {
-		if (hasRawSdkMethod(this.client, "get")) return this.requestRaw("get", { url: "/session/status" });
-		return unwrapSdkData(await this.client.session.status(void 0, SDK_OPTIONS));
+		return this.loadSessionStatuses();
 	}
 	async listProjects() {
 		if (hasRawSdkMethod(this.client, "get")) return this.requestRaw("get", { url: "/project" });
@@ -394,7 +409,7 @@ var OpenCodeClient = class {
 			structured
 		};
 		let bestCandidate = selectPromptResponseCandidate([data], candidateOptions) ?? data;
-		const deadlineAt = Date.now() + PROMPT_MESSAGE_POLL_TIMEOUT_MS;
+		const deadlineAt = Date.now() + this.promptRequestTimeouts.totalPollMs;
 		let idleStatusSeen = false;
 		let attempt = 0;
 		while (true) {
@@ -412,7 +427,7 @@ var OpenCodeClient = class {
 					if (!shouldPollPromptMessage(next, structured)) return bestCandidate;
 				}
 			}
-			const latest = await this.findLatestPromptResponse(input.sessionId, candidateOptions);
+			const latest = await this.findLatestPromptResponse(input.sessionId, candidateOptions, "poll-messages");
 			if (latest) {
 				bestCandidate = selectPromptResponseCandidate([bestCandidate, latest], candidateOptions) ?? bestCandidate;
 				if (!shouldPollPromptMessage(bestCandidate, structured)) return bestCandidate;
@@ -423,57 +438,114 @@ var OpenCodeClient = class {
 			}
 			if (Date.now() >= deadlineAt) break;
 		}
-		const latest = await this.findLatestPromptResponse(input.sessionId, candidateOptions);
-		return selectPromptResponseCandidate([bestCandidate, latest], candidateOptions) ?? bestCandidate;
+		const latest = await this.findLatestPromptResponse(input.sessionId, candidateOptions, "final-scan");
+		const resolved = selectPromptResponseCandidate([bestCandidate, latest], candidateOptions) ?? bestCandidate;
+		if (shouldPollPromptMessage(resolved, structured)) {
+			const error = createOpenCodePromptTimeoutError({
+				sessionId: input.sessionId,
+				stage: "final-scan",
+				timeoutMs: this.promptRequestTimeouts.totalPollMs,
+				messageId: messageId ?? void 0
+			});
+			this.logPromptRequestFailure(error, {
+				sessionId: input.sessionId,
+				stage: "final-scan",
+				timeoutMs: this.promptRequestTimeouts.totalPollMs,
+				messageId
+			});
+			throw error;
+		}
+		return resolved;
 	}
 	async fetchPromptMessage(sessionId, messageId) {
 		try {
-			if (hasRawSdkMethod(this.client, "get")) return normalizePromptResponse(await this.requestRaw("get", {
-				url: "/session/{sessionID}/message/{messageID}",
-				path: {
+			return await this.runPromptRequestWithTimeout({
+				sessionId,
+				stage: "poll-message",
+				timeoutMs: this.promptRequestTimeouts.pollRequestMs,
+				messageId
+			}, async (signal) => {
+				if (hasRawSdkMethod(this.client, "get")) return normalizePromptResponse(await this.requestRaw("get", {
+					url: "/session/{sessionID}/message/{messageID}",
+					path: {
+						sessionID: sessionId,
+						messageID: messageId
+					},
+					signal
+				}));
+				if (typeof this.client.session.message !== "function") return null;
+				return normalizePromptResponse(unwrapSdkData(await this.client.session.message({
 					sessionID: sessionId,
 					messageID: messageId
-				}
-			}));
-			if (typeof this.client.session.message !== "function") return null;
-			return normalizePromptResponse(unwrapSdkData(await this.client.session.message({
-				sessionID: sessionId,
-				messageID: messageId
-			}, SDK_OPTIONS)));
-		} catch {
+				}, {
+					...SDK_OPTIONS,
+					signal
+				})));
+			});
+		} catch (error) {
+			this.logPromptRequestFailure(error, {
+				sessionId,
+				stage: "poll-message",
+				timeoutMs: this.promptRequestTimeouts.pollRequestMs,
+				messageId
+			});
 			return null;
 		}
 	}
 	async captureKnownMessageIds(sessionId) {
-		const messages = await this.fetchRecentPromptMessages(sessionId);
+		const messages = await this.fetchRecentPromptMessages(sessionId, "capture-known-messages");
 		if (!messages) return /* @__PURE__ */ new Set();
 		return new Set(messages.map((message) => extractMessageId(message.info)).filter((id) => typeof id === "string" && id.length > 0));
 	}
-	async fetchRecentPromptMessages(sessionId) {
+	async fetchRecentPromptMessages(sessionId, stage) {
 		try {
-			if (hasRawSdkMethod(this.client, "get")) return normalizePromptResponses(await this.requestRaw("get", {
-				url: "/session/{sessionID}/message",
-				path: { sessionID: sessionId },
-				query: { limit: PROMPT_MESSAGE_POLL_LIMIT }
-			}));
-			if (typeof this.client.session.messages !== "function") return null;
-			return normalizePromptResponses(unwrapSdkData(await this.client.session.messages({
-				sessionID: sessionId,
-				limit: PROMPT_MESSAGE_POLL_LIMIT
-			}, SDK_OPTIONS)));
-		} catch {
+			return await this.runPromptRequestWithTimeout({
+				sessionId,
+				stage,
+				timeoutMs: this.promptRequestTimeouts.pollRequestMs
+			}, async (signal) => {
+				if (hasRawSdkMethod(this.client, "get")) return normalizePromptResponses(await this.requestRaw("get", {
+					url: "/session/{sessionID}/message",
+					path: { sessionID: sessionId },
+					query: { limit: PROMPT_MESSAGE_POLL_LIMIT },
+					signal
+				}));
+				if (typeof this.client.session.messages !== "function") return null;
+				return normalizePromptResponses(unwrapSdkData(await this.client.session.messages({
+					sessionID: sessionId,
+					limit: PROMPT_MESSAGE_POLL_LIMIT
+				}, {
+					...SDK_OPTIONS,
+					signal
+				})));
+			});
+		} catch (error) {
+			this.logPromptRequestFailure(error, {
+				sessionId,
+				stage,
+				timeoutMs: this.promptRequestTimeouts.pollRequestMs
+			});
 			return null;
 		}
 	}
 	async fetchPromptSessionStatus(sessionId) {
 		try {
-			return (await this.getSessionStatuses())[sessionId] ?? null;
-		} catch {
+			return (await this.runPromptRequestWithTimeout({
+				sessionId,
+				stage: "poll-status",
+				timeoutMs: this.promptRequestTimeouts.pollRequestMs
+			}, async (signal) => this.loadSessionStatuses(signal)))[sessionId] ?? null;
+		} catch (error) {
+			this.logPromptRequestFailure(error, {
+				sessionId,
+				stage: "poll-status",
+				timeoutMs: this.promptRequestTimeouts.pollRequestMs
+			});
 			return null;
 		}
 	}
-	async findLatestPromptResponse(sessionId, options) {
-		const messages = await this.fetchRecentPromptMessages(sessionId);
+	async findLatestPromptResponse(sessionId, options, stage) {
+		const messages = await this.fetchRecentPromptMessages(sessionId, stage);
 		if (!messages || messages.length === 0) return null;
 		return selectPromptResponseCandidate(messages, options);
 	}
@@ -497,25 +569,44 @@ var OpenCodeClient = class {
 		return unwrapSdkData(await this.client.config.providers(void 0, SDK_OPTIONS));
 	}
 	async sendPromptRequest(input, parts) {
-		if (hasRawSdkMethod(this.client, "post")) return normalizePromptResponse(await this.requestRaw("post", {
-			url: "/session/{sessionID}/message",
-			path: { sessionID: input.sessionId },
-			body: {
-				...input.agent ? { agent: input.agent } : {},
-				...input.structured ? { format: STRUCTURED_REPLY_SCHEMA } : {},
-				...input.model ? { model: input.model } : {},
-				...input.variant ? { variant: input.variant } : {},
-				parts
-			}
-		}));
-		return normalizePromptResponse(unwrapSdkData(await this.client.session.prompt({
-			sessionID: input.sessionId,
-			...input.agent ? { agent: input.agent } : {},
-			...input.structured ? { format: STRUCTURED_REPLY_SCHEMA } : {},
-			...input.model ? { model: input.model } : {},
-			...input.variant ? { variant: input.variant } : {},
-			parts
-		}, SDK_OPTIONS)));
+		try {
+			return await this.runPromptRequestWithTimeout({
+				sessionId: input.sessionId,
+				stage: "send-prompt",
+				timeoutMs: this.promptRequestTimeouts.sendMs
+			}, async (signal) => {
+				if (hasRawSdkMethod(this.client, "post")) return normalizePromptResponse(await this.requestRaw("post", {
+					url: "/session/{sessionID}/message",
+					path: { sessionID: input.sessionId },
+					body: {
+						...input.agent ? { agent: input.agent } : {},
+						...input.structured ? { format: STRUCTURED_REPLY_SCHEMA } : {},
+						...input.model ? { model: input.model } : {},
+						...input.variant ? { variant: input.variant } : {},
+						parts
+					},
+					signal
+				}));
+				return normalizePromptResponse(unwrapSdkData(await this.client.session.prompt({
+					sessionID: input.sessionId,
+					...input.agent ? { agent: input.agent } : {},
+					...input.structured ? { format: STRUCTURED_REPLY_SCHEMA } : {},
+					...input.model ? { model: input.model } : {},
+					...input.variant ? { variant: input.variant } : {},
+					parts
+				}, {
+					...SDK_OPTIONS,
+					signal
+				})));
+			});
+		} catch (error) {
+			this.logPromptRequestFailure(error, {
+				sessionId: input.sessionId,
+				stage: "send-prompt",
+				timeoutMs: this.promptRequestTimeouts.sendMs
+			});
+			throw error;
+		}
 	}
 	async requestRaw(method, options) {
 		const handler = getRawSdkClient(this.client)?.[method];
@@ -525,6 +616,67 @@ var OpenCodeClient = class {
 			...options
 		}));
 	}
+	async loadSessionStatuses(signal) {
+		if (hasRawSdkMethod(this.client, "get")) return this.requestRaw("get", {
+			url: "/session/status",
+			...signal ? { signal } : {}
+		});
+		return unwrapSdkData(await this.client.session.status(void 0, {
+			...SDK_OPTIONS,
+			...signal ? { signal } : {}
+		}));
+	}
+	async runPromptRequestWithTimeout(input, operation) {
+		const startedAt = Date.now();
+		const controller = new AbortController();
+		let timeoutHandle = null;
+		const timeoutPromise = new Promise((_, reject) => {
+			timeoutHandle = setTimeout(() => {
+				reject(createOpenCodePromptTimeoutError({
+					sessionId: input.sessionId,
+					stage: input.stage,
+					timeoutMs: input.timeoutMs,
+					messageId: input.messageId ?? void 0,
+					elapsedMs: Date.now() - startedAt
+				}));
+				controller.abort();
+			}, input.timeoutMs);
+		});
+		try {
+			return await Promise.race([operation(controller.signal), timeoutPromise]);
+		} finally {
+			if (timeoutHandle !== null) clearTimeout(timeoutHandle);
+		}
+	}
+	logPromptRequestFailure(error, input) {
+		if (error instanceof OpenCodePromptTimeoutError) {
+			this.logPromptRequest("warn", {
+				elapsedMs: error.data.elapsedMs,
+				messageId: error.data.messageId,
+				sessionId: error.data.sessionId,
+				stage: error.data.stage,
+				timeoutMs: error.data.timeoutMs
+			}, "OpenCode prompt request timed out");
+			return;
+		}
+		this.logPromptRequest("warn", {
+			error,
+			messageId: input.messageId ?? void 0,
+			sessionId: input.sessionId,
+			stage: input.stage,
+			timeoutMs: input.timeoutMs
+		}, "OpenCode prompt request failed");
+	}
+	logPromptRequest(level, extra, message) {
+		const log = this.client.app?.log;
+		if (typeof log !== "function") return;
+		log.call(this.client.app, {
+			service: PROMPT_LOG_SERVICE,
+			level,
+			message,
+			extra
+		}).catch(() => void 0);
+	}
 };
 function createOpenCodeClientFromSdkClient(client, fetchFn = fetch) {
 	return new OpenCodeClient(void 0, client, fetchFn);
@@ -696,6 +848,13 @@ function extractMessageId(message) {
 function delay(ms) {
 	return new Promise((resolve) => setTimeout(resolve, ms));
 }
+function createOpenCodePromptTimeoutError(input) {
+	return new OpenCodePromptTimeoutError({
+		...input,
+		elapsedMs: input.elapsedMs ?? input.timeoutMs,
+		message: input.message ?? "The OpenCode host did not finish this request in time."
+	});
+}
 function getPromptMessagePollDelayMs(attempt) {
 	return PROMPT_MESSAGE_POLL_INITIAL_DELAYS_MS[attempt] ?? PROMPT_MESSAGE_POLL_INTERVAL_MS;
 }
@@ -2336,6 +2495,7 @@ var EN_BOT_COPY = {
 		unexpected: "Unexpected error.",
 		providerAuth: "Provider authentication failed.",
 		requestAborted: "Request was aborted.",
+		promptTimeout: "OpenCode request timed out.",
 		structuredOutput: "Structured output validation failed.",
 		voiceNotConfigured: "Voice transcription is not configured.",
 		voiceDownload: "Failed to download the Telegram voice file.",
@@ -2571,6 +2731,7 @@ var ZH_CN_BOT_COPY = {
 		unexpected: "发生未知错误。",
 		providerAuth: "Provider 认证失败。",
 		requestAborted: "请求已中止。",
+		promptTimeout: "OpenCode 响应超时。",
 		structuredOutput: "结构化输出校验失败。",
 		voiceNotConfigured: "未配置语音转写服务。",
 		voiceDownload: "下载 Telegram 语音文件失败。",
@@ -2918,6 +3079,10 @@ function normalizeError(error, copy) {
 		message: copy.errors.requestAborted,
 		cause: extractMessage(error.data) ?? null
 	};
+	if (isNamedError(error, "OpenCodePromptTimeoutError")) return {
+		message: copy.errors.promptTimeout,
+		cause: null
+	};
 	if (isNamedError(error, "StructuredOutputError")) return {
 		message: copy.errors.structuredOutput,
 		cause: joinNonEmptyParts([extractMessage(error.data), extractRetries(error.data)])