1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
|
// 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::OnDestruct() {
delete this;
}
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::WebElement result_element = start_node.querySelector(web_selector);
int result_acc_obj_id = 0;
if (!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
|
