"""Catalog of ready-to-run cloud images for the quick VM flow.

Each entry points to the vendor's official cloud image. Images boot in seconds
and are configured through cloud-init (user, password, SSH key) — no installer.
"""
import re
from typing import Any

# osinfo ids must exist in osinfo-db shipped with Ubuntu 24.04+/Debian 13.
CLOUD_IMAGES: list[dict[str, Any]] = [
    {
        "id": "ubuntu-26.04",
        "title": "Ubuntu Server 26.04 LTS",
        "vendor": "Ubuntu",
        "hint": "Самая свежая LTS, поддержка до 2031 года. Рекомендуем.",
        "osinfo": "ubuntu24.04",
        "recommended": True,
        "login": "ubuntu",
        "arches": {
            "x86_64": {"url": "https://cloud-images.ubuntu.com/resolute/current/resolute-server-cloudimg-amd64.img", "sums": "https://cloud-images.ubuntu.com/resolute/current/SHA256SUMS", "algo": "sha256"},
            "aarch64": {"url": "https://cloud-images.ubuntu.com/resolute/current/resolute-server-cloudimg-arm64.img", "sums": "https://cloud-images.ubuntu.com/resolute/current/SHA256SUMS", "algo": "sha256"},
        },
    },
    {
        "id": "ubuntu-24.04",
        "title": "Ubuntu Server 24.04 LTS",
        "vendor": "Ubuntu",
        "hint": "Проверенная LTS, поддержка до 2029 года.",
        "osinfo": "ubuntu24.04",
        "login": "ubuntu",
        "arches": {
            "x86_64": {"url": "https://cloud-images.ubuntu.com/noble/current/noble-server-cloudimg-amd64.img", "sums": "https://cloud-images.ubuntu.com/noble/current/SHA256SUMS", "algo": "sha256"},
            "aarch64": {"url": "https://cloud-images.ubuntu.com/noble/current/noble-server-cloudimg-arm64.img", "sums": "https://cloud-images.ubuntu.com/noble/current/SHA256SUMS", "algo": "sha256"},
        },
    },
    {
        "id": "debian-13",
        "title": "Debian 13",
        "vendor": "Debian",
        "hint": "Лёгкий и стабильный. Хорош для домашних сервисов.",
        "osinfo": "debian13",
        "login": "debian",
        "arches": {
            "x86_64": {"url": "https://cloud.debian.org/images/cloud/trixie/latest/debian-13-genericcloud-amd64.qcow2", "sums": "https://cloud.debian.org/images/cloud/trixie/latest/SHA512SUMS", "algo": "sha512"},
            "aarch64": {"url": "https://cloud.debian.org/images/cloud/trixie/latest/debian-13-genericcloud-arm64.qcow2", "sums": "https://cloud.debian.org/images/cloud/trixie/latest/SHA512SUMS", "algo": "sha512"},
        },
    },
    {
        "id": "alpine-3.23",
        "title": "Alpine Linux 3.23",
        "vendor": "Alpine",
        "hint": "Очень маленький: 200 МБ на диске, стартует мгновенно.",
        "osinfo": "alpinelinux3.21",
        "login": "alpine",
        "arches": {
            "x86_64": {"url": "https://dl-cdn.alpinelinux.org/alpine/v3.23/releases/cloud/nocloud_alpine-3.23.0-x86_64-uefi-cloudinit-r0.qcow2", "sums": "https://dl-cdn.alpinelinux.org/alpine/v3.23/releases/cloud/nocloud_alpine-3.23.0-x86_64-uefi-cloudinit-r0.qcow2.sha512", "algo": "sha512", "uefi": True},
            "aarch64": {"url": "https://dl-cdn.alpinelinux.org/alpine/v3.23/releases/cloud/nocloud_alpine-3.23.0-aarch64-uefi-cloudinit-r0.qcow2", "sums": "https://dl-cdn.alpinelinux.org/alpine/v3.23/releases/cloud/nocloud_alpine-3.23.0-aarch64-uefi-cloudinit-r0.qcow2.sha512", "algo": "sha512", "uefi": True},
        },
    },
]

