123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395 |
- 'use strict';
- var whichTypedArray = require('which-typed-array');
- var taSlice = require('typedarray.prototype.slice');
- var gopd = require('gopd');
- // TODO: use call-bind, is-date, is-regex, is-string, is-boolean-object, is-number-object
- function toS(obj) { return Object.prototype.toString.call(obj); }
- function isDate(obj) { return toS(obj) === '[object Date]'; }
- function isRegExp(obj) { return toS(obj) === '[object RegExp]'; }
- function isError(obj) { return toS(obj) === '[object Error]'; }
- function isBoolean(obj) { return toS(obj) === '[object Boolean]'; }
- function isNumber(obj) { return toS(obj) === '[object Number]'; }
- function isString(obj) { return toS(obj) === '[object String]'; }
- // TODO: use isarray
- var isArray = Array.isArray || function isArray(xs) {
- return Object.prototype.toString.call(xs) === '[object Array]';
- };
- // TODO: use for-each?
- function forEach(xs, fn) {
- if (xs.forEach) { return xs.forEach(fn); }
- for (var i = 0; i < xs.length; i++) {
- fn(xs[i], i, xs);
- }
- return void undefined;
- }
- // TODO: use object-keys
- var objectKeys = Object.keys || function keys(obj) {
- var res = [];
- for (var key in obj) { res[res.length] = key; } // eslint-disable-line no-restricted-syntax
- return res;
- };
- var propertyIsEnumerable = Object.prototype.propertyIsEnumerable;
- var getOwnPropertySymbols = Object.getOwnPropertySymbols; // eslint-disable-line id-length
- // TODO: use reflect.ownkeys and filter out non-enumerables
- function ownEnumerableKeys(obj) {
- var res = objectKeys(obj);
- // Include enumerable symbol properties.
- if (getOwnPropertySymbols) {
- var symbols = getOwnPropertySymbols(obj);
- for (var i = 0; i < symbols.length; i++) {
- if (propertyIsEnumerable.call(obj, symbols[i])) {
- res[res.length] = symbols[i];
- }
- }
- }
- return res;
- }
- // TODO: use object.hasown
- var hasOwnProperty = Object.prototype.hasOwnProperty || function (obj, key) {
- return key in obj;
- };
- function isWritable(object, key) {
- if (typeof gopd !== 'function') {
- return true;
- }
- var desc = gopd(object, key);
- return !desc || !desc.writable;
- }
- function copy(src, options) {
- if (typeof src === 'object' && src !== null) {
- var dst;
- if (isArray(src)) {
- dst = [];
- } else if (isDate(src)) {
- dst = new Date(src.getTime ? src.getTime() : src);
- } else if (isRegExp(src)) {
- dst = new RegExp(src);
- } else if (isError(src)) {
- dst = { message: src.message };
- } else if (isBoolean(src) || isNumber(src) || isString(src)) {
- dst = Object(src);
- } else {
- var ta = whichTypedArray(src);
- if (ta) {
- return taSlice(src);
- } else if (Object.create && Object.getPrototypeOf) {
- dst = Object.create(Object.getPrototypeOf(src));
- } else if (src.constructor === Object) {
- dst = {};
- } else {
- var proto = (src.constructor && src.constructor.prototype)
- || src.__proto__
- || {};
- var T = function T() {}; // eslint-disable-line func-style, func-name-matching
- T.prototype = proto;
- dst = new T();
- }
- }
- var iteratorFunction = options.includeSymbols ? ownEnumerableKeys : objectKeys;
- forEach(iteratorFunction(src), function (key) {
- dst[key] = src[key];
- });
- return dst;
- }
- return src;
- }
- /** @type {TraverseOptions} */
- var emptyNull = { __proto__: null };
- function walk(root, cb) {
- var path = [];
- var parents = [];
- var alive = true;
- var options = arguments.length > 2 ? arguments[2] : emptyNull;
- var iteratorFunction = options.includeSymbols ? ownEnumerableKeys : objectKeys;
- var immutable = !!options.immutable;
- return (function walker(node_) {
- var node = immutable ? copy(node_, options) : node_;
- var modifiers = { __proto__: null };
- var keepGoing = true;
- var state = {
- node: node,
- node_: node_,
- path: [].concat(path),
- parent: parents[parents.length - 1],
- parents: parents,
- key: path[path.length - 1],
- removedKeys: { __proto__: null },
- isRoot: path.length === 0,
- level: path.length,
- circular: null,
- update: function (x, stopHere) {
- if (!state.isRoot) {
- state.parent.node[state.key] = x;
- }
- state.node = x;
- if (stopHere) { keepGoing = false; }
- },
- delete: function (stopHere) {
- delete state.parent.node[state.key];
- state.parent.removedKeys[state.key] = true;
- if (stopHere) { keepGoing = false; }
- },
- remove: function (stopHere) {
- if (isArray(state.parent.node)) {
- state.parent.node.splice(state.key, 1);
- state.parent.removedKeys[state.key] = true;
- if (stopHere) { keepGoing = false; }
- } else {
- state.delete(stopHere);
- }
- },
- keys: null,
- before: function (f) { modifiers.before = f; },
- after: function (f) { modifiers.after = f; },
- pre: function (f) { modifiers.pre = f; },
- post: function (f) { modifiers.post = f; },
- stop: function () { alive = false; },
- block: function () { keepGoing = false; },
- };
- if (!alive) { return state; }
- function updateState() {
- if (typeof state.node === 'object' && state.node !== null) {
- if (!state.keys || state.node_ !== state.node) {
- state.keys = iteratorFunction(state.node);
- }
- state.isLeaf = state.keys.length === 0;
- for (var i = 0; i < parents.length; i++) {
- if (parents[i].node_ === node_) {
- state.circular = parents[i];
- break; // eslint-disable-line no-restricted-syntax
- }
- }
- } else {
- state.isLeaf = true;
- state.keys = null;
- }
- state.notLeaf = !state.isLeaf;
- state.notRoot = !state.isRoot;
- }
- updateState();
- // use return values to update if defined
- var ret = cb.call(state, state.node);
- if (ret !== undefined && state.update) { state.update(ret); }
- if (modifiers.before) { modifiers.before.call(state, state.node); }
- if (!keepGoing) { return state; }
- if (
- typeof state.node === 'object'
- && state.node !== null
- && !state.circular
- ) {
- parents[parents.length] = state;
- updateState();
- forEach(state.keys, function (key, i) {
- var prevIsRemoved = (i - 1) in state.removedKeys;
- if (prevIsRemoved) {
- key = state.keys[i - 1]; // eslint-disable-line no-param-reassign
- }
- path[path.length] = (key);
- if (modifiers.pre) { modifiers.pre.call(state, state.node[key], key); }
- var child = walker(state.node[key]);
- if (
- immutable
- && hasOwnProperty.call(state.node, key)
- && !isWritable(state.node, key)
- && !prevIsRemoved
- ) {
- state.node[key] = child.node;
- }
- child.isLast = i === state.keys.length - 1;
- child.isFirst = i === 0;
- if (modifiers.post) { modifiers.post.call(state, child); }
- path.pop();
- });
- parents.pop();
- }
- if (modifiers.after) { modifiers.after.call(state, state.node); }
- return state;
- }(root)).node;
- }
- /** @typedef {{ immutable?: boolean, includeSymbols?: boolean }} TraverseOptions */
- /**
- * A traverse constructor
- * @param {object} obj - the object to traverse
- * @param {TraverseOptions | undefined} [options] - options for the traverse
- * @constructor
- */
- function Traverse(obj) {
- /** @type {TraverseOptions} */
- this.options = arguments.length > 1 ? arguments[1] : emptyNull;
- this.value = obj;
- }
- /** @type {(ps: PropertyKey[]) => Traverse['value']} */
- Traverse.prototype.get = function (ps) {
- var node = this.value;
- for (var i = 0; node && i < ps.length; i++) {
- var key = ps[i];
- if (
- !hasOwnProperty.call(node, key)
- || (!this.options.includeSymbols && typeof key === 'symbol')
- ) {
- return void undefined;
- }
- node = node[key];
- }
- return node;
- };
- /** @type {(ps: PropertyKey[]) => boolean} */
- Traverse.prototype.has = function (ps) {
- var node = this.value;
- // TODO: remove ps.length check
- if (!node && ps.length > 0) {
- return false;
- }
- for (var i = 0; node && i < ps.length; i++) {
- var key = ps[i];
- if (!hasOwnProperty.call(node, key) || (!this.options.includeSymbols && typeof key === 'symbol')) {
- return false;
- }
- node = node[key];
- }
- return true;
- };
- Traverse.prototype.set = function (ps, value) {
- var node = this.value;
- for (var i = 0; i < ps.length - 1; i++) {
- var key = ps[i];
- if (!hasOwnProperty.call(node, key)) { node[key] = {}; }
- node = node[key];
- }
- node[ps[i]] = value;
- return value;
- };
- Traverse.prototype.map = function (cb) {
- return walk(this.value, cb, { __proto__: null, immutable: true, includeSymbols: !!this.options.includeSymbols });
- };
- Traverse.prototype.forEach = function (cb) {
- this.value = walk(this.value, cb, this.options);
- return this.value;
- };
- Traverse.prototype.reduce = function (cb, init) {
- var skip = arguments.length === 1;
- var acc = skip ? this.value : init;
- this.forEach(function (x) {
- if (!this.isRoot || !skip) {
- acc = cb.call(this, acc, x);
- }
- });
- return acc;
- };
- Traverse.prototype.paths = function () {
- var acc = [];
- this.forEach(function () {
- acc[acc.length] = this.path;
- });
- return acc;
- };
- Traverse.prototype.nodes = function () {
- var acc = [];
- this.forEach(function () {
- acc[acc.length] = this.node;
- });
- return acc;
- };
- Traverse.prototype.clone = function () {
- var parents = [];
- var nodes = [];
- var options = this.options;
- if (whichTypedArray(this.value)) {
- return taSlice(this.value);
- }
- return (function clone(src) {
- for (var i = 0; i < parents.length; i++) {
- if (parents[i] === src) {
- return nodes[i];
- }
- }
- if (typeof src === 'object' && src !== null) {
- var dst = copy(src, options);
- parents[parents.length] = (src);
- nodes[nodes.length] = (dst);
- var iteratorFunction = options.includeSymbols ? ownEnumerableKeys : objectKeys;
- forEach(iteratorFunction(src), function (key) {
- dst[key] = clone(src[key]);
- });
- parents.pop();
- nodes.pop();
- return dst;
- }
- return src;
- }(this.value));
- };
- /** @type {(obj: object, options?: TraverseOptions) => Traverse} */
- function traverse(obj) {
- var options = arguments.length > 1 ? arguments[1] : emptyNull;
- return new Traverse(obj, options);
- }
- // TODO: replace with object.assign?
- forEach(ownEnumerableKeys(Traverse.prototype), function (key) {
- traverse[key] = function (obj) {
- var args = [].slice.call(arguments, 1);
- var t = new Traverse(obj);
- return t[key].apply(t, args);
- };
- });
- module.exports = traverse;
|