feat: add jiti support

Author: folliehiyuki

index.js

@@ -215,6 +215,12 @@ var extensions = {
         hook(Object.assign({}, config, { extensions: '.jsx' }));
       },
     },
+    {
+      module: 'jiti/lib/jiti-register.mjs',
+      register: function () {
+        process.env.JITI_JSX = true
+      },
+    },
     'sucrase/register/jsx',
   ],
   '.litcoffee': 'coffeescript/register',
@@ -366,6 +372,7 @@ var extensions = {
   '.ts': [
     'ts-node/register',
     'sucrase/register/ts',
+    'jiti/lib/jiti-register.mjs',
     {
       module: '@babel/register',
       register: function (hook, config) {
@@ -425,10 +432,16 @@ var extensions = {
       },
     },
   ],
-  '.cts': ['ts-node/register'],
+  '.cts': ['ts-node/register', 'jiti/lib/jiti-register.mjs'],
   '.tsx': [
     'ts-node/register',
     'sucrase/register/tsx',
+    {
+      module: 'jiti/lib/jiti-register.mjs',
+      register: function () {
+        process.env.JITI_JSX = true
+      },
+    },
     {
       module: '@babel/register',
       register: function (hook, config) {

test/fixtures/cts/1/package.json

@@ -0,0 +1,5 @@
+{
+  "dependencies": {
+    "jiti": "^2.4.2"
+  }
+}

test/fixtures/cts/1/test.cts

@@ -0,0 +1,19 @@
+var test: {
+  data: {
+    trueKey: boolean;
+    falseKey: boolean;
+    subKey: {
+      subProp: number;
+    };
+  };
+} = {
+  data: {
+    trueKey: true,
+    falseKey: false,
+    subKey: {
+      subProp: 1,
+    },
+  },
+};
+
+export default test;

test/fixtures/cts/1/tsconfig.json

@@ -0,0 +1,11 @@
+{
+  "compilerOptions": {
+    "target": "es5",
+    "module": "commonjs",
+    "declaration": false,
+    "noImplicitAny": false,
+    "removeComments": true,
+    "sourceMap": true,
+    "outDir": ".tmp"
+  }
+}

test/fixtures/jsx/2/package.json

@@ -0,0 +1,5 @@
+{
+  "dependencies": {
+    "jiti": "^2.4.2"
+  }
+}

test/fixtures/jsx/2/test.jsx

@@ -0,0 +1,17 @@
+const React = {
+  createElement: function (Component) {
+    return Component();
+  },
+};
+
+// Test harmony arrow functions
+const Component = () => {
+  var trueKey = true;
+  var falseKey = false;
+  var subKey = { subProp: 1 };
+  // Test harmony object short notation
+  return { data: { trueKey, falseKey, subKey } };
+};
+
+// Test JSX syntax
+module.exports = <Component />;

test/fixtures/ts/5/package.json

@@ -0,0 +1,5 @@
+{
+  "dependencies": {
+    "jiti": "^2.4.2"
+  }
+}

test/fixtures/ts/5/test.ts

@@ -0,0 +1,15 @@
+var test = {
+  data: {
+    trueKey: true,
+    falseKey: false,
+    subKey: {
+      subProp: 1,
+    },
+  },
+};
+
+var main = {
+  default: test,
+};
+
+export = main;

test/fixtures/tsx/5/package.json

@@ -0,0 +1,5 @@
+{
+  "dependencies": {
+    "jiti": "^2.4.2"
+  }
+}

test/fixtures/tsx/5/test.tsx

@@ -0,0 +1,18 @@
+const React = {
+  createElement(Component: () => any) {
+    return Component();
+  },
+};
+
+// Test harmony arrow functions.
+const Component = () => {
+  var trueKey: boolean = true;
+  var falseKey: boolean = false;
+  var subKey = { subProp: 1 };
+
+  // Test harmony object short notation.
+  return { data: { trueKey, falseKey, subKey } };
+};
+
+// Test TSX syntax.
+export default <Component />;