import { describe, expect, it } from "vitest";
import {
  renderMainCompletion,
  renderMainCompletionHelp,
  renderMainHelp,
  resolveMainCompletionCommand,
} from "../../src/cli/mainHelp.js";

describe("tdmcp top-level help", () => {
  it("documents the primary binary commands without starting the MCP server", () => {
    const help = renderMainHelp();

    expect(help).toContain("Usage: tdmcp");
    expect(help).toContain("serve");
    expect(help).toContain("--http --port");
    expect(help).toContain("install-bridge");
    expect(help).toContain("--verify/--wait");
    expect(help).toContain("install-client");
    expect(help).toContain("chat");
    expect(help).toContain("telegram");
    expect(help).toContain("creative-rag");
    expect(help).toContain("project-rag");
    expect(help).toContain("dashboard");
    expect(help).toContain("packages");
    expect(help).toContain("search [query]");
    expect(help).toContain("install <lib>");
    expect(help).toContain("packages path");
    expect(help).toContain("completion <shell>");
  });

  it("prints top-level shell completions including package commands", () => {
    const completion = renderMainCompletion("bash");

    expect(completion).toContain("complete -F _tdmcp tdmcp");
    expect(completion).toContain("install-bridge");
    expect(completion).toContain("install-client");
    expect(completion).toContain("telegram");
    expect(completion).toContain("creative-rag");
    expect(completion).toContain("project-rag");
    expect(completion).toContain("search");
    expect(completion).toContain("install");
    expect(completion).toContain("packages");
    expect(completion).toContain("--dry-run");
    expect(completion).toContain("--json");
  });

  it("documents the completion subcommand help", () => {
    const help = renderMainCompletionHelp();

    expect(help).toContain("Usage: tdmcp completion <bash|zsh|fish>");
    expect(help).toContain("bash");
    expect(help).toContain("zsh");
    expect(help).toContain("fish");
  });

  it("resolves completion help for help flags and omitted shells", () => {
    for (const shell of [undefined, "--help", "-h"]) {
      const result = resolveMainCompletionCommand(shell);

      expect(result.stdout).toContain("Usage: tdmcp completion <bash|zsh|fish>");
      expect(result.stderr).toBeUndefined();
      expect(result.exitCode).toBeUndefined();
    }
  });

  it("does not suggest packages path as a fake top-level path command", () => {
    const completion = renderMainCompletion("bash");
    const wordLists = [...(completion?.matchAll(/compgen -W '([^']+)'/g) ?? [])].map(
      (match) => match[1]?.split(" ") ?? [],
    );
    const words = wordLists.find((list) => list.includes("packages")) ?? [];

    expect(words).toContain("packages");
    expect(words).toContain("--path");
    expect(words).not.toContain("path");
  });

  it("suggests path and help flags in the bash completion context after packages", () => {
    const completion = renderMainCompletion("bash");

    expect(completion).toContain("COMP_WORDS[COMP_CWORD - 1]");
    expect(completion).toContain('== "packages"');
    expect(completion).toContain("compgen -W 'path --help -h'");
  });

  it("returns undefined for unsupported completion shells", () => {
    expect(renderMainCompletion("powershell")).toBeUndefined();
  });

  it("resolves unsupported completion shells as a command error", () => {
    expect(resolveMainCompletionCommand("powershell")).toEqual({
      stderr: 'Unsupported shell for completion. Use "bash", "zsh", or "fish".\n',
      exitCode: 2,
    });
  });
});