Support for Regex in src

README.md

@@ -45,8 +45,8 @@ Via `.babelrc` or babel-loader.
         dest: 'lib1'
     },
     {
-        src: 'test2',
-        dest: 'lib2'
+        src: /test\/(\w+)/,
+        dest: 'test/$1/src'
     }
 ]
 ```

lib/index.js

@@ -14,8 +14,16 @@ function replaceImportPathPlugin(path, t, opts) {
     }
 
     var node = path.node;
-    if (node && node.source.value === src) {
-        node.source.value = dest;
+    if (!node || !src) {
+        return;
+    }
+    if (typeof src === 'string') {
+        if (node.source.value === src) {
+            node.source.value = dest;
+        }
+    } else {
+        // src is a regex
+        node.source.value = node.source.value.replace(src, dest);
     }
 }
 

src/index.js

@@ -6,8 +6,15 @@ function replaceImportPathPlugin (path, t, opts) {
     }
 
     const node = path.node;
-    if (node && node.source.value === src) {
-        node.source.value = dest;
+    if (!node || !src) {
+        return;
+    }
+    if (typeof src === 'string') {
+        if (node.source.value === src) {
+            node.source.value = dest;
+        }
+    } else { // src is a regex
+        node.source.value = node.source.value.replace(src, dest);
     }
 }
 

test/index.js

@@ -24,4 +24,15 @@ describe('replace import path', () => {
         const expected = 'import A from "testaaaa";';
         expect(expected === result.code).to.be.true;
     });
+
+    it('replace with regex', () => {
+        const result = babel.transform('import A from "test/dir"', {
+            plugins: [[replaceImportPath, {
+                src: /test\/(\w+)/,
+                dest: 'test/$1/src'
+            }]]
+        });
+        const expected = 'import A from "test/dir/src";';
+        expect(result.code).to.be.eq(expected);
+    });
 });