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
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
// Copyright 2023 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "content/browser/devtools/protocol/device_access_handler.h"
namespace content {
namespace protocol {
// static
std::vector<DeviceAccessHandler*> DeviceAccessHandler::ForAgentHost(
DevToolsAgentHostImpl* host) {
return host->HandlersByName<DeviceAccessHandler>(
DeviceAccess::Metainfo::domainName);
}
DeviceAccessHandler::DeviceAccessHandler()
: DevToolsDomainHandler(DeviceAccess::Metainfo::domainName) {}
DeviceAccessHandler::~DeviceAccessHandler() = default;
void DeviceAccessHandler::Wire(UberDispatcher* dispatcher) {
frontend_.emplace(dispatcher->channel());
DeviceAccess::Dispatcher::wire(dispatcher, this);
}
DispatchResponse DeviceAccessHandler::Enable() {
enabled_ = true;
return Response::Success();
}
DispatchResponse DeviceAccessHandler::Disable() {
enabled_ = false;
request_prompt_infos_.clear();
return Response::Success();
}
DispatchResponse DeviceAccessHandler::SelectPrompt(const String& in_id,
const String& in_deviceId) {
if (!enabled_) {
return Response::ServerError("DeviceAccess domain is not enabled");
}
auto* prompt_info = FindRequest(in_id);
if (!prompt_info) {
return Response::InvalidParams("Cannot find request with id");
}
if (!prompt_info->SelectDevice(in_deviceId)) {
return Response::InvalidParams("Cannot find device with deviceId");
}
return Response::Success();
}
DispatchResponse DeviceAccessHandler::CancelPrompt(const String& in_id) {
if (!enabled_) {
return Response::ServerError("DeviceAccess domain is not enabled");
}
auto* prompt_info = FindRequest(in_id);
if (!prompt_info) {
return Response::InvalidParams("Cannot find request with id");
}
prompt_info->Cancel();
return Response::Success();
}
const std::string& DeviceAccessHandler::FindOrAddRequestId(
DevtoolsDeviceRequestPromptInfo* prompt_info) {
auto it = std::ranges::find_if(
request_prompt_infos_,
[prompt_info](auto& pair) { return pair.second == prompt_info; });
if (it != request_prompt_infos_.end()) {
return it->first;
}
auto request_id = base::UnguessableToken::Create().ToString();
auto did_insert =
request_prompt_infos_.insert_or_assign(request_id, prompt_info);
CHECK(did_insert.second);
return did_insert.first->first;
}
DevtoolsDeviceRequestPromptInfo* DeviceAccessHandler::FindRequest(
const String& requestId) {
auto it = request_prompt_infos_.find(requestId);
if (it != request_prompt_infos_.end()) {
return it->second;
}
return nullptr;
}
void DeviceAccessHandler::UpdateDeviceRequestPrompt(
DevtoolsDeviceRequestPromptInfo* prompt_info) {
if (enabled_) {
auto request_id = FindOrAddRequestId(prompt_info);
auto devices = std::make_unique<
std::vector<std::unique_ptr<DeviceAccess::PromptDevice>>>();
for (auto& device : prompt_info->GetDevices()) {
devices->push_back(DeviceAccess::PromptDevice::Create()
.SetId(device.id)
.SetName(device.name)
.Build());
}
frontend_->DeviceRequestPrompted(request_id, std::move(devices));
}
}
void DeviceAccessHandler::CleanUpDeviceRequestPrompt(
DevtoolsDeviceRequestPromptInfo* prompt_info) {
if (enabled_) {
base::EraseIf(request_prompt_infos_,
[&](auto& pair) { return pair.second == prompt_info; });
}
}
} // namespace protocol
} // namespace content
|
