1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/ui/webui/uber/uber_ui.h"
6
7 #include "base/stl_util.h"
8 #include "base/strings/utf_string_conversions.h"
9 #include "chrome/browser/chrome_notification_types.h"
10 #include "chrome/browser/extensions/extension_service.h"
11 #include "chrome/browser/profiles/profile.h"
12 #include "chrome/browser/ui/webui/chrome_web_ui_controller_factory.h"
13 #include "chrome/browser/ui/webui/extensions/extensions_ui.h"
14 #include "chrome/browser/ui/webui/options/options_ui.h"
15 #include "chrome/common/extensions/extension_set.h"
16 #include "chrome/common/extensions/manifest_url_handler.h"
17 #include "chrome/common/url_constants.h"
18 #include "content/public/browser/navigation_controller.h"
19 #include "content/public/browser/navigation_entry.h"
20 #include "content/public/browser/notification_source.h"
21 #include "content/public/browser/web_contents.h"
22 #include "content/public/browser/web_ui.h"
23 #include "content/public/browser/web_ui_data_source.h"
24 #include "grit/browser_resources.h"
25 #include "grit/chromium_strings.h"
26 #include "grit/generated_resources.h"
27
28 using content::NavigationController;
29 using content::NavigationEntry;
30 using content::RenderViewHost;
31 using content::WebContents;
32
33 namespace {
34
CreateUberHTMLSource()35 content::WebUIDataSource* CreateUberHTMLSource() {
36 content::WebUIDataSource* source =
37 content::WebUIDataSource::Create(chrome::kChromeUIUberHost);
38
39 source->SetUseJsonJSFormatV2();
40 source->SetJsonPath("strings.js");
41 source->AddResourcePath("uber.js", IDR_UBER_JS);
42 source->AddResourcePath("uber_utils.js", IDR_UBER_UTILS_JS);
43 source->SetDefaultResource(IDR_UBER_HTML);
44 source->OverrideContentSecurityPolicyFrameSrc("frame-src chrome:;");
45
46 // Hack alert: continue showing "Loading..." until a real title is set.
47 source->AddLocalizedString("pageTitle", IDS_TAB_LOADING_TITLE);
48
49 source->AddString("extensionsFrameURL",
50 ASCIIToUTF16(chrome::kChromeUIExtensionsFrameURL));
51 source->AddString("extensionsHost",
52 ASCIIToUTF16(chrome::kChromeUIExtensionsHost));
53 source->AddString("helpFrameURL",
54 ASCIIToUTF16(chrome::kChromeUIHelpFrameURL));
55 source->AddString("helpHost",
56 ASCIIToUTF16(chrome::kChromeUIHelpHost));
57 source->AddString("historyFrameURL",
58 ASCIIToUTF16(chrome::kChromeUIHistoryFrameURL));
59 source->AddString("historyHost",
60 ASCIIToUTF16(chrome::kChromeUIHistoryHost));
61 source->AddString("settingsFrameURL",
62 ASCIIToUTF16(chrome::kChromeUISettingsFrameURL));
63 source->AddString("settingsHost",
64 ASCIIToUTF16(chrome::kChromeUISettingsHost));
65
66 return source;
67 }
68
69 // Determines whether the user has an active extension of the given type.
HasExtensionType(Profile * profile,const char * extensionType)70 bool HasExtensionType(Profile* profile, const char* extensionType) {
71 const ExtensionSet* extensionSet =
72 profile->GetExtensionService()->extensions();
73
74 for (ExtensionSet::const_iterator iter = extensionSet->begin();
75 iter != extensionSet->end(); ++iter) {
76 extensions::URLOverrides::URLOverrideMap map =
77 extensions::URLOverrides::GetChromeURLOverrides(iter->get());
78 extensions::URLOverrides::URLOverrideMap::const_iterator result =
79 map.find(std::string(extensionType));
80
81 if (result != map.end())
82 return true;
83 }
84
85 return false;
86 }
87
CreateUberFrameHTMLSource(Profile * profile)88 content::WebUIDataSource* CreateUberFrameHTMLSource(Profile* profile) {
89 content::WebUIDataSource* source =
90 content::WebUIDataSource::Create(chrome::kChromeUIUberFrameHost);
91
92 source->SetUseJsonJSFormatV2();
93 source->SetJsonPath("strings.js");
94 source->AddResourcePath("uber_frame.js", IDR_UBER_FRAME_JS);
95 source->SetDefaultResource(IDR_UBER_FRAME_HTML);
96
97 // TODO(jhawkins): Attempt to get rid of IDS_SHORT_PRODUCT_OS_NAME.
98 #if defined(OS_CHROMEOS)
99 source->AddLocalizedString("shortProductName", IDS_SHORT_PRODUCT_OS_NAME);
100 #else
101 source->AddLocalizedString("shortProductName", IDS_SHORT_PRODUCT_NAME);
102 #endif // defined(OS_CHROMEOS)
103
104 source->AddString("extensionsHost",
105 ASCIIToUTF16(chrome::kChromeUIExtensionsHost));
106 source->AddLocalizedString("extensionsDisplayName",
107 IDS_MANAGE_EXTENSIONS_SETTING_WINDOWS_TITLE);
108 source->AddString("helpHost",
109 ASCIIToUTF16(chrome::kChromeUIHelpHost));
110 source->AddLocalizedString("helpDisplayName", IDS_HELP_TITLE);
111 source->AddString("historyHost",
112 ASCIIToUTF16(chrome::kChromeUIHistoryHost));
113 source->AddLocalizedString("historyDisplayName", IDS_HISTORY_TITLE);
114 source->AddString("settingsHost",
115 ASCIIToUTF16(chrome::kChromeUISettingsHost));
116 source->AddLocalizedString("settingsDisplayName", IDS_SETTINGS_TITLE);
117 bool overridesHistory = HasExtensionType(profile,
118 chrome::kChromeUIHistoryHost);
119 source->AddString("overridesHistory",
120 ASCIIToUTF16(overridesHistory ? "yes" : "no"));
121 source->DisableDenyXFrameOptions();
122 source->OverrideContentSecurityPolicyFrameSrc("frame-src chrome:;");
123
124 return source;
125 }
126
127 } // namespace
128
UberUI(content::WebUI * web_ui)129 UberUI::UberUI(content::WebUI* web_ui) : WebUIController(web_ui) {
130 Profile* profile = Profile::FromWebUI(web_ui);
131 content::WebUIDataSource::Add(profile, CreateUberHTMLSource());
132
133 RegisterSubpage(chrome::kChromeUIExtensionsFrameURL);
134 RegisterSubpage(chrome::kChromeUIHelpFrameURL);
135 RegisterSubpage(chrome::kChromeUIHistoryFrameURL);
136 RegisterSubpage(chrome::kChromeUISettingsFrameURL);
137 RegisterSubpage(chrome::kChromeUIUberFrameURL);
138 }
139
~UberUI()140 UberUI::~UberUI() {
141 STLDeleteValues(&sub_uis_);
142 }
143
RegisterSubpage(const std::string & page_url)144 void UberUI::RegisterSubpage(const std::string& page_url) {
145 content::WebUI* webui =
146 web_ui()->GetWebContents()->CreateWebUI(GURL(page_url));
147
148 webui->SetFrameXPath("//iframe[starts-with(@src,'" + page_url + "')]");
149 sub_uis_[page_url] = webui;
150 }
151
RenderViewCreated(RenderViewHost * render_view_host)152 void UberUI::RenderViewCreated(RenderViewHost* render_view_host) {
153 for (SubpageMap::iterator iter = sub_uis_.begin(); iter != sub_uis_.end();
154 ++iter) {
155 iter->second->GetController()->RenderViewCreated(render_view_host);
156 }
157 }
158
RenderViewReused(RenderViewHost * render_view_host)159 void UberUI::RenderViewReused(RenderViewHost* render_view_host) {
160 for (SubpageMap::iterator iter = sub_uis_.begin(); iter != sub_uis_.end();
161 ++iter) {
162 iter->second->GetController()->RenderViewReused(render_view_host);
163 }
164 }
165
OverrideHandleWebUIMessage(const GURL & source_url,const std::string & message,const ListValue & args)166 bool UberUI::OverrideHandleWebUIMessage(const GURL& source_url,
167 const std::string& message,
168 const ListValue& args) {
169 // Find the appropriate subpage and forward the message.
170 SubpageMap::iterator subpage = sub_uis_.find(source_url.GetOrigin().spec());
171 if (subpage == sub_uis_.end()) {
172 // The message was sent from the uber page itself.
173 DCHECK_EQ(std::string(chrome::kChromeUIUberHost), source_url.host());
174 return false;
175 }
176
177 // The message was sent from a subpage.
178 // TODO(jam) fix this to use interface
179 // return subpage->second->GetController()->OverrideHandleWebUIMessage(
180 // source_url, message, args);
181 subpage->second->ProcessWebUIMessage(source_url, message, args);
182 return true;
183 }
184
185 // UberFrameUI
186
UberFrameUI(content::WebUI * web_ui)187 UberFrameUI::UberFrameUI(content::WebUI* web_ui) : WebUIController(web_ui) {
188 Profile* profile = Profile::FromWebUI(web_ui);
189 content::WebUIDataSource::Add(profile, CreateUberFrameHTMLSource(profile));
190
191 // Register as an observer for when extensions are loaded and unloaded.
192 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOADED,
193 content::Source<Profile>(profile));
194 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED,
195 content::Source<Profile>(profile));
196 }
197
~UberFrameUI()198 UberFrameUI::~UberFrameUI() {
199 }
200
Observe(int type,const content::NotificationSource & source,const content::NotificationDetails & details)201 void UberFrameUI::Observe(int type, const content::NotificationSource& source,
202 const content::NotificationDetails& details) {
203 switch (type) {
204 // We listen for notifications that indicate an extension has been loaded
205 // (i.e., has been installed and/or enabled) or unloaded (i.e., has been
206 // uninstalled and/or disabled). If one of these events has occurred, then
207 // we must update the behavior of the History navigation element so that
208 // it opens the history extension if one is installed and enabled or
209 // opens the default history page if one is uninstalled or disabled.
210 case chrome::NOTIFICATION_EXTENSION_LOADED:
211 case chrome::NOTIFICATION_EXTENSION_UNLOADED: {
212 Profile* profile = Profile::FromWebUI(web_ui());
213 bool overrides_history =
214 HasExtensionType(profile, chrome::kChromeUIHistoryHost);
215 web_ui()->CallJavascriptFunction(
216 "uber_frame.setNavigationOverride",
217 base::StringValue(chrome::kChromeUIHistoryHost),
218 base::StringValue(overrides_history ? "yes" : "no"));
219 break;
220 }
221 default:
222 NOTREACHED();
223 }
224 }
225