chore: chain refactoring

Author: onyb

app/api/common/annotations.py

@@ -1,4 +1,6 @@
-COIN_TYPE_DESCRIPTION = "SLIP-44 coin type"
-CHAIN_ID_DESCRIPTION = "0x-prefixed unique hex identifier for the blockchain network (e.g. '0x1' for Ethereum, '0x2105' for Base)"
+COIN_TYPE_DESCRIPTION = "SLIP-44 coin type (e.g. 'ETH', 'BTC', 'SOL', etc.)"
+CHAIN_ID_DESCRIPTION = "Unique identifier for the blockchain network (e.g. '0x1' for Ethereum, '0x65' for Solana, 'bitcoin_mainnet' for Bitcoin, etc.)"
 ADDRESS_DESCRIPTION = "0x-prefixed contract address of the token in case of EVM chains, base58 encoded address for Solana"
-VS_CURRENCY_DESCRIPTION = "The target currency to get the price in (e.g. 'USD', 'EUR')"
+VS_CURRENCY_DESCRIPTION = (
+    "The target fiat currency to get the price in (e.g. 'USD', 'EUR')"
+)

app/api/common/models.py

@@ -21,24 +21,124 @@ class CoinType(str, Enum):
     ZEC = "ZEC"
 
 
-class ChainId(str, Enum):
-    ARBITRUM = "0xa4b1"
-    AVALANCHE = "0xa86a"
-    BASE = "0x2105"
-    BNB_CHAIN = "0x38"
-    ETHEREUM = "0x1"
-    OPTIMISM = "0xa"
-    POLYGON = "0x89"
-    SOLANA = "0x65"
-
-
-ChainIdCoinTypeMap = {
-    ChainId.ARBITRUM: CoinType.ETH,
-    ChainId.AVALANCHE: CoinType.ETH,
-    ChainId.BASE: CoinType.ETH,
-    ChainId.BNB_CHAIN: CoinType.ETH,
-    ChainId.ETHEREUM: CoinType.ETH,
-    ChainId.OPTIMISM: CoinType.ETH,
-    ChainId.POLYGON: CoinType.ETH,
-    ChainId.SOLANA: CoinType.SOL,
-}
+class _c(BaseModel):
+    coin: CoinType
+    chain_id: str
+    simplehash_id: str
+    alchemy_id: str
+    has_nft_support: bool
+
+
+class Chain(Enum):
+    # EVM chains
+    ETHEREUM = _c(
+        coin=CoinType.ETH,
+        chain_id="0x1",
+        simplehash_id="ethereum",
+        alchemy_id="eth-mainnet",
+        has_nft_support=True,
+    )
+    ARBITRUM = _c(
+        coin=CoinType.ETH,
+        chain_id="0xa4b1",
+        simplehash_id="arbitrum",
+        alchemy_id="arb-mainnet",
+        has_nft_support=True,
+    )
+    AVALANCHE = _c(
+        coin=CoinType.ETH,
+        chain_id="0xa86a",
+        simplehash_id="avalanche",
+        alchemy_id="avax-mainnet",
+        has_nft_support=True,
+    )
+    BASE = _c(
+        coin=CoinType.ETH,
+        chain_id="0x2105",
+        simplehash_id="base",
+        alchemy_id="base-mainnet",
+        has_nft_support=True,
+    )
+    BNB_CHAIN = _c(
+        coin=CoinType.ETH,
+        chain_id="0x38",
+        simplehash_id="bsc",
+        alchemy_id="bnb-mainnet",
+        has_nft_support=False,
+    )
+    OPTIMISM = _c(
+        coin=CoinType.ETH,
+        chain_id="0xa",
+        simplehash_id="optimism",
+        alchemy_id="opt-mainnet",
+        has_nft_support=True,
+    )
+    POLYGON = _c(
+        coin=CoinType.ETH,
+        chain_id="0x89",
+        simplehash_id="polygon",
+        alchemy_id="polygon-mainnet",
+        has_nft_support=True,
+    )
+
+    # Non-EVM chains
+    BITCOIN = _c(
+        coin=CoinType.BTC,
+        chain_id="bitcoin_mainnet",
+        simplehash_id="bitcoin",
+        alchemy_id="bitcoin-mainnet",
+        has_nft_support=False,
+    )
+    SOLANA = _c(
+        coin=CoinType.SOL,
+        chain_id="0x65",
+        simplehash_id="solana",
+        alchemy_id="sol-mainnet",
+        has_nft_support=True,
+    )
+    FILECOIN = _c(
+        coin=CoinType.FIL,
+        chain_id="f",
+        simplehash_id="filecoin",
+        alchemy_id="filecoin-mainnet",
+        has_nft_support=False,
+    )
+    CARDANO = _c(
+        coin=CoinType.ADA,
+        chain_id="cardano_mainnet",
+        simplehash_id="cardano",
+        alchemy_id="cardano-mainnet",
+        has_nft_support=False,
+    )
+    ZCASH = _c(
+        coin=CoinType.ZEC,
+        chain_id="zcash_mainnet",
+        simplehash_id="zcash",
+        alchemy_id="zcash-mainnet",
+        has_nft_support=False,
+    )
+
+    def __getattr__(self, name):
+        """Delegate attribute access to the chain info"""
+        # Only delegate specific attributes to avoid recursion
+        if name in [
+            "coin",
+            "chain_id",
+            "simplehash_id",
+            "alchemy_id",
+            "has_nft_support",
+        ]:
+            return getattr(self.value, name)
+
+        return super().__getattr__(name)
+
+    @staticmethod
+    def get(coin: str, chain_id: str):
+        for chain in Chain:
+            if chain.coin.value == coin.upper() and chain.chain_id == chain_id.lower():
+                return chain
+
+        return None
+
+    def __repr__(self):
+        return f"<{self.__class__.__name__}.{self.name}: coin={self.coin.value} chain_id={self.chain_id}>"

app/api/nft/models.py

@@ -90,29 +90,6 @@ class AlchemyNFTResponse(BaseModel):
     model_config = ConfigDict(alias_generator=to_camel)
 
 
-class AlchemyChain(str, Enum):
-    ETHEREUM = "eth-mainnet"
-    POLYGON = "polygon-mainnet"
-    BASE = "base-mainnet"
-    OPTIMISM = "opt-mainnet"
-    ARBITRUM = "arb-mainnet"
-    SOLANA = "sol-mainnet"
-    BSC = "bnb-mainnet"
-    AVALANCHE = "avax-mainnet"
-
-
-# SimpleHash Models
-class SimpleHashChain(str, Enum):
-    ETHEREUM = "ethereum"
-    BSC = "bsc"
-    AVALANCHE = "avalanche"
-    POLYGON = "polygon"
-    BASE = "base"
-    OPTIMISM = "optimism"
-    ARBITRUM = "arbitrum"
-    SOLANA = "solana"
-
-
 class SimpleHashTokenType(str, Enum):
     ERC721 = "ERC721"
     ERC1155 = "ERC1155"
@@ -139,7 +116,7 @@ class SimpleHashExtraMetadata(BaseModel):
 
 
 class SimpleHashNFT(BaseModel):
-    chain: SimpleHashChain
+    chain: str
     contract_address: str
     token_id: str | None = None
     name: str | None = None