from . import (
    MetaBase,
)
from pydantic import Field
from datetime import datetime
from enum import IntEnum, Enum
from .enum import StrEnum
from typing import Optional, List, Dict, Any, Iterator, Iterable, NamedTuple
from collections import namedtuple
from urllib.parse import urljoin, urlencode, urlparse, urlunparse

# namedtuple to match the internal signature of urlunparse


class JavaRuntimeOS(StrEnum):
    MacOsX64 = "mac-os-x64"
    MacOsX86 = "mac-os-x86" # rare
    MacOsArm64 = "mac-os-arm64"
    # MacOsArm32 = "mac-os-arm32" # doesn't exsist
    LinuxX64 = "linux-x64"
    LinuxX86 = "linux-x86"
    LinuxArm64 = "linux-arm64"
    LinuxArm32 = "linux-arm32"
    WindowsX64 = "windows-x64"
    WindowsX86 = "windows-x86"
    WindowsArm64 = "windows-arm64"
    WindowsArm32 = "windows-arm32"
    Unknown = "unknown"


class JavaRuntimeDownloadType(StrEnum):
    Manifest = "manifest"
    Archive = "archive"


class JavaVersionMeta(MetaBase):
    major: int
    minor: int
    security: int
    build: Optional[int]

    def __str__(self):
        ver = f"{self.major}.{self.minor}.{self.security}"
        if self.build is not None:
            ver = f"{ver}+{self.build}"
        return ver


class JavaChecksumType(StrEnum):
    Sha1 = "sha1"
    Sha256 = "sha256"


class JavaChecksumMeta(MetaBase):
    type: JavaChecksumType
    hash: str


class JavaRuntimeMeta(MetaBase):
    name: str
    vender: str
    url: str
    release_time: datetime = Field(alias="releaseTime")
    checksum: Optional[JavaChecksumMeta]
    recomended: bool
    download_type: JavaRuntimeDownloadType = Field(alias="downloadType")


class JavaRuntimeMap(MetaBase):
    __root__: dict[JavaRuntimeOS, list[JavaRuntimeMeta]] = {
        os: [] for os in JavaRuntimeOS if os != JavaRuntimeOS.Unknown
    }

    def __iter__(self) -> Iterator[JavaRuntimeOS]:
        return iter(self.__root__)

    def __getitem__(self, item) -> list[JavaRuntimeMeta]:
        return self.__root__[item]

    def __len__(self):
        return len(self.__root__)


class URLComponents(NamedTuple):
    scheme: str
    netloc: str
    url: str
    path: str
    query: str
    fragment: str


class APIQuery(MetaBase):
    def to_query(self):
        set_parts: dict[str, Any] = {}
        for key, value in self.dict().items():
            if value is not None:
                if isinstance(value, Enum):
                    set_parts[key] = value.value
                elif isinstance(value, list):
                    if len(value) > 0:
                        set_parts[key] = value
                elif isinstance(value, datetime):
                    set_parts[key] = value.isoformat()
                else:
                    set_parts[key] = value
        return urlencode(set_parts, doseq=True)


class AdoptiumJvmImpl(StrEnum):
    Hostspot = "hotspot"


class AdoptiumVendor(StrEnum):
    Eclipse = "eclipse"


class AdoptiumArchitecture(StrEnum):
    X64 = "x64"
    X86 = "x86"
    X32 = "x32"
    Ppc64 = "ppc64"
    Ppc64le = "ppc64le"
    S390x = "s390x"
    Aarch64 = "aarch64"
    Arm = "arm"
    Sparcv9 = "sparcv9"
    Riscv64 = "riscv64"


class AdoptiumReleaseType(StrEnum):
    GenralAccess = "ga"
    EarlyAccess = "ea"


class AdoptiumSortMethod(StrEnum):
    Default = "DEFAULT"
    Date = "DATE"


class AdoptiumSortOrder(StrEnum):
    Asc = "ASC"
    Desc = "DESC"


