handle import export with babel-transform plugin

Author: ShMcK

lib/compiler.js

@@ -0,0 +1,12 @@
+"use strict";
+var ts = require('typescript');
+function compiler(str) {
+    return ts.transpile(str, {
+        module: ts.ModuleKind.CommonJS,
+        target: 'es5',
+        allowJs: true,
+        allowUnreachableCode: true,
+    });
+}
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.default = compiler;

lib/import-paths.js

@@ -2,15 +2,15 @@
 var path_1 = require('path');
 var importPathRegex = /require\(["'](.+)["']\);?$|^import.+from\s?["'](.+)["'];?$/;
 var relativePathRegex = /^\./;
-function fixImportPaths(str) {
+function fixImportPaths(dir, str) {
     return str.split('\n').map(function (line) {
         var isMatch = line.match(importPathRegex);
         if (!isMatch) {
             return line;
         }
         var importPath = isMatch[1] || isMatch[2];
         if (importPath.match(relativePathRegex)) {
-            return line.replace(importPath, path_1.join(process.env.DIR, importPath));
+            return line.replace(importPath, path_1.join(dir, importPath));
         }
         return line;
     }).join('\n');

lib/runner-process.js

@@ -45,6 +45,7 @@ function runnerProcess(config) {
         '--harmony',
         '--no-colors',
         '--timeout=3000',
+        '--require babelhook',
         ("--reporter=" + path_1.join(__dirname, 'reporter.js')),
         constants_1.testPath
     ], options);

lib/write-test.js

@@ -3,17 +3,16 @@ var fs_1 = require('fs');
 var process_console_log_1 = require('process-console-log');
 var exists_1 = require('./exists');
 var constants_1 = require('./constants');
-var ts = require('typescript');
 var import_paths_1 = require('./import-paths');
+var compiler_1 = require('./compiler');
 function writeTest(config, testString) {
     return new Promise(function (resolve, reject) {
-        var testStringWithFixedImportPaths = import_paths_1.default(testString);
+        var testStringWithFixedImportPaths = import_paths_1.default(config.dir, testString);
         var output = ''
             .concat(process_console_log_1.logger)
             .concat(exists_1.default(config.dir))
-            .concat(ts.transpile(testStringWithFixedImportPaths, {
-            module: ts.ModuleKind.CommonJS
-        }));
+            .concat("require(\"babel-register\")({\"plugins\": [[\"transform-es2015-modules-commonjs\", {\"strict\": true,\"loose\": true}]]});")
+            .concat(compiler_1.default(testStringWithFixedImportPaths));
         fs_1.writeFile(constants_1.testPath, output, function (err) {
             if (err) {
                 reject(err);

package.json

@@ -2,30 +2,32 @@
   "name": "mocha-coderoad",
   "version": "0.9.3",
   "description": "mocha test runner & reporter for atom-coderoad",
-  "main": "lib/index.js",
-  "scripts": {
-    "test": "echo \"Error: no test specified\" && exit 1"
-  },
-  "repository": {
-    "type": "git",
-    "url": "git+https://github.com/coderoad/mocha-coderoad.git"
-  },
-  "files": [
-    "lib",
-    "package.json",
-    "*.md"
-  ],
   "keywords": [
     "coderoad",
     "mocha"
   ],
-  "author": "Shawn McKay <shawn.j.mckay@gmail.com>",
-  "license": "ISC",
+  "homepage": "https://github.com/coderoad/mocha-coderoad#readme",
   "bugs": {
     "url": "https://github.com/coderoad/mocha-coderoad/issues"
   },
-  "homepage": "https://github.com/coderoad/mocha-coderoad#readme",
+  "license": "ISC",
+  "author": "Shawn McKay <shawn.j.mckay@gmail.com>",
+  "files": [
+    "lib",
+    "package.json",
+    "*.md"
+  ],
+  "main": "lib/index.js",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/coderoad/mocha-coderoad.git"
+  },
+  "scripts": {
+    "test": "echo \"Error: no test specified\" && exit 1"
+  },
   "dependencies": {
+    "babel-plugin-transform-es2015-modules-commonjs": "^6.11.5",
+    "babel-register": "^6.11.6",
     "mocha": "^3.0.1",
     "node-file-exists": "^1.1.0",
     "process-console-log": "^0.2.2",

src/compiler.ts

@@ -0,0 +1,11 @@
+import * as ts from 'typescript';
+
+// run code through TS compiler with options
+export default function compiler(str: string): string {
+  return ts.transpile(str, {
+    module: ts.ModuleKind.CommonJS,
+    target: 'es5',
+    allowJs: true,
+    allowUnreachableCode: true,
+  });
+}

src/import-paths.ts

@@ -9,7 +9,7 @@ import { join } from 'path';
 const importPathRegex = /require\(["'](.+)["']\);?$|^import.+from\s?["'](.+)["'];?$/;
 const relativePathRegex = /^\./;
 
-export default function fixImportPaths(str: string): string {
+export default function fixImportPaths(dir: string, str: string): string {
   return str.split('\n').map(line => {
     const isMatch = line.match(importPathRegex);
     if (!isMatch) {
@@ -21,7 +21,7 @@ export default function fixImportPaths(str: string): string {
     // // relative path
     if (importPath.match(relativePathRegex)) {
       // process.env.DIR
-      return line.replace(importPath, join(process.env.DIR, importPath));
+      return line.replace(importPath, join(dir, importPath));
     }
 		// no match, return line
     return line;

src/runner-process.ts

@@ -53,6 +53,7 @@ export default function runnerProcess(config: CR.Config) {
     '--harmony',
     '--no-colors',
     '--timeout=3000',
+    '--require babelhook',
     `--reporter=${join(__dirname, 'reporter.js')}`,
     testPath
 

src/write-test.ts

@@ -2,25 +2,24 @@ import {writeFile} from 'fs';
 import {logger} from 'process-console-log';
 import exists from './exists';
 import {testPath} from './constants';
-import * as ts from 'typescript';
 import importPaths from './import-paths';
+import compiler from './compiler';
 
 export default function writeTest(config, testString: string) {
   return new Promise((resolve, reject) => {
     // fix import paths relative to project dir instead of test runner
-    const testStringWithFixedImportPaths = importPaths(testString);
+    const testStringWithFixedImportPaths = importPaths(config.dir, testString);
 
     const output = ''
         // append logger
         .concat(logger)
         // exists polyfill for file/folder exists checks
         .concat(exists(config.dir))
+        // babel hook to handle import / export in other files
+        .concat(`require("babel-register")({"plugins": [["transform-es2015-modules-commonjs", {"strict": true,"loose": true}]]});`)
         // compile using ts
-        .concat(
-          ts.transpile(testStringWithFixedImportPaths, {
-            module: ts.ModuleKind.CommonJS
-          })
-        );
+        .concat(compiler(testStringWithFixedImportPaths));
+
 		// write test file
     writeFile(testPath, output, (err) => {
       if (err) { reject(err); }

tsconfig.json

@@ -15,6 +15,7 @@
         "src/typings/**/*.d.ts"
     ],
     "files": [
+        "src/compiler.ts",
         "src/constants.ts",
         "src/exists.ts",
         "src/import-paths.ts",