feat: proxy beacon

aura-nw · Aug 29, 2024 · 929bff5 · 929bff5
1 parent e4b7773
commit 929bff5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/src/services/evm/crawl_contract_evm.service.ts b/src/services/evm/crawl_contract_evm.service.ts
@@ -301,22 +301,22 @@ export default class CrawlSmartContractEVMService extends BullableService {
           this.contractHelper.detectProxyContractByByteCode(
             addr,
             byteCode,
-            EIPProxyContractSupportByteCode.EIP_1967_IMPLEMENTATION.SLOT
+            EIPProxyContractSupportByteCode.EIP_1967_IMPLEMENTATION.TYPE
           ),
           this.contractHelper.detectProxyContractByByteCode(
             addr,
             byteCode,
-            EIPProxyContractSupportByteCode.EIP_1822_IMPLEMENTATION.SLOT
+            EIPProxyContractSupportByteCode.EIP_1822_IMPLEMENTATION.TYPE
           ),
           this.contractHelper.detectProxyContractByByteCode(
             addr,
             byteCode,
-            EIPProxyContractSupportByteCode.OPEN_ZEPPELIN_IMPLEMENTATION.SLOT
+            EIPProxyContractSupportByteCode.OPEN_ZEPPELIN_IMPLEMENTATION.TYPE
           ),
           this.contractHelper.detectProxyContractByByteCode(
             addr,
             byteCode,
-            EIPProxyContractSupportByteCode.EIP_1167_IMPLEMENTATION.SLOT
+            EIPProxyContractSupportByteCode.EIP_1167_IMPLEMENTATION.TYPE
           ),
         ]).catch(() => Promise.resolve(null));
       })

diff --git a/src/services/evm/helpers/contract_helper.ts b/src/services/evm/helpers/contract_helper.ts
@@ -20,13 +20,17 @@ export class ContractHelper {
   public async detectProxyContractByByteCode(
     contractAddress: string,
     byteCode: string,
-    byteCodeSlot: string,
+    type: string,
     blockHeight?: number | string
   ): Promise<DetectEVMProxyContract> {
     const resultReturn: DetectEVMProxyContract = {
       logicContractAddress: '',
       EIP: '',
     };
+    const byteCodeSlot = Object.values(EIPProxyContractSupportByteCode).find(
+      (e) => e.TYPE === type
+    )?.SLOT;
+    if (!byteCodeSlot) throw Error('Type proxy not supported');
     const result = byteCode.includes(byteCodeSlot);
 
     if (!result) throw Error('Not proxy contract!');