@zhushanwen/pi-goal 0.2.0 → 0.4.0

package/src/tool-handler.ts

@@ -1,252 +0,0 @@
-/**
- * Goal 扩展 — Tool 执行处理器和共享 helpers
- *
- * 从 index.ts 提取的 executeGoalAction 及其依赖的辅助函数、类型和 schema。
- */
-
-import { StringEnum } from "@mariozechner/pi-ai";
-import type { CustomEntry, ExtensionAPI, ExtensionContext, SessionEntry } from "@mariozechner/pi-coding-agent";
-import { type Static,Type } from "typebox";
-
-import { ACTION_HANDLERS } from "./action-handlers.js";
-import {
-	MS_PER_SECOND,
-	SECONDS_PER_MINUTE,
-} from "./constants";
-import {
-	getCompletedCount,
-	getElapsedTimeSeconds,
-	GOAL_TASK_STATUSES,
-	type GoalRuntimeState,
-	type GoalTask,
-	isTerminalStatus,
-	serializeState,
-	SUBTASK_STATUSES,
-} from "./state";
-import { renderStatusLine, renderTerminalStatusLine,renderWidgetLines } from "./widget";
-
-// ── 常量 ─────────────────────────────────────────────
-
-const ENTRY_TYPE = "goal-state";
-
-export const HISTORY_ENTRY_TYPE = "goal-history";
-
-// ── Session State Interface ──────────────────────────
-
-export interface GoalSession {
-	state: GoalRuntimeState | null;
-	tasksCompletedAtAgentStart: number;
-	hasPendingInjection: boolean;
-	/** 防重入标志：handleAgentEnd / handleBeforeAgentStart 等事件处理器入口检查 */
-	isProcessing: boolean;
-}
-
-// ── Stale Context Detection ──────────────────────────
-
-const STALE_CONTEXT_PATTERNS = ["aborted", "context canceled", "stale context", "stalecontext", "extension context no longer active"];
-
-/** 检查错误是否表示 stale / canceled context（如 session 重建 / compact 后） */
-export function isStaleContextError(error: Error | unknown): boolean {
-	const msg = error instanceof Error ? error.message : String(error);
-	const lower = msg.toLowerCase();
-	return STALE_CONTEXT_PATTERNS.some((p) => lower.includes(p));
-}
-
-// ── Tool Parameter Schema ────────────────────────────
-
-export const GoalManagerParams = Type.Object({
-	action: StringEnum([
-		"create_tasks",
-		"add_tasks",
-		"update_tasks",
-		"list_tasks",
-		"complete_goal",
-		"cancel_goal",
-		"report_blocked",
-		"add_subtasks",
-		"update_subtasks",
-		"delete_subtasks",
-	] as const),
-	tasks: Type.Optional(Type.Array(Type.String(), { description: "Task descriptions. Each must be a one-line summary (max 60 chars), no newlines or markdown" })),
-	updates: Type.Optional(Type.Array(Type.Object({
-		taskId: Type.Number(),
-		status: StringEnum(GOAL_TASK_STATUSES),
-		evidence: Type.Optional(Type.String()),
-	}))),
-	taskId: Type.Optional(Type.Number({ description: "Task ID (required for subtask operations)" })),
-	texts: Type.Optional(Type.Array(Type.String(), { description: "Subtask text list (for add_subtasks)" })),
-	subUpdates: Type.Optional(Type.Array(Type.Object({
-		subId: Type.Number(),
-		status: StringEnum(SUBTASK_STATUSES),
-	}))),
-	subIds: Type.Optional(Type.Array(Type.Number(), { description: "Subtask ID list (for delete_subtasks)" })),
-	evidence: Type.Optional(Type.String({ description: "Evidence for completion (required for complete_goal)" })),
-	reason: Type.Optional(Type.String({ description: "Reason for being blocked (required for report_blocked)" })),
-	cancelReason: Type.Optional(Type.String({ description: "Why the user wants to cancel (required for cancel_goal)" })),
-});
-
-// ── Tool Details Types ───────────────────────────────
-
-export interface GoalManagerDetails {
-	action: string;
-	tasks: GoalTask[];
-	goalId: string;
-	status: string;
-	_render?: {
-		type: "task-list" | "summary-table" | "progress" | "code-block";
-		summary?: string;
-		data: unknown;
-	};
-}
-
-// ── Module-level Helpers ─────────────────────────────
-
-export function isGoalEntry(entry: SessionEntry): entry is CustomEntry<GoalRuntimeState> {
-	return entry.type === "custom" && (entry as CustomEntry).customType === ENTRY_TYPE;
-}
-
-export function persistGoalState(pi: ExtensionAPI, session: GoalSession, _ctx: ExtensionContext): void {
-	if (!session.state) return;
-	const now = Date.now();
-	if (session.state.timeStartedAt > 0) {
-		session.state.timeUsedSeconds += (now - session.state.timeStartedAt) / MS_PER_SECOND;
-		session.state.timeStartedAt = now;
-	}
-	pi.appendEntry(ENTRY_TYPE, serializeState(session.state));
-}
-
-export function writeGoalHistoryEntry(pi: ExtensionAPI, session: GoalSession): void {
-	const state = session.state;
-	if (!state) return;
-	pi.appendEntry(HISTORY_ENTRY_TYPE, {
-		goalId: state.goalId,
-		objective: state.objective,
-		status: state.status,
-		completedTasks: getCompletedCount(state.tasks),
-		totalTasks: state.tasks.length,
-		elapsedSeconds: Math.floor(getElapsedTimeSeconds(state)),
-		timestamp: Date.now(),
-	});
-}
-
-export function updateWidget(session: GoalSession, ctx: ExtensionContext): void {
-	if (!session.state || session.state.status === "cancelled") {
-		ctx.ui.setWidget("goal", undefined);
-		ctx.ui.setStatus("goal", undefined);
-		return;
-	}
-
-	// 终态折叠为单行 status bar
-	if (isTerminalStatus(session.state.status)) {
-		const statusText = renderTerminalStatusLine(session.state, ctx.ui.theme);
-		if (statusText) {
-			ctx.ui.setStatus("goal", statusText);
-		}
-		ctx.ui.setWidget("goal", undefined);
-		return;
-	}
-
-	ctx.ui.setStatus("goal", renderStatusLine(session.state, ctx.ui.theme));
-	ctx.ui.setWidget("goal", renderWidgetLines(session.state, ctx.ui.theme));
-}
-
-export function clearGoalSession(session: GoalSession, ctx: ExtensionContext): void {
-	session.state = null;
-	session.tasksCompletedAtAgentStart = 0;
-	session.hasPendingInjection = false;
-	ctx.ui.setWidget("goal", undefined);
-	ctx.ui.setStatus("goal", undefined);
-}
-
-// ── Result Builder ───────────────────────────────────
-
-export function makeGoalResult(session: GoalSession, text: string) {
-	const state = session.state;
-	if (!state) {
-		// P1-1: 不再抛异常，返回标准 isError 结果
-		return errorResult("No active goal");
-	}
-	const budgetInfo: string[] = [];
-	if (state.budget.tokenBudget) {
-		const remaining = Math.max(state.budget.tokenBudget - state.tokensUsed, 0);
-		budgetInfo.push(`Token: ${state.tokensUsed}/${state.budget.tokenBudget} (${remaining} remaining)`);
-	}
-	if (state.budget.timeBudgetMinutes) {
-		const elapsed = getElapsedTimeSeconds(state);
-		const remaining = Math.max(state.budget.timeBudgetMinutes * SECONDS_PER_MINUTE - elapsed, 0);
-		budgetInfo.push(`Time: ${Math.floor(elapsed / SECONDS_PER_MINUTE)}m/${state.budget.timeBudgetMinutes}m (${Math.floor(remaining / SECONDS_PER_MINUTE)}m remaining)`);
-	}
-	const suffix = budgetInfo.length > 0 ? `\n\n[Budget] ${budgetInfo.join(" | ")}` : "";
-	return {
-		content: [{ type: "text" as const, text: text + suffix }],
-		details: {
-			action: "update",
-			tasks: state.tasks.map((t) => ({ ...t })),
-			goalId: state.goalId,
-			status: state.status,
-			_render: {
-				type: "task-list" as const,
-				summary: `${getCompletedCount(state.tasks)}/${state.tasks.length} completed`,
-				data: {
-					items: state.tasks.map((t) => ({
-						id: t.id,
-						text: t.description,
-						status: t.status,
-						evidence: t.evidence,
-						subtasks: t.subtasks?.map((s) => ({
-							id: s.id,
-							text: s.text,
-							status: s.status,
-						})),
-					})),
-					meta: {
-						...(state.budget.tokenBudget ? { "Token": `${state.tokensUsed}/${state.budget.tokenBudget}` } : {}),
-						...(state.budget.timeBudgetMinutes ? { "Time": `${Math.floor(getElapsedTimeSeconds(state) / SECONDS_PER_MINUTE)}m/${state.budget.timeBudgetMinutes}m` } : {}),
-						"Turn": `${state.currentTurnIndex}/${state.budget.maxTurns}`,
-					},
-				},
-			},
-		} satisfies GoalManagerDetails,
-	};
-}
-
-// ── Tool Execute Handler ──────────────────────────────
-
-/**
- * 执行 goal_manager tool action 的分发入口。
- *
- * 行为不变 —— 所有 case 逻辑已委托到 action-handlers.ts 中的独立函数。
- * 这里仅做：状态检查 + signal 守卫 + action 查表分发。
- */
-export async function executeGoalAction(
-	pi: ExtensionAPI,
-	session: GoalSession,
-	params: Static<typeof GoalManagerParams>,
-	ctx: ExtensionContext,
-	signal?: AbortSignal,
-) {
-	const state = session.state;
-	if (!state) {
-		return errorResult("Goal mode not active. Use /goal <objective> to start.");
-	}
-
-	// P1-4: signal 透传 — 若已 abort 直接返回错误
-	if (signal?.aborted) {
-		return errorResult("Tool call aborted by signal.");
-	}
-
-	// 委托到 action-handlers.ts 中的具体处理器
-	const handler = ACTION_HANDLERS[params.action];
-	if (!handler) {
-		return errorResult(`Unknown action: ${params.action}`);
-	}
-	return handler({ pi, session, state, params, ctx });
-}
-
-/** 构造标准的错误结果（避免重复的 content 模板） */
-export function errorResult(message: string) {
-	return {
-		content: [{ type: "text" as const, text: message }],
-		isError: true as const,
-	};
-}

