import type {
	ExtensionAPI,
	ExtensionCommandContext,
} from "@earendil-works/pi-coding-agent";
import { ENSURE_ASK_FOR_DOCS_ENTRY } from "../../chat-mode/ensure-ask-for-docs.js";
import { buildBugLocatePrompt } from "./prompts.js";
import { readTapdSessionState } from "../sessions/session-state.js";

const DOCS_WORKFLOW_MODE_RULES = [
	"",
	"## 模式约束",
	"不要调用 enter_plan_mode，也不要写入或编辑 session 的 plan.md。",
	"只将文档写入上文指定的 .pi/docs 路径（及同目录约定文档）；不要修改业务代码。",
].join("\n");

function withAdditionalInstructions(body: string, extra?: string): string {
	const trimmed = extra?.trim();
	if (!trimmed) return body;
	return `${body}\n\n## 用户补充要求与参考资料\n\n${trimmed}\n\n请将以上补充要求和 @ 引用文件一并纳入本次任务。`;
}

export async function sendTapdWorkflowPrompt(
	pi: ExtensionAPI,
	ctx: ExtensionCommandContext,
	prompt: string,
	additionalInstructions?: string,
): Promise<boolean> {
	if (!ctx.isIdle()) {
		ctx.ui.notify("Agent 正在执行，请稍后再试", "warning");
		return false;
	}

	const state = readTapdSessionState(ctx.sessionManager.getEntries());
	if (state?.kind === "bug") {
		ctx.ui.notify("当前是 Bug 会话，请执行 /tapd bug 定位缺陷原因", "warning");
		return false;
	}

	// 请求 chat-mode 在本轮 agent 启动前切到 Ask（可写 .pi/docs，避免 Plan 拦 design.md）。
	pi.appendEntry(ENSURE_ASK_FOR_DOCS_ENTRY, { version: 1 });

	// This command is registered by the extension instance bound to the current
	// session, so use its current pi. Never retain the ReplacedSessionContext
	// from the newSession() callback for a later command invocation.
	pi.sendUserMessage(
		withAdditionalInstructions(
			`${prompt}${DOCS_WORKFLOW_MODE_RULES}`,
			additionalInstructions,
		),
	);
	return true;
}

export async function locateTapdBug(
	pi: ExtensionAPI,
	ctx: ExtensionCommandContext,
	additionalInstructions?: string,
): Promise<void> {
	if (!ctx.isIdle()) {
		ctx.ui.notify("Agent 正在执行，请稍后再试", "warning");
		return;
	}
	const state = readTapdSessionState(ctx.sessionManager.getEntries());
	if (!state) {
		ctx.ui.notify(
			"当前会话没有关联 TAPD 条目，请先从 TAPD 缺陷列表创建或切换会话",
			"warning",
		);
		return;
	}
	if (state.kind !== "bug") {
		ctx.ui.notify("/tapd bug 只能在 Bug 会话中执行", "warning");
		return;
	}

	// Use this command's factory pi, not a ctx captured across newSession/switchSession.
	pi.sendUserMessage(
		withAdditionalInstructions(buildBugLocatePrompt(), additionalInstructions),
	);
}