Implements export statement rewriting

graphql · Dec 6, 2019 · 316a69e · 316a69e
1 parent d27e528
commit 316a69e
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/.babelrc.js b/.babelrc.js
@@ -22,6 +22,7 @@ module.exports = {
           return {
             visitor: {
               ImportDeclaration: function(path, state) {
+                if (!path.node.source) return;
                 var source = path.node.source.value;
                 if (source.match(/^\.{0,2}\//) && !source.endsWith('.es.js')) {
                   path.replaceWith(
@@ -32,6 +33,19 @@ module.exports = {
                   );
                 }
               },
+              ExportNamedDeclaration: function(path, state) {
+                if (!path.node.source) return;
+                const source = path.node.source.value;
+                if (source.match(/^\.{0,2}\//) && !source.endsWith('.es.js')) {
+                  path.replaceWith(
+                    types.exportNamedDeclaration(
+                      path.node.declaration,
+                      path.node.specifiers,
+                      types.stringLiteral(source + '.es.js'),
+                    ),
+                  );
+                }
+              },
             },
           };
         },