feat: proxy slot utils

kroma-network · Dec 29, 2023 · f5cfee1 · f5cfee1
1 parent 38f022c
commit f5cfee1
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/helper/eco-proxy.ts b/helper/eco-proxy.ts
@@ -40,11 +40,21 @@ export class ProxyFactory extends AsyncConstructor {
     const inst = await logic.attach(proxy) as EcoProxiedInstance<CT>;
 
     inst.ecoProxy = proxy;
-    inst.ecoProxyAdmin = this.proxyAdminFactory.attach(await ethers.provider.getStorage(proxy, this.ADMIN_SLOT)) as EcoProxyAdmin;
+    inst.ecoProxyAdmin = this.proxyAdminFactory.attach(await this.getAdminAddress(proxy)) as EcoProxyAdmin;
     inst.ecoLogic = logic;
 
     return inst;
   }
+
+  async getImplAddress<CT extends BaseContract>(logic:CT) {
+    const slotData = await ethers.provider.getStorage(logic, this.IMPLEMENTATION_SLOT)
+    return ethers.getAddress("0x" + slotData.slice(-40));
+  }
+
+  async getAdminAddress<CT extends BaseContract>(logic:CT) {
+    const slotData = await ethers.provider.getStorage(logic, this.ADMIN_SLOT)
+    return ethers.getAddress("0x" + slotData.slice(-40));
+  }
 }
 
 async function test() {