Merge pull request #27 from rwjblue/typescript-support

Add typescript support to the default generated codemod.

Author: rondale-sc

commands/local/generate/codemod.js

@@ -20,15 +20,18 @@ module.exports.handler = function handler(options) {
   fs.outputFileSync(
     `${codemodDir}/index.js`,
     stripIndent`
+      const { getParser } = require('codemod-cli').jscodeshift;
+
       module.exports = function transformer(file, api) {
-        const j = api.jscodeshift;
+        const j = getParser(api);
 
         return j(file.source)
           .find(j.Identifier)
           .forEach(path => {
-            j(path).replaceWith(
-              j.identifier(path.node.name.split('').reverse().join(''))
-            );
+            path.node.name = path.node.name
+              .split('')
+              .reverse()
+              .join('');
           })
           .toSource();
       }

package.json

@@ -17,6 +17,7 @@
     "test": "qunit tests/**/*-test.js"
   },
   "dependencies": {
+    "@babel/parser": "^7.0.0-beta.51",
     "chalk": "^2.4.1",
     "common-tags": "^1.8.0",
     "execa": "^0.10.0",
@@ -26,6 +27,7 @@
     "import-local": "^1.0.0",
     "jscodeshift": "^0.5.1",
     "pkg-up": "^2.0.0",
+    "recast": "^0.15.0",
     "yargs": "^11.0.0"
   },
   "devDependencies": {

src/index.js

@@ -2,8 +2,10 @@
 
 const TestSupport = require('./test-support');
 const BinSupport = require('./bin-support');
+const TransformSupport = require('./transform-support');
 
 module.exports = {
   runTransformTest: TestSupport.runTransformTest,
   runTransform: BinSupport.runTransform,
+  jscodeshift: TransformSupport.jscodeshift,
 };

src/transform-support.js

@@ -0,0 +1,22 @@
+'use strict';
+
+function getJSCodeshiftParser(api) {
+  try {
+    let parser = require('recast/parsers/typescript');
+
+    return api.jscodeshift.withParser(parser);
+  } catch (e) {
+    // eslint-disable-next-line
+    console.log(
+      'could not load typescript aware parser, falling back to standard recast parser...'
+    );
+
+    return api.jscodeshift;
+  }
+}
+
+module.exports = {
+  jscodeshift: {
+    getParser: getJSCodeshiftParser,
+  },
+};

tests/cli-test.js

@@ -256,17 +256,19 @@ QUnit.module('codemod-cli', function(hooks) {
         wrap(function*(assert) {
           userProject.write({
             foo: {
-              'something.js': 'let blah = bar',
-              'other.js': 'let blah = bar',
+              'something.js': 'let blah = bar;',
+              'other.js': 'let blah = bar;',
+              'otherthing.ts': 'let blah: Map = bar;',
             },
           });
 
-          yield CodemodCLI.runTransform(codemodProject.path('bin'), 'main', 'foo/*thing.js');
+          yield CodemodCLI.runTransform(codemodProject.path('bin'), 'main', 'foo/*thing.[jt]s');
 
           assert.deepEqual(userProject.read(), {
             foo: {
-              'something.js': 'let halb = rab',
-              'other.js': 'let blah = bar',
+              'something.js': 'let halb = rab;',
+              'other.js': 'let blah = bar;',
+              'otherthing.ts': 'let halb: paM = rab;',
             },
           });
         })

yarn.lock

@@ -16,6 +16,10 @@
     esutils "^2.0.2"
     js-tokens "^3.0.0"
 
+"@babel/parser@^7.0.0-beta.51":
+  version "7.0.0-beta.51"
+  resolved "https://registry.yarnpkg.com/@babel/parser/-/parser-7.0.0-beta.51.tgz#27cec2df409df60af58270ed8f6aa55409ea86f6"
+
 "@mrmlnc/readdir-enhanced@^2.2.1":
   version "2.2.1"
   resolved "https://registry.yarnpkg.com/@mrmlnc/readdir-enhanced/-/readdir-enhanced-2.2.1.tgz#524af240d1a360527b730475ecfa1344aa540dde"