class AdoptiumImageType(StrEnum):
    Jdk = "jdk"
    Jre = "jre"
    Testimage = "testimage"
    Debugimage = "debugimage"
    Staticlibs = "staticlibs"
    Sources = "sources"
    Sbom = "sbom"


class AdoptiumHeapSize(StrEnum):
    Normal = "normal"
    Large = "large"


class AdoptiumProject(StrEnum):
    Jdk = "jdk"
    Valhalla = "valhalla"
    Metropolis = "metropolis"
    Jfr = "jfr"
    Shenandoah = "shenandoah"


class AdoptiumCLib(StrEnum):
    Musl = "musl"
    Glibc = "glibc"


class AdoptiumOs(StrEnum):
    Linux = "linux"
    Windows = "windows"
    Mac = "mac"
    Solaris = "solaris"
    Aix = "aix"
    AlpineLinux = "alpine-linux"


ADOPTIUM_API_BASE = " https://api.adoptium.net"
ADOPTIUM_API_FEATURE_RELEASES = f"{ADOPTIUM_API_BASE}/v3/assets/feature_releases/{{feature_version}}/{{release_type}}"
# ?image_type={{image_type}}&heap_size={{heap_size}}&project={{project}}&vender={{vender}}&page_size={{page_size}}&page={{page}}&sort_method={{sort_method}}&sort_order={{sort_order}}
ADOPTIUM_API_AVAILABLE_RELEASES = f"{ADOPTIUM_API_BASE}/v3/info/available_releases"


class AdoptiumAPIFeatureReleasesQuery(APIQuery):
    architecture: Optional[AdoptiumArchitecture] = None
    before: Optional[datetime] = None
    c_lib: Optional[AdoptiumCLib] = None
    heap_size: Optional[AdoptiumHeapSize] = AdoptiumHeapSize.Normal
    image_type: Optional[AdoptiumImageType] = None
    jvm_impl: Optional[AdoptiumJvmImpl] = None
    os: Optional[AdoptiumOs] = None
    page_size: int = 10
    page: int = 0
    project: Optional[AdoptiumProject] = AdoptiumProject.Jdk
    sort_method: Optional[AdoptiumSortMethod] = AdoptiumSortMethod.Default
    sort_order: Optional[AdoptiumSortOrder] = AdoptiumSortOrder.Desc
    vender: Optional[AdoptiumVendor] = AdoptiumVendor.Eclipse


def adoptiumAPIFeatureReleasesUrl(
    feature: int,
    release_type: AdoptiumReleaseType = AdoptiumReleaseType.GenralAccess,
    query: AdoptiumAPIFeatureReleasesQuery = AdoptiumAPIFeatureReleasesQuery(),
):
    url = urlparse(
        ADOPTIUM_API_FEATURE_RELEASES.format(
            feature_version=feature,
            release_type=release_type.value,
        )
    )
    return urlunparse(url._replace(query=query.to_query()))


class AdoptiumAvailableReleases(MetaBase):
    available_releases: list[int]
    available_lts_releases: list[int]
    most_recent_lts: Optional[int]
    most_recent_feature_release: Optional[int]
    most_recent_feature_version: Optional[int]
    tip_version: Optional[int]


class AdoptiumFile(MetaBase):
    name: str
    link: str
    size: Optional[int]


class AdoptiumPackage(AdoptiumFile):
    checksum: Optional[str]
    checksum_link: Optional[str]
    signature_link: Optional[str]
    download_count: Optional[int]
    metadata_link: Optional[str]


class AdoptiumBinary(MetaBase):
    os: str
    architecture: AdoptiumArchitecture
    image_type: AdoptiumImageType
    c_lib: Optional[AdoptiumCLib]
    jvm_impl: AdoptiumJvmImpl
    package: Optional[AdoptiumPackage]
    installer: Optional[AdoptiumPackage]
    heap_size: AdoptiumHeapSize
    download_count: Optional[int]
    updated_at: datetime
    scm_ref: Optional[str]
    project: AdoptiumProject


