from __future__ import annotations


TEAM_FLAG_URL_MAP = {
    "Australia": "https://flagcdn.com/w40/au.png",
    "Brazil": "https://flagcdn.com/w40/br.png",
    "Canada": "https://flagcdn.com/w40/ca.png",
    "China": "https://flagcdn.com/w40/cn.png",
    "Chinese Taipei": "https://flagcdn.com/w40/tw.png",
    "Colombia": "https://flagcdn.com/w40/co.png",
    "Cuba": "https://flagcdn.com/w40/cu.png",
    "Czechia": "https://flagcdn.com/w40/cz.png",
    "Dominican Republic": "https://flagcdn.com/w40/do.png",
    "Great Britain": "https://flagcdn.com/w40/gb.png",
    "Israel": "https://flagcdn.com/w40/il.png",
    "Italy": "https://flagcdn.com/w40/it.png",
    "Japan": "https://flagcdn.com/w40/jp.png",
    "Korea": "https://flagcdn.com/w40/kr.png",
    "Mexico": "https://flagcdn.com/w40/mx.png",
    "Netherlands": "https://flagcdn.com/w40/nl.png",
    "Nicaragua": "https://flagcdn.com/w40/ni.png",
    "Panama": "https://flagcdn.com/w40/pa.png",
    "Puerto Rico": "https://flagcdn.com/w40/pr.png",
    "United States": "https://flagcdn.com/w40/us.png",
    "Venezuela": "https://flagcdn.com/w40/ve.png",
}


MLB_TEAM_LOGOS = {
    "angels": "https://a.espncdn.com/i/teamlogos/mlb/500/ana.png",
    "astros": "https://a.espncdn.com/i/teamlogos/mlb/500/hou.png",
    "athletics": "https://a.espncdn.com/i/teamlogos/mlb/500/oak.png",
    "blue jays": "https://a.espncdn.com/i/teamlogos/mlb/500/tor.png",
    "braves": "https://a.espncdn.com/i/teamlogos/mlb/500/atl.png",
    "brewers": "https://a.espncdn.com/i/teamlogos/mlb/500/mil.png",
    "cardinals": "https://a.espncdn.com/i/teamlogos/mlb/500/stl.png",
    "cubs": "https://a.espncdn.com/i/teamlogos/mlb/500/chc.png",
    "diamondbacks": "https://a.espncdn.com/i/teamlogos/mlb/500/ari.png",
    "dodgers": "https://a.espncdn.com/i/teamlogos/mlb/500/lad.png",
    "giants": "https://a.espncdn.com/i/teamlogos/mlb/500/sf.png",
    "guardians": "https://a.espncdn.com/i/teamlogos/mlb/500/cle.png",
    "mariners": "https://a.espncdn.com/i/teamlogos/mlb/500/sea.png",
    "marlins": "https://a.espncdn.com/i/teamlogos/mlb/500/mia.png",
    "mets": "https://a.espncdn.com/i/teamlogos/mlb/500/nym.png",
    "nationals": "https://a.espncdn.com/i/teamlogos/mlb/500/wsh.png",
    "orioles": "https://a.espncdn.com/i/teamlogos/mlb/500/bal.png",
    "padres": "https://a.espncdn.com/i/teamlogos/mlb/500/sd.png",
    "phillies": "https://a.espncdn.com/i/teamlogos/mlb/500/phi.png",
    "pirates": "https://a.espncdn.com/i/teamlogos/mlb/500/pit.png",
    "rangers": "https://a.espncdn.com/i/teamlogos/mlb/500/tex.png",
    "rays": "https://a.espncdn.com/i/teamlogos/mlb/500/tb.png",
    "red sox": "https://a.espncdn.com/i/teamlogos/mlb/500/bos.png",
    "reds": "https://a.espncdn.com/i/teamlogos/mlb/500/cin.png",
    "rockies": "https://a.espncdn.com/i/teamlogos/mlb/500/col.png",
    "royals": "https://a.espncdn.com/i/teamlogos/mlb/500/kc.png",
    "tigers": "https://a.espncdn.com/i/teamlogos/mlb/500/det.png",
    "twins": "https://a.espncdn.com/i/teamlogos/mlb/500/min.png",
    "white sox": "https://a.espncdn.com/i/teamlogos/mlb/500/cws.png",
    "yankees": "https://a.espncdn.com/i/teamlogos/mlb/500/nyy.png",
}


def get_team_flag_url(team_name: str) -> str:
    return TEAM_FLAG_URL_MAP.get(str(team_name).strip(), "")


def get_team_logo_url(team_name: str) -> str:
    team_norm = str(team_name or "").strip().lower()

    for key, url in MLB_TEAM_LOGOS.items():
        if team_norm == key or team_norm.endswith(f" {key}"):
            return url

    return ""