[DevTools] Accessibility: Show siblings and children of selected node

third_party/WebKit/Source/devtools/front_end/accessibility/AccessibilityModel.js

Index: third_party/WebKit/Source/devtools/front_end/accessibility/AccessibilityModel.js
diff --git a/third_party/WebKit/Source/devtools/front_end/accessibility/AccessibilityModel.js b/third_party/WebKit/Source/devtools/front_end/accessibility/AccessibilityModel.js
index 0a3da2903e75bdd2f7452dc9c2f8ad2a35e70baf..5ce4be890f3a23b2e3cbf4c42072e06136a781da 100644
--- a/third_party/WebKit/Source/devtools/front_end/accessibility/AccessibilityModel.js
+++ b/third_party/WebKit/Source/devtools/front_end/accessibility/AccessibilityModel.js
@@ -115,6 +115,121 @@ WebInspector.AccessibilityNode.prototype = {
         return this._accessibilityModel.axNodeForId(this._parentId);
     },
 
+    /**
+     * @return {boolean}
+     */
+    isDOMNode: function()
+    {
+        return !!this._domNodeId;
+    },
+
+    /**
+     * @return {?WebInspector.DOMNode}
+     */
+    domNode: function()
+    {
+        return this._domNode || null;
+    },
+
+    /**
+     * @param {?WebInspector.DOMNode} domNode
+     */
+    _setDOMNode: function(domNode)
+    {
+        this._domNode = domNode;
+        if (domNode)
+            this._accessibilityModel.setAXNodeForDOMNode(domNode, this);
+    },
+
+    /**
+     * @return {!Promise<?WebInspector.DOMNode>}
+     */
+    resolveDOMNode: function()
+    {
+        if (this._domNode)
+            return Promise.resolve(this._domNode);
+        if (!this._domNodeId)
+            return Promise.resolve(/** @type {?WebInspector.DOMNode} */ (null));
+
+        var deferredNode = new WebInspector.DeferredDOMNode(this.target(), this._domNodeId);
+        return deferredNode.resolvePromise()
+            .then((domNode) => {
+                this._setDOMNode(domNode);
+                return domNode;
+            });
+    },
+
+    /**
+     * @return {?number}
+     */
+    domNodeId: function()
+    {
+        return this._domNodeId || null;
+    },
+
+    /**
+     * @return {!Array<!WebInspector.AccessibilityNode>}
+     */
+    children: function()
+    {
+        var children = [];
+        if (!this._childIds)
+            return children;
+
+        for (var childId of this._childIds) {
+            var child = this._accessibilityModel.axNodeForId(childId);
+            if (child)
+                children.push(child);
+        }
+
+        return children;
+    },
+
+    /**
+     * @return {number}
+     */
+    numChildren: function()
+    {
+        if (!this._childIds)
+            return 0;
+        return this._childIds.length;
+    },
+
+    /**
+     * @return {boolean}
+     */
+    hasOnlyUnloadedChildren: function()
+    {
+        if (!this._childIds || !this._childIds.length)
+            return false;
+
+        return !this._childIds.some((id) => this._accessibilityModel.axNodeForId(id) !== undefined);
+    },
+
+    /**
+     * TODO(aboxhall): Remove once protocol is stable.
+     * @param {!WebInspector.AccessibilityNode} inspectedNode
+     * @param {string=} leadingSpace
+     * @return {string}
+     */
+    printSelfAndChildren: function(inspectedNode, leadingSpace)
+    {
+        var string = leadingSpace || "";
+        if (this._role)
+            string += this._role.value;
+        else
+            string += "<no role>";
+        string += (this._name ? " " + this._name.value : "");
+        string += " " + this._id;
+        if (this._domNode)
+            string += " (" + this._domNode.nodeName() + ")";
+        if (this === inspectedNode)
+            string += " *";
+        for (var child of this.children())
+            string += "\n" + child.printSelfAndChildren(inspectedNode, (leadingSpace || "") + "  ");
+        return string;
+    },
+
     __proto__: WebInspector.SDKObject.prototype
 };
 
@@ -130,41 +245,32 @@ WebInspector.AccessibilityModel = function(target)
 
     /** @type {!Map<string, !WebInspector.AccessibilityNode>} */
     this._axIdToAXNode = new Map();
