renamer.js 3.4 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113
  1. "use strict";
  2. exports.__esModule = true;
  3. var _classCallCheck2 = require("babel-runtime/helpers/classCallCheck");
  4. var _classCallCheck3 = _interopRequireDefault(_classCallCheck2);
  5. var _binding = require("../binding");
  6. var _binding2 = _interopRequireDefault(_binding);
  7. var _babelTypes = require("babel-types");
  8. var t = _interopRequireWildcard(_babelTypes);
  9. function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } }
  10. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  11. var renameVisitor = {
  12. ReferencedIdentifier: function ReferencedIdentifier(_ref, state) {
  13. var node = _ref.node;
  14. if (node.name === state.oldName) {
  15. node.name = state.newName;
  16. }
  17. },
  18. Scope: function Scope(path, state) {
  19. if (!path.scope.bindingIdentifierEquals(state.oldName, state.binding.identifier)) {
  20. path.skip();
  21. }
  22. },
  23. "AssignmentExpression|Declaration": function AssignmentExpressionDeclaration(path, state) {
  24. var ids = path.getOuterBindingIdentifiers();
  25. for (var name in ids) {
  26. if (name === state.oldName) ids[name].name = state.newName;
  27. }
  28. }
  29. };
  30. var Renamer = function () {
  31. function Renamer(binding, oldName, newName) {
  32. (0, _classCallCheck3.default)(this, Renamer);
  33. this.newName = newName;
  34. this.oldName = oldName;
  35. this.binding = binding;
  36. }
  37. Renamer.prototype.maybeConvertFromExportDeclaration = function maybeConvertFromExportDeclaration(parentDeclar) {
  38. var exportDeclar = parentDeclar.parentPath.isExportDeclaration() && parentDeclar.parentPath;
  39. if (!exportDeclar) return;
  40. var isDefault = exportDeclar.isExportDefaultDeclaration();
  41. if (isDefault && (parentDeclar.isFunctionDeclaration() || parentDeclar.isClassDeclaration()) && !parentDeclar.node.id) {
  42. parentDeclar.node.id = parentDeclar.scope.generateUidIdentifier("default");
  43. }
  44. var bindingIdentifiers = parentDeclar.getOuterBindingIdentifiers();
  45. var specifiers = [];
  46. for (var name in bindingIdentifiers) {
  47. var localName = name === this.oldName ? this.newName : name;
  48. var exportedName = isDefault ? "default" : name;
  49. specifiers.push(t.exportSpecifier(t.identifier(localName), t.identifier(exportedName)));
  50. }
  51. if (specifiers.length) {
  52. var aliasDeclar = t.exportNamedDeclaration(null, specifiers);
  53. if (parentDeclar.isFunctionDeclaration()) {
  54. aliasDeclar._blockHoist = 3;
  55. }
  56. exportDeclar.insertAfter(aliasDeclar);
  57. exportDeclar.replaceWith(parentDeclar.node);
  58. }
  59. };
  60. Renamer.prototype.rename = function rename(block) {
  61. var binding = this.binding,
  62. oldName = this.oldName,
  63. newName = this.newName;
  64. var scope = binding.scope,
  65. path = binding.path;
  66. var parentDeclar = path.find(function (path) {
  67. return path.isDeclaration() || path.isFunctionExpression();
  68. });
  69. if (parentDeclar) {
  70. this.maybeConvertFromExportDeclaration(parentDeclar);
  71. }
  72. scope.traverse(block || scope.block, renameVisitor, this);
  73. if (!block) {
  74. scope.removeOwnBinding(oldName);
  75. scope.bindings[newName] = binding;
  76. this.binding.identifier.name = newName;
  77. }
  78. if (binding.type === "hoisted") {}
  79. };
  80. return Renamer;
  81. }();
  82. exports.default = Renamer;
  83. module.exports = exports["default"];