autoload: Fix onProgress label, add abiFromBytecode override

shazow · Oct 17, 2023 · d8bb770 · d8bb770
1 parent cb1d397
commit d8bb770
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/src/auto.ts b/src/auto.ts
@@ -34,6 +34,7 @@ export type AutoloadResult = {
 export type AutoloadConfig = {
     provider: AnyProvider;
 
+    abiFromBytecode?: (bytecode: string) => Promise<ABI>;
     abiLoader?: ABILoader|false;
     signatureLookup?: SignatureLookup|false;
 
@@ -118,8 +119,12 @@ export async function autoload(address: string, config: AutoloadConfig): Promise
     }
 
     // Load from code
-    onProgress("getCode", {address});
-    result.abi = abiFromBytecode(program);
+    onProgress("abiFromBytecode", {address});
+    if (config.abiFromBytecode) {
+        result.abi = await config.abiFromBytecode(bytecode);
+    } else {
+        result.abi = abiFromBytecode(program);
+    }
 
     if (!config.enableExperimentalMetadata) {
         result.abi = stripUnreliableABI(result.abi);