+    this._domNodeToAXNode = new Map();
 };
 
 WebInspector.AccessibilityModel.prototype = {
-
     /**
-     * @param {string} axId
-     * @return {?WebInspector.AccessibilityNode}
+     * @param {!WebInspector.DOMNode} node
+     * @return {!Promise<?WebInspector.AccessibilityNode>}
      */
-    axNodeForId: function(axId)
+    setDOMNode: function(node)
     {
-        return this._axIdToAXNode.get(axId);
-    },
+        this._axIdToAXNode.clear();
+        this._inspectedDOMNode = node;
 
-    /**
-     * @param {string} axId
-     * @param {!WebInspector.AccessibilityNode} axNode
-     */
-    _setAXNodeForAXId: function(axId, axNode)
-    {
-        this._axIdToAXNode.set(axId, axNode);
+        return this._requestPartialAXTree(node);
     },
 
     /**
      * @param {!WebInspector.DOMNode} node
-     * @return {!Promise<?Array<!WebInspector.AccessibilityNode>>}
+     * @return {!Promise}
      */
-    getAXNodeChain: function(node)
+    _requestPartialAXTree: function(node)
     {
-        this._axIdToAXNode.clear();
-
         /**
          * @this {WebInspector.AccessibilityModel}
          * @param {?string} error
          * @param {!Array<!AccessibilityAgent.AXNode>=} payloads
-         * @return {?Array<!WebInspector.AccessibilityNode>}
          */
         function parsePayload(error, payloads)
         {
@@ -174,19 +280,83 @@ WebInspector.AccessibilityModel.prototype = {
             }
 
             if (!payloads)
-                return null;
+                return;
 
-            var nodes = [];
             for (var payload of payloads)
-                nodes.push(new WebInspector.AccessibilityNode(this, payload));
-
-            return nodes;
+                new WebInspector.AccessibilityNode(this, payload);
         }
-        return this._agent.getAXNodeChain(node.id, true, parsePayload.bind(this));
+        return this._agent.getPartialAXTree(node.id, parsePayload.bind(this));
+    },
+
+    /**
+     * @return {?WebInspector.AccessibilityNode}
+     */
+    getInspectedAXNode: function()
+    {
+        return this._domNodeToAXNode.get(this._inspectedDOMNode);
+    },
+
+    /**
+     * @param {string} axId
+     * @return {?WebInspector.AccessibilityNode}
+     */
+    axNodeForId: function(axId)
+    {
+        return this._axIdToAXNode.get(axId);
+    },
+
+    /**
+     * @param {string} axId
+     * @param {!WebInspector.AccessibilityNode} axNode
+     */
+    _setAXNodeForAXId: function(axId, axNode)
+    {
+        this._axIdToAXNode.set(axId, axNode);
+    },
+
+    /**
+     * @param {!WebInspector.DOMNode} domNode
+     * @return {?WebInspector.AccessibilityNode}
+     */
+    axNodeForDOMNode: function(domNode)
+    {
+        return this._domNodeToAXNode.get(domNode);
+    },
+
+    /**
+     * @param {!WebInspector.DOMNode} domNode
+     * @param {!WebInspector.AccessibilityNode} axNode
+     */
+    setAXNodeForDOMNode: function(domNode, axNode)
+    {
+        this._domNodeToAXNode.set(domNode, axNode);
+    },
+
+    // TODO(aboxhall): Remove once protocol is stable.
+    logTree: function()
+    {
+        var inspectedNode = this._domNodeToAXNode.get(this._inspectedDOMNode);
+        if (!inspectedNode)
+            return;
+        var rootNode = inspectedNode;
+        while (rootNode.parentNode())
+            rootNode = rootNode.parentNode();
+        console.log(rootNode.printSelfAndChildren(inspectedNode));
+    },
+
+    /**
+     * @return {!Promise}
+     */
+    resolveAllDOMNodes: function()
+    {
+        var promises = [];
+        for (var axNode of this._axIdToAXNode.values())
+            promises.push(axNode.resolveDOMNode());
+        return Promise.all(promises);
     },
 
     __proto__: WebInspector.SDKModel.prototype
-};
+}
 
 WebInspector.AccessibilityModel._symbol = Symbol("AccessibilityModel");
 /**