forked from zhurui/management
130 lines
3.4 KiB
JavaScript
130 lines
3.4 KiB
JavaScript
|
"use strict";
|
||
|
|
||
|
Object.defineProperty(exports, "__esModule", {
|
||
|
value: true
|
||
|
});
|
||
|
exports.default = traverse;
|
||
|
Object.defineProperty(exports, "NodePath", {
|
||
|
enumerable: true,
|
||
|
get: function () {
|
||
|
return _path.default;
|
||
|
}
|
||
|
});
|
||
|
Object.defineProperty(exports, "Scope", {
|
||
|
enumerable: true,
|
||
|
get: function () {
|
||
|
return _scope.default;
|
||
|
}
|
||
|
});
|
||
|
Object.defineProperty(exports, "Hub", {
|
||
|
enumerable: true,
|
||
|
get: function () {
|
||
|
return _hub.default;
|
||
|
}
|
||
|
});
|
||
|
exports.visitors = void 0;
|
||
|
|
||
|
var _context = _interopRequireDefault(require("./context"));
|
||
|
|
||
|
var visitors = _interopRequireWildcard(require("./visitors"));
|
||
|
|
||
|
exports.visitors = visitors;
|
||
|
|
||
|
function _includes() {
|
||
|
const data = _interopRequireDefault(require("lodash/includes"));
|
||
|
|
||
|
_includes = function () {
|
||
|
return data;
|
||
|
};
|
||
|
|
||
|
return data;
|
||
|
}
|
||
|
|
||
|
function t() {
|
||
|
const data = _interopRequireWildcard(require("@babel/types"));
|
||
|
|
||
|
t = function () {
|
||
|
return data;
|
||
|
};
|
||
|
|
||
|
return data;
|
||
|
}
|
||
|
|
||
|
var cache = _interopRequireWildcard(require("./cache"));
|
||
|
|
||
|
var _path = _interopRequireDefault(require("./path"));
|
||
|
|
||
|
var _scope = _interopRequireDefault(require("./scope"));
|
||
|
|
||
|
var _hub = _interopRequireDefault(require("./hub"));
|
||
|
|
||
|
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)) { var desc = Object.defineProperty && Object.getOwnPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : {}; if (desc.get || desc.set) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } } newObj.default = obj; return newObj; } }
|
||
|
|
||
|
function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
|
||
|
|
||
|
function traverse(parent, opts, scope, state, parentPath) {
|
||
|
if (!parent) return;
|
||
|
if (!opts) opts = {};
|
||
|
|
||
|
if (!opts.noScope && !scope) {
|
||
|
if (parent.type !== "Program" && parent.type !== "File") {
|
||
|
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.");
|
||
|
}
|
||
|
}
|
||
|
|
||
|
visitors.explode(opts);
|
||
|
traverse.node(parent, opts, scope, state, parentPath);
|
||
|
}
|
||
|
|
||
|
traverse.visitors = visitors;
|
||
|
traverse.verify = visitors.verify;
|
||
|
traverse.explode = visitors.explode;
|
||
|
|
||
|
traverse.cheap = function (node, enter) {
|
||
|
return t().traverseFast(node, enter);
|
||
|
};
|
||
|
|
||
|
traverse.node = function (node, opts, scope, state, parentPath, skipKeys) {
|
||
|
const keys = t().VISITOR_KEYS[node.type];
|
||
|
if (!keys) return;
|
||
|
const context = new _context.default(scope, opts, state, parentPath);
|
||
|
|
||
|
for (const key of keys) {
|
||
|
if (skipKeys && skipKeys[key]) continue;
|
||
|
if (context.visit(node, key)) return;
|
||
|
}
|
||
|
};
|
||
|
|
||
|
traverse.clearNode = function (node, opts) {
|
||
|
t().removeProperties(node, opts);
|
||
|
cache.path.delete(node);
|
||
|
};
|
||
|
|
||
|
traverse.removeProperties = function (tree, opts) {
|
||
|
t().traverseFast(tree, traverse.clearNode, opts);
|
||
|
return tree;
|
||
|
};
|
||
|
|
||
|
function hasBlacklistedType(path, state) {
|
||
|
if (path.node.type === state.type) {
|
||
|
state.has = true;
|
||
|
path.stop();
|
||
|
}
|
||
|
}
|
||
|
|
||
|
traverse.hasType = function (tree, type, blacklistTypes) {
|
||
|
if ((0, _includes().default)(blacklistTypes, tree.type)) return false;
|
||
|
if (tree.type === type) return true;
|
||
|
const state = {
|
||
|
has: false,
|
||
|
type: type
|
||
|
};
|
||
|
traverse(tree, {
|
||
|
noScope: true,
|
||
|
blacklist: blacklistTypes,
|
||
|
enter: hasBlacklistedType
|
||
|
}, null, state);
|
||
|
return state.has;
|
||
|
};
|
||
|
|
||
|
traverse.cache = cache;
|