class AdoptiumVersion(MetaBase):
    major: Optional[int]
    minor: Optional[int]
    security: Optional[int]
    patch: Optional[int]
    pre: Optional[str]
    adopt_build_number: Optional[int]
    semver: str
    openjdk_version: str
    build: Optional[int]
    optional: Optional[str]


class AdoptiumRelease(MetaBase):
    release_id: str = Field(alias="id")
    release_link: str
    release_name: str
    timestamp: datetime
    updated_at: datetime
    binaries: list[AdoptiumBinary]
    download_count: Optional[int]
    release_type: str
    vendor: AdoptiumVendor
    version_data: AdoptiumVersion
    source: Optional[AdoptiumFile]
    release_notes: Optional[AdoptiumFile]


class AdoptiumReleases(MetaBase):
    __root__: list[AdoptiumRelease]

    def __iter__(self) -> Iterator[AdoptiumRelease]:
        return iter(self.__root__)

    def __getitem__(self, item) -> AdoptiumRelease:
        return self.__root__[item]

    def append(self, rls: AdoptiumRelease):
        self.__root__.append(rls)


class AzulProduct(StrEnum):
    Zulu = "zulu"


class AzulAvailabilityType(StrEnum):
    SA = "SA"
    CA = "CA"
    NV = "NV"
    _LA = "LA"


class AzulJavaPackageType(StrEnum):
    Jdk = "jdk"
    Jre = "jre"


class AzulReleaseType(StrEnum):
    CPU = "CPU"
    PSU = "PSU"
    LU = "LU"


class AzulOs(StrEnum):
    Linux = "linux"
    Macos = "macos"
    Qnx = "qnx"
    Windows = "windows"
    Solaris = "solaris"


class AzulLibCType(StrEnum):
    Glibc = "glibc"
    Uclibc = "uclibc"
    Musl = "musl"


class AzulCPUGen(StrEnum):
    V5 = "v5"
    V6kV6kz = "v6k_v6kz"
    V6t2 = "v6t2"
    V7 = "v7"
    V8 = "v8"


class AzulArch(StrEnum):
    Arm = "arm"
    X86 = "x86"
    Mips = "mips"
    Ppc = "ppc"
    Sparcv9 = "sparcv9"
    Sparc = "sparc"


class AzulHwBitness(IntEnum):
    X32 = 32
    X64 = 64


class AzulAbi(StrEnum):
    HardFloat = "hard_float"
    SoftFloat = "soft_float"
    Spe = "spe"
    Any = "any"


class AzulArchiveType(StrEnum):
    Deb = "deb"
    Rpm = "rpm"
    Dmg = "dmg"
    Targz = "tar.gz"
    Zip = "zip"
    Cab = "cab"
    Msi = "msi"


class AzulReleaseStatus(StrEnum):
    Eval = "eval"
    Ea = "ea"
    Ga = "ga"
    Both = "both"


class AzulSupportTerm(StrEnum):
    Sts = "sts"
    Mts = "mts"
    Lts = "lts"


class AzulCertifications(StrEnum):
    Tck = "tck"
    _Aqavit = "aqavit"
    none = "none"


class AzulSignatureType(StrEnum):
    Openpgp = "openpgp"


class AzulOsQueryParam(StrEnum):
    Macos = "macos"
    Windows = "windows"
    Linux = "linux"
    LinuxMusl = "linux-musl"
    LinuxGlibc = "linux-glibc"
    Qnx = "qnx"
    Solaris = "solaris"


class AzulArchQueryParam(StrEnum):
    X86 = "x86"
    X64 = "x64"
    Amd64 = "amd64"
    I686 = "i686"
    Arm = "arm"
    Aarch64 = "aarch64"
    Aarch32 = "aarch32"
    Aarch32sf = "aarch32sf"
    Aarch32hf = "aarch32hf"
    Ppc = "ppc"
    Ppc64 = "ppc64"
    Ppc64hf = "ppc64hf"
    Ppc32 = "ppc32"
    Ppc32spe = "ppc32spe"
    Ppc32hf = "ppc32hf"
    Sparc = "sparc"
    Sparc32 = "sparc32"
    Sparcv9 = "sparcv9"
    Sparcv9_64 = "sparcv9-64"


