index.js 9.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395
  1. 'use strict';
  2. var whichTypedArray = require('which-typed-array');
  3. var taSlice = require('typedarray.prototype.slice');
  4. var gopd = require('gopd');
  5. // TODO: use call-bind, is-date, is-regex, is-string, is-boolean-object, is-number-object
  6. function toS(obj) { return Object.prototype.toString.call(obj); }
  7. function isDate(obj) { return toS(obj) === '[object Date]'; }
  8. function isRegExp(obj) { return toS(obj) === '[object RegExp]'; }
  9. function isError(obj) { return toS(obj) === '[object Error]'; }
  10. function isBoolean(obj) { return toS(obj) === '[object Boolean]'; }
  11. function isNumber(obj) { return toS(obj) === '[object Number]'; }
  12. function isString(obj) { return toS(obj) === '[object String]'; }
  13. // TODO: use isarray
  14. var isArray = Array.isArray || function isArray(xs) {
  15. return Object.prototype.toString.call(xs) === '[object Array]';
  16. };
  17. // TODO: use for-each?
  18. function forEach(xs, fn) {
  19. if (xs.forEach) { return xs.forEach(fn); }
  20. for (var i = 0; i < xs.length; i++) {
  21. fn(xs[i], i, xs);
  22. }
  23. return void undefined;
  24. }
  25. // TODO: use object-keys
  26. var objectKeys = Object.keys || function keys(obj) {
  27. var res = [];
  28. for (var key in obj) { res[res.length] = key; } // eslint-disable-line no-restricted-syntax
  29. return res;
  30. };
  31. var propertyIsEnumerable = Object.prototype.propertyIsEnumerable;
  32. var getOwnPropertySymbols = Object.getOwnPropertySymbols; // eslint-disable-line id-length
  33. // TODO: use reflect.ownkeys and filter out non-enumerables
  34. function ownEnumerableKeys(obj) {
  35. var res = objectKeys(obj);
  36. // Include enumerable symbol properties.
  37. if (getOwnPropertySymbols) {
  38. var symbols = getOwnPropertySymbols(obj);
  39. for (var i = 0; i < symbols.length; i++) {
  40. if (propertyIsEnumerable.call(obj, symbols[i])) {
  41. res[res.length] = symbols[i];
  42. }
  43. }
  44. }
  45. return res;
  46. }
  47. // TODO: use object.hasown
  48. var hasOwnProperty = Object.prototype.hasOwnProperty || function (obj, key) {
  49. return key in obj;
  50. };
  51. function isWritable(object, key) {
  52. if (typeof gopd !== 'function') {
  53. return true;
  54. }
  55. var desc = gopd(object, key);
  56. return !desc || !desc.writable;
  57. }
  58. function copy(src, options) {
  59. if (typeof src === 'object' && src !== null) {
  60. var dst;
  61. if (isArray(src)) {
  62. dst = [];
  63. } else if (isDate(src)) {
  64. dst = new Date(src.getTime ? src.getTime() : src);
  65. } else if (isRegExp(src)) {
  66. dst = new RegExp(src);
  67. } else if (isError(src)) {
  68. dst = { message: src.message };
  69. } else if (isBoolean(src) || isNumber(src) || isString(src)) {
  70. dst = Object(src);
  71. } else {
  72. var ta = whichTypedArray(src);
  73. if (ta) {
  74. return taSlice(src);
  75. } else if (Object.create && Object.getPrototypeOf) {
  76. dst = Object.create(Object.getPrototypeOf(src));
  77. } else if (src.constructor === Object) {
  78. dst = {};
  79. } else {
  80. var proto = (src.constructor && src.constructor.prototype)
  81. || src.__proto__
  82. || {};
  83. var T = function T() {}; // eslint-disable-line func-style, func-name-matching
  84. T.prototype = proto;
  85. dst = new T();
  86. }
  87. }
  88. var iteratorFunction = options.includeSymbols ? ownEnumerableKeys : objectKeys;
  89. forEach(iteratorFunction(src), function (key) {
  90. dst[key] = src[key];
  91. });
  92. return dst;
  93. }
  94. return src;
  95. }
  96. /** @type {TraverseOptions} */
  97. var emptyNull = { __proto__: null };
  98. function walk(root, cb) {
  99. var path = [];
  100. var parents = [];
  101. var alive = true;
  102. var options = arguments.length > 2 ? arguments[2] : emptyNull;
  103. var iteratorFunction = options.includeSymbols ? ownEnumerableKeys : objectKeys;
  104. var immutable = !!options.immutable;
  105. return (function walker(node_) {
  106. var node = immutable ? copy(node_, options) : node_;
  107. var modifiers = { __proto__: null };
  108. var keepGoing = true;
  109. var state = {
  110. node: node,
  111. node_: node_,
  112. path: [].concat(path),
  113. parent: parents[parents.length - 1],
  114. parents: parents,
  115. key: path[path.length - 1],
  116. removedKeys: { __proto__: null },
  117. isRoot: path.length === 0,
  118. level: path.length,
  119. circular: null,
  120. update: function (x, stopHere) {
  121. if (!state.isRoot) {
  122. state.parent.node[state.key] = x;
  123. }
  124. state.node = x;
  125. if (stopHere) { keepGoing = false; }
  126. },
  127. delete: function (stopHere) {
  128. delete state.parent.node[state.key];
  129. state.parent.removedKeys[state.key] = true;
  130. if (stopHere) { keepGoing = false; }
  131. },
  132. remove: function (stopHere) {
  133. if (isArray(state.parent.node)) {
  134. state.parent.node.splice(state.key, 1);
  135. state.parent.removedKeys[state.key] = true;
  136. if (stopHere) { keepGoing = false; }
  137. } else {
  138. state.delete(stopHere);
  139. }
  140. },
  141. keys: null,
  142. before: function (f) { modifiers.before = f; },
  143. after: function (f) { modifiers.after = f; },
  144. pre: function (f) { modifiers.pre = f; },
  145. post: function (f) { modifiers.post = f; },
  146. stop: function () { alive = false; },
  147. block: function () { keepGoing = false; },
  148. };
  149. if (!alive) { return state; }
  150. function updateState() {
  151. if (typeof state.node === 'object' && state.node !== null) {
  152. if (!state.keys || state.node_ !== state.node) {
  153. state.keys = iteratorFunction(state.node);
  154. }
  155. state.isLeaf = state.keys.length === 0;
  156. for (var i = 0; i < parents.length; i++) {
  157. if (parents[i].node_ === node_) {
  158. state.circular = parents[i];
  159. break; // eslint-disable-line no-restricted-syntax
  160. }
  161. }
  162. } else {
  163. state.isLeaf = true;
  164. state.keys = null;
  165. }
  166. state.notLeaf = !state.isLeaf;
  167. state.notRoot = !state.isRoot;
  168. }
  169. updateState();
  170. // use return values to update if defined
  171. var ret = cb.call(state, state.node);
  172. if (ret !== undefined && state.update) { state.update(ret); }
  173. if (modifiers.before) { modifiers.before.call(state, state.node); }
  174. if (!keepGoing) { return state; }
  175. if (
  176. typeof state.node === 'object'
  177. && state.node !== null
  178. && !state.circular
  179. ) {
  180. parents[parents.length] = state;
  181. updateState();
  182. forEach(state.keys, function (key, i) {
  183. var prevIsRemoved = (i - 1) in state.removedKeys;
  184. if (prevIsRemoved) {
  185. key = state.keys[i - 1]; // eslint-disable-line no-param-reassign
  186. }
  187. path[path.length] = (key);
  188. if (modifiers.pre) { modifiers.pre.call(state, state.node[key], key); }
  189. var child = walker(state.node[key]);
  190. if (
  191. immutable
  192. && hasOwnProperty.call(state.node, key)
  193. && !isWritable(state.node, key)
  194. && !prevIsRemoved
  195. ) {
  196. state.node[key] = child.node;
  197. }
  198. child.isLast = i === state.keys.length - 1;
  199. child.isFirst = i === 0;
  200. if (modifiers.post) { modifiers.post.call(state, child); }
  201. path.pop();
  202. });
  203. parents.pop();
  204. }
  205. if (modifiers.after) { modifiers.after.call(state, state.node); }
  206. return state;
  207. }(root)).node;
  208. }
  209. /** @typedef {{ immutable?: boolean, includeSymbols?: boolean }} TraverseOptions */
  210. /**
  211. * A traverse constructor
  212. * @param {object} obj - the object to traverse
  213. * @param {TraverseOptions | undefined} [options] - options for the traverse
  214. * @constructor
  215. */
  216. function Traverse(obj) {
  217. /** @type {TraverseOptions} */
  218. this.options = arguments.length > 1 ? arguments[1] : emptyNull;
  219. this.value = obj;
  220. }
  221. /** @type {(ps: PropertyKey[]) => Traverse['value']} */
  222. Traverse.prototype.get = function (ps) {
  223. var node = this.value;
  224. for (var i = 0; node && i < ps.length; i++) {
  225. var key = ps[i];
  226. if (
  227. !hasOwnProperty.call(node, key)
  228. || (!this.options.includeSymbols && typeof key === 'symbol')
  229. ) {
  230. return void undefined;
  231. }
  232. node = node[key];
  233. }
  234. return node;
  235. };
  236. /** @type {(ps: PropertyKey[]) => boolean} */
  237. Traverse.prototype.has = function (ps) {
  238. var node = this.value;
  239. // TODO: remove ps.length check
  240. if (!node && ps.length > 0) {
  241. return false;
  242. }
  243. for (var i = 0; node && i < ps.length; i++) {
  244. var key = ps[i];
  245. if (!hasOwnProperty.call(node, key) || (!this.options.includeSymbols && typeof key === 'symbol')) {
  246. return false;
  247. }
  248. node = node[key];
  249. }
  250. return true;
  251. };
  252. Traverse.prototype.set = function (ps, value) {
  253. var node = this.value;
  254. for (var i = 0; i < ps.length - 1; i++) {
  255. var key = ps[i];
  256. if (!hasOwnProperty.call(node, key)) { node[key] = {}; }
  257. node = node[key];
  258. }
  259. node[ps[i]] = value;
  260. return value;
  261. };
  262. Traverse.prototype.map = function (cb) {
  263. return walk(this.value, cb, { __proto__: null, immutable: true, includeSymbols: !!this.options.includeSymbols });
  264. };
  265. Traverse.prototype.forEach = function (cb) {
  266. this.value = walk(this.value, cb, this.options);
  267. return this.value;
  268. };
  269. Traverse.prototype.reduce = function (cb, init) {
  270. var skip = arguments.length === 1;
  271. var acc = skip ? this.value : init;
  272. this.forEach(function (x) {
  273. if (!this.isRoot || !skip) {
  274. acc = cb.call(this, acc, x);
  275. }
  276. });
  277. return acc;
  278. };
  279. Traverse.prototype.paths = function () {
  280. var acc = [];
  281. this.forEach(function () {
  282. acc[acc.length] = this.path;
  283. });
  284. return acc;
  285. };
  286. Traverse.prototype.nodes = function () {
  287. var acc = [];
  288. this.forEach(function () {
  289. acc[acc.length] = this.node;
  290. });
  291. return acc;
  292. };
  293. Traverse.prototype.clone = function () {
  294. var parents = [];
  295. var nodes = [];
  296. var options = this.options;
  297. if (whichTypedArray(this.value)) {
  298. return taSlice(this.value);
  299. }
  300. return (function clone(src) {
  301. for (var i = 0; i < parents.length; i++) {
  302. if (parents[i] === src) {
  303. return nodes[i];
  304. }
  305. }
  306. if (typeof src === 'object' && src !== null) {
  307. var dst = copy(src, options);
  308. parents[parents.length] = (src);
  309. nodes[nodes.length] = (dst);
  310. var iteratorFunction = options.includeSymbols ? ownEnumerableKeys : objectKeys;
  311. forEach(iteratorFunction(src), function (key) {
  312. dst[key] = clone(src[key]);
  313. });
  314. parents.pop();
  315. nodes.pop();
  316. return dst;
  317. }
  318. return src;
  319. }(this.value));
  320. };
  321. /** @type {(obj: object, options?: TraverseOptions) => Traverse} */
  322. function traverse(obj) {
  323. var options = arguments.length > 1 ? arguments[1] : emptyNull;
  324. return new Traverse(obj, options);
  325. }
  326. // TODO: replace with object.assign?
  327. forEach(ownEnumerableKeys(Traverse.prototype), function (key) {
  328. traverse[key] = function (obj) {
  329. var args = [].slice.call(arguments, 1);
  330. var t = new Traverse(obj);
  331. return t[key].apply(t, args);
  332. };
  333. });
  334. module.exports = traverse;