import argparse
import shlex
import subprocess
from dataclasses import dataclass
from pathlib import Path

from ..errors import ClanError
from ..nix import nix_config, nix_eval, nix_metadata


@dataclass
class FlakeConfig:
    flake_url: str | Path
    flake_attr: str

    icon: str | None
    description: str | None
    last_updated: str
    revision: str | None


def inspect_flake(flake_url: str | Path, flake_attr: str) -> FlakeConfig:
    config = nix_config()
    system = config["system"]

    cmd = nix_eval(
        [
            f'{flake_url}#clanInternals.machines."{system}"."{flake_attr}".config.clanCore.clanIcon'
        ]
    )

    proc = subprocess.run(cmd, check=True, text=True, stdout=subprocess.PIPE)
    assert proc.stdout is not None
    if proc.returncode != 0:
        raise ClanError(
            f"""
command: {shlex.join(cmd)}
exit code: {proc.returncode}
stdout:
{proc.stdout}
"""
        )
    res = proc.stdout.strip()
    if res == "null":
        icon_path = None
    else:
        icon_path = res.strip('"')

    meta = nix_metadata(flake_url)

    return FlakeConfig(
        flake_url=flake_url,
        flake_attr=flake_attr,
        icon=icon_path,
        description=meta.get("description"),
        last_updated=meta["lastModified"],
        revision=meta.get("revision"),
    )


@dataclass
class InspectOptions:
    machine: str
    flake: Path


def inspect_command(args: argparse.Namespace) -> None:
    inspect_options = InspectOptions(
        machine=args.machine,
        flake=args.flake or Path.cwd(),
    )
    res = inspect_flake(
        flake_url=inspect_options.flake, flake_attr=inspect_options.machine
    )
    print("Icon:", res.icon)
    print("Description:", res.description)
    print("Last updated:", res.last_updated)
    print("Revision:", res.revision)


def register_inspect_parser(parser: argparse.ArgumentParser) -> None:
    parser.add_argument("--machine", type=str, default="defaultVM")
    parser.set_defaults(func=inspect_command)