Add typescript support to jscodeshift based transforms.

Author: Robert Jackson

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();
       }

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

@@ -213,17 +213,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;',
             },
           });
         })