Add implementation to ProxyType (#2)

ethui · Feb 5, 2025 · f3ce623 · f3ce623
1 parent 87a270c
commit f3ce623
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/lib.rs b/src/lib.rs
@@ -21,6 +21,20 @@ pub enum ProxyType {
     Comptroller(Address),
 }
 
+impl ProxyType {
+    pub fn implementation(&self) -> Address {
+        match self {
+            ProxyType::Eip1167(addr)
+            | ProxyType::Eip1967Direct(addr)
+            | ProxyType::Eip1967Beacon(addr)
+            | ProxyType::OpenZeppelin(addr)
+            | ProxyType::Eip897(addr)
+            | ProxyType::Safe(addr)
+            | ProxyType::Comptroller(addr) => *addr,
+        }
+    }
+}
+
 pub async fn detect_proxy<N, P: Provider<N>>(
     address: Address,
     provider: &P,