"""Click commands for inspecting connected browser clients."""
from __future__ import annotations

import os
import sys

import click
from rich.console import Console

from browser_cli.client import (
  BrowserNotConnected,
  REGISTRY_PATH,
  active_browser_targets,
  collect_browser_clients,
  send_command,
)
from browser_cli.commands.rendering import print_browser_grouped_table_rows
from browser_cli.registry import load_registry

console = Console()

def _rename_target_profile(target_browser: str | None) -> str | None:
  if target_browser:
    return target_browser

  active = active_browser_targets()
  if len(active) == 1:
    return active[0].profile
  return None

def _ensure_unique_browser_alias(alias: str, target_browser: str | None) -> None:
  target_profile = _rename_target_profile(target_browser)
  profiles: dict[str, str] = load_registry(REGISTRY_PATH)

  if alias in profiles and alias != target_profile:
    raise click.ClickException(f"Browser alias '{alias}' already exists")


@click.group("clients", invoke_without_command=True)
@click.pass_context
def clients_group(ctx):
  """Inspect and manage connected browser clients."""
  if ctx.invoked_subcommand is not None:
    return

  browser_alias = (ctx.obj or {}).get("browser")
  remote = (ctx.obj or {}).get("remote") or os.environ.get("BROWSER_CLI_REMOTE")
  key = (ctx.obj or {}).get("key")

  try:
    all_clients = collect_browser_clients(
      browser_alias=browser_alias,
      remote=remote,
      key=key,
      registry_path=REGISTRY_PATH,
    )
  except (BrowserNotConnected, RuntimeError) as e:
    console.print(f"[red]Error:[/red] {e}")
    sys.exit(1)

  if not all_clients:
    console.print("[yellow]No browser clients found. Start a browser with the extension enabled first.[/yellow]")
    sys.exit(1)

  _print_clients(all_clients)


def _print_clients(all_clients: list) -> None:
  groups = {c.get("profileGroup") for c in all_clients if c.get("profileGroup")}
  grouped = bool(groups and groups != {"local"})
  columns = [
    ("Browser", lambda item: item.get("name", "")),
    ("Version", lambda item: item.get("version", "")),
    ("Extension Version", lambda item: item.get("extensionVersion", "")),
  ]
  print_browser_grouped_table_rows(
    all_clients,
    columns,
    console=console,
    empty_message="[yellow]No browser clients found. Start a browser with the extension enabled first.[/yellow]",
    browser_getter=lambda item: item.get("profile", ""),
    group_getter=lambda item: item.get("profileGroup", "") if grouped else "",
    browser_header="Profile",
  )

@clients_group.command("rename")
@click.option(
  "--browser",
  "target_browser",
  default=None,
  metavar="ALIAS",
  help="Browser profile alias to rename. Overrides the global --browser option for this command.",
)
@click.argument("alias")
@click.pass_context
def cmd_clients_rename(ctx, target_browser, alias):
  """Set the profile alias used to identify this browser instance."""
  root_obj = ctx.find_root().obj or {}
  selected_browser = target_browser or root_obj.get("browser")
  try:
    _ensure_unique_browser_alias(alias, selected_browser)
    send_command("clients.rename_profile", {"alias": alias}, profile=selected_browser)
  except BrowserNotConnected as e:
    console.print(f"[red]Error:[/red] {e}")
    sys.exit(1)
  console.print(f"[green]Profile renamed to '{alias}'[/green]")