Chromium Code Reviews| Index: chrome/browser/ui/webui/uber/uber_ui.cc |
| diff --git a/chrome/browser/ui/webui/uber/uber_ui.cc b/chrome/browser/ui/webui/uber/uber_ui.cc |
| index 59b4755bb660f46ea447ea5ff71868ee6056f195..873d74f78b850a1cc929ad0d010f3ec383fc1bf8 100644 |
| --- a/chrome/browser/ui/webui/uber/uber_ui.cc |
| +++ b/chrome/browser/ui/webui/uber/uber_ui.cc |
| @@ -6,15 +6,20 @@ |
| #include "base/stl_util.h" |
| #include "base/utf_string_conversions.h" |
| +#include "chrome/browser/extensions/extension_service.h" |
|
Mark Mentovai
2012/04/09 22:55:58
This too.
|
| #include "chrome/browser/profiles/profile.h" |
| #include "chrome/browser/ui/webui/chrome_url_data_manager.h" |
| #include "chrome/browser/ui/webui/chrome_web_ui_data_source.h" |
| #include "chrome/browser/ui/webui/chrome_web_ui_controller_factory.h" |
| #include "chrome/browser/ui/webui/extensions/extensions_ui.h" |
| #include "chrome/browser/ui/webui/options2/options_ui2.h" |
| +#include "chrome/common/chrome_notification_types.h" |
| +#include "chrome/common/extensions/extension.h" |
| +#include "chrome/common/extensions/extension_set.h" |
| #include "chrome/common/url_constants.h" |
| #include "content/public/browser/navigation_controller.h" |
| #include "content/public/browser/navigation_entry.h" |
| +#include "content/public/browser/notification_source.h" |
| #include "content/public/browser/web_contents.h" |
| #include "content/public/browser/web_ui.h" |
| #include "grit/browser_resources.h" |
| @@ -60,7 +65,25 @@ ChromeWebUIDataSource* CreateUberHTMLSource() { |
| return source; |
| } |
| -ChromeWebUIDataSource* CreateUberFrameHTMLSource() { |
| +// Determines whether the user has an active extension of the given type. |
| +bool HasExtensionType(Profile* profile, const char* extensionType) { |
| + const ExtensionSet* extensionSet = |
| + profile->GetExtensionService()->extensions(); |
| + |
| + for (ExtensionSet::const_iterator iter = extensionSet->begin(); |
| + iter != extensionSet->end(); ++iter) { |
| + Extension::URLOverrideMap map = (*iter)->GetChromeURLOverrides(); |
| + Extension::URLOverrideMap::const_iterator result = |
| + map.find(std::string(extensionType)); |
| + |
| + if (result != map.end()) |
| + return true; |
| + } |
| + |
| + return false; |
| +} |
| + |
| +ChromeWebUIDataSource* CreateUberFrameHTMLSource(Profile* profile) { |
| ChromeWebUIDataSource* source = |
| new ChromeWebUIDataSource(chrome::kChromeUIUberFrameHost); |
| @@ -88,6 +111,10 @@ ChromeWebUIDataSource* CreateUberFrameHTMLSource() { |
| source->AddString("settingsHost", |
| ASCIIToUTF16(chrome::kChromeUISettingsHost)); |
| source->AddLocalizedString("settingsDisplayName", IDS_SETTINGS_TITLE); |
| + bool overridesHistory = HasExtensionType(profile, |
| + chrome::kChromeUIHistoryHost); |
| + source->AddString("overridesHistory", |
| + ASCIIToUTF16(overridesHistory ? "yes" : "no")); |
| return source; |
| } |
| @@ -170,8 +197,41 @@ bool UberUI::OverrideHandleWebUIMessage(const GURL& source_url, |
| UberFrameUI::UberFrameUI(content::WebUI* web_ui) : WebUIController(web_ui) { |
| Profile* profile = Profile::FromWebUI(web_ui); |
| profile->GetChromeURLDataManager()->AddDataSource( |
| - CreateUberFrameHTMLSource()); |
| + CreateUberFrameHTMLSource(profile)); |
| + |
| + // Register as an observer for when extensions are loaded and unloaded. |
| + registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOADED, |
| + content::Source<Profile>(profile)); |
| + registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED, |
| + content::Source<Profile>(profile)); |
| } |
| UberFrameUI::~UberFrameUI() { |
| } |
| + |
| +void UberFrameUI::Observe(int type, const content::NotificationSource& source, |
| + const content::NotificationDetails& details) { |
| + switch (type) { |
| + // We listen for notifications that indicate an extension has been loaded |
| + // (i.e., has been installed and/or enabled) or unloaded (i.e., has been |
| + // uninstalled and/or disabled). If one of these events has occurred, then |
| + // we must update the behavior of the History navigation element so that |
| + // it opens the history extension if one is installed and enabled or |
| + // opens the default history page if one is uninstalled or disabled. |
| + case chrome::NOTIFICATION_EXTENSION_LOADED: |
| + case chrome::NOTIFICATION_EXTENSION_UNLOADED: { |
| + Profile* profile = Profile::FromWebUI(web_ui()); |
| + bool overridesHistory = HasExtensionType(profile, |
| + chrome::kChromeUIHistoryHost); |
| + scoped_ptr<Value> controlsValue( |
| + Value::CreateStringValue(chrome::kChromeUIHistoryHost)); |
| + scoped_ptr<Value> overrideValue( |
| + Value::CreateStringValue(overridesHistory ? "yes" : "no")); |
| + web_ui()->CallJavascriptFunction( |
| + "uber_frame.setNavigationOverride", *controlsValue, *overrideValue); |
| + break; |
| + } |
| + default: |
| + NOTREACHED(); |
| + } |
| +} |