Restrict value choices for taco domain cli parameter, and fix initiat…

…e_ritual bug to adjust providers list based on provided domain.

deployment/constants.py

@@ -1,8 +1,7 @@
 from pathlib import Path
 
-from ape import project
-
 import deployment
+from ape import project
 
 DEPLOYMENT_DIR = Path(deployment.__file__).parent
 CONSTRUCTOR_PARAMS_DIR = DEPLOYMENT_DIR / "constructor_params"
@@ -15,6 +14,16 @@
 PROXY_NAME = "TransparentUpgradeableProxy"
 OZ_DEPENDENCY = project.dependencies["openzeppelin"]["4.9.1"]
 
+#
+# Domains
+#
+LYNX = "lynx"
+TAPIR = "tapir"
+MAINNET = "mainnet"
+
+SUPPORTED_TACO_DOMAINS = [LYNX, TAPIR, MAINNET]
+
+
 LYNX_NODES = {
     # staking provider -> operator
     "0xb15d5a4e2be34f4be154a1b08a94ab920ffd8a41": "0x890069745E9497C6f99Db68C4588deC5669F3d3E",

scripts/authorize_enrico.py

@@ -3,6 +3,7 @@
 import click
 from ape import project
 from ape.cli import NetworkBoundCommand, account_option, network_option
+from deployment.constants import SUPPORTED_TACO_DOMAINS
 from deployment.params import Transactor
 from deployment.registry import contracts_from_registry
 from deployment.utils import check_plugins, registry_filepath_from_domain
@@ -16,7 +17,7 @@
     "--domain",
     "-d",
     help="TACo domain",
-    type=click.STRING,
+    type=click.Choice(SUPPORTED_TACO_DOMAINS),
     required=True,
 )
 @click.option(

scripts/grant_initiator_role.py

@@ -3,6 +3,7 @@
 import click
 from ape import networks, project
 from ape.cli import NetworkBoundCommand, account_option, network_option
+from deployment.constants import SUPPORTED_TACO_DOMAINS
 from deployment.params import Transactor
 from deployment.registry import contracts_from_registry
 from deployment.utils import check_plugins, registry_filepath_from_domain
@@ -15,7 +16,7 @@
     "--domain",
     "-d",
     help="TACo domain",
-    type=click.STRING,
+    type=click.Choice(SUPPORTED_TACO_DOMAINS),
     required=True,
 )
 def cli(network, account, domain):

scripts/initiate_ritual.py

@@ -3,7 +3,7 @@
 import click
 from ape import project
 from ape.cli import NetworkBoundCommand, account_option, network_option
-from deployment.constants import LYNX_NODES
+from deployment.constants import LYNX, LYNX_NODES, SUPPORTED_TACO_DOMAINS, TAPIR, TAPIR_NODES
 from deployment.params import Transactor
 from deployment.registry import contracts_from_registry
 from deployment.utils import check_plugins, registry_filepath_from_domain
@@ -16,7 +16,7 @@
     "--domain",
     "-d",
     help="TACo domain",
-    type=click.STRING,
+    type=click.Choice(SUPPORTED_TACO_DOMAINS),
     required=True,
 )
 @click.option(
@@ -34,6 +34,14 @@ def cli(domain, duration, network, account):
     print(f"Using account: {account}")
     transactor = Transactor(account=account)
 
+    if domain == LYNX:
+        providers = list(sorted(LYNX_NODES.keys()))
+    elif domain == TAPIR:
+        providers = list(sorted(TAPIR_NODES.keys()))
+    else:
+        # mainnet sampling not currently supported
+        raise ValueError(f"Sampling of providers not supported for domain '{domain}'")
+
     registry_filepath = registry_filepath_from_domain(domain=domain)
 
     chain_id = project.chain_manager.chain_id
@@ -42,7 +50,6 @@ def cli(domain, duration, network, account):
 
     global_allow_list = deployments[project.GlobalAllowList.contract_type.name]
     authority = transactor.get_account().address
-    providers = list(sorted(LYNX_NODES.keys()))
 
     while True:
         transactor.transact(

scripts/verify.py

@@ -1,6 +1,7 @@
 import click
 from ape import networks
 from ape.cli import NetworkBoundCommand, network_option
+from deployment.constants import SUPPORTED_TACO_DOMAINS
 from deployment.registry import contracts_from_registry
 from deployment.utils import registry_filepath_from_domain, verify_contracts
 
@@ -11,7 +12,7 @@
     "--domain",
     "-d",
     help="TACo domain",
-    type=click.STRING,
+    type=click.Choice(SUPPORTED_TACO_DOMAINS),
     required=True,
 )
 def cli(network, domain):