index.js 3.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. Object.defineProperty(exports, "NodePath", {
  6. enumerable: true,
  7. get: function () {
  8. return _path.default;
  9. }
  10. });
  11. Object.defineProperty(exports, "Scope", {
  12. enumerable: true,
  13. get: function () {
  14. return _scope.default;
  15. }
  16. });
  17. Object.defineProperty(exports, "Hub", {
  18. enumerable: true,
  19. get: function () {
  20. return _hub.default;
  21. }
  22. });
  23. exports.visitors = exports.default = void 0;
  24. var _context = _interopRequireDefault(require("./context"));
  25. var visitors = _interopRequireWildcard(require("./visitors"));
  26. exports.visitors = visitors;
  27. var t = _interopRequireWildcard(require("@babel/types"));
  28. var cache = _interopRequireWildcard(require("./cache"));
  29. var _path = _interopRequireDefault(require("./path"));
  30. var _scope = _interopRequireDefault(require("./scope"));
  31. var _hub = _interopRequireDefault(require("./hub"));
  32. function _getRequireWildcardCache() { if (typeof WeakMap !== "function") return null; var cache = new WeakMap(); _getRequireWildcardCache = function () { return cache; }; return cache; }
  33. function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
  34. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  35. function traverse(parent, opts = {}, scope, state, parentPath) {
  36. if (!parent) return;
  37. if (!opts.noScope && !scope) {
  38. if (parent.type !== "Program" && parent.type !== "File") {
  39. throw new Error("You must pass a scope and parentPath unless traversing a Program/File. " + `Instead of that you tried to traverse a ${parent.type} node without ` + "passing scope and parentPath.");
  40. }
  41. }
  42. if (!t.VISITOR_KEYS[parent.type]) {
  43. return;
  44. }
  45. visitors.explode(opts);
  46. traverse.node(parent, opts, scope, state, parentPath);
  47. }
  48. var _default = traverse;
  49. exports.default = _default;
  50. traverse.visitors = visitors;
  51. traverse.verify = visitors.verify;
  52. traverse.explode = visitors.explode;
  53. traverse.cheap = function (node, enter) {
  54. return t.traverseFast(node, enter);
  55. };
  56. traverse.node = function (node, opts, scope, state, parentPath, skipKeys) {
  57. const keys = t.VISITOR_KEYS[node.type];
  58. if (!keys) return;
  59. const context = new _context.default(scope, opts, state, parentPath);
  60. for (const key of keys) {
  61. if (skipKeys && skipKeys[key]) continue;
  62. if (context.visit(node, key)) return;
  63. }
  64. };
  65. traverse.clearNode = function (node, opts) {
  66. t.removeProperties(node, opts);
  67. cache.path.delete(node);
  68. };
  69. traverse.removeProperties = function (tree, opts) {
  70. t.traverseFast(tree, traverse.clearNode, opts);
  71. return tree;
  72. };
  73. function hasDenylistedType(path, state) {
  74. if (path.node.type === state.type) {
  75. state.has = true;
  76. path.stop();
  77. }
  78. }
  79. traverse.hasType = function (tree, type, denylistTypes) {
  80. if (denylistTypes != null && denylistTypes.includes(tree.type)) return false;
  81. if (tree.type === type) return true;
  82. const state = {
  83. has: false,
  84. type: type
  85. };
  86. traverse(tree, {
  87. noScope: true,
  88. denylist: denylistTypes,
  89. enter: hasDenylistedType
  90. }, null, state);
  91. return state.has;
  92. };
  93. traverse.cache = cache;