89 lines
3.9 KiB
TypeScript
89 lines
3.9 KiB
TypeScript
|
import { tool, type ToolDefinition } from "@opencode-ai/plugin"
|
||
|
|
import type { BackgroundOutputManager, BackgroundOutputClient } from "../types"
|
||
|
|
import type { BackgroundOutputArgs } from "../types"
|
||
|
|
import { BACKGROUND_OUTPUT_DESCRIPTION } from "../constants"
|
||
|
|
import { formatTaskStatus, formatTaskResult, formatFullSession } from "./formatters"
|
||
|
|
import { delay } from "./utils"
|
||
|
|
|
||
|
|
export function createBackgroundOutput(manager: BackgroundOutputManager, client: BackgroundOutputClient): ToolDefinition {
|
||
|
|
return tool({
|
||
|
|
description: BACKGROUND_OUTPUT_DESCRIPTION,
|
||
|
|
args: {
|
||
|
|
task_id: tool.schema.string().describe("Task ID to get output from"),
|
||
|
|
block: tool.schema.boolean().optional().describe("Wait for completion (default: false). System notifies when done, so blocking is rarely needed."),
|
||
|
|
timeout: tool.schema.number().optional().describe("Max wait time in ms (default: 60000, max: 600000)"),
|
||
|
|
full_session: tool.schema.boolean().optional().describe("Return full session messages with filters (default: false)"),
|
||
|
|
include_thinking: tool.schema.boolean().optional().describe("Include thinking/reasoning parts in full_session output (default: false)"),
|
||
|
|
message_limit: tool.schema.number().optional().describe("Max messages to return (capped at 100)"),
|
||
|
|
since_message_id: tool.schema.string().optional().describe("Return messages after this message ID (exclusive)"),
|
||
|
|
include_tool_results: tool.schema.boolean().optional().describe("Include tool results in full_session output (default: false)"),
|
||
|
|
thinking_max_chars: tool.schema.number().optional().describe("Max characters for thinking content (default: 2000)"),
|
||
|
|
},
|
||
|
|
async execute(args: BackgroundOutputArgs) {
|
||
|
|
try {
|
||
|
|
const task = manager.getTask(args.task_id)
|
||
|
|
if (!task) {
|
||
|
|
return `Task not found: ${args.task_id}`
|
||
|
|
}
|
||
|
|
|
||
|
|
if (args.full_session === true) {
|
||
|
|
return await formatFullSession(task, client, {
|
||
|
|
includeThinking: args.include_thinking === true,
|
||
|
|
messageLimit: args.message_limit,
|
||
|
|
sinceMessageId: args.since_message_id,
|
||
|
|
includeToolResults: args.include_tool_results === true,
|
||
|
|
thinkingMaxChars: args.thinking_max_chars,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
const shouldBlock = args.block === true
|
||
|
|
const timeoutMs = Math.min(args.timeout ?? 60000, 600000)
|
||
|
|
|
||
|
|
// Already completed: return result immediately (regardless of block flag)
|
||
|
|
if (task.status === "completed") {
|
||
|
|
return await formatTaskResult(task, client)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Error or cancelled: return status immediately
|
||
|
|
if (task.status === "error" || task.status === "cancelled") {
|
||
|
|
return formatTaskStatus(task)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Non-blocking and still running: return status
|
||
|
|
if (!shouldBlock) {
|
||
|
|
return formatTaskStatus(task)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Blocking: poll until completion or timeout
|
||
|
|
const startTime = Date.now()
|
||
|
|
|
||
|
|
while (Date.now() - startTime < timeoutMs) {
|
||
|
|
await delay(1000)
|
||
|
|
|
||
|
|
const currentTask = manager.getTask(args.task_id)
|
||
|
|
if (!currentTask) {
|
||
|
|
return `Task was deleted: ${args.task_id}`
|
||
|
|
}
|
||
|
|
|
||
|
|
if (currentTask.status === "completed") {
|
||
|
|
return await formatTaskResult(currentTask, client)
|
||
|
|
}
|
||
|
|
|
||
|
|
if (currentTask.status === "error" || currentTask.status === "cancelled") {
|
||
|
|
return formatTaskStatus(currentTask)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Timeout exceeded: return current status
|
||
|
|
const finalTask = manager.getTask(args.task_id)
|
||
|
|
if (!finalTask) {
|
||
|
|
return `Task was deleted: ${args.task_id}`
|
||
|
|
}
|
||
|
|
return `Timeout exceeded (${timeoutMs}ms). Task still ${finalTask.status}.\n\n${formatTaskStatus(finalTask)}`
|
||
|
|
} catch (error) {
|
||
|
|
return `Error getting output: ${error instanceof Error ? error.message : String(error)}`
|
||
|
|
}
|
||
|
|
},
|
||
|
|
})
|
||
|
|
}
|