OLD | NEW |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/ui/webui/tracking_ui.h" | 5 #include "chrome/browser/ui/webui/profiler_ui.h" |
6 | 6 |
7 #include <string> | 7 #include <string> |
8 | 8 |
9 // When testing the javacript code, it is cumbersome to have to keep | 9 // When testing the javacript code, it is cumbersome to have to keep |
10 // re-building the resouces package and reloading the browser. To solve | 10 // re-building the resouces package and reloading the browser. To solve |
11 // this, enable the following flag to read the webapp's source files | 11 // this, enable the following flag to read the webapp's source files |
12 // directly off disk, so all you have to do is refresh the page to | 12 // directly off disk, so all you have to do is refresh the page to |
13 // test the modifications. | 13 // test the modifications. |
14 //#define USE_SOURCE_FILES_DIRECTLY | 14 //#define USE_SOURCE_FILES_DIRECTLY |
15 | 15 |
(...skipping 15 matching lines...) Expand all Loading... |
31 #include "base/path_service.h" | 31 #include "base/path_service.h" |
32 #endif // USE_SOURCE_FILES_DIRECTLY | 32 #endif // USE_SOURCE_FILES_DIRECTLY |
33 | 33 |
34 using content::BrowserThread; | 34 using content::BrowserThread; |
35 using chrome_browser_metrics::TrackingSynchronizer; | 35 using chrome_browser_metrics::TrackingSynchronizer; |
36 | 36 |
37 namespace { | 37 namespace { |
38 | 38 |
39 #ifdef USE_SOURCE_FILES_DIRECTLY | 39 #ifdef USE_SOURCE_FILES_DIRECTLY |
40 | 40 |
41 class TrackingWebUIDataSource : public ChromeURLDataManager::DataSource { | 41 class ProfilerWebUIDataSource : public ChromeURLDataManager::DataSource { |
42 public: | 42 public: |
43 TrackingWebUIDataSource() | 43 ProfilerWebUIDataSource() |
44 : DataSource(chrome::kChromeUITrackingHost2, MessageLoop::current()) { | 44 : DataSource(chrome::kChromeUIProfilerHost, MessageLoop::current()) { |
45 } | 45 } |
46 | 46 |
47 protected: | 47 protected: |
48 // ChromeURLDataManager | 48 // ChromeURLDataManager |
49 virtual std::string GetMimeType(const std::string& path) const OVERRIDE { | 49 virtual std::string GetMimeType(const std::string& path) const OVERRIDE { |
50 if (EndsWith(path, ".js", false)) | 50 if (EndsWith(path, ".js", false)) |
51 return "application/javascript"; | 51 return "application/javascript"; |
52 return "text/html"; | 52 return "text/html"; |
53 } | 53 } |
54 | 54 |
55 virtual void StartDataRequest(const std::string& path, | 55 virtual void StartDataRequest(const std::string& path, |
56 bool is_incognito, | 56 bool is_incognito, |
57 int request_id) OVERRIDE { | 57 int request_id) OVERRIDE { |
58 FilePath base_path; | 58 FilePath base_path; |
59 PathService::Get(base::DIR_SOURCE_ROOT, &base_path); | 59 PathService::Get(base::DIR_SOURCE_ROOT, &base_path); |
60 base_path = base_path.AppendASCII("chrome"); | 60 base_path = base_path.AppendASCII("chrome"); |
61 base_path = base_path.AppendASCII("browser"); | 61 base_path = base_path.AppendASCII("browser"); |
62 base_path = base_path.AppendASCII("resources"); | 62 base_path = base_path.AppendASCII("resources"); |
63 | 63 |
64 // If no resource was specified, default to tracking.html. | 64 // If no resource was specified, default to profiler.html. |
65 std::string filename = path.empty() ? "tracking.html" : path; | 65 std::string filename = path.empty() ? "profiler.html" : path; |
66 | 66 |
67 FilePath file_path; | 67 FilePath file_path; |
68 file_path = base_path.AppendASCII(filename); | 68 file_path = base_path.AppendASCII(filename); |
69 | 69 |
70 // Read the file synchronously and send it as the response. | 70 // Read the file synchronously and send it as the response. |
71 base::ThreadRestrictions::ScopedAllowIO allow; | 71 base::ThreadRestrictions::ScopedAllowIO allow; |
72 std::string file_contents; | 72 std::string file_contents; |
73 if (!file_util::ReadFileToString(file_path, &file_contents)) | 73 if (!file_util::ReadFileToString(file_path, &file_contents)) |
74 LOG(ERROR) << "Couldn't read file: " << file_path.value(); | 74 LOG(ERROR) << "Couldn't read file: " << file_path.value(); |
75 scoped_refptr<base::RefCountedString> response = | 75 scoped_refptr<base::RefCountedString> response = |
76 new base::RefCountedString(); | 76 new base::RefCountedString(); |
77 response->data() = file_contents; | 77 response->data() = file_contents; |
78 SendResponse(request_id, response); | 78 SendResponse(request_id, response); |
79 } | 79 } |
80 | 80 |
81 private: | 81 private: |
82 DISALLOW_COPY_AND_ASSIGN(TrackingWebUIDataSource); | 82 DISALLOW_COPY_AND_ASSIGN(ProfilerWebUIDataSource); |
83 }; | 83 }; |
84 | 84 |
85 ChromeURLDataManager::DataSource* CreateTrackingHTMLSource() { | 85 ChromeURLDataManager::DataSource* CreateProfilerHTMLSource() { |
86 return new TrackingWebUIDataSource(); | 86 return new ProfilerWebUIDataSource(); |
87 } | 87 } |
88 | 88 |
89 #else // USE_SOURCE_FILES_DIRECTLY | 89 #else // USE_SOURCE_FILES_DIRECTLY |
90 | 90 |
91 ChromeWebUIDataSource* CreateTrackingHTMLSource() { | 91 ChromeWebUIDataSource* CreateProfilerHTMLSource() { |
92 // TODO(eroman): Use kChromeUITrackingHost instead of kChromeUITrackingHost2 | |
93 // once migration to webui is complete. | |
94 ChromeWebUIDataSource* source = | 92 ChromeWebUIDataSource* source = |
95 new ChromeWebUIDataSource(chrome::kChromeUITrackingHost2); | 93 new ChromeWebUIDataSource(chrome::kChromeUIProfilerHost); |
96 | 94 |
97 source->set_json_path("strings.js"); | 95 source->set_json_path("strings.js"); |
98 source->add_resource_path("tracking.js", IDR_TRACKING_JS); | 96 source->add_resource_path("profiler.js", IDR_PROFILER_JS); |
99 source->set_default_resource(IDR_TRACKING_HTML); | 97 source->set_default_resource(IDR_PROFILER_HTML); |
100 return source; | 98 return source; |
101 } | 99 } |
102 | 100 |
103 #endif | 101 #endif |
104 | 102 |
105 // This class receives javascript messages from the renderer. | 103 // This class receives javascript messages from the renderer. |
106 // Note that the WebUI infrastructure runs on the UI thread, therefore all of | 104 // Note that the WebUI infrastructure runs on the UI thread, therefore all of |
107 // this class's methods are expected to run on the UI thread. | 105 // this class's methods are expected to run on the UI thread. |
108 class TrackingMessageHandler : public WebUIMessageHandler { | 106 class ProfilerMessageHandler : public WebUIMessageHandler { |
109 public: | 107 public: |
110 TrackingMessageHandler() {} | 108 ProfilerMessageHandler() {} |
111 | 109 |
112 // WebUIMessageHandler implementation. | 110 // WebUIMessageHandler implementation. |
113 virtual WebUIMessageHandler* Attach(WebUI* web_ui) OVERRIDE; | 111 virtual WebUIMessageHandler* Attach(WebUI* web_ui) OVERRIDE; |
114 virtual void RegisterMessages() OVERRIDE; | 112 virtual void RegisterMessages() OVERRIDE; |
115 | 113 |
116 // Messages. | 114 // Messages. |
117 void OnGetData(const ListValue* list); | 115 void OnGetData(const ListValue* list); |
118 void OnResetData(const ListValue* list); | 116 void OnResetData(const ListValue* list); |
119 | 117 |
120 private: | 118 private: |
121 DISALLOW_COPY_AND_ASSIGN(TrackingMessageHandler); | 119 DISALLOW_COPY_AND_ASSIGN(ProfilerMessageHandler); |
122 }; | 120 }; |
123 | 121 |
124 WebUIMessageHandler* TrackingMessageHandler::Attach(WebUI* web_ui) { | 122 WebUIMessageHandler* ProfilerMessageHandler::Attach(WebUI* web_ui) { |
125 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 123 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
126 WebUIMessageHandler* result = WebUIMessageHandler::Attach(web_ui); | 124 WebUIMessageHandler* result = WebUIMessageHandler::Attach(web_ui); |
127 return result; | 125 return result; |
128 } | 126 } |
129 | 127 |
130 void TrackingMessageHandler::RegisterMessages() { | 128 void ProfilerMessageHandler::RegisterMessages() { |
131 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 129 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
132 | 130 |
133 web_ui_->RegisterMessageCallback("getData", | 131 web_ui_->RegisterMessageCallback("getData", |
134 base::Bind(&TrackingMessageHandler::OnGetData,base::Unretained(this))); | 132 base::Bind(&ProfilerMessageHandler::OnGetData,base::Unretained(this))); |
135 web_ui_->RegisterMessageCallback("resetData", | 133 web_ui_->RegisterMessageCallback("resetData", |
136 base::Bind(&TrackingMessageHandler::OnResetData, | 134 base::Bind(&ProfilerMessageHandler::OnResetData, |
137 base::Unretained(this))); | 135 base::Unretained(this))); |
138 } | 136 } |
139 | 137 |
140 void TrackingMessageHandler::OnGetData(const ListValue* list) { | 138 void ProfilerMessageHandler::OnGetData(const ListValue* list) { |
141 TrackingUI* tracking_ui = reinterpret_cast<TrackingUI*>(web_ui_); | 139 ProfilerUI* profiler_ui = reinterpret_cast<ProfilerUI*>(web_ui_); |
142 tracking_ui->GetData(); | 140 profiler_ui->GetData(); |
143 } | 141 } |
144 | 142 |
145 void TrackingMessageHandler::OnResetData(const ListValue* list) { | 143 void ProfilerMessageHandler::OnResetData(const ListValue* list) { |
146 tracked_objects::ThreadData::ResetAllThreadData(); | 144 tracked_objects::ThreadData::ResetAllThreadData(); |
147 } | 145 } |
148 | 146 |
149 } // namespace | 147 } // namespace |
150 | 148 |
151 TrackingUI::TrackingUI(TabContents* contents) : ChromeWebUI(contents) { | 149 ProfilerUI::ProfilerUI(TabContents* contents) : ChromeWebUI(contents) { |
152 ui_weak_ptr_factory_.reset(new base::WeakPtrFactory<TrackingUI>(this)); | 150 ui_weak_ptr_factory_.reset(new base::WeakPtrFactory<ProfilerUI>(this)); |
153 ui_weak_ptr_ = ui_weak_ptr_factory_->GetWeakPtr(); | 151 ui_weak_ptr_ = ui_weak_ptr_factory_->GetWeakPtr(); |
154 | 152 |
155 AddMessageHandler((new TrackingMessageHandler())->Attach(this)); | 153 AddMessageHandler((new ProfilerMessageHandler())->Attach(this)); |
156 | 154 |
157 // Set up the chrome://tracking2/ source. | 155 // Set up the chrome://profiler/ source. |
158 Profile::FromBrowserContext(contents->browser_context())-> | 156 Profile::FromBrowserContext(contents->browser_context())-> |
159 GetChromeURLDataManager()->AddDataSource(CreateTrackingHTMLSource()); | 157 GetChromeURLDataManager()->AddDataSource(CreateProfilerHTMLSource()); |
160 } | 158 } |
161 | 159 |
162 TrackingUI::~TrackingUI() { | 160 ProfilerUI::~ProfilerUI() { |
163 } | 161 } |
164 | 162 |
165 void TrackingUI::GetData() { | 163 void ProfilerUI::GetData() { |
166 TrackingSynchronizer::FetchTrackingDataAsynchronously(ui_weak_ptr_); | 164 TrackingSynchronizer::FetchTrackingDataAsynchronously(ui_weak_ptr_); |
167 } | 165 } |
168 | 166 |
169 void TrackingUI::ReceivedData(base::Value* value) { | 167 void ProfilerUI::ReceivedData(base::Value* value) { |
170 // Send the data to the renderer. | 168 // Send the data to the renderer. |
171 scoped_ptr<Value> data_values(value); | 169 scoped_ptr<Value> data_values(value); |
172 CallJavascriptFunction("g_browserBridge.receivedData", *data_values.get()); | 170 CallJavascriptFunction("g_browserBridge.receivedData", *data_values.get()); |
173 } | 171 } |
174 | 172 |
OLD | NEW |