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
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/ui/webui/omnibox_popup/omnibox_popup_ui.h"
#include <atomic>
#include <string_view>
#include "base/feature_list.h"
#include "base/strings/string_number_conversions.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser_finder.h"
#include "chrome/browser/ui/browser_window.h"
#include "chrome/browser/ui/location_bar/location_bar.h"
#include "chrome/browser/ui/webui/favicon_source.h"
#include "chrome/browser/ui/webui/metrics_reporter/metrics_reporter_service.h"
#include "chrome/browser/ui/webui/sanitized_image_source.h"
#include "chrome/browser/ui/webui/searchbox/realbox_handler.h"
#include "chrome/common/webui_url_constants.h"
#include "chrome/grit/omnibox_popup_resources.h"
#include "chrome/grit/omnibox_popup_resources_map.h"
#include "components/favicon_base/favicon_url_parser.h"
#include "components/omnibox/browser/omnibox_view.h"
#include "content/public/browser/web_ui_data_source.h"
#include "ui/webui/color_change_listener/color_change_handler.h"
#include "ui/webui/webui_util.h"
bool OmniboxPopupUIConfig::IsWebUIEnabled(
content::BrowserContext* browser_context) {
return base::FeatureList::IsEnabled(omnibox::kWebUIOmniboxPopup);
}
OmniboxPopupUI::OmniboxPopupUI(content::WebUI* web_ui)
: ui::MojoWebUIController(web_ui, /*enable_chrome_send=*/true),
profile_(Profile::FromWebUI(web_ui)) {
content::WebUIDataSource* source = content::WebUIDataSource::CreateAndAdd(
Profile::FromWebUI(web_ui), chrome::kChromeUIOmniboxPopupHost);
RealboxHandler::SetupWebUIDataSource(source, Profile::FromWebUI(web_ui));
webui::SetupWebUIDataSource(source, kOmniboxPopupResources,
IDR_OMNIBOX_POPUP_OMNIBOX_POPUP_HTML);
webui::EnableTrustedTypesCSP(source);
content::URLDataSource::Add(profile_,
std::make_unique<SanitizedImageSource>(profile_));
content::URLDataSource::Add(
profile_, std::make_unique<FaviconSource>(
profile_, chrome::FaviconUrlFormat::kFavicon2));
}
OmniboxPopupUI::~OmniboxPopupUI() = default;
WEB_UI_CONTROLLER_TYPE_IMPL(OmniboxPopupUI)
void OmniboxPopupUI::BindInterface(
content::RenderFrameHost* host,
mojo::PendingReceiver<searchbox::mojom::PageHandler> pending_page_handler) {
// Extract SessionID from URL to select the omnibox that initiated page load.
const GURL& url = host->GetWebUI()->GetWebContents()->GetLastCommittedURL();
SessionID id = SessionID::InvalidValue();
if (url.is_valid() && url.has_query()) {
std::string_view spec(url.query_piece());
url::Component query, key, value;
query.len = static_cast<int>(spec.size());
while (url::ExtractQueryKeyValue(spec, &query, &key, &value)) {
if (key.is_nonempty() && value.is_nonempty()) {
const std::string_view key_piece = spec.substr(key.begin, key.len);
constexpr char kSessionIdKey[] = "session_id";
if (key_piece == kSessionIdKey) {
const std::string_view value_piece =
spec.substr(value.begin, value.len);
int value_int = 0;
if (base::StringToInt(value_piece, &value_int)) {
id = SessionID::FromSerializedValue(value_int);
}
break;
}
}
}
}
if (id.is_valid()) {
if (Browser* browser = chrome::FindBrowserWithID(id)) {
OmniboxController* controller =
browser->window()->GetLocationBar()->GetOmniboxView()->controller();
MetricsReporterService* metrics_reporter_service =
MetricsReporterService::GetFromWebContents(
web_ui()->GetWebContents());
handler_ = std::make_unique<RealboxHandler>(
std::move(pending_page_handler), Profile::FromWebUI(web_ui()),
web_ui()->GetWebContents(),
metrics_reporter_service->metrics_reporter(), controller);
}
}
}
void OmniboxPopupUI::BindInterface(
mojo::PendingReceiver<color_change_listener::mojom::PageHandler>
pending_receiver) {
color_provider_handler_ = std::make_unique<ui::ColorChangeHandler>(
web_ui()->GetWebContents(), std::move(pending_receiver));
}
|
