common.py

from typing import NewType, Optional, TypedDict

from pydantic import BaseModel, Field, validator
from web3 import Web3

CHAIN_NAMES_BY_ID = {
    "1": "ethereum",
    "10": "optimism",
    "100": "gnosis",
    "10000": "smartbch",
    "-1": "solana",
    "-2": "near",
    "1024": "clover",
    "11297108109": "palm",
    "122": "fuse",
    "128": "heco",
    "1284": "moonbeam",
    "1285": "moonriver",
    "1287": "moonbase",
    "1313161554": "aurora",
    "137": "polygon",
    "1666600000": "harmony",
    "1666700000": "harmony-testnet",
    "20": "elastos",
    "25": "cronos",
    "250": "ftm",
    "256": "heco-testnet",
    "288": "boba",
    "3": "ropsten",
    "321": "kcc",
    "361": "theta",
    "4": "rinkeby",
    "40": "telos",
    "4002": "ftmtest",
    "42": "kovan",
    "42161": "arbitrum",
    "42220": "celo",
    "43113": "fuji",
    "43114": "avax",
    "4689": "iotex",
    "592": "astar",
    "5": "goerli",
    "11155111": "sepolia",
    "56": "bsc",
    "1818": "cube",
    "65": "okex-testnet",
    "66": "okex",
    "70": "hoo",
    "80001": "mumbai",
    "82": "meter",
    "88": "tomochain",
    "97": "bsc-testnet",
    "9001": "evmos",
}

Address = NewType("Address", str)

ChainId = NewType("ChainId", int)


class Token(BaseModel):
    symbol: str
    name: str
    address: Address
    decimals: int = Field(..., alias="tokenDecimal")
    chainId: ChainId
    logoURI: Optional[str]
    coingeckoId: Optional[str]
    listedIn: list[str] = []

    class Config:
        allow_population_by_field_name = True

    def __init__(self, **data):
        super().__init__(
            logoURI=(
                data.pop("logoURI", None)
                or data.pop("logo", None)
                or data.pop("icon", None)
                or data.pop("image", None)
            ),
            **data,
        )

    # if logo.startswith('//'):
    # logo = 'ht

    @validator("address")
    def addr_checksum(cls, v: str):
        v = v.strip()
        if Web3.is_address(v):
            if "#" in v:
                v = v.split("#")[0]
            return Web3.to_checksum_address(v)
        return v


NATIVE_ADDR_0xe = "0xeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeeee"
NATIVE_ADDR_0x0 = Address("0x0000000000000000000000000000000000000000")

NATIVE_MATIC_ADDR = "0x0000000000000000000000000000000000001010"