feat: replace lib similarly to what we do for node_modules

defi-wonderland · Nov 26, 2024 · 6993ff6 · 6993ff6
1 parent 632efe8
commit 6993ff6
Show file tree

Hide file tree

Showing 4 changed files with 34 additions and 3 deletions.
diff --git a/dist/index.js b/dist/index.js
diff --git a/dist/index.js.map b/dist/index.js.map
diff --git a/src/transformRemappings.ts b/src/transformRemappings.ts
@@ -37,7 +37,7 @@ export const transformRemappings = (file: string): string => {
         line = line.replace(remappingOrigin, remappingDestination);
       }
 
-      line = line.replace(/(['""]).*node_modules\//, `$1`);
+      line = line.replace(/(['""]).*(?:node_modules|lib)\//, `$1`);
 
       return line;
     })

diff --git a/test/unit/transformRemappings.spec.ts b/test/unit/transformRemappings.spec.ts
@@ -115,6 +115,37 @@ describe('transformRemappings', () => {
     });
   });
 
+  describe('with remapping for path inside lib/', function () {
+    before(() => {
+      mock({
+        'remappings.txt': `
+          solmate/=lib/solmate/src/`,
+      });
+    });
+
+    it('should apply remapping and then remove lib from path', () => {
+      const fileContent = `
+                                  import {FixedPointMathLib} from 'solmate/utils/FixedPointMathLib.sol';
+                                  import 'solmate/utils/FixedPointMathLib.sol';
+                                  import "solmate/utils/FixedPointMathLib.sol";
+                                  import {FixedPointMathLib as FPL} from 'solmate/utils/FixedPointMathLib.sol';
+                                  import * as FPL from "solmate/utils/FixedPointMathLib.sol";
+                                `;
+
+      const transformedContent = transformRemappings(fileContent);
+
+      expect(transformedContent).to.include(
+        `import {FixedPointMathLib} from 'solmate/src/utils/FixedPointMathLib.sol';`,
+      );
+      expect(transformedContent).to.include(`import 'solmate/src/utils/FixedPointMathLib.sol';`);
+      expect(transformedContent).to.include(`import "solmate/src/utils/FixedPointMathLib.sol";`);
+      expect(transformedContent).to.include(
+        `import {FixedPointMathLib as FPL} from 'solmate/src/utils/FixedPointMathLib.sol';`,
+      );
+      expect(transformedContent).to.include(`import * as FPL from "solmate/src/utils/FixedPointMathLib.sol";`);
+    });
+  });
+
   it('should leave unrelated imports alone', () => {
     const fileContent = `
       import './Contract.sol';