Go generate using chain type map

aptos.go

@@ -96,3 +96,10 @@ func AptosChainBySelector(selector uint64) (AptosChain, bool) {
 	chain, exist := aptosChainsBySelector[selector]
 	return chain, exist
 }
+
+func AptosNetworkTypeFromChainId(chainId uint64) (NetworkType, error) {
+	if chainDetails, exist := aptosSelectorsMap[chainId]; exist {
+		return chainDetails.NetworkType, nil
+	}
+	return "", fmt.Errorf("chain network type not found for chain %v", chainId)
+}

canton.go

@@ -104,3 +104,10 @@ func CantonChainBySelector(selector uint64) (CantonChain, bool) {
 	chain, exists := cantonChainsBySelector[selector]
 	return chain, exists
 }
+
+func CantonNetworkTypeFromChainId(chainId string) (NetworkType, error) {
+	if chainDetails, exist := cantonChainsByChainId[chainId]; exist {
+		return chainDetails.NetworkType, nil
+	}
+	return "", fmt.Errorf("chain network type not found for chain %v", chainId)
+}

evm.go

@@ -85,6 +85,13 @@ func EvmChainIdToChainSelector() map[uint64]uint64 {
 	return copyMap
 }
 
+func EvmNetworkTypeFromChainId(chainId uint64) (NetworkType, error) {
+	if chainDetails, exist := evmChainIdToChainSelector[chainId]; exist {
+		return chainDetails.NetworkType, nil
+	}
+	return "", fmt.Errorf("chain network type not found for chain %d", chainId)
+}
+
 // Deprecated, this only supports EVM chains, use the chain agnostic `GetChainIDFromSelector` instead
 func ChainIdFromSelector(chainSelectorId uint64) (uint64, error) {
 	for k, v := range evmChainIdToChainSelector {

genchains_aptos.go

@@ -86,7 +86,7 @@ func genChainsSourceCode() (string, error) {
 		if err != nil {
 			return "", err
 		}
-		networkType, err := chain_selectors.GetNetworkType(chainSel)
+		networkType, err := chain_selectors.AptosNetworkTypeFromChainId(ChainID)
 		if err != nil {
 			return "", err
 		}

genchains_canton.go

@@ -86,7 +86,7 @@ func genChainsSourceCode() (string, error) {
 		if err != nil {
 			return "", err
 		}
-		networkType, err := chain_selectors.GetNetworkType(chainSel)
+		networkType, err := chain_selectors.CantonNetworkTypeFromChainId(chainID)
 		if err != nil {
 			return "", err
 		}

genchains_evm.go

@@ -86,7 +86,8 @@ func genChainsSourceCode() (string, error) {
 		if err != nil {
 			return "", err
 		}
-		networkType, err := chain_selectors.GetNetworkType(chainSel)
+
+		networkType, err := chain_selectors.EvmNetworkTypeFromChainId(evmChainID)
 		if err != nil {
 			return "", err
 		}

genchains_solana.go

@@ -87,7 +87,7 @@ func genChainsSourceCode() (string, error) {
 		if err != nil {
 			return "", err
 		}
-		networkType, err := chain_selectors.GetNetworkType(chainSel)
+		networkType, err := chain_selectors.SolanaNetworkTypeFromChainId(ChainID)
 		if err != nil {
 			return "", err
 		}

genchains_starknet.go

@@ -86,7 +86,7 @@ func genChainsSourceCode() (string, error) {
 			return "", err
 		}
 
-		networkType, err := chain_selectors.GetNetworkType(chainSel)
+		networkType, err := chain_selectors.StarknetNetworkTypeFromChainId(ChainID)
 		if err != nil {
 			return "", err
 		}

genchains_sui.go

@@ -87,7 +87,7 @@ func genChainsSourceCode() (string, error) {
 			return "", err
 		}
 
-		networkType, err := chain_selectors.GetNetworkType(chainSel)
+		networkType, err := chain_selectors.SuiNetworkTypeFromChainId(ChainID)
 		if err != nil {
 			return "", err
 		}

genchains_ton.go

@@ -92,7 +92,7 @@ func genChainsSourceCode() (string, error) {
 			return "", err
 		}
 
-		networkType, err := chain_selectors.GetNetworkType(chainSel)
+		networkType, err := chain_selectors.TonNetworkTypeFromChainId(ChainID)
 		if err != nil {
 			return "", err
 		}

genchains_tron.go

@@ -92,7 +92,7 @@ func genChainsSourceCode() (string, error) {
 			return "", err
 		}
 
-		networkType, err := chain_selectors.GetNetworkType(chainSel)
+		networkType, err := chain_selectors.TronNetworkTypeFromChainId(ChainID)
 		if err != nil {
 			return "", err
 		}

solana.go

@@ -123,3 +123,10 @@ func SolanaChainBySelector(selector uint64) (SolanaChain, bool) {
 
 	return chain, exists
 }
+
+func SolanaNetworkTypeFromChainId(chainId string) (NetworkType, error) {
+	if chainDetails, exist := solanaChainIdToChainSelector[chainId]; exist {
+		return chainDetails.NetworkType, nil
+	}
+	return "", fmt.Errorf("chain network type not found for chain %v", chainId)
+}

starknet.go

@@ -89,3 +89,10 @@ func StarknetChainBySelector(selector uint64) (StarknetChain, bool) {
 
 	return chain, exists
 }
+
+func StarknetNetworkTypeFromChainId(chainId string) (NetworkType, error) {
+	if chainDetails, exist := starknetSelectorsMap[chainId]; exist {
+		return chainDetails.NetworkType, nil
+	}
+	return "", fmt.Errorf("chain network type not found for chain %v", chainId)
+}

sui.go

@@ -96,3 +96,10 @@ func SuiChainBySelector(selector uint64) (SuiChain, bool) {
 	chain, exist := suiChainsBySelector[selector]
 	return chain, exist
 }
+
+func SuiNetworkTypeFromChainId(chainId uint64) (NetworkType, error) {
+	if chainDetails, exist := suiSelectorsMap[chainId]; exist {
+		return chainDetails.NetworkType, nil
+	}
+	return "", fmt.Errorf("chain network type not found for chain %d", chainId)
+}

ton.go

@@ -76,3 +76,10 @@ func TonChainIdFromSelector(selector uint64) (int32, error) {
 
 	return chainId, nil
 }
+
+func TonNetworkTypeFromChainId(chainId int32) (NetworkType, error) {
+	if chainDetails, exist := tonSelectorsMap[chainId]; exist {
+		return chainDetails.NetworkType, nil
+	}
+	return "", fmt.Errorf("chain network type not found for chain %v", chainId)
+}

tron.go

@@ -76,3 +76,10 @@ func TronChainIdFromSelector(selector uint64) (uint64, error) {
 
 	return chainId, nil
 }
+
+func TronNetworkTypeFromChainId(chainId uint64) (NetworkType, error) {
+	if chainDetails, exist := tronSelectorsMap[chainId]; exist {
+		return chainDetails.NetworkType, nil
+	}
+	return "", fmt.Errorf("chain network type not found for chain %v", chainId)
+}