feat: exports expressions

closes #7

Author: anion155

src/index.ts

@@ -13,6 +13,17 @@ const transformer = <T extends ts.Node>(_: ts.Program) => {
       resolve: paths[key][0] // TODO should check if is not empty
     }));
     let fileDir = "";
+    function findFileInPaths(text: string) {
+      for (const path of regPaths) {
+        const match = text.match(path.regexp);
+        if (match) {
+          const out = path.resolve.replace(/\*/g, match[1]);
+          const file = slash(relative(fileDir, resolve(baseUrl, out)));
+          return file;
+        }
+      }
+      return null;
+    }
     function visit(node: ts.Node): ts.Node {
       if (ts.isSourceFile(node)) {
         fileDir = dirname(node.fileName);
@@ -22,11 +33,8 @@ const transformer = <T extends ts.Node>(_: ts.Program) => {
         ts.isImportDeclaration(node) &&
         ts.isStringLiteral(node.moduleSpecifier)
       ) {
-        for (const path of regPaths) {
-          const match = node.moduleSpecifier.text.match(path.regexp);
-          if (match === null) continue;
-          const out = path.resolve.replace(/\*/g, match[1]);
-          const file = slash(relative(fileDir, resolve(baseUrl, out)));
+        const file = findFileInPaths(node.moduleSpecifier.text);
+        if (file) {
           return ts.updateImportDeclaration(
             node,
             node.decorators,
@@ -37,6 +45,23 @@ const transformer = <T extends ts.Node>(_: ts.Program) => {
           );
         }
       }
+      if (
+        ts.isExportDeclaration(node) &&
+        node.moduleSpecifier &&
+        ts.isStringLiteral(node.moduleSpecifier)
+      ) {
+        const file = findFileInPaths(node.moduleSpecifier.text);
+        if (file) {
+          return ts.updateExportDeclaration(
+            node,
+            node.decorators,
+            node.modifiers,
+            node.exportClause,
+            // If it's in the same level or below add the ./
+            ts.createLiteral(file[0] === "." ? file : `./${file}`)
+          );
+        }
+      }
       return ts.visitEachChild(node, visit, context);
     }
     return ts.visitNode(rootNode, visit);

tests/core/index.ts

@@ -1,5 +1,4 @@
-import { sum } from "@utils/sum";
-import { subs } from "@utils/subs";
+import { sum, subs } from '@utils';
 
 sum(2, 3);
 subs(2, 3);

tests/tsconfig.json

@@ -7,7 +7,8 @@
 
     "baseUrl": "./",
     "paths": {
-      "@utils/*": ["utils/*"]
+      "@utils/*": ["utils/*"],
+      "@utils": ["utils"]
     },
 
     "esModuleInterop": true,

tests/utils/index.ts

@@ -0,0 +1,2 @@
+export * from '@utils/sum';
+export * from '@utils/subs';