import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
import type { TUI } from "@earendil-works/pi-tui";
import { ModelBorderEditor } from "./editor.ts";
import { SingleLineStatusbar } from "./footer.ts";
import { SilentHeader } from "./header.ts";

const SPINNER_FRAMES = ["⠋", "⠙", "⠹", "⠸", "⠼", "⠴", "⠦", "⠧", "⠇", "⠏"];

export default function (pi: ExtensionAPI) {
	let statusbar: SingleLineStatusbar | undefined;
	let activeTui: TUI | undefined;
	let spinnerIndex = 0;
	let spinnerTimer: ReturnType<typeof setInterval> | undefined;
	let currentToolName: string | undefined;
	let agentStartMs: number | null = null;
	let latestTps: number | undefined;

	const startSpinner = () => {
		if (spinnerTimer) return;
		spinnerTimer = setInterval(() => {
			spinnerIndex = (spinnerIndex + 1) % SPINNER_FRAMES.length;
			activeTui?.requestRender();
		}, 80);
	};

	const stopSpinner = () => {
		if (spinnerTimer) {
			clearInterval(spinnerTimer);
			spinnerTimer = undefined;
		}
	};

	const getSpinnerFrame = () => SPINNER_FRAMES[spinnerIndex];

	const invalidate = () => statusbar?.invalidate();
	const refreshGit = () => void statusbar?.refreshGit();
	const rerender = () => activeTui?.requestRender();

	// 监听核心生命周期与交互事件
	pi.on("agent_start", () => {
		agentStartMs = Date.now();
		startSpinner();
		rerender();
	});
	pi.on("agent_end", (event) => {
		if (agentStartMs !== null) {
			const elapsedMs = Date.now() - agentStartMs;
			agentStartMs = null;
			if (elapsedMs > 0) {
				let output = 0;
				for (const msg of event.messages ?? []) {
					if ((msg as any)?.role === "assistant" && (msg as any)?.usage?.output) {
						output += (msg as any).usage.output;
					}
				}
				if (output > 0) {
					latestTps = output / (elapsedMs / 1000);
				}
			}
		}
		stopSpinner();
		currentToolName = undefined;
		invalidate();
		refreshGit();
		rerender();
	});
	pi.on("agent_settled", () => {
		stopSpinner();
		currentToolName = undefined;
		invalidate();
		rerender();
	});
	pi.on("turn_end", () => {
		currentToolName = undefined;
		invalidate();
		refreshGit();
		rerender();
	});
	pi.on("message_end", () => {
		invalidate();
		rerender();
	});
	pi.on("tool_execution_start", (event) => {
		currentToolName = event.toolName;
		startSpinner();
		rerender();
	});
	pi.on("tool_execution_end", () => {
		currentToolName = undefined;
		rerender();
	});
	pi.on("model_select", () => rerender());
	pi.on("thinking_level_select", () => rerender());
	pi.on("session_info_changed", () => rerender());
	pi.on("session_compact", () => {
		invalidate();
		rerender();
	});
	pi.on("session_shutdown", () => {
		stopSpinner();
		currentToolName = undefined;
		activeTui = undefined;
	});

	pi.on("session_start", (_event, ctx) => {
		// 1. 关闭内置的多余 working... 行，避免动画冲突并省出空间
		ctx.ui.setWorkingVisible(false);

		// 2. 头部：静音 Pi 内置 verbose 启动头，仅保留初始资源卡片美化；正文交还原生渲染
		ctx.ui.setHeader((tui, theme) => {
			activeTui = tui;
			return new SilentHeader(tui, theme);
		});

		// 3. 遥测栏：极致单行状态栏
		ctx.ui.setFooter((tui, theme, footerData) => {
			activeTui = tui;
			statusbar = new SingleLineStatusbar(ctx, tui, theme, footerData);
			return statusbar;
		});

		// 4. 输入框：圆角边框 + 模型身份与工具执行动态反馈 + 右上角生成速率 tok/s
		ctx.ui.setEditorComponent((tui, theme, keybindings) => {
			activeTui = tui;
			return new ModelBorderEditor(
				tui,
				theme,
				keybindings,
				ctx,
				ctx.ui.theme,
				getSpinnerFrame,
				() => currentToolName,
				() => latestTps,
			);
		});
	});
}