blob: 49dab41ab352dce439c3acdb92d938831f8d7ef3 (
plain)
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
|
// Copyright 2018 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 "content/browser/ukm_internals_ui.h"
#include <stddef.h>
#include <memory>
#include <string>
#include <utility>
#include "base/bind.h"
#include "components/ukm/debug/ukm_debug_data_extractor.h"
#include "components/ukm/ukm_service.h"
#include "content/grit/content_resources.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/browser/content_browser_client.h"
#include "content/public/browser/web_contents.h"
#include "content/public/browser/web_ui.h"
#include "content/public/browser/web_ui_data_source.h"
#include "content/public/browser/web_ui_message_handler.h"
#include "content/public/common/content_client.h"
#include "content/public/common/url_constants.h"
namespace content {
namespace {
WebUIDataSource* CreateUkmHTMLSource() {
WebUIDataSource* source = WebUIDataSource::Create(kChromeUIUkmHost);
source->AddResourcePath("ukm_internals.js", IDR_UKM_INTERNALS_JS);
source->AddResourcePath("ukm_internals.css", IDR_UKM_INTERNALS_CSS);
source->SetDefaultResource(IDR_UKM_INTERNALS_HTML);
return source;
}
// This class receives javascript messages from the renderer.
// Note that the WebUI infrastructure runs on the UI thread, therefore all of
// this class's methods are expected to run on the UI thread.
class UkmMessageHandler : public WebUIMessageHandler {
public:
explicit UkmMessageHandler(const ukm::UkmService* ukm_service);
~UkmMessageHandler() override;
// WebUIMessageHandler:
void RegisterMessages() override;
private:
void HandleRequestUkmData(const base::ListValue* args);
const ukm::UkmService* ukm_service_;
DISALLOW_COPY_AND_ASSIGN(UkmMessageHandler);
};
UkmMessageHandler::UkmMessageHandler(const ukm::UkmService* ukm_service)
: ukm_service_(ukm_service) {}
UkmMessageHandler::~UkmMessageHandler() {}
void UkmMessageHandler::HandleRequestUkmData(const base::ListValue* args) {
AllowJavascript();
// Identifies the callback, used for when resolving.
std::string callback_id;
args->GetString(0, &callback_id);
base::Value ukm_debug_data =
ukm::debug::UkmDebugDataExtractor::GetStructuredData(ukm_service_);
ResolveJavascriptCallback(base::Value(callback_id),
std::move(ukm_debug_data));
}
void UkmMessageHandler::RegisterMessages() {
DCHECK_CURRENTLY_ON(BrowserThread::UI);
// We can use base::Unretained() here, as both the callback and this class are
// owned by UkmInternalsUI.
web_ui()->RegisterMessageCallback(
"requestUkmData",
base::BindRepeating(&UkmMessageHandler::HandleRequestUkmData,
base::Unretained(this)));
}
} // namespace
// Changes to this class should be in sync with its iOS equivalent
// ios/chrome/browser/ui/webui/ukm_internals_ui.cc
UkmInternalsUI::UkmInternalsUI(WebUI* web_ui) : WebUIController(web_ui) {
ukm::UkmService* ukm_service = GetContentClient()->browser()->GetUkmService();
web_ui->AddMessageHandler(std::make_unique<UkmMessageHandler>(ukm_service));
// Set up the chrome://ukm/ source.
BrowserContext* browser_context =
web_ui->GetWebContents()->GetBrowserContext();
WebUIDataSource::Add(browser_context, CreateUkmHTMLSource());
}
} // namespace content
|
