from typing import Dict, Union from urllib.parse import urljoin from httpx import URL as _URL from simnet.errors import RegionNotSupported, NotSupported from simnet.utils.enum_ import Region, Game URLTypes = Union["URL", str] __all__ = ( "URL", "BaseRoute", "Route", "InternationalRoute", "GameRoute", "RECORD_URL", "GACHA_INFO_URL", "AUTH_URL", "PASSPORT_URL", "WEB_ACCOUNT_URL", "AUTH_KEY_URL", "HK4E_LOGIN_URL", "REWARD_URL", "TAKUMI_URL", "CALCULATOR_URL", "DETAIL_LEDGER_URL", "INFO_LEDGER_URL", "HK4E_URL", "CODE_URL", "YSULOG_URL", "QRCODE_URL", ) class URL(_URL): """A subclass of httpx's URL class, with additional convenience methods for URL manipulation.""" def join(self, url: URLTypes) -> "URL": """ Join the current URL with the given URL. Args: url (Union[URL, str]): The URL to join with. Returns: URL: A new URL instance representing the joined URL. """ return URL(urljoin(str(self), str(URL(url)))) def __truediv__(self, url: URLTypes) -> "URL": """ Append the given URL to the current URL using the '/' operator. Args: url (Union[URL, str]): The URL to append. Returns: URL: A new URL instance representing the joined URL. """ return URL(urljoin(str(self) + "/", str(URL(url)))) def __bool__(self): """Return True if the URL is not empty. Returns: bool: True if the URL is not empty. """ return str(self) != "" def replace(self, old: str, new: str) -> "URL": """ Replace a substring in the URL. Args: old (str): The substring to replace. new (str): The new substring to replace with. Returns: URL: A new URL instance with the substring replaced. """ return URL(str(self).replace(old, new)) class BaseRoute: """A base class for defining routes with useful metadata.""" class Route(BaseRoute): """A standard route with a single URL.""" url: URL def __init__(self, url: str) -> None: """ Initialize a Route instance. Args: url (str): The URL for this route. """ self.url = URL(url) def get_url(self) -> URL: """ Get the URL for this route. Returns: URL: The URL for this route. """ return self.url def __truediv__(self, other: str) -> URL: """ Append the given URL to this route using the '/' operator. Args: other (Union[URL, str]): The URL to append. Returns: URL: A new URL instance representing the joined URL. """ return self.url / other class InternationalRoute(BaseRoute): """A route with URLs for both the overseas and Chinese regions.""" urls: Dict[Region, URL] def __init__(self, overseas: str, chinese: str) -> None: """ Initialize an InternationalRoute instance. Args: overseas (str): The URL for the overseas region. chinese (str): The URL for the Chinese region. """ self.urls = { Region.OVERSEAS: URL(overseas), Region.CHINESE: URL(chinese), } def get_url(self, region: Region) -> URL: """ Get the URL for the given region. Args: region (Region): The region to get the URL for. Returns: URL: The URL for the given region. Raises: RegionNotSupported: If the given region is not supported. """ if not self.urls[region]: raise RegionNotSupported(f"URL does not support {region.name} region.") return self.urls[region] class GameRoute(BaseRoute): """A route with URLs for different games and regions.""" urls: Dict[Region, Dict[Game, URL]] def __init__( self, overseas: Dict[str, str], chinese: Dict[str, str], ) -> None: """ Initialize a GameRoute instance. Args: overseas (Dict[str, str]): A dictionary mapping game names to URLs for the overseas region. chinese (Dict[str, str]): A dictionary mapping game names to URLs for the Chinese region. """ self.urls = { Region.OVERSEAS: {Game(game): URL(url) for game, url in overseas.items()}, Region.CHINESE: {Game(game): URL(url) for game, url in chinese.items()}, } def get_url(self, region: Region, game: Game) -> URL: """ Get the URL for the given region and game. Args: region (Region): The region to get the URL for. game (Game): The game to get the URL for. Returns: URL: The URL for the given region and game. Raises: RegionNotSupported: If the given region is not supported. GameNotSupported: If the given game is not supported. """ if not self.urls[region]: raise RegionNotSupported(f"URL does not support {region.name} region.") if not self.urls[region][game]: raise NotSupported(f"URL does not support {game.name} game for {region.name} region.") return self.urls[region][game] RECORD_URL = InternationalRoute( overseas="https://bbs-api-os.hoyolab.com/game_record", chinese="https://api-takumi-record.mihoyo.com/game_record/app", ) GACHA_INFO_URL = GameRoute( overseas=dict( genshin="https://hk4e-api-os.hoyoverse.com/event/gacha_info/api", hkrpg="https://api-os-takumi.mihoyo.com/common/gacha_record/api/", ), chinese=dict( genshin="https://hk4e-api.mihoyo.com/event/gacha_info/api", hkrpg="https://api-takumi.mihoyo.com/common/gacha_record/api", ), ) AUTH_URL = InternationalRoute( overseas="https://api-os-takumi.mihoyo.com/auth/api", chinese="https://api-takumi.mihoyo.com/auth/api", ) PASSPORT_URL = InternationalRoute( overseas="https://api-account-os.hoyoverse.com/account/auth/api/", chinese="https://passport-api.mihoyo.com/account/auth/api/", ) WEB_ACCOUNT_URL = InternationalRoute( overseas="https://webapi-os.account.hoyoverse.com/Api/", chinese="https://webapi.account.mihoyo.com/Api/", ) AUTH_KEY_URL = InternationalRoute( overseas="https://sg-public-api.hoyoverse.com/binding/api/genAuthKey", chinese="https://api-takumi.mihoyo.com/binding/api/genAuthKey", ) HK4E_LOGIN_URL = InternationalRoute( overseas="https://sg-public-api.hoyoverse.com/common/badge/v1/login/account", chinese="https://api-takumi.mihoyo.com/common/badge/v1/login/account", ) REWARD_URL = GameRoute( overseas=dict( genshin="https://sg-hk4e-api.hoyolab.com/event/sol/?act_id=e202102251931481", honkai3rd="https://sg-public-api.hoyolab.com/event/mani/?act_id=e202110291205111", hkrpg="https://sg-public-api.hoyolab.com/event/luna/os/?act_id=e202303301540311", ), chinese=dict( genshin="https://api-takumi.mihoyo.com/event/bbs_sign_reward/?act_id=e202009291139501", honkai3rd="https://api-takumi.mihoyo.com/event/luna/?act_id=e202207181446311", hkrpg="https://api-takumi.mihoyo.com/event/luna/?act_id=e202304121516551", ), ) TAKUMI_URL = InternationalRoute( overseas="https://api-os-takumi.mihoyo.com/", chinese="https://api-takumi.mihoyo.com/", ) CALCULATOR_URL = InternationalRoute( overseas="https://sg-public-api.hoyoverse.com/event/calculateos/", chinese="https://api-takumi.mihoyo.com/event/e20200928calculate/v1/", ) DETAIL_LEDGER_URL = GameRoute( overseas=dict( genshin="https://sg-hk4e-api.hoyolab.com/event/ysledgeros/month_detail", hkrpg="https://sg-public-api.hoyolab.com/event/srledger/month_detail", ), chinese=dict( genshin="https://hk4e-api.mihoyo.com/event/ys_ledger/monthDetail", hkrpg="https://api-takumi.mihoyo.com/event/srledger/month_detail", ), ) INFO_LEDGER_URL = GameRoute( overseas=dict( genshin="https://sg-hk4e-api.hoyolab.com/event/ysledgeros/month_info", hkrpg="https://sg-public-api.hoyolab.com/event/srledger/month_info", ), chinese=dict( genshin="https://hk4e-api.mihoyo.com/event/ys_ledger/monthInfo", hkrpg="https://api-takumi.mihoyo.com/event/srledger/month_info", ), ) YSULOG_URL = InternationalRoute( overseas="https://hk4e-api-os.hoyoverse.com/common/hk4e_self_help_query/User/", chinese="https://hk4e-api.mihoyo.com/common/hk4e_self_help_query/User/", ) HK4E_URL = Route("https://sg-hk4e-api.hoyoverse.com/common/hk4e_global/") CODE_URL = Route("https://sg-hk4e-api.hoyoverse.com/common/apicdkey/api/webExchangeCdkey") QRCODE_URL = Route("https://hk4e-sdk.mihoyo.com/hk4e_cn/combo/panda/qrcode")