traverse.js 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.default = traverse;
  6. var _util = require("@glimmer/util");
  7. var _visitorKeys = _interopRequireDefault(require("../v1/visitor-keys"));
  8. var _errors = require("./errors");
  9. var _path = _interopRequireDefault(require("./path"));
  10. function _interopRequireDefault(obj) { return obj && obj.__esModule ? obj : { default: obj }; }
  11. function getEnterFunction(handler) {
  12. if (typeof handler === 'function') {
  13. return handler;
  14. } else {
  15. return handler.enter;
  16. }
  17. }
  18. function getExitFunction(handler) {
  19. if (typeof handler === 'function') {
  20. return undefined;
  21. } else {
  22. return handler.exit;
  23. }
  24. }
  25. function getKeyHandler(handler, key) {
  26. let keyVisitor = typeof handler !== 'function' ? handler.keys : undefined;
  27. if (keyVisitor === undefined) return;
  28. let keyHandler = keyVisitor[key];
  29. if (keyHandler !== undefined) {
  30. return keyHandler;
  31. }
  32. return keyVisitor.All;
  33. }
  34. function getNodeHandler(visitor, nodeType) {
  35. if (nodeType === 'Template' || nodeType === 'Block') {
  36. if (visitor.Program) {
  37. if (false
  38. /* LOCAL_DEBUG */
  39. ) {
  40. false && !false && (0, _util.deprecate)(`The 'Program' visitor node is deprecated. Use 'Template' or 'Block' instead (node was '${nodeType}') `);
  41. }
  42. return visitor.Program;
  43. }
  44. }
  45. let handler = visitor[nodeType];
  46. if (handler !== undefined) {
  47. return handler;
  48. }
  49. return visitor.All;
  50. }
  51. function visitNode(visitor, path) {
  52. let {
  53. node,
  54. parent,
  55. parentKey
  56. } = path;
  57. let handler = getNodeHandler(visitor, node.type);
  58. let enter;
  59. let exit;
  60. if (handler !== undefined) {
  61. enter = getEnterFunction(handler);
  62. exit = getExitFunction(handler);
  63. }
  64. let result;
  65. if (enter !== undefined) {
  66. result = enter(node, path);
  67. }
  68. if (result !== undefined && result !== null) {
  69. if (JSON.stringify(node) === JSON.stringify(result)) {
  70. result = undefined;
  71. } else if (Array.isArray(result)) {
  72. visitArray(visitor, result, parent, parentKey);
  73. return result;
  74. } else {
  75. let path = new _path.default(result, parent, parentKey);
  76. return visitNode(visitor, path) || result;
  77. }
  78. }
  79. if (result === undefined) {
  80. let keys = _visitorKeys.default[node.type];
  81. for (let i = 0; i < keys.length; i++) {
  82. let key = keys[i]; // we know if it has child keys we can widen to a ParentNode
  83. visitKey(visitor, handler, path, key);
  84. }
  85. if (exit !== undefined) {
  86. result = exit(node, path);
  87. }
  88. }
  89. return result;
  90. }
  91. function get(node, key) {
  92. return node[key];
  93. }
  94. function set(node, key, value) {
  95. node[key] = value;
  96. }
  97. function visitKey(visitor, handler, path, key) {
  98. let {
  99. node
  100. } = path;
  101. let value = get(node, key);
  102. if (!value) {
  103. return;
  104. }
  105. let keyEnter;
  106. let keyExit;
  107. if (handler !== undefined) {
  108. let keyHandler = getKeyHandler(handler, key);
  109. if (keyHandler !== undefined) {
  110. keyEnter = getEnterFunction(keyHandler);
  111. keyExit = getExitFunction(keyHandler);
  112. }
  113. }
  114. if (keyEnter !== undefined) {
  115. if (keyEnter(node, key) !== undefined) {
  116. throw (0, _errors.cannotReplaceOrRemoveInKeyHandlerYet)(node, key);
  117. }
  118. }
  119. if (Array.isArray(value)) {
  120. visitArray(visitor, value, path, key);
  121. } else {
  122. let keyPath = new _path.default(value, path, key);
  123. let result = visitNode(visitor, keyPath);
  124. if (result !== undefined) {
  125. // TODO: dynamically check the results by having a table of
  126. // expected node types in value space, not just type space
  127. // eslint-disable-next-line @typescript-eslint/no-explicit-any
  128. assignKey(node, key, value, result);
  129. }
  130. }
  131. if (keyExit !== undefined) {
  132. if (keyExit(node, key) !== undefined) {
  133. throw (0, _errors.cannotReplaceOrRemoveInKeyHandlerYet)(node, key);
  134. }
  135. }
  136. }
  137. function visitArray(visitor, array, parent, parentKey) {
  138. for (let i = 0; i < array.length; i++) {
  139. let node = array[i];
  140. let path = new _path.default(node, parent, parentKey);
  141. let result = visitNode(visitor, path);
  142. if (result !== undefined) {
  143. i += spliceArray(array, i, result) - 1;
  144. }
  145. }
  146. }
  147. function assignKey(node, key, value, result) {
  148. if (result === null) {
  149. throw (0, _errors.cannotRemoveNode)(value, node, key);
  150. } else if (Array.isArray(result)) {
  151. if (result.length === 1) {
  152. set(node, key, result[0]);
  153. } else {
  154. if (result.length === 0) {
  155. throw (0, _errors.cannotRemoveNode)(value, node, key);
  156. } else {
  157. throw (0, _errors.cannotReplaceNode)(value, node, key);
  158. }
  159. }
  160. } else {
  161. set(node, key, result);
  162. }
  163. }
  164. function spliceArray(array, index, result) {
  165. if (result === null) {
  166. array.splice(index, 1);
  167. return 0;
  168. } else if (Array.isArray(result)) {
  169. array.splice(index, 1, ...result);
  170. return result.length;
  171. } else {
  172. array.splice(index, 1, result);
  173. return 1;
  174. }
  175. }
  176. function traverse(node, visitor) {
  177. let path = new _path.default(node);
  178. visitNode(visitor, path);
  179. }
  180. //# sourceMappingURL=data:application/json;charset=utf-8;base64,eyJ2ZXJzaW9uIjozLCJzb3VyY2VzIjpbIi4uLy4uLy4uLy4uLy4uL3BhY2thZ2VzL0BnbGltbWVyL3N5bnRheC9saWIvdHJhdmVyc2FsL3RyYXZlcnNlLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7QUFDQTs7QUFHQTs7QUFDQTs7QUFLQTs7OztBQVNBLFNBQUEsZ0JBQUEsQ0FBQSxPQUFBLEVBQ2dEO0FBRTlDLE1BQUksT0FBQSxPQUFBLEtBQUosVUFBQSxFQUFtQztBQUNqQyxXQUFBLE9BQUE7QUFERixHQUFBLE1BRU87QUFDTCxXQUFPLE9BQU8sQ0FBZCxLQUFBO0FBQ0Q7QUFDRjs7QUFRRCxTQUFBLGVBQUEsQ0FBQSxPQUFBLEVBQ2dEO0FBRTlDLE1BQUksT0FBQSxPQUFBLEtBQUosVUFBQSxFQUFtQztBQUNqQyxXQUFBLFNBQUE7QUFERixHQUFBLE1BRU87QUFDTCxXQUFPLE9BQU8sQ0FBZCxJQUFBO0FBQ0Q7QUFDRjs7QUFFRCxTQUFBLGFBQUEsQ0FBQSxPQUFBLEVBQUEsR0FBQSxFQUVRO0FBRU4sTUFBSSxVQUFVLEdBQUcsT0FBQSxPQUFBLEtBQUEsVUFBQSxHQUFnQyxPQUFPLENBQXZDLElBQUEsR0FBakIsU0FBQTtBQUNBLE1BQUksVUFBVSxLQUFkLFNBQUEsRUFBOEI7QUFFOUIsTUFBSSxVQUFVLEdBQUcsVUFBVSxDQUEzQixHQUEyQixDQUEzQjs7QUFDQSxNQUFJLFVBQVUsS0FBZCxTQUFBLEVBQThCO0FBQzVCLFdBQUEsVUFBQTtBQUNEOztBQUNELFNBQU8sVUFBVSxDQUFqQixHQUFBO0FBQ0Q7O0FBT0QsU0FBQSxjQUFBLENBQUEsT0FBQSxFQUFBLFFBQUEsRUFFcUI7QUFFbkIsTUFBSSxRQUFRLEtBQVIsVUFBQSxJQUEyQixRQUFRLEtBQXZDLE9BQUEsRUFBcUQ7QUFDbkQsUUFBSSxPQUFPLENBQVgsT0FBQSxFQUFxQjtBQUNuQixVQUFBO0FBQUE7QUFBQSxRQUFpQjtBQUFBLG1CQUFBLENBQUEsS0FBQSxJQUNmLHFCQUNFLDBGQUEwRixRQUY3RSxLQUNmLENBRGU7QUFJaEI7O0FBRUQsYUFBTyxPQUFPLENBQWQsT0FBQTtBQUNEO0FBQ0Y7O0FBRUQsTUFBSSxPQUFPLEdBQUcsT0FBTyxDQUFyQixRQUFxQixDQUFyQjs7QUFDQSxNQUFJLE9BQU8sS0FBWCxTQUFBLEVBQTJCO0FBQ3pCLFdBQUEsT0FBQTtBQUNEOztBQUNELFNBQU8sT0FBTyxDQUFkLEdBQUE7QUFDRDs7QUFFRCxTQUFBLFNBQUEsQ0FBQSxPQUFBLEVBQUEsSUFBQSxFQUVxQjtBQUVuQixNQUFJO0FBQUEsSUFBQSxJQUFBO0FBQUEsSUFBQSxNQUFBO0FBQWdCLElBQUE7QUFBaEIsTUFBSixJQUFBO0FBRUEsTUFBSSxPQUFPLEdBQXFCLGNBQWMsQ0FBQSxPQUFBLEVBQVUsSUFBSSxDQUE1RCxJQUE4QyxDQUE5QztBQUNBLE1BQUEsS0FBQTtBQUNBLE1BQUEsSUFBQTs7QUFFQSxNQUFJLE9BQU8sS0FBWCxTQUFBLEVBQTJCO0FBQ3pCLElBQUEsS0FBSyxHQUFHLGdCQUFnQixDQUF4QixPQUF3QixDQUF4QjtBQUNBLElBQUEsSUFBSSxHQUFHLGVBQWUsQ0FBdEIsT0FBc0IsQ0FBdEI7QUFDRDs7QUFFRCxNQUFBLE1BQUE7O0FBQ0EsTUFBSSxLQUFLLEtBQVQsU0FBQSxFQUF5QjtBQUN2QixJQUFBLE1BQU0sR0FBRyxLQUFLLENBQUEsSUFBQSxFQUFkLElBQWMsQ0FBZDtBQUNEOztBQUVELE1BQUksTUFBTSxLQUFOLFNBQUEsSUFBd0IsTUFBTSxLQUFsQyxJQUFBLEVBQTZDO0FBQzNDLFFBQUksSUFBSSxDQUFKLFNBQUEsQ0FBQSxJQUFBLE1BQXlCLElBQUksQ0FBSixTQUFBLENBQTdCLE1BQTZCLENBQTdCLEVBQXFEO0FBQ25ELE1BQUEsTUFBTSxHQUFOLFNBQUE7QUFERixLQUFBLE1BRU8sSUFBSSxLQUFLLENBQUwsT0FBQSxDQUFKLE1BQUksQ0FBSixFQUEyQjtBQUNoQyxNQUFBLFVBQVUsQ0FBQSxPQUFBLEVBQUEsTUFBQSxFQUFBLE1BQUEsRUFBVixTQUFVLENBQVY7QUFDQSxhQUFBLE1BQUE7QUFGSyxLQUFBLE1BR0E7QUFDTCxVQUFJLElBQUksR0FBRyxJQUFBLGFBQUEsQ0FBQSxNQUFBLEVBQUEsTUFBQSxFQUFYLFNBQVcsQ0FBWDtBQUNBLGFBQU8sU0FBUyxDQUFBLE9BQUEsRUFBVCxJQUFTLENBQVQsSUFBUCxNQUFBO0FBQ0Q7QUFDRjs7QUFFRCxNQUFJLE1BQU0sS0FBVixTQUFBLEVBQTBCO0FBQ3hCLFFBQUksSUFBSSxHQUFHLHFCQUFZLElBQUksQ0FBM0IsSUFBVyxDQUFYOztBQUVBLFNBQUssSUFBSSxDQUFDLEdBQVYsQ0FBQSxFQUFnQixDQUFDLEdBQUcsSUFBSSxDQUF4QixNQUFBLEVBQWlDLENBQWpDLEVBQUEsRUFBc0M7QUFDcEMsVUFBSSxHQUFHLEdBQUcsSUFBSSxDQURzQixDQUN0QixDQUFkLENBRG9DLENBRXBDOztBQUNBLE1BQUEsUUFBUSxDQUFBLE9BQUEsRUFBQSxPQUFBLEVBQUEsSUFBQSxFQUFSLEdBQVEsQ0FBUjtBQUNEOztBQUVELFFBQUksSUFBSSxLQUFSLFNBQUEsRUFBd0I7QUFDdEIsTUFBQSxNQUFNLEdBQUcsSUFBSSxDQUFBLElBQUEsRUFBYixJQUFhLENBQWI7QUFDRDtBQUNGOztBQUVELFNBQUEsTUFBQTtBQUNEOztBQUVELFNBQUEsR0FBQSxDQUFBLElBQUEsRUFBQSxHQUFBLEVBRXVDO0FBRXJDLFNBQVEsSUFBSSxDQUFaLEdBQVksQ0FBWjtBQUNEOztBQUVELFNBQUEsR0FBQSxDQUFBLElBQUEsRUFBQSxHQUFBLEVBQUEsS0FBQSxFQUFrRjtBQUNoRixFQUFBLElBQUksQ0FBSixHQUFJLENBQUosR0FBQSxLQUFBO0FBQ0Q7O0FBRUQsU0FBQSxRQUFBLENBQUEsT0FBQSxFQUFBLE9BQUEsRUFBQSxJQUFBLEVBQUEsR0FBQSxFQUl1QztBQUVyQyxNQUFJO0FBQUUsSUFBQTtBQUFGLE1BQUosSUFBQTtBQUVBLE1BQUksS0FBSyxHQUFHLEdBQUcsQ0FBQSxJQUFBLEVBQWYsR0FBZSxDQUFmOztBQUNBLE1BQUksQ0FBSixLQUFBLEVBQVk7QUFDVjtBQUNEOztBQUVELE1BQUEsUUFBQTtBQUNBLE1BQUEsT0FBQTs7QUFFQSxNQUFJLE9BQU8sS0FBWCxTQUFBLEVBQTJCO0FBQ3pCLFFBQUksVUFBVSxHQUFHLGFBQWEsQ0FBQSxPQUFBLEVBQTlCLEdBQThCLENBQTlCOztBQUNBLFFBQUksVUFBVSxLQUFkLFNBQUEsRUFBOEI7QUFDNUIsTUFBQSxRQUFRLEdBQUcsZ0JBQWdCLENBQTNCLFVBQTJCLENBQTNCO0FBQ0EsTUFBQSxPQUFPLEdBQUcsZUFBZSxDQUF6QixVQUF5QixDQUF6QjtBQUNEO0FBQ0Y7O0FBRUQsTUFBSSxRQUFRLEtBQVosU0FBQSxFQUE0QjtBQUMxQixRQUFJLFFBQVEsQ0FBQSxJQUFBLEVBQVIsR0FBUSxDQUFSLEtBQUosU0FBQSxFQUF1QztBQUNyQyxZQUFNLGtEQUFvQyxJQUFwQyxFQUFOLEdBQU0sQ0FBTjtBQUNEO0FBQ0Y7O0FBRUQsTUFBSSxLQUFLLENBQUwsT0FBQSxDQUFKLEtBQUksQ0FBSixFQUEwQjtBQUN4QixJQUFBLFVBQVUsQ0FBQSxPQUFBLEVBQUEsS0FBQSxFQUFBLElBQUEsRUFBVixHQUFVLENBQVY7QUFERixHQUFBLE1BRU87QUFDTCxRQUFJLE9BQU8sR0FBRyxJQUFBLGFBQUEsQ0FBQSxLQUFBLEVBQUEsSUFBQSxFQUFkLEdBQWMsQ0FBZDtBQUNBLFFBQUksTUFBTSxHQUFHLFNBQVMsQ0FBQSxPQUFBLEVBQXRCLE9BQXNCLENBQXRCOztBQUNBLFFBQUksTUFBTSxLQUFWLFNBQUEsRUFBMEI7QUFDeEI7QUFDQTtBQUNBO0FBQ0EsTUFBQSxTQUFTLENBQUEsSUFBQSxFQUFBLEdBQUEsRUFBQSxLQUFBLEVBQVQsTUFBUyxDQUFUO0FBQ0Q7QUFDRjs7QUFFRCxNQUFJLE9BQU8sS0FBWCxTQUFBLEVBQTJCO0FBQ3pCLFFBQUksT0FBTyxDQUFBLElBQUEsRUFBUCxHQUFPLENBQVAsS0FBSixTQUFBLEVBQXNDO0FBQ3BDLFlBQU0sa0RBQW9DLElBQXBDLEVBQU4sR0FBTSxDQUFOO0FBQ0Q7QUFDRjtBQUNGOztBQUVELFNBQUEsVUFBQSxDQUFBLE9BQUEsRUFBQSxLQUFBLEVBQUEsTUFBQSxFQUFBLFNBQUEsRUFJMEI7QUFFeEIsT0FBSyxJQUFJLENBQUMsR0FBVixDQUFBLEVBQWdCLENBQUMsR0FBRyxLQUFLLENBQXpCLE1BQUEsRUFBa0MsQ0FBbEMsRUFBQSxFQUF1QztBQUNyQyxRQUFJLElBQUksR0FBRyxLQUFLLENBQWhCLENBQWdCLENBQWhCO0FBQ0EsUUFBSSxJQUFJLEdBQUcsSUFBQSxhQUFBLENBQUEsSUFBQSxFQUFBLE1BQUEsRUFBWCxTQUFXLENBQVg7QUFDQSxRQUFJLE1BQU0sR0FBRyxTQUFTLENBQUEsT0FBQSxFQUF0QixJQUFzQixDQUF0Qjs7QUFDQSxRQUFJLE1BQU0sS0FBVixTQUFBLEVBQTBCO0FBQ3hCLE1BQUEsQ0FBQyxJQUFJLFdBQVcsQ0FBQSxLQUFBLEVBQUEsQ0FBQSxFQUFYLE1BQVcsQ0FBWCxHQUFMLENBQUE7QUFDRDtBQUNGO0FBQ0Y7O0FBRUQsU0FBQSxTQUFBLENBQUEsSUFBQSxFQUFBLEdBQUEsRUFBQSxLQUFBLEVBQUEsTUFBQSxFQUk4QjtBQUU1QixNQUFJLE1BQU0sS0FBVixJQUFBLEVBQXFCO0FBQ25CLFVBQU0sOEJBQWdCLEtBQWhCLEVBQWdCLElBQWhCLEVBQU4sR0FBTSxDQUFOO0FBREYsR0FBQSxNQUVPLElBQUksS0FBSyxDQUFMLE9BQUEsQ0FBSixNQUFJLENBQUosRUFBMkI7QUFDaEMsUUFBSSxNQUFNLENBQU4sTUFBQSxLQUFKLENBQUEsRUFBeUI7QUFDdkIsTUFBQSxHQUFHLENBQUEsSUFBQSxFQUFBLEdBQUEsRUFBWSxNQUFNLENBQXJCLENBQXFCLENBQWxCLENBQUg7QUFERixLQUFBLE1BRU87QUFDTCxVQUFJLE1BQU0sQ0FBTixNQUFBLEtBQUosQ0FBQSxFQUF5QjtBQUN2QixjQUFNLDhCQUFnQixLQUFoQixFQUFnQixJQUFoQixFQUFOLEdBQU0sQ0FBTjtBQURGLE9BQUEsTUFFTztBQUNMLGNBQU0sK0JBQWlCLEtBQWpCLEVBQWlCLElBQWpCLEVBQU4sR0FBTSxDQUFOO0FBQ0Q7QUFDRjtBQVRJLEdBQUEsTUFVQTtBQUNMLElBQUEsR0FBRyxDQUFBLElBQUEsRUFBQSxHQUFBLEVBQUgsTUFBRyxDQUFIO0FBQ0Q7QUFDRjs7QUFFRCxTQUFBLFdBQUEsQ0FBQSxLQUFBLEVBQUEsS0FBQSxFQUFBLE1BQUEsRUFBaUc7QUFDL0YsTUFBSSxNQUFNLEtBQVYsSUFBQSxFQUFxQjtBQUNuQixJQUFBLEtBQUssQ0FBTCxNQUFBLENBQUEsS0FBQSxFQUFBLENBQUE7QUFDQSxXQUFBLENBQUE7QUFGRixHQUFBLE1BR08sSUFBSSxLQUFLLENBQUwsT0FBQSxDQUFKLE1BQUksQ0FBSixFQUEyQjtBQUNoQyxJQUFBLEtBQUssQ0FBTCxNQUFBLENBQUEsS0FBQSxFQUFBLENBQUEsRUFBdUIsR0FBdkIsTUFBQTtBQUNBLFdBQU8sTUFBTSxDQUFiLE1BQUE7QUFGSyxHQUFBLE1BR0E7QUFDTCxJQUFBLEtBQUssQ0FBTCxNQUFBLENBQUEsS0FBQSxFQUFBLENBQUEsRUFBQSxNQUFBO0FBQ0EsV0FBQSxDQUFBO0FBQ0Q7QUFDRjs7QUFFYSxTQUFBLFFBQUEsQ0FBQSxJQUFBLEVBQUEsT0FBQSxFQUF5RDtBQUNyRSxNQUFJLElBQUksR0FBRyxJQUFBLGFBQUEsQ0FBWCxJQUFXLENBQVg7QUFDQSxFQUFBLFNBQVMsQ0FBQSxPQUFBLEVBQVQsSUFBUyxDQUFUO0FBQ0QiLCJzb3VyY2VzQ29udGVudCI6WyJpbXBvcnQgeyBMT0NBTF9ERUJVRyB9IGZyb20gJ0BnbGltbWVyL2xvY2FsLWRlYnVnLWZsYWdzJztcbmltcG9ydCB7IGRlcHJlY2F0ZSB9IGZyb20gJ0BnbGltbWVyL3V0aWwnO1xuXG5pbXBvcnQgKiBhcyBBU1R2MSBmcm9tICcuLi92MS9hcGknO1xuaW1wb3J0IHZpc2l0b3JLZXlzLCB7IFZpc2l0b3JLZXksIFZpc2l0b3JLZXlzIH0gZnJvbSAnLi4vdjEvdmlzaXRvci1rZXlzJztcbmltcG9ydCB7XG4gIGNhbm5vdFJlbW92ZU5vZGUsXG4gIGNhbm5vdFJlcGxhY2VOb2RlLFxuICBjYW5ub3RSZXBsYWNlT3JSZW1vdmVJbktleUhhbmRsZXJZZXQsXG59IGZyb20gJy4vZXJyb3JzJztcbmltcG9ydCBXYWxrZXJQYXRoIGZyb20gJy4vcGF0aCc7XG5pbXBvcnQgeyBLZXlIYW5kbGVyLCBLZXlUcmF2ZXJzYWwsIE5vZGVIYW5kbGVyLCBOb2RlVHJhdmVyc2FsLCBOb2RlVmlzaXRvciB9IGZyb20gJy4vdmlzaXRvcic7XG5cbmZ1bmN0aW9uIGdldEVudGVyRnVuY3Rpb248TiBleHRlbmRzIEFTVHYxLk5vZGU+KFxuICBoYW5kbGVyOiBOb2RlVHJhdmVyc2FsPE4+XG4pOiBOb2RlSGFuZGxlcjxOPiB8IHVuZGVmaW5lZDtcbmZ1bmN0aW9uIGdldEVudGVyRnVuY3Rpb248TiBleHRlbmRzIEFTVHYxLk5vZGUsIEsgZXh0ZW5kcyBWaXNpdG9yS2V5PE4+PihcbiAgaGFuZGxlcjogS2V5VHJhdmVyc2FsPE4sIEs+XG4pOiBLZXlIYW5kbGVyPE4sIEs+IHwgdW5kZWZpbmVkO1xuZnVuY3Rpb24gZ2V0RW50ZXJGdW5jdGlvbjxOIGV4dGVuZHMgQVNUdjEuTm9kZSwgSyBleHRlbmRzIFZpc2l0b3JLZXk8Tj4+KFxuICBoYW5kbGVyOiBOb2RlVHJhdmVyc2FsPE4+IHwgS2V5VHJhdmVyc2FsPE4sIEs+XG4pOiBOb2RlSGFuZGxlcjxOPiB8IEtleUhhbmRsZXI8TiwgSz4gfCB1bmRlZmluZWQge1xuICBpZiAodHlwZW9mIGhhbmRsZXIgPT09ICdmdW5jdGlvbicpIHtcbiAgICByZXR1cm4gaGFuZGxlcjtcbiAgfSBlbHNlIHtcbiAgICByZXR1cm4gaGFuZGxlci5lbnRlciBhcyBOb2RlSGFuZGxlcjxOPiB8IEtleUhhbmRsZXI8TiwgSz47XG4gIH1cbn1cblxuZnVuY3Rpb24gZ2V0RXhpdEZ1bmN0aW9uPE4gZXh0ZW5kcyBBU1R2MS5Ob2RlPihcbiAgaGFuZGxlcjogTm9kZVRyYXZlcnNhbDxOPlxuKTogTm9kZUhhbmRsZXI8Tj4gfCB1bmRlZmluZWQ7XG5mdW5jdGlvbiBnZXRFeGl0RnVuY3Rpb248TiBleHRlbmRzIEFTVHYxLk5vZGUsIEsgZXh0ZW5kcyBWaXNpdG9yS2V5PE4+PihcbiAgaGFuZGxlcjogS2V5VHJhdmVyc2FsPE4sIEs+XG4pOiBLZXlIYW5kbGVyPE4sIEs+IHwgdW5kZWZpbmVkO1xuZnVuY3Rpb24gZ2V0RXhpdEZ1bmN0aW9uPE4gZXh0ZW5kcyBBU1R2MS5Ob2RlLCBLIGV4dGVuZHMgVmlzaXRvcktleTxOPj4oXG4gIGhhbmRsZXI6IE5vZGVUcmF2ZXJzYWw8Tj4gfCBLZXlUcmF2ZXJzYWw8TiwgSz5cbik6IE5vZGVIYW5kbGVyPE4+IHwgS2V5SGFuZGxlcjxOLCBLPiB8IHVuZGVmaW5lZCB7XG4gIGlmICh0eXBlb2YgaGFuZGxlciA9PT0gJ2Z1bmN0aW9uJykge1xuICAgIHJldHVybiB1bmRlZmluZWQ7XG4gIH0gZWxzZSB7XG4gICAgcmV0dXJuIGhhbmRsZXIuZXhpdCBhcyBOb2RlSGFuZGxlcjxOPiB8IEtleUhhbmRsZXI8TiwgSz47XG4gIH1cbn1cblxuZnVuY3Rpb24gZ2V0S2V5SGFuZGxlcjxOIGV4dGVuZHMgQVNUdjEuTm9kZSwgSyBleHRlbmRzIFZpc2l0b3JLZXk8Tj4+KFxuICBoYW5kbGVyOiBOb2RlVHJhdmVyc2FsPE4+LFxuICBrZXk6IEtcbik6IEtleVRyYXZlcnNhbDxOLCBLPiB8IEtleVRyYXZlcnNhbDxOLCBWaXNpdG9yS2V5PE4+PiB8IHVuZGVmaW5lZCB7XG4gIGxldCBrZXlWaXNpdG9yID0gdHlwZW9mIGhhbmRsZXIgIT09ICdmdW5jdGlvbicgPyBoYW5kbGVyLmtleXMgOiB1bmRlZmluZWQ7XG4gIGlmIChrZXlWaXNpdG9yID09PSB1bmRlZmluZWQpIHJldHVybjtcblxuICBsZXQga2V5SGFuZGxlciA9IGtleVZpc2l0b3Jba2V5XTtcbiAgaWYgKGtleUhhbmRsZXIgIT09IHVuZGVmaW5lZCkge1xuICAgIHJldHVybiBrZXlIYW5kbGVyIGFzIEtleVRyYXZlcnNhbDxOLCBLPjtcbiAgfVxuICByZXR1cm4ga2V5VmlzaXRvci5BbGw7XG59XG5cbmZ1bmN0aW9uIGdldE5vZGVIYW5kbGVyPE4gZXh0ZW5kcyBBU1R2MS5Ob2RlPihcbiAgdmlzaXRvcjogTm9kZVZpc2l0b3IsXG4gIG5vZGVUeXBlOiBOWyd0eXBlJ11cbik6IE5vZGVUcmF2ZXJzYWw8Tj47XG5mdW5jdGlvbiBnZXROb2RlSGFuZGxlcih2aXNpdG9yOiBOb2RlVmlzaXRvciwgbm9kZVR5cGU6ICdBbGwnKTogTm9kZVRyYXZlcnNhbDxBU1R2MS5Ob2RlPjtcbmZ1bmN0aW9uIGdldE5vZGVIYW5kbGVyPE4gZXh0ZW5kcyBBU1R2MS5Ob2RlPihcbiAgdmlzaXRvcjogTm9kZVZpc2l0b3IsXG4gIG5vZGVUeXBlOiBOWyd0eXBlJ11cbik6IE5vZGVUcmF2ZXJzYWw8QVNUdjEuTm9kZT4gfCB1bmRlZmluZWQge1xuICBpZiAobm9kZVR5cGUgPT09ICdUZW1wbGF0ZScgfHwgbm9kZVR5cGUgPT09ICdCbG9jaycpIHtcbiAgICBpZiAodmlzaXRvci5Qcm9ncmFtKSB7XG4gICAgICBpZiAoTE9DQUxfREVCVUcpIHtcbiAgICAgICAgZGVwcmVjYXRlKFxuICAgICAgICAgIGBUaGUgJ1Byb2dyYW0nIHZpc2l0b3Igbm9kZSBpcyBkZXByZWNhdGVkLiBVc2UgJ1RlbXBsYXRlJyBvciAnQmxvY2snIGluc3RlYWQgKG5vZGUgd2FzICcke25vZGVUeXBlfScpIGBcbiAgICAgICAgKTtcbiAgICAgIH1cblxuICAgICAgcmV0dXJuIHZpc2l0b3IuUHJvZ3JhbSBhcyBOb2RlVHJhdmVyc2FsPEFTVHYxLk5vZGU+O1xuICAgIH1cbiAgfVxuXG4gIGxldCBoYW5kbGVyID0gdmlzaXRvcltub2RlVHlwZV07XG4gIGlmIChoYW5kbGVyICE9PSB1bmRlZmluZWQpIHtcbiAgICByZXR1cm4gKGhhbmRsZXIgYXMgdW5rbm93bikgYXMgTm9kZVRyYXZlcnNhbDxBU1R2MS5Ob2RlPjtcbiAgfVxuICByZXR1cm4gdmlzaXRvci5BbGw7XG59XG5cbmZ1bmN0aW9uIHZpc2l0Tm9kZTxOIGV4dGVuZHMgQVNUdjEuTm9kZT4oXG4gIHZpc2l0b3I6IE5vZGVWaXNpdG9yLFxuICBwYXRoOiBXYWxrZXJQYXRoPE4+XG4pOiBBU1R2MS5Ob2RlIHwgQVNUdjEuTm9kZVtdIHwgdW5kZWZpbmVkIHwgbnVsbCB8IHZvaWQge1xuICBsZXQgeyBub2RlLCBwYXJlbnQsIHBhcmVudEtleSB9ID0gcGF0aDtcblxuICBsZXQgaGFuZGxlcjogTm9kZVRyYXZlcnNhbDxOPiA9IGdldE5vZGVIYW5kbGVyKHZpc2l0b3IsIG5vZGUudHlwZSk7XG4gIGxldCBlbnRlcjtcbiAgbGV0IGV4aXQ7XG5cbiAgaWYgKGhhbmRsZXIgIT09IHVuZGVmaW5lZCkge1xuICAgIGVudGVyID0gZ2V0RW50ZXJGdW5jdGlvbihoYW5kbGVyKTtcbiAgICBleGl0ID0gZ2V0RXhpdEZ1bmN0aW9uKGhhbmRsZXIpO1xuICB9XG5cbiAgbGV0IHJlc3VsdDogQVNUdjEuTm9kZSB8IEFTVHYxLk5vZGVbXSB8IHVuZGVmaW5lZCB8IG51bGwgfCB2b2lkO1xuICBpZiAoZW50ZXIgIT09IHVuZGVmaW5lZCkge1xuICAgIHJlc3VsdCA9IGVudGVyKG5vZGUsIHBhdGgpO1xuICB9XG5cbiAgaWYgKHJlc3VsdCAhPT0gdW5kZWZpbmVkICYmIHJlc3VsdCAhPT0gbnVsbCkge1xuICAgIGlmIChKU09OLnN0cmluZ2lmeShub2RlKSA9PT0gSlNPTi5zdHJpbmdpZnkocmVzdWx0KSkge1xuICAgICAgcmVzdWx0ID0gdW5kZWZpbmVkO1xuICAgIH0gZWxzZSBpZiAoQXJyYXkuaXNBcnJheShyZXN1bHQpKSB7XG4gICAgICB2aXNpdEFycmF5KHZpc2l0b3IsIHJlc3VsdCwgcGFyZW50LCBwYXJlbnRLZXkpO1xuICAgICAgcmV0dXJuIHJlc3VsdDtcbiAgICB9IGVsc2Uge1xuICAgICAgbGV0IHBhdGggPSBuZXcgV2Fsa2VyUGF0aChyZXN1bHQsIHBhcmVudCwgcGFyZW50S2V5KTtcbiAgICAgIHJldHVybiB2aXNpdE5vZGUodmlzaXRvciwgcGF0aCkgfHwgcmVzdWx0O1xuICAgIH1cbiAgfVxuXG4gIGlmIChyZXN1bHQgPT09IHVuZGVmaW5lZCkge1xuICAgIGxldCBrZXlzID0gdmlzaXRvcktleXNbbm9kZS50eXBlXTtcblxuICAgIGZvciAobGV0IGkgPSAwOyBpIDwga2V5cy5sZW5ndGg7IGkrKykge1xuICAgICAgbGV0IGtleSA9IGtleXNbaV0gYXMgVmlzaXRvcktleXNbTlsndHlwZSddXSAmIGtleW9mIE47XG4gICAgICAvLyB3ZSBrbm93IGlmIGl0IGhhcyBjaGlsZCBrZXlzIHdlIGNhbiB3aWRlbiB0byBhIFBhcmVudE5vZGVcbiAgICAgIHZpc2l0S2V5KHZpc2l0b3IsIGhhbmRsZXIsIHBhdGgsIGtleSk7XG4gICAgfVxuXG4gICAgaWYgKGV4aXQgIT09IHVuZGVmaW5lZCkge1xuICAgICAgcmVzdWx0ID0gZXhpdChub2RlLCBwYXRoKTtcbiAgICB9XG4gIH1cblxuICByZXR1cm4gcmVzdWx0O1xufVxuXG5mdW5jdGlvbiBnZXQ8TiBleHRlbmRzIEFTVHYxLk5vZGU+KFxuICBub2RlOiBOLFxuICBrZXk6IFZpc2l0b3JLZXlzW05bJ3R5cGUnXV0gJiBrZXlvZiBOXG4pOiBBU1R2MS5Ob2RlIHwgQVNUdjEuTm9kZVtdIHtcbiAgcmV0dXJuIChub2RlW2tleV0gYXMgdW5rbm93bikgYXMgQVNUdjEuTm9kZSB8IEFTVHYxLk5vZGVbXTtcbn1cblxuZnVuY3Rpb24gc2V0PE4gZXh0ZW5kcyBBU1R2MS5Ob2RlLCBLIGV4dGVuZHMga2V5b2YgTj4obm9kZTogTiwga2V5OiBLLCB2YWx1ZTogTltLXSk6IHZvaWQge1xuICBub2RlW2tleV0gPSB2YWx1ZTtcbn1cblxuZnVuY3Rpb24gdmlzaXRLZXk8TiBleHRlbmRzIEFTVHYxLk5vZGU+KFxuICB2aXNpdG9yOiBOb2RlVmlzaXRvcixcbiAgaGFuZGxlcjogTm9kZVRyYXZlcnNhbDxOPixcbiAgcGF0aDogV2Fsa2VyUGF0aDxOPixcbiAga2V5OiBWaXNpdG9yS2V5c1tOWyd0eXBlJ11dICYga2V5b2YgTlxuKSB7XG4gIGxldCB7IG5vZGUgfSA9IHBhdGg7XG5cbiAgbGV0IHZhbHVlID0gZ2V0KG5vZGUsIGtleSk7XG4gIGlmICghdmFsdWUpIHtcbiAgICByZXR1cm47XG4gIH1cblxuICBsZXQga2V5RW50ZXI7XG4gIGxldCBrZXlFeGl0O1xuXG4gIGlmIChoYW5kbGVyICE9PSB1bmRlZmluZWQpIHtcbiAgICBsZXQga2V5SGFuZGxlciA9IGdldEtleUhhbmRsZXIoaGFuZGxlciwga2V5KTtcbiAgICBpZiAoa2V5SGFuZGxlciAhPT0gdW5kZWZpbmVkKSB7XG4gICAgICBrZXlFbnRlciA9IGdldEVudGVyRnVuY3Rpb24oa2V5SGFuZGxlcik7XG4gICAgICBrZXlFeGl0ID0gZ2V0RXhpdEZ1bmN0aW9uKGtleUhhbmRsZXIpO1xuICAgIH1cbiAgfVxuXG4gIGlmIChrZXlFbnRlciAhPT0gdW5kZWZpbmVkKSB7XG4gICAgaWYgKGtleUVudGVyKG5vZGUsIGtleSkgIT09IHVuZGVmaW5lZCkge1xuICAgICAgdGhyb3cgY2Fubm90UmVwbGFjZU9yUmVtb3ZlSW5LZXlIYW5kbGVyWWV0KG5vZGUsIGtleSk7XG4gICAgfVxuICB9XG5cbiAgaWYgKEFycmF5LmlzQXJyYXkodmFsdWUpKSB7XG4gICAgdmlzaXRBcnJheSh2aXNpdG9yLCB2YWx1ZSwgcGF0aCwga2V5KTtcbiAgfSBlbHNlIHtcbiAgICBsZXQga2V5UGF0aCA9IG5ldyBXYWxrZXJQYXRoKHZhbHVlLCBwYXRoLCBrZXkpO1xuICAgIGxldCByZXN1bHQgPSB2aXNpdE5vZGUodmlzaXRvciwga2V5UGF0aCk7XG4gICAgaWYgKHJlc3VsdCAhPT0gdW5kZWZpbmVkKSB7XG4gICAgICAvLyBUT0RPOiBkeW5hbWljYWxseSBjaGVjayB0aGUgcmVzdWx0cyBieSBoYXZpbmcgYSB0YWJsZSBvZlxuICAgICAgLy8gZXhwZWN0ZWQgbm9kZSB0eXBlcyBpbiB2YWx1ZSBzcGFjZSwgbm90IGp1c3QgdHlwZSBzcGFjZVxuICAgICAgLy8gZXNsaW50LWRpc2FibGUtbmV4dC1saW5lIEB0eXBlc2NyaXB0LWVzbGludC9uby1leHBsaWNpdC1hbnlcbiAgICAgIGFzc2lnbktleShub2RlLCBrZXksIHZhbHVlLCByZXN1bHQgYXMgYW55KTtcbiAgICB9XG4gIH1cblxuICBpZiAoa2V5RXhpdCAhPT0gdW5kZWZpbmVkKSB7XG4gICAgaWYgKGtleUV4aXQobm9kZSwga2V5KSAhPT0gdW5kZWZpbmVkKSB7XG4gICAgICB0aHJvdyBjYW5ub3RSZXBsYWNlT3JSZW1vdmVJbktleUhhbmRsZXJZZXQobm9kZSwga2V5KTtcbiAgICB9XG4gIH1cbn1cblxuZnVuY3Rpb24gdmlzaXRBcnJheShcbiAgdmlzaXRvcjogTm9kZVZpc2l0b3IsXG4gIGFycmF5OiBBU1R2MS5Ob2RlW10sXG4gIHBhcmVudDogV2Fsa2VyUGF0aDxBU1R2MS5Ob2RlPiB8IG51bGwsXG4gIHBhcmVudEtleTogc3RyaW5nIHwgbnVsbFxuKSB7XG4gIGZvciAobGV0IGkgPSAwOyBpIDwgYXJyYXkubGVuZ3RoOyBpKyspIHtcbiAgICBsZXQgbm9kZSA9IGFycmF5W2ldO1xuICAgIGxldCBwYXRoID0gbmV3IFdhbGtlclBhdGgobm9kZSwgcGFyZW50LCBwYXJlbnRLZXkpO1xuICAgIGxldCByZXN1bHQgPSB2aXNpdE5vZGUodmlzaXRvciwgcGF0aCk7XG4gICAgaWYgKHJlc3VsdCAhPT0gdW5kZWZpbmVkKSB7XG4gICAgICBpICs9IHNwbGljZUFycmF5KGFycmF5LCBpLCByZXN1bHQpIC0gMTtcbiAgICB9XG4gIH1cbn1cblxuZnVuY3Rpb24gYXNzaWduS2V5PE4gZXh0ZW5kcyBBU1R2MS5Ob2RlLCBLIGV4dGVuZHMgVmlzaXRvcktleTxOPj4oXG4gIG5vZGU6IE4sXG4gIGtleTogSyxcbiAgdmFsdWU6IEFTVHYxLk5vZGUsXG4gIHJlc3VsdDogTltLXSB8IFtOW0tdXSB8IG51bGxcbikge1xuICBpZiAocmVzdWx0ID09PSBudWxsKSB7XG4gICAgdGhyb3cgY2Fubm90UmVtb3ZlTm9kZSh2YWx1ZSwgbm9kZSwga2V5KTtcbiAgfSBlbHNlIGlmIChBcnJheS5pc0FycmF5KHJlc3VsdCkpIHtcbiAgICBpZiAocmVzdWx0Lmxlbmd0aCA9PT0gMSkge1xuICAgICAgc2V0KG5vZGUsIGtleSwgcmVzdWx0WzBdKTtcbiAgICB9IGVsc2Uge1xuICAgICAgaWYgKHJlc3VsdC5sZW5ndGggPT09IDApIHtcbiAgICAgICAgdGhyb3cgY2Fubm90UmVtb3ZlTm9kZSh2YWx1ZSwgbm9kZSwga2V5KTtcbiAgICAgIH0gZWxzZSB7XG4gICAgICAgIHRocm93IGNhbm5vdFJlcGxhY2VOb2RlKHZhbHVlLCBub2RlLCBrZXkpO1xuICAgICAgfVxuICAgIH1cbiAgfSBlbHNlIHtcbiAgICBzZXQobm9kZSwga2V5LCByZXN1bHQpO1xuICB9XG59XG5cbmZ1bmN0aW9uIHNwbGljZUFycmF5KGFycmF5OiBBU1R2MS5Ob2RlW10sIGluZGV4OiBudW1iZXIsIHJlc3VsdDogQVNUdjEuTm9kZSB8IEFTVHYxLk5vZGVbXSB8IG51bGwpIHtcbiAgaWYgKHJlc3VsdCA9PT0gbnVsbCkge1xuICAgIGFycmF5LnNwbGljZShpbmRleCwgMSk7XG4gICAgcmV0dXJuIDA7XG4gIH0gZWxzZSBpZiAoQXJyYXkuaXNBcnJheShyZXN1bHQpKSB7XG4gICAgYXJyYXkuc3BsaWNlKGluZGV4LCAxLCAuLi5yZXN1bHQpO1xuICAgIHJldHVybiByZXN1bHQubGVuZ3RoO1xuICB9IGVsc2Uge1xuICAgIGFycmF5LnNwbGljZShpbmRleCwgMSwgcmVzdWx0KTtcbiAgICByZXR1cm4gMTtcbiAgfVxufVxuXG5leHBvcnQgZGVmYXVsdCBmdW5jdGlvbiB0cmF2ZXJzZShub2RlOiBBU1R2MS5Ob2RlLCB2aXNpdG9yOiBOb2RlVmlzaXRvcik6IHZvaWQge1xuICBsZXQgcGF0aCA9IG5ldyBXYWxrZXJQYXRoKG5vZGUpO1xuICB2aXNpdE5vZGUodmlzaXRvciwgcGF0aCk7XG59XG4iXSwic291cmNlUm9vdCI6IiJ9