# OS families for the "install from ISO" flow. virt-install picks the right
# virtual hardware from osinfo: Windows gets SATA/e1000e (works without extra
# drivers), win11 additionally gets UEFI + TPM.
OS_TYPES: list[dict[str, Any]] = [
    {"id": "linux", "title": "Linux", "hint": "Ubuntu, Debian, Fedora и другие", "osinfo": "detect=on,require=off"},
    {"id": "windows11", "title": "Windows 11 / Server 2025", "hint": "Нужны UEFI и TPM — включим сами", "osinfo": "win11"},
    {"id": "windows10", "title": "Windows 10 / Server 2019–2022", "hint": "Классическая загрузка", "osinfo": "win10"},
    {"id": "other", "title": "Другая система", "hint": "FreeBSD, старые ОС, экзотика", "osinfo": "generic"},
]


# Установщик Windows узнаём по метке тома ISO (у Microsoft вида CCCOMA_X64FRE_RU-RU_DV9) или по имени файла.
WINDOWS_RE = re.compile(r"\bWIN(?:DOWS)?\s?(?:1[01]|7|8|SERVER)\b|\bWINDOWS\b|\b(?:X64|X86|A64|ARM64)FRE[EV]?\b")
WINDOWS11_RE = re.compile(r"\bWIN(?:DOWS)?\s?11\b|\bSERVER\s?2025\b|\b(?:22000|22621|22631|26100|26200)\b")
WINDOWS10_RE = re.compile(r"\bWIN(?:DOWS)?\s?(?:10|7|8)\b|\bLTSC\b|\bSERVER\s?20(?:16|19|22)\b|\b1\d{4}\b|\b20348\b")


def guess_os_type(volume_label: str, filename: str) -> str:
    """Тип системы для установочного ISO: windows11 / windows10, если это установщик Windows, иначе "".

    Windows неизвестной версии — windows11: UEFI и TPM подходят и для Windows 10, а без них не встанет 11."""
    text = re.sub(r"[^A-Z0-9]+", " ", f"{volume_label} {filename}".upper())
    if not WINDOWS_RE.search(text):
        return ""
    if WINDOWS11_RE.search(text):
        return "windows11"
    return "windows10" if WINDOWS10_RE.search(text) else "windows11"


# Название системы для карточки образа в форме создания машины.
DISTROS = (
    ("UBUNTU", "Ubuntu"), ("DEBIAN", "Debian"), ("FEDORA", "Fedora"), ("ALPINE", "Alpine"), ("CENTOS", "CentOS"),
    ("ROCKY", "Rocky Linux"), ("ALMALINUX", "AlmaLinux"), ("OPENSUSE", "openSUSE"), ("ARCHLINUX", "Arch Linux"),
    ("LINUXMINT", "Linux Mint"), ("KALI", "Kali Linux"), ("ASTRA", "Astra Linux"), ("REDOS", "РЕД ОС"),
    ("PROXMOX", "Proxmox"), ("TRUENAS", "TrueNAS"), ("FREEBSD", "FreeBSD"), ("PFSENSE", "pfSense"),
    ("OPNSENSE", "OPNsense"), ("OPENWRT", "OpenWrt"),
)


def iso_system_name(volume_label: str, filename: str) -> str:
    """«Windows 10», «Ubuntu»… или "", если систему по образу не узнать."""
    guess = guess_os_type(volume_label, filename)
    if guess:
        return "Windows 11" if guess == "windows11" else "Windows 10"
    words = set(re.sub(r"[^A-Z0-9]+", " ", f"{volume_label} {filename}".upper()).split())
    return next((title for key, title in DISTROS if key in words), "")


def cloud_image(image_id: str) -> dict[str, Any] | None:
    return next((item for item in CLOUD_IMAGES if item["id"] == image_id), None)


def os_type(type_id: str) -> dict[str, Any]:
    return next((item for item in OS_TYPES if item["id"] == type_id), OS_TYPES[0])


def image_filename(entry: dict[str, Any], arch: str) -> str:
    url = entry["arches"][arch]["url"]
    return url.rsplit("/", 1)[-1]


def entry_for_filename(name: str) -> dict[str, Any] | None:
    """Catalog entry whose download has this file name (any architecture)."""
    for entry in CLOUD_IMAGES:
        for arch in entry["arches"].values():
            if arch["url"].rsplit("/", 1)[-1] == name:
                return entry
    return None