| 1 | /**
|
|---|
| 2 | * @license
|
|---|
| 3 | * Copyright Google LLC All Rights Reserved.
|
|---|
| 4 | *
|
|---|
| 5 | * Use of this source code is governed by an MIT-style license that can be
|
|---|
| 6 | * found in the LICENSE file at https://angular.io/license
|
|---|
| 7 | */
|
|---|
| 8 | import { __awaiter } from "tslib";
|
|---|
| 9 | import { ComponentHarness, HarnessPredicate, parallel } from '@angular/cdk/testing';
|
|---|
| 10 | import { MatTreeNodeHarness } from './node-harness';
|
|---|
| 11 | /** Harness for interacting with a standard mat-tree in tests. */
|
|---|
| 12 | export class MatTreeHarness extends ComponentHarness {
|
|---|
| 13 | /**
|
|---|
| 14 | * Gets a `HarnessPredicate` that can be used to search for a tree with specific attributes.
|
|---|
| 15 | * @param options Options for narrowing the search
|
|---|
| 16 | * @return a `HarnessPredicate` configured with the given options.
|
|---|
| 17 | */
|
|---|
| 18 | static with(options = {}) {
|
|---|
| 19 | return new HarnessPredicate(MatTreeHarness, options);
|
|---|
| 20 | }
|
|---|
| 21 | /** Gets all of the nodes in the tree. */
|
|---|
| 22 | getNodes(filter = {}) {
|
|---|
| 23 | return __awaiter(this, void 0, void 0, function* () {
|
|---|
| 24 | return this.locatorForAll(MatTreeNodeHarness.with(filter))();
|
|---|
| 25 | });
|
|---|
| 26 | }
|
|---|
| 27 | /**
|
|---|
| 28 | * Gets an object representation for the visible tree structure
|
|---|
| 29 | * If a node is under an unexpanded node it will not be included.
|
|---|
| 30 | * Eg.
|
|---|
| 31 | * Tree (all nodes expanded):
|
|---|
| 32 | * `
|
|---|
| 33 | * <mat-tree>
|
|---|
| 34 | * <mat-tree-node>Node 1<mat-tree-node>
|
|---|
| 35 | * <mat-nested-tree-node>
|
|---|
| 36 | * Node 2
|
|---|
| 37 | * <mat-nested-tree-node>
|
|---|
| 38 | * Node 2.1
|
|---|
| 39 | * <mat-tree-node>
|
|---|
| 40 | * Node 2.1.1
|
|---|
| 41 | * <mat-tree-node>
|
|---|
| 42 | * <mat-nested-tree-node>
|
|---|
| 43 | * <mat-tree-node>
|
|---|
| 44 | * Node 2.2
|
|---|
| 45 | * <mat-tree-node>
|
|---|
| 46 | * <mat-nested-tree-node>
|
|---|
| 47 | * </mat-tree>`
|
|---|
| 48 | *
|
|---|
| 49 | * Tree structure:
|
|---|
| 50 | * {
|
|---|
| 51 | * children: [
|
|---|
| 52 | * {
|
|---|
| 53 | * text: 'Node 1',
|
|---|
| 54 | * children: [
|
|---|
| 55 | * {
|
|---|
| 56 | * text: 'Node 2',
|
|---|
| 57 | * children: [
|
|---|
| 58 | * {
|
|---|
| 59 | * text: 'Node 2.1',
|
|---|
| 60 | * children: [{text: 'Node 2.1.1'}]
|
|---|
| 61 | * },
|
|---|
| 62 | * {text: 'Node 2.2'}
|
|---|
| 63 | * ]
|
|---|
| 64 | * }
|
|---|
| 65 | * ]
|
|---|
| 66 | * }
|
|---|
| 67 | * ]
|
|---|
| 68 | * };
|
|---|
| 69 | */
|
|---|
| 70 | getTreeStructure() {
|
|---|
| 71 | return __awaiter(this, void 0, void 0, function* () {
|
|---|
| 72 | const nodes = yield this.getNodes();
|
|---|
| 73 | const nodeInformation = yield parallel(() => nodes.map(node => {
|
|---|
| 74 | return parallel(() => [node.getLevel(), node.getText(), node.isExpanded()]);
|
|---|
| 75 | }));
|
|---|
| 76 | return this._getTreeStructure(nodeInformation, 1, true);
|
|---|
| 77 | });
|
|---|
| 78 | }
|
|---|
| 79 | /**
|
|---|
| 80 | * Recursively collect the structured text of the tree nodes.
|
|---|
| 81 | * @param nodes A list of tree nodes
|
|---|
| 82 | * @param level The level of nodes that are being accounted for during this iteration
|
|---|
| 83 | * @param parentExpanded Whether the parent of the first node in param nodes is expanded
|
|---|
| 84 | */
|
|---|
| 85 | _getTreeStructure(nodes, level, parentExpanded) {
|
|---|
| 86 | var _a, _b, _c;
|
|---|
| 87 | const result = {};
|
|---|
| 88 | for (let i = 0; i < nodes.length; i++) {
|
|---|
| 89 | const [nodeLevel, text, expanded] = nodes[i];
|
|---|
| 90 | const nextNodeLevel = (_b = (_a = nodes[i + 1]) === null || _a === void 0 ? void 0 : _a[0]) !== null && _b !== void 0 ? _b : -1;
|
|---|
| 91 | // Return the accumulated value for the current level once we reach a shallower level node
|
|---|
| 92 | if (nodeLevel < level) {
|
|---|
| 93 | return result;
|
|---|
| 94 | }
|
|---|
| 95 | // Skip deeper level nodes during this iteration, they will be picked up in a later iteration
|
|---|
| 96 | if (nodeLevel > level) {
|
|---|
| 97 | continue;
|
|---|
| 98 | }
|
|---|
| 99 | // Only add to representation if it is visible (parent is expanded)
|
|---|
| 100 | if (parentExpanded) {
|
|---|
| 101 | // Collect the data under this node according to the following rules:
|
|---|
| 102 | // 1. If the next node in the list is a sibling of the current node add it to the child list
|
|---|
| 103 | // 2. If the next node is a child of the current node, get the sub-tree structure for the
|
|---|
| 104 | // child and add it under this node
|
|---|
| 105 | // 3. If the next node has a shallower level, we've reached the end of the child nodes for
|
|---|
| 106 | // the current parent.
|
|---|
| 107 | if (nextNodeLevel === level) {
|
|---|
| 108 | this._addChildToNode(result, { text });
|
|---|
| 109 | }
|
|---|
| 110 | else if (nextNodeLevel > level) {
|
|---|
| 111 | let children = (_c = this._getTreeStructure(nodes.slice(i + 1), nextNodeLevel, expanded)) === null || _c === void 0 ? void 0 : _c.children;
|
|---|
| 112 | let child = children ? { text, children } : { text };
|
|---|
| 113 | this._addChildToNode(result, child);
|
|---|
| 114 | }
|
|---|
| 115 | else {
|
|---|
| 116 | this._addChildToNode(result, { text });
|
|---|
| 117 | return result;
|
|---|
| 118 | }
|
|---|
| 119 | }
|
|---|
| 120 | }
|
|---|
| 121 | return result;
|
|---|
| 122 | }
|
|---|
| 123 | _addChildToNode(result, child) {
|
|---|
| 124 | result.children ? result.children.push(child) : result.children = [child];
|
|---|
| 125 | }
|
|---|
| 126 | }
|
|---|
| 127 | /** The selector for the host element of a `MatTableHarness` instance. */
|
|---|
| 128 | MatTreeHarness.hostSelector = '.mat-tree';
|
|---|
| 129 | //# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidHJlZS1oYXJuZXNzLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiLi4vLi4vLi4vLi4vLi4vLi4vLi4vc3JjL21hdGVyaWFsL3RyZWUvdGVzdGluZy90cmVlLWhhcm5lc3MudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6IkFBQUE7Ozs7OztHQU1HOztBQUVILE9BQU8sRUFBQyxnQkFBZ0IsRUFBRSxnQkFBZ0IsRUFBRSxRQUFRLEVBQUMsTUFBTSxzQkFBc0IsQ0FBQztBQUNsRixPQUFPLEVBQUMsa0JBQWtCLEVBQUMsTUFBTSxnQkFBZ0IsQ0FBQztBQVFsRCxpRUFBaUU7QUFDakUsTUFBTSxPQUFPLGNBQWUsU0FBUSxnQkFBZ0I7SUFJbEQ7Ozs7T0FJRztJQUNILE1BQU0sQ0FBQyxJQUFJLENBQUMsVUFBOEIsRUFBRTtRQUMxQyxPQUFPLElBQUksZ0JBQWdCLENBQUMsY0FBYyxFQUFFLE9BQU8sQ0FBQyxDQUFDO0lBQ3ZELENBQUM7SUFFRCx5Q0FBeUM7SUFDbkMsUUFBUSxDQUFDLFNBQWlDLEVBQUU7O1lBQ2hELE9BQU8sSUFBSSxDQUFDLGFBQWEsQ0FBQyxrQkFBa0IsQ0FBQyxJQUFJLENBQUMsTUFBTSxDQUFDLENBQUMsRUFBRSxDQUFDO1FBQy9ELENBQUM7S0FBQTtJQUVEOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7T0EwQ0c7SUFDRyxnQkFBZ0I7O1lBQ3BCLE1BQU0sS0FBSyxHQUFHLE1BQU0sSUFBSSxDQUFDLFFBQVEsRUFBRSxDQUFDO1lBQ3BDLE1BQU0sZUFBZSxHQUFHLE1BQU0sUUFBUSxDQUFDLEdBQUcsRUFBRSxDQUFDLEtBQUssQ0FBQyxHQUFHLENBQUMsSUFBSSxDQUFDLEVBQUU7Z0JBQzVELE9BQU8sUUFBUSxDQUFDLEdBQUcsRUFBRSxDQUFDLENBQUMsSUFBSSxDQUFDLFFBQVEsRUFBRSxFQUFFLElBQUksQ0FBQyxPQUFPLEVBQUUsRUFBRSxJQUFJLENBQUMsVUFBVSxFQUFFLENBQUMsQ0FBQyxDQUFDO1lBQzlFLENBQUMsQ0FBQyxDQUFDLENBQUM7WUFDSixPQUFPLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxlQUFlLEVBQUUsQ0FBQyxFQUFFLElBQUksQ0FBQyxDQUFDO1FBQzFELENBQUM7S0FBQTtJQUVEOzs7OztPQUtHO0lBQ0ssaUJBQWlCLENBQUMsS0FBa0MsRUFBRSxLQUFhLEVBQ3hDLGNBQXVCOztRQUN4RCxNQUFNLE1BQU0sR0FBYSxFQUFFLENBQUM7UUFDNUIsS0FBSyxJQUFJLENBQUMsR0FBRyxDQUFDLEVBQUUsQ0FBQyxHQUFHLEtBQUssQ0FBQyxNQUFNLEVBQUUsQ0FBQyxFQUFFLEVBQUU7WUFDckMsTUFBTSxDQUFDLFNBQVMsRUFBRSxJQUFJLEVBQUUsUUFBUSxDQUFDLEdBQUcsS0FBSyxDQUFDLENBQUMsQ0FBQyxDQUFDO1lBQzdDLE1BQU0sYUFBYSxHQUFHLE1BQUEsTUFBQSxLQUFLLENBQUMsQ0FBQyxHQUFHLENBQUMsQ0FBQywwQ0FBRyxDQUFDLENBQUMsbUNBQUksQ0FBQyxDQUFDLENBQUM7WUFFOUMsMEZBQTBGO1lBQzFGLElBQUksU0FBUyxHQUFHLEtBQUssRUFBRTtnQkFDckIsT0FBTyxNQUFNLENBQUM7YUFDZjtZQUNELDZGQUE2RjtZQUM3RixJQUFJLFNBQVMsR0FBRyxLQUFLLEVBQUU7Z0JBQ3JCLFNBQVM7YUFDVjtZQUNELG1FQUFtRTtZQUNuRSxJQUFJLGNBQWMsRUFBRTtnQkFDbEIscUVBQXFFO2dCQUNyRSw0RkFBNEY7Z0JBQzVGLHlGQUF5RjtnQkFDekYsc0NBQXNDO2dCQUN0QywwRkFBMEY7Z0JBQzFGLHlCQUF5QjtnQkFDekIsSUFBSSxhQUFhLEtBQUssS0FBSyxFQUFFO29CQUMzQixJQUFJLENBQUMsZUFBZSxDQUFDLE1BQU0sRUFBRSxFQUFDLElBQUksRUFBQyxDQUFDLENBQUM7aUJBQ3RDO3FCQUFNLElBQUksYUFBYSxHQUFHLEtBQUssRUFBRTtvQkFDaEMsSUFBSSxRQUFRLEdBQUcsTUFBQSxJQUFJLENBQUMsaUJBQWlCLENBQUMsS0FBSyxDQUFDLEtBQUssQ0FBQyxDQUFDLEdBQUcsQ0FBQyxDQUFDLEVBQ3RELGFBQWEsRUFDYixRQUFRLENBQUMsMENBQUUsUUFBUSxDQUFDO29CQUN0QixJQUFJLEtBQUssR0FBRyxRQUFRLENBQUMsQ0FBQyxDQUFDLEVBQUMsSUFBSSxFQUFFLFFBQVEsRUFBQyxDQUFDLENBQUMsQ0FBQyxFQUFDLElBQUksRUFBQyxDQUFDO29CQUNqRCxJQUFJLENBQUMsZUFBZSxDQUFDLE1BQU0sRUFBRSxLQUFLLENBQUMsQ0FBQztpQkFDckM7cUJBQU07b0JBQ0wsSUFBSSxDQUFDLGVBQWUsQ0FBQyxNQUFNLEVBQUUsRUFBQyxJQUFJLEVBQUMsQ0FBQyxDQUFDO29CQUNyQyxPQUFPLE1BQU0sQ0FBQztpQkFDZjthQUNGO1NBQ0Y7UUFDRCxPQUFPLE1BQU0sQ0FBQztJQUNoQixDQUFDO0lBRU8sZUFBZSxDQUFDLE1BQWdCLEVBQUUsS0FBZTtRQUN2RCxNQUFNLENBQUMsUUFBUSxDQUFDLENBQUMsQ0FBQyxNQUFNLENBQUMsUUFBUSxDQUFDLElBQUksQ0FBQyxLQUFLLENBQUMsQ0FBQyxDQUFDLENBQUMsTUFBTSxDQUFDLFFBQVEsR0FBRyxDQUFDLEtBQUssQ0FBQyxDQUFDO0lBQzVFLENBQUM7O0FBcEhELHlFQUF5RTtBQUNsRSwyQkFBWSxHQUFHLFdBQVcsQ0FBQyIsInNvdXJjZXNDb250ZW50IjpbIi8qKlxuICogQGxpY2Vuc2VcbiAqIENvcHlyaWdodCBHb29nbGUgTExDIEFsbCBSaWdodHMgUmVzZXJ2ZWQuXG4gKlxuICogVXNlIG9mIHRoaXMgc291cmNlIGNvZGUgaXMgZ292ZXJuZWQgYnkgYW4gTUlULXN0eWxlIGxpY2Vuc2UgdGhhdCBjYW4gYmVcbiAqIGZvdW5kIGluIHRoZSBMSUNFTlNFIGZpbGUgYXQgaHR0cHM6Ly9hbmd1bGFyLmlvL2xpY2Vuc2VcbiAqL1xuXG5pbXBvcnQge0NvbXBvbmVudEhhcm5lc3MsIEhhcm5lc3NQcmVkaWNhdGUsIHBhcmFsbGVsfSBmcm9tICdAYW5ndWxhci9jZGsvdGVzdGluZyc7XG5pbXBvcnQge01hdFRyZWVOb2RlSGFybmVzc30gZnJvbSAnLi9ub2RlLWhhcm5lc3MnO1xuaW1wb3J0IHtUcmVlSGFybmVzc0ZpbHRlcnMsIFRyZWVOb2RlSGFybmVzc0ZpbHRlcnN9IGZyb20gJy4vdHJlZS1oYXJuZXNzLWZpbHRlcnMnO1xuXG5leHBvcnQgdHlwZSBUZXh0VHJlZSA9IHtcbiAgdGV4dD86IHN0cmluZztcbiAgY2hpbGRyZW4/OiBUZXh0VHJlZVtdO1xufTtcblxuLyoqIEhhcm5lc3MgZm9yIGludGVyYWN0aW5nIHdpdGggYSBzdGFuZGFyZCBtYXQtdHJlZSBpbiB0ZXN0cy4gKi9cbmV4cG9ydCBjbGFzcyBNYXRUcmVlSGFybmVzcyBleHRlbmRzIENvbXBvbmVudEhhcm5lc3Mge1xuICAvKiogVGhlIHNlbGVjdG9yIGZvciB0aGUgaG9zdCBlbGVtZW50IG9mIGEgYE1hdFRhYmxlSGFybmVzc2AgaW5zdGFuY2UuICovXG4gIHN0YXRpYyBob3N0U2VsZWN0b3IgPSAnLm1hdC10cmVlJztcblxuICAvKipcbiAgICogR2V0cyBhIGBIYXJuZXNzUHJlZGljYXRlYCB0aGF0IGNhbiBiZSB1c2VkIHRvIHNlYXJjaCBmb3IgYSB0cmVlIHdpdGggc3BlY2lmaWMgYXR0cmlidXRlcy5cbiAgICogQHBhcmFtIG9wdGlvbnMgT3B0aW9ucyBmb3IgbmFycm93aW5nIHRoZSBzZWFyY2hcbiAgICogQHJldHVybiBhIGBIYXJuZXNzUHJlZGljYXRlYCBjb25maWd1cmVkIHdpdGggdGhlIGdpdmVuIG9wdGlvbnMuXG4gICAqL1xuICBzdGF0aWMgd2l0aChvcHRpb25zOiBUcmVlSGFybmVzc0ZpbHRlcnMgPSB7fSk6IEhhcm5lc3NQcmVkaWNhdGU8TWF0VHJlZUhhcm5lc3M+IHtcbiAgICByZXR1cm4gbmV3IEhhcm5lc3NQcmVkaWNhdGUoTWF0VHJlZUhhcm5lc3MsIG9wdGlvbnMpO1xuICB9XG5cbiAgLyoqIEdldHMgYWxsIG9mIHRoZSBub2RlcyBpbiB0aGUgdHJlZS4gKi9cbiAgYXN5bmMgZ2V0Tm9kZXMoZmlsdGVyOiBUcmVlTm9kZUhhcm5lc3NGaWx0ZXJzID0ge30pOiBQcm9taXNlPE1hdFRyZWVOb2RlSGFybmVzc1tdPiB7XG4gICAgcmV0dXJuIHRoaXMubG9jYXRvckZvckFsbChNYXRUcmVlTm9kZUhhcm5lc3Mud2l0aChmaWx0ZXIpKSgpO1xuICB9XG5cbiAgLyoqXG4gICAqIEdldHMgYW4gb2JqZWN0IHJlcHJlc2VudGF0aW9uIGZvciB0aGUgdmlzaWJsZSB0cmVlIHN0cnVjdHVyZVxuICAgKiBJZiBhIG5vZGUgaXMgdW5kZXIgYW4gdW5leHBhbmRlZCBub2RlIGl0IHdpbGwgbm90IGJlIGluY2x1ZGVkLlxuICAgKiBFZy5cbiAgICogVHJlZSAoYWxsIG5vZGVzIGV4cGFuZGVkKTpcbiAgICogYFxuICAgKiA8bWF0LXRyZWU+XG4gICAqICAgPG1hdC10cmVlLW5vZGU+Tm9kZSAxPG1hdC10cmVlLW5vZGU+XG4gICAqICAgPG1hdC1uZXN0ZWQtdHJlZS1ub2RlPlxuICAgKiAgICAgTm9kZSAyXG4gICAqICAgICA8bWF0LW5lc3RlZC10cmVlLW5vZGU+XG4gICAqICAgICAgIE5vZGUgMi4xXG4gICAqICAgICAgIDxtYXQtdHJlZS1ub2RlPlxuICAgKiAgICAgICAgIE5vZGUgMi4xLjFcbiAgICogICAgICAgPG1hdC10cmVlLW5vZGU+XG4gICAqICAgICA8bWF0LW5lc3RlZC10cmVlLW5vZGU+XG4gICAqICAgICA8bWF0LXRyZWUtbm9kZT5cbiAgICogICAgICAgTm9kZSAyLjJcbiAgICogICAgIDxtYXQtdHJlZS1ub2RlPlxuICAgKiAgIDxtYXQtbmVzdGVkLXRyZWUtbm9kZT5cbiAgICogPC9tYXQtdHJlZT5gXG4gICAqXG4gICAqIFRyZWUgc3RydWN0dXJlOlxuICAgKiB7XG4gICAqICBjaGlsZHJlbjogW1xuICAgKiAgICB7XG4gICAqICAgICAgdGV4dDogJ05vZGUgMScsXG4gICAqICAgICAgY2hpbGRyZW46IFtcbiAgICogICAgICAgIHtcbiAgICogICAgICAgICAgdGV4dDogJ05vZGUgMicsXG4gICAqICAgICAgICAgIGNoaWxkcmVuOiBbXG4gICAqICAgICAgICAgICAge1xuICAgKiAgICAgICAgICAgICAgdGV4dDogJ05vZGUgMi4xJyxcbiAgICogICAgICAgICAgICAgIGNoaWxkcmVuOiBbe3RleHQ6ICdOb2RlIDIuMS4xJ31dXG4gICAqICAgICAgICAgICAgfSxcbiAgICogICAgICAgICAgICB7dGV4dDogJ05vZGUgMi4yJ31cbiAgICogICAgICAgICAgXVxuICAgKiAgICAgICAgfVxuICAgKiAgICAgIF1cbiAgICogICAgfVxuICAgKiAgXVxuICAgKiB9O1xuICAgKi9cbiAgYXN5bmMgZ2V0VHJlZVN0cnVjdHVyZSgpOiBQcm9taXNlPFRleHRUcmVlPiB7XG4gICAgY29uc3Qgbm9kZXMgPSBhd2FpdCB0aGlzLmdldE5vZGVzKCk7XG4gICAgY29uc3Qgbm9kZUluZm9ybWF0aW9uID0gYXdhaXQgcGFyYWxsZWwoKCkgPT4gbm9kZXMubWFwKG5vZGUgPT4ge1xuICAgICAgcmV0dXJuIHBhcmFsbGVsKCgpID0+IFtub2RlLmdldExldmVsKCksIG5vZGUuZ2V0VGV4dCgpLCBub2RlLmlzRXhwYW5kZWQoKV0pO1xuICAgIH0pKTtcbiAgICByZXR1cm4gdGhpcy5fZ2V0VHJlZVN0cnVjdHVyZShub2RlSW5mb3JtYXRpb24sIDEsIHRydWUpO1xuICB9XG5cbiAgLyoqXG4gICAqIFJlY3Vyc2l2ZWx5IGNvbGxlY3QgdGhlIHN0cnVjdHVyZWQgdGV4dCBvZiB0aGUgdHJlZSBub2Rlcy5cbiAgICogQHBhcmFtIG5vZGVzIEEgbGlzdCBvZiB0cmVlIG5vZGVzXG4gICAqIEBwYXJhbSBsZXZlbCBUaGUgbGV2ZWwgb2Ygbm9kZXMgdGhhdCBhcmUgYmVpbmcgYWNjb3VudGVkIGZvciBkdXJpbmcgdGhpcyBpdGVyYXRpb25cbiAgICogQHBhcmFtIHBhcmVudEV4cGFuZGVkIFdoZXRoZXIgdGhlIHBhcmVudCBvZiB0aGUgZmlyc3Qgbm9kZSBpbiBwYXJhbSBub2RlcyBpcyBleHBhbmRlZFxuICAgKi9cbiAgcHJpdmF0ZSBfZ2V0VHJlZVN0cnVjdHVyZShub2RlczogW251bWJlciwgc3RyaW5nLCBib29sZWFuXVtdLCBsZXZlbDogbnVtYmVyLFxuICAgICAgICAgICAgICAgICAgICAgICAgICAgICAgICAgICAgIHBhcmVudEV4cGFuZGVkOiBib29sZWFuKTogVGV4dFRyZWUge1xuICAgIGNvbnN0IHJlc3VsdDogVGV4dFRyZWUgPSB7fTtcbiAgICBmb3IgKGxldCBpID0gMDsgaSA8IG5vZGVzLmxlbmd0aDsgaSsrKSB7XG4gICAgICBjb25zdCBbbm9kZUxldmVsLCB0ZXh0LCBleHBhbmRlZF0gPSBub2Rlc1tpXTtcbiAgICAgIGNvbnN0IG5leHROb2RlTGV2ZWwgPSBub2Rlc1tpICsgMV0/LlswXSA/PyAtMTtcblxuICAgICAgLy8gUmV0dXJuIHRoZSBhY2N1bXVsYXRlZCB2YWx1ZSBmb3IgdGhlIGN1cnJlbnQgbGV2ZWwgb25jZSB3ZSByZWFjaCBhIHNoYWxsb3dlciBsZXZlbCBub2RlXG4gICAgICBpZiAobm9kZUxldmVsIDwgbGV2ZWwpIHtcbiAgICAgICAgcmV0dXJuIHJlc3VsdDtcbiAgICAgIH1cbiAgICAgIC8vIFNraXAgZGVlcGVyIGxldmVsIG5vZGVzIGR1cmluZyB0aGlzIGl0ZXJhdGlvbiwgdGhleSB3aWxsIGJlIHBpY2tlZCB1cCBpbiBhIGxhdGVyIGl0ZXJhdGlvblxuICAgICAgaWYgKG5vZGVMZXZlbCA+IGxldmVsKSB7XG4gICAgICAgIGNvbnRpbnVlO1xuICAgICAgfVxuICAgICAgLy8gT25seSBhZGQgdG8gcmVwcmVzZW50YXRpb24gaWYgaXQgaXMgdmlzaWJsZSAocGFyZW50IGlzIGV4cGFuZGVkKVxuICAgICAgaWYgKHBhcmVudEV4cGFuZGVkKSB7XG4gICAgICAgIC8vIENvbGxlY3QgdGhlIGRhdGEgdW5kZXIgdGhpcyBub2RlIGFjY29yZGluZyB0byB0aGUgZm9sbG93aW5nIHJ1bGVzOlxuICAgICAgICAvLyAxLiBJZiB0aGUgbmV4dCBub2RlIGluIHRoZSBsaXN0IGlzIGEgc2libGluZyBvZiB0aGUgY3VycmVudCBub2RlIGFkZCBpdCB0byB0aGUgY2hpbGQgbGlzdFxuICAgICAgICAvLyAyLiBJZiB0aGUgbmV4dCBub2RlIGlzIGEgY2hpbGQgb2YgdGhlIGN1cnJlbnQgbm9kZSwgZ2V0IHRoZSBzdWItdHJlZSBzdHJ1Y3R1cmUgZm9yIHRoZVxuICAgICAgICAvLyAgICBjaGlsZCBhbmQgYWRkIGl0IHVuZGVyIHRoaXMgbm9kZVxuICAgICAgICAvLyAzLiBJZiB0aGUgbmV4dCBub2RlIGhhcyBhIHNoYWxsb3dlciBsZXZlbCwgd2UndmUgcmVhY2hlZCB0aGUgZW5kIG9mIHRoZSBjaGlsZCBub2RlcyBmb3JcbiAgICAgICAgLy8gICAgdGhlIGN1cnJlbnQgcGFyZW50LlxuICAgICAgICBpZiAobmV4dE5vZGVMZXZlbCA9PT0gbGV2ZWwpIHtcbiAgICAgICAgICB0aGlzLl9hZGRDaGlsZFRvTm9kZShyZXN1bHQsIHt0ZXh0fSk7XG4gICAgICAgIH0gZWxzZSBpZiAobmV4dE5vZGVMZXZlbCA+IGxldmVsKSB7XG4gICAgICAgICAgbGV0IGNoaWxkcmVuID0gdGhpcy5fZ2V0VHJlZVN0cnVjdHVyZShub2Rlcy5zbGljZShpICsgMSksXG4gICAgICAgICAgICBuZXh0Tm9kZUxldmVsLFxuICAgICAgICAgICAgZXhwYW5kZWQpPy5jaGlsZHJlbjtcbiAgICAgICAgICBsZXQgY2hpbGQgPSBjaGlsZHJlbiA/IHt0ZXh0LCBjaGlsZHJlbn0gOiB7dGV4dH07XG4gICAgICAgICAgdGhpcy5fYWRkQ2hpbGRUb05vZGUocmVzdWx0LCBjaGlsZCk7XG4gICAgICAgIH0gZWxzZSB7XG4gICAgICAgICAgdGhpcy5fYWRkQ2hpbGRUb05vZGUocmVzdWx0LCB7dGV4dH0pO1xuICAgICAgICAgIHJldHVybiByZXN1bHQ7XG4gICAgICAgIH1cbiAgICAgIH1cbiAgICB9XG4gICAgcmV0dXJuIHJlc3VsdDtcbiAgfVxuXG4gIHByaXZhdGUgX2FkZENoaWxkVG9Ob2RlKHJlc3VsdDogVGV4dFRyZWUsIGNoaWxkOiBUZXh0VHJlZSkge1xuICAgIHJlc3VsdC5jaGlsZHJlbiA/IHJlc3VsdC5jaGlsZHJlbi5wdXNoKGNoaWxkKSA6IHJlc3VsdC5jaGlsZHJlbiA9IFtjaGlsZF07XG4gIH1cbn1cbiJdfQ== |
|---|
