fix(ts-transform-plugin): commonjs module fix #1

zerkalica · May 26, 2018 · f5aee9a · f5aee9a
1 parent 1aa3762
commit f5aee9a
Showing 1 changed file with 11 additions and 7 deletions.
diff --git a/packages/ts-transform-paths/src/index.ts b/packages/ts-transform-paths/src/index.ts
@@ -63,15 +63,19 @@ function importPathVisitor(
     fixNode.pos = cachedPos
     fixNode.end = cachedPos + newStr.length
 
-    // const newSpec = ts.createLiteral(newImport)
+    const newSpec = ts.createLiteral(newImport)
 
-    // if (ts.isImportDeclaration(node)) return ts.updateImportDeclaration(
-    //     node, undefined, undefined, undefined, newSpec
-    // )
+    if (ts.isImportDeclaration(node)) return ts.updateImportDeclaration(
+        node, undefined, undefined, undefined, newSpec
+    )
+
+    if (ts.isExportDeclaration(node)) return ts.updateExportDeclaration(
+        node, undefined, undefined, undefined, newSpec
+    )
 
-    // if (ts.isExportDeclaration(node)) return ts.updateExportDeclaration(
-    //     node, undefined, undefined, undefined, newSpec
-    // )
+    if (ts.isCallExpression(node)) return ts.updateCall(
+        node, node.expression, undefined, [newSpec]
+    )
 }
 
 function patchEmitFiles(host: any): ts.TransformerFactory<ts.SourceFile>[] {