Implement AutomationNode.querySelector().

extensions/renderer/api/automation/automation_api_helper.cc

Index: extensions/renderer/api/automation/automation_api_helper.cc
diff --git a/extensions/renderer/api/automation/automation_api_helper.cc b/extensions/renderer/api/automation/automation_api_helper.cc
new file mode 100644
index 0000000000000000000000000000000000000000..75593c4bfa14d103d0350db92b86f3b62f865c55
--- /dev/null
+++ b/extensions/renderer/api/automation/automation_api_helper.cc
@@ -0,0 +1,70 @@
+// 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 <string>
+
+#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) {
+  if (!render_view() || !render_view()->GetWebView() ||
+      !render_view()->GetWebView()->mainFrame()) {
+    LOG(WARNING) << "No main frame";
+    // TODO(aboxhall): return error result
+  }
+  blink::WebDocument document =

[Devlin, 2014/10/29 21:23:44]

I don't know blink accessibility code well enough to fully grok from here to
line 65, so I'll trust the other reviewers here.

[aboxhall, 2014/10/30 18:34:18]

Acknowledged.

+      render_view()->GetWebView()->mainFrame()->document();
+  if (document.isNull()) {
+    LOG(WARNING) << "Document is null";
+    // TODO(aboxhall): return error result
+  }
+  blink::WebNode node = document;
+  if (acc_obj_id > 0) {
+    const blink::WebAXObject& obj =
+        document.accessibilityObjectFromID(acc_obj_id);
+    node = obj.node();
+  }
+  blink::WebString web_selector(selector);
+  blink::WebExceptionCode ec = 0;
+  blink::WebElement result = node.querySelector(web_selector, ec);
+  int result_acc_obj_id = 0;
+  if (!ec && !result.isNull()) {
+    blink::WebAXObject ax_obj = result.accessibilityObject();
+    if (!ax_obj.isDetached())
+      result_acc_obj_id = ax_obj.axID();
+  }
+  Send(new ExtensionHostMsg_AutomationQuerySelector_Result(
+      routing_id(), request_id, "", result_acc_obj_id));
+}
+
+}  // namespace extensions