"""Assign global sales regions from ISO2 country codes."""

from __future__ import annotations

from typing import Any, Dict, Optional

# Core region sets
NA_CODES = {"US", "CA", "MX"}

LATAM_CODES = {
    "AG",
    "AI",
    "AR",
    "AW",
    "BB",
    "BL",
    "BM",
    "BO",
    "BQ",
    "BR",
    "BS",
    "BZ",
    "CL",
    "CO",
    "CR",
    "CU",
    "CW",
    "DM",
    "DO",
    "EC",
    "FK",
    "GD",
    "GF",
    "GP",
    "GT",
    "GY",
    "HN",
    "HT",
    "JM",
    "KN",
    "KY",
    "LC",
    "MF",
    "MQ",
    "MS",
    "NI",
    "PA",
    "PE",
    "PM",
    "PR",
    "PY",
    "SR",
    "SV",
    "SX",
    "TC",
    "TT",
    "UY",
    "VC",
    "VE",
    "VG",
    "VI",
}

APJ_CODES = {
    "AF",
    "AS",
    "AU",
    "BD",
    "BN",
    "BT",
    "CC",
    "CK",
    "CN",
    "CX",
    "FJ",
    "FM",
    "GU",
    "HK",
    "ID",
    "IN",
    "JP",
    "KG",
    "KH",
    "KI",
    "KP",
    "KR",
    "KZ",
    "LA",
    "LK",
    "MH",
    "MM",
    "MN",
    "MO",
    "MP",
    "MV",
    "MY",
    "NC",
    "NF",
    "NP",
    "NR",
    "NU",
    "NZ",
    "PF",
    "PG",
    "PH",
    "PK",
    "PN",
    "PW",
    "SB",
    "SG",
    "TH",
    "TJ",
    "TK",
    "TL",
    "TM",
    "TO",
    "TV",
    "TW",
    "UM",
    "UZ",
    "VN",
    "VU",
    "WF",
    "WS",
}

EMEA_CODES = {
    "AD",
    "AE",
    "AL",
    "AM",
    "AO",
    "AT",
    "AX",
    "AZ",
    "BA",
    "BE",
    "BF",
    "BG",
    "BH",
    "BI",
    "BJ",
    "BW",
    "BY",
    "CD",
    "CF",
    "CG",
    "CH",
    "CI",
    "CM",
    "CV",
    "CY",
    "CZ",
    "DE",
    "DJ",
    "DK",
    "DZ",
    "EE",
    "EG",
    "EH",
    "ER",
    "ES",
    "ET",
    "FI",
    "FO",
    "FR",
    "GA",
    "GB",
    "GE",
    "GG",
    "GH",
    "GI",
    "GL",
    "GM",
    "GN",
    "GQ",
    "GR",
    "GW",
    "HR",
    "HU",
    "IE",
    "IL",
    "IM",
    "IO",
    "IQ",
    "IR",
    "IS",
    "IT",
    "JE",
    "JO",
    "KE",
    "KM",
    "KW",
    "LB",
    "LI",
    "LR",
    "LS",
    "LT",
    "LU",
    "LV",
    "LY",
    "MA",
    "MC",
    "MD",
    "ME",
    "MG",
    "MK",
    "ML",
    "MR",
    "MT",
    "MU",
    "MW",
    "MZ",
    "NA",
    "NE",
    "NG",
    "NL",
    "NO",
    "OM",
    "PL",
    "PS",
    "PT",
    "QA",
    "RE",
    "RO",
    "RS",
    "RU",
    "RW",
    "SA",
    "SC",
    "SD",
    "SE",
    "SH",
    "SI",
    "SJ",
    "SK",
    "SL",
    "SM",
    "SN",
    "SO",
    "SS",
    "ST",
    "SY",
    "SZ",
    "TD",
    "TG",
    "TN",
    "TR",
    "TZ",
    "UA",
    "UG",
    "VA",
    "XK",
    "YE",
    "YT",
    "ZA",
    "ZM",
    "ZW",
}

# Synonyms frequently seen in CRM exports
SYNONYMS = {
    "UNITED STATES": "US",
    "UNITED STATES OF AMERICA": "US",
    "USA": "US",
    "UNITED KINGDOM": "GB",
    "GREAT BRITAIN": "GB",
    "BRITAIN": "GB",
    "UK": "GB",
    "EN": "GB",
    "ENG": "GB",
    "SCOTLAND": "GB",
    "WALES": "GB",
    "UAE": "AE",
    "KSA": "SA",
    "SOUTH KOREA": "KR",
    "NORTH KOREA": "KP",
    "VIET NAM": "VN",
    "SAUDI ARABIA": "SA",
    "UNITED ARAB EMIRATES": "AE",
    "SOUTH AFRICA": "ZA",
    "MEXICO": "MX",
    "CANADA": "CA",
    "AUSTRALIA": "AU",
    "NEW ZEALAND": "NZ",
}

REGION_BY_CODE = {code: "NA" for code in NA_CODES}
REGION_BY_CODE.update({code: "LATAM" for code in LATAM_CODES})
REGION_BY_CODE.update({code: "EMEA" for code in EMEA_CODES})
REGION_BY_CODE.update({code: "APJ" for code in APJ_CODES})


def _normalize_country_code(value: Optional[str]) -> str:
    if not value:
        return ""
    candidates = str(value).strip().upper()
    if not candidates:
        return ""
    return SYNONYMS.get(candidates, candidates)


def _assign_global_region(code: str) -> str:
    return REGION_BY_CODE.get(code, "")


def assign_global_region(value: str | None) -> Dict[str, Any]:
    """Return the FoundryMatch global region for a 2-letter ISO country code."""

    code = _normalize_country_code(value)
    if not code:
        return {"value": "", "meta": {"reason": "empty", "input": value}}
    if len(code) != 2:
        return {"value": "", "meta": {"reason": "invalid_code", "input": value}}

    region = _assign_global_region(code)
    if not region:
        return {"value": "", "meta": {"reason": "unknown_code", "code": code}}

    return {"value": region, "meta": {"code": code, "region": region}}
