UNPKG

d2-ui

Version:
300 lines (232 loc) 7.59 kB
/*istanbul ignore next*/"use strict"; exports.__esModule = true; var _getIterator2 = require("babel-runtime/core-js/get-iterator"); var _getIterator3 = _interopRequireDefault(_getIterator2); exports.call = call; /*istanbul ignore next*/exports._call = _call; /*istanbul ignore next*/exports.isBlacklisted = isBlacklisted; /*istanbul ignore next*/exports.visit = visit; /*istanbul ignore next*/exports.skip = skip; /*istanbul ignore next*/exports.skipKey = skipKey; /*istanbul ignore next*/exports.stop = stop; /*istanbul ignore next*/exports.setScope = setScope; /*istanbul ignore next*/exports.setContext = setContext; /*istanbul ignore next*/exports.resync = resync; /*istanbul ignore next*/exports._resyncParent = _resyncParent; /*istanbul ignore next*/exports._resyncKey = _resyncKey; /*istanbul ignore next*/exports._resyncList = _resyncList; /*istanbul ignore next*/exports._resyncRemoved = _resyncRemoved; /*istanbul ignore next*/exports.popContext = popContext; /*istanbul ignore next*/exports.pushContext = pushContext; /*istanbul ignore next*/exports.setup = setup; /*istanbul ignore next*/exports.setKey = setKey; /*istanbul ignore next*/exports.requeue = requeue; /*istanbul ignore next*/exports._getQueueContexts = _getQueueContexts; var /*istanbul ignore next*/_index = require("../index"); /*istanbul ignore next*/ var _index2 = _interopRequireDefault(_index); function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; } function call(key) { var opts = this.opts; this.debug(function () /*istanbul ignore next*/{ return key; }); if (this.node) { if (this._call(opts[key])) return true; } if (this.node) { return this._call(opts[this.node.type] && opts[this.node.type][key]); } return false; } // This file contains methods responsible for maintaining a TraversalContext. function _call(fns) { if (!fns) return false; for ( /*istanbul ignore next*/var _iterator = fns, _isArray = Array.isArray(_iterator), _i = 0, _iterator = _isArray ? _iterator : (0, _getIterator3.default)(_iterator);;) { /*istanbul ignore next*/ var _ref; if (_isArray) { if (_i >= _iterator.length) break; _ref = _iterator[_i++]; } else { _i = _iterator.next(); if (_i.done) break; _ref = _i.value; } var fn = _ref; if (!fn) continue; var node = this.node; if (!node) return true; var ret = fn.call(this.state, this, this.state); if (ret) throw new Error( /*istanbul ignore next*/"Unexpected return value from visitor method " + fn); // node has been replaced, it will have been requeued if (this.node !== node) return true; if (this.shouldStop || this.shouldSkip || this.removed) return true; } return false; } function isBlacklisted() { var blacklist = this.opts.blacklist; return blacklist && blacklist.indexOf(this.node.type) > -1; } function visit() { if (!this.node) { return false; } if (this.isBlacklisted()) { return false; } if (this.opts.shouldSkip && this.opts.shouldSkip(this)) { return false; } if (this.call("enter") || this.shouldSkip) { this.debug(function () /*istanbul ignore next*/{ return "Skip..."; }); return this.shouldStop; } this.debug(function () /*istanbul ignore next*/{ return "Recursing into..."; }); /*istanbul ignore next*/_index2.default.node(this.node, this.opts, this.scope, this.state, this, this.skipKeys); this.call("exit"); return this.shouldStop; } function skip() { this.shouldSkip = true; } function skipKey(key) { this.skipKeys[key] = true; } function stop() { this.shouldStop = true; this.shouldSkip = true; } function setScope() { if (this.opts && this.opts.noScope) return; var target = this.context && this.context.scope; if (!target) { var path = this.parentPath; while (path && !target) { if (path.opts && path.opts.noScope) return; target = path.scope; path = path.parentPath; } } this.scope = this.getScope(target); if (this.scope) this.scope.init(); } function setContext(context) { this.shouldSkip = false; this.shouldStop = false; this.removed = false; this.skipKeys = {}; if (context) { this.context = context; this.state = context.state; this.opts = context.opts; } this.setScope(); return this; } /** * Here we resync the node paths `key` and `container`. If they've changed according * to what we have stored internally then we attempt to resync by crawling and looking * for the new values. */ function resync() { if (this.removed) return; this._resyncParent(); this._resyncList(); this._resyncKey(); //this._resyncRemoved(); } function _resyncParent() { if (this.parentPath) { this.parent = this.parentPath.node; } } function _resyncKey() { if (!this.container) return; if (this.node === this.container[this.key]) return; // grrr, path key is out of sync. this is likely due to a modification to the AST // not done through our path APIs if (Array.isArray(this.container)) { for (var i = 0; i < this.container.length; i++) { if (this.container[i] === this.node) { return this.setKey(i); } } } else { for (var key in this.container) { if (this.container[key] === this.node) { return this.setKey(key); } } } // ¯\_(ツ)_/¯ who knows where it's gone lol this.key = null; } function _resyncList() { if (!this.parent || !this.inList) return; var newContainer = this.parent[this.listKey]; if (this.container === newContainer) return; // container is out of sync. this is likely the result of it being reassigned this.container = newContainer || null; } function _resyncRemoved() { if (this.key == null || !this.container || this.container[this.key] !== this.node) { this._markRemoved(); } } function popContext() { this.contexts.pop(); this.setContext(this.contexts[this.contexts.length - 1]); } function pushContext(context) { this.contexts.push(context); this.setContext(context); } function setup(parentPath, container, listKey, key) { this.inList = !!listKey; this.listKey = listKey; this.parentKey = listKey || key; this.container = container; this.parentPath = parentPath || this.parentPath; this.setKey(key); } function setKey(key) { this.key = key; this.node = this.container[this.key]; this.type = this.node && this.node.type; } function requeue() { /*istanbul ignore next*/var pathToQueue = arguments.length <= 0 || arguments[0] === undefined ? this : arguments[0]; if (pathToQueue.removed) return; // TODO(loganfsmyth): This should be switched back to queue in parent contexts // automatically once T2892 and T7160 have been resolved. See T7166. // let contexts = this._getQueueContexts(); var contexts = this.contexts; for ( /*istanbul ignore next*/var _iterator2 = contexts, _isArray2 = Array.isArray(_iterator2), _i2 = 0, _iterator2 = _isArray2 ? _iterator2 : (0, _getIterator3.default)(_iterator2);;) { /*istanbul ignore next*/ var _ref2; if (_isArray2) { if (_i2 >= _iterator2.length) break; _ref2 = _iterator2[_i2++]; } else { _i2 = _iterator2.next(); if (_i2.done) break; _ref2 = _i2.value; } var context = _ref2; context.maybeQueue(pathToQueue); } } function _getQueueContexts() { var path = this; var contexts = this.contexts; while (!contexts.length) { path = path.parentPath; contexts = path.contexts; } return contexts; }