AZUL_API_BASE = "https://api.azul.com/metadata/v1"
AZUL_API_PACKAGES = f"{AZUL_API_BASE}/zulu/packages/"
AZUL_API_PACKAGE_DETAIL = f"{AZUL_API_BASE}/zulu/packages/{{package_uuid}}"


class AzulApiPackagesQuery(APIQuery):
    java_version: Optional[str] = None
    os: Optional[AzulOsQueryParam] = None
    arch: Optional[AzulArchQueryParam] = None
    archive_type: Optional[AzulArchiveType] = None
    java_package_type: Optional[AzulJavaPackageType] = None
    javafx_bundled: Optional[bool] = None
    crac_supported: Optional[bool] = None
    support_term: Optional[AzulSupportTerm] = None
    release_type: Optional[AzulReleaseType] = None
    latest: Optional[bool] = None
    distro_version: Optional[str] = None
    java_package_features: list[str] = []
    release_status: Optional[AzulReleaseStatus] = None
    availability_types: list[AzulAvailabilityType] = None
    certifications: list[AzulCertifications] = None
    include_fields: list[str] = []
    page: int = 0
    page_size: int = 100


def azulApiPackagesUrl(query: AzulApiPackagesQuery = AzulApiPackagesQuery()):
    url = urlparse(AZUL_API_PACKAGES)
    return urlunparse(url._replace(query=query.to_query()))


def azulApiPackageDetailUrl(package_uuid: str):
    return AZUL_API_PACKAGE_DETAIL.format(package_uuid=package_uuid)


class ZuluSignatureDetail(MetaBase):
    type: AzulSignatureType
    url: str
    details: dict[str, Any]
    signature_index: int
    signature: str


class ZuluPackageDetail(MetaBase):
    package_uuid: str
    name: Optional[str]
    md5_hash: Optional[str]
    sha256_hash: Optional[str]
    build_date: datetime
    last_modified: datetime
    download_url: str
    product: AzulProduct
    availability_type: AzulAvailabilityType
    java_version: list[int]
    openjdk_build_number: Optional[int]
    java_package_type: AzulJavaPackageType
    javafx_bundled: bool
    release_type: AzulReleaseType
    os: AzulOs
    lib_c_type: Optional[AzulLibCType]
    cpu_gen: Optional[list[AzulCPUGen]]
    arch: AzulArch
    hw_bitness: AzulHwBitness
    abi: AzulAbi
    archive_type: AzulArchiveType
    release_status: AzulReleaseStatus
    support_term: AzulSupportTerm
    certifications: Optional[list[AzulCertifications]]
    latest: Optional[bool]
    size: int
    distro_version: list[int]
    signatures: list[ZuluSignatureDetail]


class ZuluPackageList(MetaBase):
    package_uuid: str
    name: Optional[str]
    java_version: list[int]
    openjdk_build_number: Optional[int]
    latest: Optional[bool]
    download_url: str
    product: Optional[AzulProduct]
    distro_version: list[int]
    availability_type: Optional[AzulAvailabilityType]


class ZuluPackages(MetaBase):
    __root__: list[ZuluPackageList]

    def __iter__(self) -> Iterator[ZuluPackageList]:
        return iter(self.__root__)

    def __getitem__(self, item) -> ZuluPackageList:
        return self.__root__[item]

    def append(self, pkg: ZuluPackageList):
        self.__root__.append(pkg)


class ZuluPackagesDetail(MetaBase):
    __root__: list[ZuluPackageDetail]

    def __iter__(self) -> Iterator[ZuluPackageDetail]:
        return iter(self.__root__)

    def __getitem__(self, item) -> ZuluPackageDetail:
        return self.__root__[item]

    def append(self, pkg: ZuluPackageDetail):
        self.__root__.append(pkg)


MOJANG_OS_NAMES = ["mac-os", "linux", "windows"]

MOJANG_OS_ARCHITECTURES = [
    "x64" "x86",
    "arm64",
    "arm32",
]