import yargs from "yargs"
import type { CommandModule } from "yargs"
import * as Log from "@opencode-ai/core/util/log"

type Cmd = CommandModule<any, any>

const ANSI_REGEX = /\x1b\[[0-9;]*m/g

function strip(text: string): string {
  return text.replace(ANSI_REGEX, "")
}

function extractCommandName(cmd: Cmd): string | undefined {
  const raw = typeof cmd.command === "string" ? cmd.command : cmd.command?.[0]
  if (!raw) return undefined
  if (raw.startsWith("$0")) return raw.slice(2).trim() || ""
  return raw.split(/[\s[<]/)[0]
}

async function getHelpText(name: string, cmd: Cmd): Promise<string> {
  const inst = yargs([])
    .scriptName(name ? `kilo ${name}` : "kilo")
    .wrap(null)
  if (cmd.builder) {
    if (typeof cmd.builder === "function") {
      ;(cmd.builder as any)(inst)
    } else {
      inst.options(cmd.builder as any)
    }
  }
  if (cmd.describe) {
    inst.usage(typeof cmd.describe === "string" ? cmd.describe : "")
  }
  const help = await inst.getHelp()
  return strip(help)
}

async function getSubcommands(
  name: string,
  builder: ((y: any) => any) | undefined,
  depth = 0,
): Promise<Array<{ name: string; hidden: boolean; help: string }>> {
  if (!builder || typeof builder !== "function") return []
  if (depth > 4) return [] // guard against infinite recursion

  const inst = yargs([]).scriptName(`kilo ${name}`).wrap(null)
  builder(inst)

  const result: Array<{ name: string; hidden: boolean; help: string }> = []

  try {
    // yargs 18 internal API — verified against yargs@18.0.0
    // If these internals change, the catch block below will log a warning
    // and subcommand help will be omitted (top-level help still works)
    const internal = (inst as any).getInternalMethods()
    const cmdInstance = internal.getCommandInstance()
    const handlers = cmdInstance.getCommandHandlers()

    for (const [sub, handler] of Object.entries(handlers as Record<string, any>)) {
      if (sub === "$0") continue

      const full = `${name} ${sub}`
      const subInst = yargs([]).scriptName(`kilo ${full}`).wrap(null)

      if (handler.builder && typeof handler.builder === "function") {
        handler.builder(subInst)
      } else if (handler.builder && typeof handler.builder === "object") {
        subInst.options(handler.builder)
      }

      if (handler.description) {
        subInst.usage(handler.description)
      }

      const help = strip(await subInst.getHelp())
      result.push({
        name: full,
        hidden: handler.description === false,
        help,
      })

      // recurse into sub-subcommands
      const deeper = await getSubcommands(
        full,
        typeof handler.builder === "function" ? handler.builder : undefined,
        depth + 1,
      )
      result.push(...deeper)
    }
  } catch (err) {
    Log.Default.warn("failed to extract subcommands via yargs internals", { err })
  }

  return result
}

function formatMarkdown(
  sections: Array<{
    name: string
    hidden: boolean
    help: string
    subs: Array<{ name: string; hidden: boolean; help: string }>
  }>,
): string {
  const parts: string[] = []

  for (const section of sections) {
    parts.push(`## ${section.name ? `kilo ${section.name}` : "kilo"}`)
    parts.push("")
    if (section.hidden) {
      parts.push("> **Internal command** — not intended for direct use.")
      parts.push("")
    }
    parts.push("```")
    parts.push(section.help)
    parts.push("```")
    parts.push("")

    for (const sub of section.subs) {
      parts.push(`### kilo ${sub.name}`)
      parts.push("")
      if (sub.hidden) {
        parts.push("> **Internal command** — not intended for direct use.")
        parts.push("")
      }
      parts.push("```")
      parts.push(sub.help)
      parts.push("```")
      parts.push("")
    }
  }

  return parts.join("\n")
}

function formatText(
  sections: Array<{
    name: string
    hidden: boolean
    help: string
    subs: Array<{ name: string; hidden: boolean; help: string }>
  }>,
): string {
  const parts: string[] = []
  const rule = "=".repeat(80)

  for (const section of sections) {
    parts.push(rule)
    const display = section.name ? `kilo ${section.name}` : "kilo"
    const label = section.hidden ? `${display} [internal]` : display
    parts.push(label)
    parts.push(rule)
    parts.push("")
    parts.push(section.help)
    parts.push("")

    for (const sub of section.subs) {
      const sublabel = sub.hidden ? `--- kilo ${sub.name} [internal] ---` : `--- kilo ${sub.name} ---`
      parts.push(sublabel)
      parts.push("")
      parts.push(sub.help)
      parts.push("")
    }
  }

  return parts.join("\n")
}

async function loadCommands(): Promise<Cmd[]> {
  const { commands } = await import("./commands")
  return commands as Cmd[]
}

export async function generateHelp(options: {
  command?: string
  all?: boolean
  format?: "md" | "text"
  commands?: Cmd[]
}): Promise<string> {
  const format = options.format ?? "md"

  const cmds = options.commands ?? (await loadCommands())
  const relevant = (() => {
    if (options.command) return cmds.filter((c) => extractCommandName(c) === options.command)
    if (options.all) return cmds.filter((c) => extractCommandName(c) !== undefined && c.describe)
    return []
  })()

  if (options.command && relevant.length === 0) {
    throw new Error(`unknown command: ${options.command}`)
  }

  const sections: Array<{
    name: string
    hidden: boolean
    help: string
    subs: Array<{ name: string; hidden: boolean; help: string }>
  }> = []

  for (const cmd of relevant) {
    const name = extractCommandName(cmd)!
    const help = await getHelpText(name, cmd)
    const hidden = (cmd as any).hidden === true
    const subs = await getSubcommands(name, typeof cmd.builder === "function" ? cmd.builder : undefined)

    sections.push({ name, hidden, help, subs })
  }

  return format === "md" ? formatMarkdown(sections) : formatText(sections)
}

export async function generateCommandTable(options?: { commands?: Cmd[] }) {
  const cmds = options?.commands ?? (await loadCommands())

  const rows: Array<{ display: string; description: string }> = []

  for (const cmd of cmds) {
    const raw = typeof cmd.command === "string" ? cmd.command : cmd.command?.[0]
    if (!raw) continue
    if (!cmd.describe) continue

    const display = raw.startsWith("$0") ? "kilo" + raw.slice(2) : "kilo " + raw

    rows.push({
      display: display.trim(),
      description: typeof cmd.describe === "string" ? cmd.describe : "",
    })
  }

  // The repo enforces unpadded markdown table separators via
  // script/check-md-table-padding.ts (see AGENTS.md). Padded `| --- | --- |`
  // separators fail CI, so emit the compact form here.
  const lines = ["| Command | Description |", "|---|---|"]

  for (const row of rows) {
    lines.push(`| \`${row.display}\` | ${row.description} |`)
  }

  return lines.join("\n") + "\n"
}
