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
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "headless/lib/browser/protocol/target_handler.h"
#include "build/build_config.h"
#include "headless/lib/browser/headless_browser_context_impl.h"
#include "headless/lib/browser/headless_browser_impl.h"
#include "headless/lib/browser/headless_web_contents_impl.h"
#include "ui/gfx/geometry/size.h"
namespace headless {
namespace protocol {
TargetHandler::TargetHandler(HeadlessBrowserImpl* browser)
: browser_(browser) {}
TargetHandler::~TargetHandler() = default;
void TargetHandler::Wire(UberDispatcher* dispatcher) {
Target::Dispatcher::wire(dispatcher, this);
}
Response TargetHandler::Disable() {
return Response::Success();
}
Response TargetHandler::CreateTarget(const std::string& url,
Maybe<int> width,
Maybe<int> height,
Maybe<std::string> context_id,
Maybe<bool> enable_begin_frame_control,
Maybe<bool> new_window,
Maybe<bool> background,
Maybe<bool> for_tab,
std::string* out_target_id) {
#if BUILDFLAG(IS_MAC)
if (enable_begin_frame_control.value_or(false)) {
return Response::ServerError(
"BeginFrameControl is not supported on MacOS yet");
}
#endif
HeadlessBrowserContext* context;
if (context_id.has_value()) {
context = browser_->GetBrowserContextForId(context_id.value());
if (!context)
return Response::InvalidParams("browserContextId");
} else {
context = browser_->GetDefaultBrowserContext();
if (!context) {
return Response::ServerError(
"You specified no |browserContextId|, but "
"there is no default browser context set on "
"HeadlessBrowser");
}
}
GURL gurl(url);
if (gurl.is_empty()) {
gurl = GURL(url::kAboutBlankURL);
}
HeadlessWebContentsImpl* web_contents_impl = HeadlessWebContentsImpl::From(
context->CreateWebContentsBuilder()
.SetInitialURL(gurl)
.SetWindowSize(gfx::Size(
width.value_or(browser_->options()->window_size.width()),
height.value_or(browser_->options()->window_size.height())))
.SetEnableBeginFrameControl(
enable_begin_frame_control.value_or(false))
.SetUseTabTarget(for_tab.value_or(false))
.Build());
*out_target_id = web_contents_impl->GetDevToolsAgentHostId();
return Response::Success();
}
Response TargetHandler::CloseTarget(const std::string& target_id,
bool* out_success) {
HeadlessWebContents* web_contents =
browser_->GetWebContentsForDevToolsAgentHostId(target_id);
*out_success = false;
if (web_contents) {
web_contents->Close();
*out_success = true;
}
return Response::Success();
}
} // namespace protocol
} // namespace headless
|
