fix: allow overriding default import map

Author: NtTestAlert

src/imports.js

@@ -45,7 +45,17 @@ export const resolveImports = async code => {
   const addressMap = {}
   const importList = extractImports(fixShorthandImports(code))
   for (const key in importList) {
-    if (defaultsByName[key]) {
+    if (
+      key !== "FlowManager" &&
+      importList[key] &&
+      importList[key].toLowerCase().startsWith("0x") &&
+      importList[key].length === 18
+    ) {
+      addressMap[key] = importList[key]
+    } else if (
+      defaultsByName[key] &&
+      !(importList[key] && importList[key] === '"DYNAMIC"')
+    ) {
       addressMap[key] = defaultsByName[key]
     } else {
       const address = await getContractAddress(key)

src/index.js

@@ -23,6 +23,7 @@ export {
   getScriptCode,
   getContractCode,
   getTransactionCode,
+  defaultsByName,
 } from "./file"
 export {sendTransaction, executeScript} from "./interaction"
 export {getFlowBalance, mintFlow} from "./flow-token"

test/integration/imports.test.js

@@ -5,8 +5,9 @@ import {
   deployContract,
   resolveImports,
   getServiceAddress,
+  getAccountAddress,
 } from "../../src"
-import {defaultsByName} from "../../src/file"
+import {defaultsByName} from "../../src"
 import {DEFAULT_TEST_TIMEOUT} from "../util/timeout.const"
 import {fixShorthandImports} from "../../src/imports"
 
@@ -32,11 +33,13 @@ describe("import resolver", () => {
   })
 
   test("use imports", async () => {
+    const Dynamic = await getAccountAddress("Dynamic")
     await deployContract({code: emptyContract("First"), name: "First"})
     await deployContract({code: emptyContract("Second"), name: "Second"})
     await deployContract({code: emptyContract("Third"), name: "Third"})
     await deployContract({code: emptyContract("A"), name: "A"})
     await deployContract({code: emptyContract("B"), name: "B"})
+    await deployContract({code: emptyContract("Dynamo"), name: "Dynamo", to: Dynamic})
 
     const code = `
             import First from 0xFIRST
@@ -47,6 +50,10 @@ describe("import resolver", () => {
             import FungibleToken from 0xFUNGIBLETOKEN
             import FlowToken from 0xFLOWTOKEN
             
+            import Dynamo from "DYNAMIC"
+            import Direct from 0x0123456789012345
+            import FlowFees from 0x0123456789012345
+            
             access(all) fun main(){}
         `
 
@@ -64,5 +71,9 @@ describe("import resolver", () => {
     expect(B).toBe(Registry)
     expect(FungibleToken).toBe(defaultsByName.FungibleToken)
     expect(FlowToken).toBe(defaultsByName.FlowToken)
+    const {Dynamo, Direct, FlowFees} = addressMap
+    expect(Dynamo).toBe(Dynamic)
+    expect(Direct).toBe('0x0123456789012345')
+    expect(FlowFees).toBe('0x0123456789012345')
   })
 })