Sorta get TS-based transforms running

Author: markerikson

.prettierignore

@@ -1,3 +1,4 @@
 **/dist/**
 **/etc/**
-**/temp/**
+**/temp/**
+**/__testfixtures__/**

package.json

@@ -50,6 +50,7 @@
     "react": "npm:18.1.0",
     "react-dom": "npm:18.1.0",
     "resolve": "1.22.1",
+    "ts-node": "10.4.0",
     "@types/react": "npm:18.0.12",
     "@types/react-dom": "npm:18.0.5",
     "@types/inquirer": "npm:8.2.1",

packages/rtk-codemods/v2.0/__testfixtures__/create-reducer-builder/basic.output.ts

@@ -1,3 +1,3 @@
-createReducer(initialState, builder => {
-  builder.addCase(todoAdded, (state, action) => {});
+createReducer(initialState, (builder) => {
+  builder.addCase(todoAdded, (state, action) => {})
 })

packages/rtk-codemods/v2.0/createReducerBuilder.ts

@@ -1,16 +1,26 @@
-// @ts-nocheck
+import { namedTypes } from 'ast-types'
+import { ExpressionKind } from 'ast-types/gen/kinds'
+import { JSCodeshift, Transform } from 'jscodeshift'
 
-function wrapInAddCaseExpression(j, member, arrowArguments) {
+function wrapInAddCaseExpression(
+  j: JSCodeshift,
+  member: namedTypes.Identifier,
+  arrowArguments: any[]
+) {
   return j.callExpression(
     j.memberExpression(member, j.identifier('addCase'), false),
     arrowArguments
   )
 }
 
-export function reducerPropsToBuilderExpression(j, defNode) {
+export function reducerPropsToBuilderExpression(
+  j: JSCodeshift,
+  defNode: namedTypes.SpreadElement | ExpressionKind
+) {
+  // @ts-ignore
   const [firstCase, ...restOfCases] = defNode.properties
 
-  const expressionStatement = restOfCases.reduce((acc, c) => {
+  const expressionStatement = restOfCases.reduce((acc: any, c: any) => {
     return wrapInAddCaseExpression(j, acc, [c.key, c.value])
   }, wrapInAddCaseExpression(j, j.identifier('builder'), [firstCase.key, firstCase.value]))
 
@@ -20,21 +30,29 @@ export function reducerPropsToBuilderExpression(j, defNode) {
   )
 }
 
-export default function transformer(file, api) {
+const transform: Transform = (file, api) => {
   const j = api.jscodeshift
 
-  return j(file.source)
-    .find(j.CallExpression, {
-      callee: { name: 'createReducer' },
-      arguments: { 1: { type: 'ObjectExpression' } },
-    })
-    .forEach((path) => {
-      j(path).replaceWith(
-        j.callExpression(j.identifier('createReducer'), [
-          path.node.arguments[0],
-          reducerPropsToBuilderExpression(j, path.node.arguments[1]),
-        ])
-      )
-    })
-    .toSource()
+  return (
+    j(file.source, {})
+      // @ts-ignore some expression mismatch
+      .find(j.CallExpression, {
+        callee: { name: 'createReducer' },
+        // @ts-ignore some expression mismatch
+        arguments: { 1: { type: 'ObjectExpression' } },
+      })
+      .forEach((path) => {
+        j(path).replaceWith(
+          j.callExpression(j.identifier('createReducer'), [
+            path.node.arguments[0],
+            reducerPropsToBuilderExpression(j, path.node.arguments[1]),
+          ])
+        )
+      })
+      .toSource({
+        arrowParensAlways: true,
+      })
+  )
 }
+
+export default transform

packages/rtk-codemods2/.babelrc

@@ -0,0 +1,3 @@
+{
+  "presets": ["@babel/preset-typescript"]
+}

packages/rtk-codemods2/bin/cli.js

@@ -1,5 +1,6 @@
 #!/usr/bin/env node
-'use strict';
+
+require('ts-node').register();
 
 require('codemod-cli').runTransform(
   __dirname,

packages/rtk-codemods2/jest.config.js

@@ -0,0 +1,6 @@
+module.exports = {
+  testEnvironment: 'node',
+  automock: false,
+  // roots: ['v2.0/__tests__'],
+  transform: { '\\.ts$': ['ts-jest'] },
+};

packages/rtk-codemods2/package.json

@@ -16,18 +16,18 @@
     "codemod-cli": "^3.2.0"
   },
   "devDependencies": {
+    "@types/jest": "^27",
+    "@types/jscodeshift": "^0.11.5",
     "coveralls": "^3.1.0",
     "eslint": "^7.25.0",
     "eslint-config-prettier": "^8.3.0",
     "eslint-plugin-node": "^11.1.0",
     "eslint-plugin-prettier": "^3.4.0",
     "jest": "^27",
-    "prettier": "^2.2.1"
+    "prettier": "^2.2.1",
+    "ts-jest": "^27"
   },
   "engines": {
     "node": ">= 14"
-  },
-  "jest": {
-    "testEnvironment": "node"
   }
 }

packages/rtk-codemods2/transforms/createReducerBuilder/__testfixtures__/basic-ts.input.ts

@@ -0,0 +1,11 @@
+createReducer(initialState, {
+  [todoAdded]: (state: SliceState, action: PayloadAction<string>) => {
+    // stuff
+  },
+});
+
+createReducer(initialState, {
+  [todoAdded](state: SliceState, action: PayloadAction<string>) {
+    // stuff
+  },
+});

packages/rtk-codemods2/transforms/createReducerBuilder/__testfixtures__/basic-ts.output.ts

@@ -0,0 +1,11 @@
+createReducer(initialState, (builder) => {
+  builder.addCase(todoAdded, (state: SliceState, action: PayloadAction<string>) => {
+    // stuff
+  });
+});
+
+createReducer(initialState, (builder) => {
+  builder.addCase(todoAdded, function(state: SliceState, action: PayloadAction<string>) {
+    // stuff
+  });
+});