package/src/widget.ts

@@ -1,186 +0,0 @@
-/**
- * Widget 渲染逻辑 — 状态栏和侧边栏任务面板
- */
-
-import type { ThemeColor } from "@mariozechner/pi-coding-agent";
-
-import { getBudgetColor,getTimeUsagePercent, getTokenUsagePercent } from "./budget.js";
-import {
-	OBJECTIVE_DISPLAY_LIMIT,
-	OBJECTIVE_TRUNCATE_KEEP,
-	PERCENT_FACTOR,
-	PROGRESS_BAR_DEFAULT_WIDTH,
-	SECONDS_PER_MINUTE,
-} from "./constants";
-import type { GoalRuntimeState } from "./state";
-import { getCompletedCount, getElapsedTimeSeconds } from "./state";
-
-/**
- * 将多行文本压缩为单行，用于 widget 渲染。
- * 多行 content 泄漏到 widget 会导致 markdown 表格/标题等破坏布局。
- */
-export function toSingleLine(text: string): string {
-	return text.replace(/\r?\n/g, " ").trim();
-}
-
-export interface ThemeLike {
-	fg: (color: ThemeColor, text: string) => string;
-	bold: (text: string) => string;
-}
-
-function renderProgressBar(pct: number, width: number = PROGRESS_BAR_DEFAULT_WIDTH): string {
-	const clamped = Math.min(Math.max(pct, 0), 1);
-	const filled = Math.round(clamped * width);
-	return "█".repeat(filled) + "░".repeat(width - filled);
-}
-
-export function renderStatusLine(state: GoalRuntimeState, th: ThemeLike): string {
-	if (state.status === "cancelled") return "";
-
-	const completedCount = getCompletedCount(state.tasks);
-	const total = state.tasks.length;
-
-	let text = th.fg("accent", `◆ Goal`) + th.fg("muted", ` ${state.currentTurnIndex}/${state.budget.maxTurns}`);
-
-	if (total > 0) {
-		text += th.fg("muted", ` | ${completedCount}/${total} tasks`);
-		const cancelledCount = state.tasks.filter(t => t.status === "cancelled").length;
-		if (cancelledCount > 0) {
-			text += th.fg("dim", `, ${cancelledCount} cancelled`);
-		}
-	}
-
-	// Budget indicators
-	if (state.budget.tokenBudget && state.budget.tokenBudget > 0) {
-		const pct = Math.round(getTokenUsagePercent(state));
-		text += th.fg(getBudgetColor(pct), ` | ${pct}% tokens`);
-	}
-	if (state.budget.timeBudgetMinutes && state.budget.timeBudgetMinutes > 0) {
-		const pct = Math.round(getTimeUsagePercent(state));
-		text += th.fg(getBudgetColor(pct), ` | ${pct}% time`);
-	}
-
-	if (state.stallCount > 0) {
-		text += th.fg("warning", ` | ⚠ ${state.stallCount} turns stalled`);
-	}
-
-	// Status suffix
-	switch (state.status) {
-		case "paused":
-			text += th.fg("warning", " | ⏸ Paused");
-			break;
-		case "blocked":
-			text += th.fg("error", " | ⊘ Blocked");
-			break;
-		case "complete":
-			text += th.fg("success", " | ✓ Completed");
-			break;
-		case "budget_limited":
-			text += th.fg("error", " | ⊗ Token budget exhausted");
-			break;
-		case "time_limited":
-			text += th.fg("error", " | ⏱ Time budget exhausted");
-			break;
-	}
-
-	return text;
-}
-
-export function renderTerminalStatusLine(state: GoalRuntimeState, th: ThemeLike): string {
-	if (state.status === "cancelled") return "";
-
-	const completedCount = getCompletedCount(state.tasks);
-	const total = state.tasks.length;
-
-	let text = th.fg("accent", "◆ Goal");
-
-	// 状态后缀
-	switch (state.status) {
-		case "complete":
-			text += th.fg("success", " ✓ Completed");
-			break;
-		case "budget_limited":
-			text += th.fg("error", " ⊗ Token budget exhausted");
-			break;
-		case "time_limited":
-			text += th.fg("error", " ⏱ Time budget exhausted");
-			break;
-		default:
-			break;
-	}
-
-	text += th.fg("muted", ` | ${completedCount}/${total} tasks`);
-
-	// 预算摘要
-	if (state.budget.tokenBudget && state.budget.tokenBudget > 0) {
-		const pct = Math.round(getTokenUsagePercent(state));
-		text += th.fg(getBudgetColor(pct), ` | ${pct}% tokens`);
-	}
-	if (state.budget.timeBudgetMinutes && state.budget.timeBudgetMinutes > 0) {
-		const pct = Math.round(getTimeUsagePercent(state));
-		text += th.fg(getBudgetColor(pct), ` | ${pct}% time`);
-	}
-
-	return text;
-}
-
-export function renderWidgetLines(state: GoalRuntimeState, th: ThemeLike): string[] {
-	if (state.status === "cancelled") return [];
-
-	const total = state.tasks.length;
-
-	// Header line
-	const header = renderStatusLine(state, th);
-	const lines: string[] = [header];
-
-	// Objective (single-line + truncated if too long)
-	const objSingleLine = toSingleLine(state.objective);
-	const objDisplay = objSingleLine.length > OBJECTIVE_DISPLAY_LIMIT
-		? objSingleLine.slice(0, OBJECTIVE_TRUNCATE_KEEP) + "..."
-		: objSingleLine;
-	lines.push(th.fg("dim", `Objective: ${objDisplay}`));
-
-	// Task list
-	if (total === 0) {
-		lines.push(th.fg("dim", "  Waiting for task list creation..."));
-	} else {
-		for (const t of state.tasks) {
-			const desc = toSingleLine(t.description);
-			if (t.status === "completed") {
-				lines.push(`  ${th.fg("success", "✓")} ${th.fg("dim", `#${t.id}`)} ${th.fg("dim", desc)}`);
-			} else if (t.status === "cancelled") {
-				lines.push(`  ${th.fg("dim", "✗")} ${th.fg("dim", `#${t.id}`)} ${th.fg("dim", desc)}`);
-			} else if (t.status === "in_progress") {
-				lines.push(`  ${th.fg("warning", "●")} ${th.fg("accent", `#${t.id}`)} ${th.fg("text", desc)}`);
-			} else {
-				lines.push(`  ${th.fg("dim", "☐")} ${th.fg("accent", `#${t.id}`)} ${th.fg("text", desc)}`);
-			}
-			// Sub-todo items
-			if (t.subtasks && t.subtasks.length > 0 && t.status !== "cancelled") {
-				for (const s of t.subtasks) {
-					const subIcon = s.status === "completed"
-						? th.fg("success", "✓")
-						: s.status === "in_progress"
-							? th.fg("warning", "●")
-							: th.fg("dim", "○");
-					const subText = s.status === "completed" ? th.fg("dim", s.text) : th.fg("muted", s.text);
-					lines.push(`    ${subIcon} ${th.fg("dim", `${t.id}.${s.id}`)} ${subText}`);
-				}
-			}
-		}
-	}
-
-	// P2-8: Budget progress bars
-	if (state.budget.tokenBudget && state.budget.tokenBudget > 0) {
-		const pct = getTokenUsagePercent(state) / PERCENT_FACTOR;
-		lines.push(`  Token: ${renderProgressBar(pct)} ${Math.round(pct * PERCENT_FACTOR)}%`);
-	}
-	if (state.budget.timeBudgetMinutes && state.budget.timeBudgetMinutes > 0) {
-		const pct = getTimeUsagePercent(state) / PERCENT_FACTOR;
-		const elapsed = getElapsedTimeSeconds(state);
-		const mins = Math.floor(elapsed / SECONDS_PER_MINUTE);
-		lines.push(`  Time: ${renderProgressBar(pct)} ${mins}/${state.budget.timeBudgetMinutes}min`);
-	}
-
-	return lines;
-}