// src/commands/subagents.ts
//
// /subagents 命令。薄壳——打开 list overlay（等同原 /subagents list [<id>]）。
//
// 解析：args[0] 直接作可选 <id>（聚焦该 record）。
// RPC 模式（xyz-agent GUI）：解析 cancel action 直接执行，不打开 TUI。

import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";

import { getSubagentService } from "../execution/subagent-service.ts";
import { displayAgentName } from "../shared/agent-ref.ts";
import { parseSubagentRpcCommand } from "./command-actions.ts";
import { LIST_LIMIT } from "./list-shared.ts";
import { createSubagentsView } from "./list-view.ts";

/** 注册 /subagents 命令（= list overlay）。 */
export function registerSubagentsCommand(pi: ExtensionAPI): void {
  pi.registerCommand("subagents", {
    description: "Subagents: /subagents [<id>] | /subagents cancel <id>",
    getArgumentCompletions(prefix: string) {
      const trimmed = prefix.trimStart();
      const parts = trimmed.split(/\s+/).filter(Boolean);

      // 第一级：cancel 动词（带尾随空格，选中后继续补 record id）
      if (parts.length <= 1) {
        return [
          { label: "cancel", value: "cancel ", description: "Cancel a running subagent" },
        ].filter((opt) => opt.label.startsWith(trimmed.toLowerCase()));
      }

      // 第二级：cancel 后补全当前 session 的 record id
      if (parts[0] === "cancel") {
        try {
          const service = getSubagentService();
          if (!service) return null;
          // collectRecords 合并内存(running) + 磁盘重建 record，按 rootSessionId 过滤。
          // cancel 只对 running 有效，但全部列出便于用户辨认（终态 record 会被 service 拒绝）。
          const records = service.collectRecords(LIST_LIMIT);
          if (records.length === 0) return null;
          return records.map((r) => ({
            label: r.id,
            value: r.id,
            description: `${displayAgentName(r.agent)} [${r.status}]`,
          }));
        } catch {
          // 拿不到运行时数据（service disposed 等）→ 静默降级，补全失败不影响 command
          return null;
        }
      }
      return null;
    },
    handler: async (argsStr: string, ctx: ExtensionCommandContext) => {
      const service = getSubagentService();
      if (!service) {
        ctx.ui.notify("subagents execution runtime not ready (session not started)", "error");
        return;
      }

      // ── RPC 模式（xyz-agent GUI）：解析 action 直接执行，不打开 TUI ──
      // hasUI 在 TUI 和 RPC 都为 true，不能用于区分；用 ctx.mode === "rpc" 判定 GUI 通道。
      if (ctx.mode === "rpc") {
        const parsed = parseSubagentRpcCommand(argsStr);
        switch (parsed.action) {
          case "cancel": {
            try {
              const ok = service.cancel(parsed.recordId);
              ctx.ui.notify(
                ok ? `Cancelled subagent ${parsed.recordId}` : `Subagent ${parsed.recordId} not found or already finished`,
                ok ? "info" : "warning",
              );
            } catch (err) {
              // service.cancel 内部 assertReady 在 session_shutdown 并发 dispose 时会抛
              const msg = err instanceof Error ? err.message : String(err);
              ctx.ui.notify(`Failed to cancel subagent ${parsed.recordId}: ${msg}`, "warning");
            }
            return;
          }
          case "cancel-missing-id":
            ctx.ui.notify("Usage: /subagents cancel <id>", "warning");
            return;
          case "noop":
            // 无 action 或未知 action：GUI 端已屏蔽此 command 入口，此处兜底
            ctx.ui.notify("View subagents in the sidebar Agents tab", "info");
            return;
          default: {
            // exhaustiveness 断言：未来新增 action verb 忘加 case 时 tsc 报错
            const _exhaustive: never = parsed;
            throw new Error(`Unhandled subagent RPC action: ${String(_exhaustive)}`);
          }
        }
      }

      // ── print/json 模式（headless）：不可交互 ──
      if (ctx.mode !== "tui") {
        ctx.ui.notify("/subagents requires interactive mode", "error");
        return;
      }

      // ── TUI 模式：打开 list overlay（原逻辑不变）──
      const args = argsStr.trim().split(/\s+/).filter(Boolean);
      await createSubagentsView(service, ctx.ui.theme, ctx, args[0]);
    },
  });
}