Implement AutomationNode.querySelector().

extensions/renderer/api/automation/automation_api_helper.cc

+// Copyright 2014 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include "extensions/renderer/api/automation/automation_api_helper.h"
+
+#include "content/public/renderer/render_view.h"
+#include "extensions/common/extension_messages.h"
+#include "third_party/WebKit/public/web/WebAXObject.h"
+#include "third_party/WebKit/public/web/WebDocument.h"
+#include "third_party/WebKit/public/web/WebElement.h"
+#include "third_party/WebKit/public/web/WebExceptionCode.h"
+#include "third_party/WebKit/public/web/WebFrame.h"
+#include "third_party/WebKit/public/web/WebNode.h"
+#include "third_party/WebKit/public/web/WebView.h"
+
+namespace extensions {
+
+AutomationApiHelper::AutomationApiHelper(content::RenderView* render_view)
+    : content::RenderViewObserver(render_view) {
+}
+
+AutomationApiHelper::~AutomationApiHelper() {
+}
+
+bool AutomationApiHelper::OnMessageReceived(const IPC::Message& message) {
+  bool handled = true;
+  IPC_BEGIN_MESSAGE_MAP(AutomationApiHelper, message)
+    IPC_MESSAGE_HANDLER(ExtensionMsg_AutomationQuerySelector, OnQuerySelector)
+    IPC_MESSAGE_UNHANDLED(handled = false)
+  IPC_END_MESSAGE_MAP()
+  return handled;
+}
+
+void AutomationApiHelper::OnQuerySelector(int request_id,
+                                          int acc_obj_id,
+                                          const base::string16& selector) {
+  ExtensionHostMsg_AutomationQuerySelector_Error error;
+  if (!render_view() || !render_view()->GetWebView() ||
+      !render_view()->GetWebView()->mainFrame()) {
+    error.value = ExtensionHostMsg_AutomationQuerySelector_Error::kNoMainFrame;
+    Send(new ExtensionHostMsg_AutomationQuerySelector_Result(
+        routing_id(), request_id, error, 0));
+    return;
+  }
+  blink::WebDocument document =
+      render_view()->GetWebView()->mainFrame()->document();
+  if (document.isNull()) {
+      error.value =
+          ExtensionHostMsg_AutomationQuerySelector_Error::kNoDocument;
+    Send(new ExtensionHostMsg_AutomationQuerySelector_Result(
+        routing_id(), request_id, error, 0));
+    return;
+  }
+  blink::WebNode start_node = document;
+  if (acc_obj_id > 0) {
+    blink::WebAXObject start_acc_obj =
+        document.accessibilityObjectFromID(acc_obj_id);
+    if (start_acc_obj.isNull()) {
+      error.value =
+          ExtensionHostMsg_AutomationQuerySelector_Error::kNodeDestroyed;
+      Send(new ExtensionHostMsg_AutomationQuerySelector_Result(
+          routing_id(), request_id, error, 0));
+      return;
+    }
+
+    start_node = start_acc_obj.node();
+    while (start_node.isNull()) {
+      start_acc_obj = start_acc_obj.parentObject();
+      start_node = start_acc_obj.node();
+    }
+  }
+  blink::WebString web_selector(selector);
+  blink::WebExceptionCode ec = 0;
+  blink::WebElement result_element = start_node.querySelector(web_selector, ec);
+  int result_acc_obj_id = 0;
+  if (!ec && !result_element.isNull()) {
+    blink::WebAXObject result_acc_obj = result_element.accessibilityObject();
+    if (!result_acc_obj.isDetached()) {
+      while (result_acc_obj.accessibilityIsIgnored())
+        result_acc_obj = result_acc_obj.parentObject();
+
+      result_acc_obj_id = result_acc_obj.axID();
+    }
+  }
+  Send(new ExtensionHostMsg_AutomationQuerySelector_Result(
+      routing_id(), request_id, error, result_acc_obj_id));
+}
+
+}  // namespace extensions