jscodeshift-rename-property.js

'use strict';

const jscodeshift = require('jscodeshift');
const { MemberExpression, Property } = jscodeshift;

jscodeshift.registerMethods({
  // exports.a => exports.b
  //
  // jscodeshift(source)
  //   .find(jscodeshift.Identifier, { name: 'a' })
  //   .renamePropertyTo('b');
  renamePropertyTo(name) {
    return this.replaceWith(nodePath => {
      const { node, parentPath } = nodePath;
      const parentNode = parentPath.value;

      // exports.a
      if (MemberExpression.check(parentPath.value) && parentNode.property.name === node.name) {
        updateNode(node, name);
      }

      // { a: 1 }
      if (Property.check(parentPath.value) && parentNode.key.value === node.value) {
        updateNode(node, name);
      }

      return node;
    });
  },
});

function updateNode(node, name) {
  switch (node.type) {
    case 'Identifier':
      node.name = name;
      break;
    case 'Literal':
      node.value = name;
      break;
    default:
  }
}