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
|
// Copyright 2015 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 "headless/lib/browser/headless_content_browser_client.h"
#include <memory>
#include <unordered_set>
#include "base/callback.h"
#include "base/json/json_reader.h"
#include "base/memory/ptr_util.h"
#include "content/public/browser/browser_context.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/render_process_host.h"
#include "content/public/browser/render_view_host.h"
#include "content/public/common/service_names.mojom.h"
#include "headless/grit/headless_lib_resources.h"
#include "headless/lib/browser/headless_browser_context_impl.h"
#include "headless/lib/browser/headless_browser_impl.h"
#include "headless/lib/browser/headless_browser_main_parts.h"
#include "headless/lib/browser/headless_devtools_manager_delegate.h"
#include "ui/base/resource/resource_bundle.h"
namespace headless {
namespace {
const char kCapabilityPath[] =
"interface_provider_specs.navigation:frame.provides.renderer";
} // namespace
HeadlessContentBrowserClient::HeadlessContentBrowserClient(
HeadlessBrowserImpl* browser)
: browser_(browser) {}
HeadlessContentBrowserClient::~HeadlessContentBrowserClient() {}
content::BrowserMainParts* HeadlessContentBrowserClient::CreateBrowserMainParts(
const content::MainFunctionParams&) {
std::unique_ptr<HeadlessBrowserMainParts> browser_main_parts =
base::MakeUnique<HeadlessBrowserMainParts>(browser_);
browser_->set_browser_main_parts(browser_main_parts.get());
return browser_main_parts.release();
}
void HeadlessContentBrowserClient::OverrideWebkitPrefs(
content::RenderViewHost* render_view_host,
content::WebPreferences* prefs) {
auto browser_context = HeadlessBrowserContextImpl::From(
render_view_host->GetProcess()->GetBrowserContext());
const base::Callback<void(headless::WebPreferences*)>& callback =
browser_context->options()->override_web_preferences_callback();
if (callback)
callback.Run(prefs);
}
content::DevToolsManagerDelegate*
HeadlessContentBrowserClient::GetDevToolsManagerDelegate() {
return new HeadlessDevToolsManagerDelegate(browser_->GetWeakPtr());
}
std::unique_ptr<base::Value>
HeadlessContentBrowserClient::GetServiceManifestOverlay(
base::StringPiece name) {
if (name != content::mojom::kBrowserServiceName ||
browser_->options()->mojo_service_names.empty())
return nullptr;
base::StringPiece manifest_template =
ui::ResourceBundle::GetSharedInstance().GetRawDataResource(
IDR_HEADLESS_BROWSER_MANIFEST_OVERLAY_TEMPLATE);
std::unique_ptr<base::Value> manifest =
base::JSONReader::Read(manifest_template);
// Add mojo_service_names to renderer capability specified in options.
base::DictionaryValue* manifest_dictionary = nullptr;
CHECK(manifest->GetAsDictionary(&manifest_dictionary));
base::ListValue* capability_list = nullptr;
CHECK(manifest_dictionary->GetList(kCapabilityPath, &capability_list));
for (std::string service_name : browser_->options()->mojo_service_names) {
capability_list->AppendString(service_name);
}
return manifest;
}
} // namespace headless
|
