From 7694b12cd7cb8615b488ea4b7b00adeeaba2d8cb Mon Sep 17 00:00:00 2001
From: derekpierre <derek.pierre@gmail.com>
Date: Wed, 11 Sep 2024 10:23:38 -0400
Subject: [PATCH] Update contract verification script to verify individual
 contracts post-deployment. The contracts can either be obtained from the
 common domain registry, or a customized registry path. Ensure that
 implementation contract for proxies are the contracts being verified.

---
 scripts/verify.py | 55 ++++++++++++++++++++++++++++++++++++++---------
 1 file changed, 45 insertions(+), 10 deletions(-)

diff --git a/scripts/verify.py b/scripts/verify.py
index 98e6a7db..67a35066 100644
--- a/scripts/verify.py
+++ b/scripts/verify.py
@@ -1,28 +1,63 @@
+from pathlib import Path
+
 import click
 from ape import networks
 from ape.cli import ConnectedProviderCommand, 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
+from deployment.utils import get_contract_container, registry_filepath_from_domain, verify_contracts
 
 
 @click.command(cls=ConnectedProviderCommand)
 @network_option(required=True)
+@click.option("--contract-name", "-c", help="Contract to verify", type=click.STRING, required=True)
 @click.option(
     "--domain",
     "-d",
-    help="TACo domain",
+    help="TACo domain; used for obtaining contract registry",
     type=click.Choice(SUPPORTED_TACO_DOMAINS),
-    required=True,
+    required=False,
+)
+@click.option(
+    "--registry-filepath",
+    "-f",
+    type=click.Path(dir_okay=False, exists=True, path_type=Path),
+    help="Registry filepath if the contract is not part of a common domain registry",
+    required=False,
 )
-def cli(network, domain):
-    """Verify deployed contracts from a registry file."""
-    registry_filepath = registry_filepath_from_domain(domain=domain)
-    contracts = contracts_from_registry(
-        registry_filepath, chain_id=networks.active_provider.chain_id
-    )
-    verify_contracts(list(contracts.values()))
+def cli(network, domain, contract_name, registry_filepath):
+    """Verify a deployed contract."""
+    if not (bool(registry_filepath) ^ bool(domain)):
+        raise click.BadOptionUsage(
+            option_name="--domain",
+            message=(
+                f"Provide either 'domain' or 'registry_filepath'; "
+                f"got {domain}, {registry_filepath}"
+            ),
+        )
+
+    registry_filepath = registry_filepath or registry_filepath_from_domain(domain=domain)
+    chain_id = networks.active_provider.chain_id
+    contracts = contracts_from_registry(registry_filepath, chain_id=chain_id)
+
+    try:
+        contract_instance = contracts[contract_name]
+    except KeyError:
+        raise ValueError(
+            f"Contract '{contract_name}' not found in registry, '{registry_filepath}', "
+            f"for chain {chain_id}"
+        )
+
+    # check whether contract is a proxy
+    proxy_info = networks.provider.network.ecosystem.get_proxy_info(contract_instance.address)
+    if proxy_info:
+        # we have an instance of a proxy contract, but need the underlying implementation
+        print(f"Proxy contract detected; verifying implementation contract at {proxy_info.target}")
+        contract_container = get_contract_container(contract_instance.contract_type.name)
+        contract_instance = contract_container.at(proxy_info.target)
+
+    verify_contracts([contract_instance])
 
 
 if __name__ == "__main__":