autoload: Fix onProgress label, add abiFromBytecode override

shazow · Oct 17, 2023 · 22dd257 · 22dd257
1 parent cb1d397
commit 22dd257
Showing 1 changed file with 11 additions and 5 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,11 +119,16 @@ export async function autoload(address: string, config: AutoloadConfig): Promise
     }
 
     // Load from code
-    onProgress("getCode", {address});
-    result.abi = abiFromBytecode(program);
-
-    if (!config.enableExperimentalMetadata) {
-        result.abi = stripUnreliableABI(result.abi);
+    onProgress("abiFromBytecode", {address});
+    if (config.abiFromBytecode) {
+        result.abi = await config.abiFromBytecode(bytecode);
+    } else {
+        result.abi = abiFromBytecode(program);
+
+        // We only strip if our default abiFromBytecode function is used.
+        if (!config.enableExperimentalMetadata) {
+            result.abi = stripUnreliableABI(result.abi);
+        }
     }
 
     let signatureLookup = config.signatureLookup;