import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";
import { worktreeMergePrompt } from "../lib/prompts.js";

/**
 * /hive:worktree-merge — command that delegates to the LLM + hive_execute_merge tool.
 *
 * Usage: /hive:worktree-merge [worktree-name | branch | index]
 */
export function registerWorktreeMergeCommand(pi: ExtensionAPI) {
  pi.registerCommand("hive:worktree-merge", {
    description: "Merge a completed worktree back into the main branch",
    handler: async (args, ctx) => {
      const worktreeArg = args?.trim() || undefined;

      // Store project dir for the tool
      pi.appendEntry("hive-merge-config", { projectDir: ctx.cwd });

      // Send prompt to LLM — it will inspect worktrees and call hive_execute_merge
      pi.sendUserMessage(worktreeMergePrompt(worktreeArg));
    },
  });
}