feat: Add Arbitrum, Optimism and Polygon support

.gitignore

@@ -105,6 +105,7 @@ venv/
 ENV/
 env.bak/
 venv.bak/
+.idea/
 
 # mkdocs documentation
 /site

ape_infura/__init__.py

@@ -2,16 +2,32 @@
 
 from .providers import Infura
 
-NETWORKS = [
-    "mainnet",
-    "ropsten",
-    "rinkeby",
-    "kovan",
-    "goerli",
-]
+NETWORKS = {
+    "ethereum": [
+        "mainnet",
+        "ropsten",
+        "rinkeby",
+        "kovan",
+        "goerli",
+    ],
+    "arbitrum": [
+        "mainnet",
+        "tesnet",
+    ],
+    "optimism": [
+        "mainnet",
+        "kovan",
+        "goerli",
+    ],
+    "polygon": [
+        "mainnet",
+        "mumbai",
+    ],
+}
 
 
 @plugins.register(plugins.ProviderPlugin)
 def providers():
-    for network_name in NETWORKS:
-        yield "ethereum", network_name, Infura
+    for ecosystem_name in NETWORKS:
+        for network_name in NETWORKS[ecosystem_name]:
+            yield ecosystem_name, network_name, Infura

ape_infura/providers.py

@@ -1,5 +1,5 @@
 import os
-from typing import Dict
+from typing import Dict, Tuple
 
 from ape.api import UpstreamProvider, Web3Provider
 from ape.exceptions import ContractLogicError, ProviderError, VirtualMachineError
@@ -8,7 +8,24 @@
 from web3.gas_strategies.rpc import rpc_gas_price_strategy
 from web3.middleware import geth_poa_middleware
 
-_ENVIRONMENT_VARIABLE_NAMES = ("WEB3_INFURA_PROJECT_ID", "WEB3_INFURA_API_KEY")
+_ENVIRONMENT_VARIABLE_OPTIONS = {
+    "ethereum": (
+        "WEB3_INFURA_PROJECT_ID",
+        "WEB3_INFURA_API_KEY",
+    ),
+    "arbitrum": (
+        "WEB3_ARBITRUM_INFURA_PROJECT_ID",
+        "WEB3_ARBITRUM_INFURA_API_KEY",
+    ),
+    "optimism": (
+        "WEB3_OPTIMISM_INFURA_PROJECT_ID",
+        "WEB3_OPTIMISM_INFURA_API_KEY",
+    ),
+    "polygon": (
+        "WEB3_POLYGON_INFURA_PROJECT_ID",
+        "WEB3_POLYGON_INFURA_API_KEY",
+    ),
+}
 
 
 class InfuraProviderError(ProviderError):
@@ -18,32 +35,38 @@ class InfuraProviderError(ProviderError):
 
 
 class MissingProjectKeyError(InfuraProviderError):
-    def __init__(self):
-        env_var_str = ", ".join([f"${n}" for n in _ENVIRONMENT_VARIABLE_NAMES])
+    def __init__(self, options: Tuple[str, str]):
+        env_var_str = ", ".join([f"${n}" for n in options])
         super().__init__(f"Must set one of {env_var_str}")
 
 
 class Infura(Web3Provider, UpstreamProvider):
-    network_uris: Dict[str, str] = {}
+    network_uris: Dict[Tuple[str, str], str] = {}
 
     @property
     def uri(self) -> str:
+        ecosystem_name = self.network.ecosystem.name
         network_name = self.network.name
-        if network_name in self.network_uris:
-            return self.network_uris[network_name]
+        if (ecosystem_name, network_name) in self.network_uris:
+            return self.network_uris[(ecosystem_name, network_name)]
 
         key = None
-        for env_var_name in _ENVIRONMENT_VARIABLE_NAMES:
+        options = _ENVIRONMENT_VARIABLE_OPTIONS[ecosystem_name]
+        for env_var_name in options:
             env_var = os.environ.get(env_var_name)
             if env_var:
                 key = env_var
                 break
 
         if not key:
-            raise MissingProjectKeyError()
-
-        network_uri = f"https://{self.network.name}.infura.io/v3/{key}"
-        self.network_uris[network_name] = network_uri
+            raise MissingProjectKeyError(options)
+
+        prefix = f"{ecosystem_name}-" if ecosystem_name != "ethereum" else ""
+        network = self.network.name
+        if ecosystem_name == 'arbitrum' and self.network.name == 'testnet':
+            network = 'rinkeby'  # currently only "testnet is supported on arbitrum"
+        network_uri = f"https://{prefix}{self.network.name}.infura.io/v3/{key}"
+        self.network_uris[(ecosystem_name, network_name)] = network_uri
         return network_uri
 
     @property

tests/test_provider.py

@@ -1,3 +1,4 @@
+import os
 from ape import networks
 
 from ape_infura.providers import Infura
@@ -7,3 +8,24 @@ def test_provider_works():
     with networks.ethereum.mainnet.use_provider("infura") as provider:
         assert isinstance(provider, Infura)
         assert provider.get_balance("0x0000000000000000000000000000000000000000") > 0
+        assert provider.uri.startswith(f"https://mainnet.infura.io/v3/")
+    # tests for different chains, commentefd due to missing requirements
+    # with networks.arbitrum.mainnet.use_provider("infura") as provider:
+    #     assert isinstance(provider, Infura)
+    #     assert provider.get_balance("0x0000000000000000000000000000000000000000") > 0
+    #     assert provider.uri.startswith(f"https://arbitrum-mainnet.infura.io/v3/")
+    #
+    # with networks.arbitrum.testnet.use_provider("infura") as provider:
+    #     assert isinstance(provider, Infura)
+    #     assert provider.get_balance("0x0000000000000000000000000000000000000000") > 0
+    #     assert provider.uri.startswith(f"https://arbitrum-rinkeby.infura.io/v3/")
+    #
+    # with networks.optimism.mainnet.use_provider("infura") as provider:
+    #     assert isinstance(provider, Infura)
+    #     assert provider.get_balance("0x0000000000000000000000000000000000000000") > 0
+    #     assert provider.uri.startswith(f"https://optimism-mainnet.infura.io/v3/")
+    #
+    # with networks.polygon.mumbai.use_provider("infura") as provider:
+    #     assert isinstance(provider, Infura)
+    #     assert provider.get_balance("0x0000000000000000000000000000000000000000") > 0
+    #     assert provider.uri.startswith(f"https://polygon-mumbai.infura.io/v3/")