| OLD | NEW |
| (Empty) |
| 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/notifications/notification_options_menu_model.h" | |
| 6 | |
| 7 #include <string> | |
| 8 | |
| 9 #include "base/logging.h" | |
| 10 #include "base/strings/utf_string_conversions.h" | |
| 11 #include "chrome/browser/browser_process.h" | |
| 12 #include "chrome/browser/extensions/extension_service.h" | |
| 13 #include "chrome/browser/notifications/balloon.h" | |
| 14 #include "chrome/browser/notifications/balloon_collection.h" | |
| 15 #include "chrome/browser/notifications/balloon_notification_ui_manager.h" | |
| 16 #include "chrome/browser/notifications/desktop_notification_service.h" | |
| 17 #include "chrome/browser/notifications/desktop_notification_service_factory.h" | |
| 18 #include "chrome/browser/notifications/notification.h" | |
| 19 #include "chrome/browser/notifications/notification_prefs_manager.h" | |
| 20 #include "chrome/browser/notifications/notification_ui_manager.h" | |
| 21 #include "chrome/browser/profiles/profile.h" | |
| 22 #include "chrome/browser/ui/browser.h" | |
| 23 #include "chrome/browser/ui/browser_finder.h" | |
| 24 #include "chrome/browser/ui/chrome_pages.h" | |
| 25 #include "chrome/browser/ui/host_desktop.h" | |
| 26 #include "chrome/common/chrome_switches.h" | |
| 27 #include "chrome/common/content_settings_types.h" | |
| 28 #include "chrome/common/url_constants.h" | |
| 29 #include "content/public/browser/web_contents_delegate.h" | |
| 30 #include "extensions/common/constants.h" | |
| 31 #include "extensions/common/extension.h" | |
| 32 #include "grit/generated_resources.h" | |
| 33 #include "ui/base/l10n/l10n_util.h" | |
| 34 | |
| 35 // Menu commands | |
| 36 const int kTogglePermissionCommand = 0; | |
| 37 const int kToggleExtensionCommand = 1; | |
| 38 const int kOpenContentSettingsCommand = 2; | |
| 39 const int kCornerSelectionSubMenu = 3; | |
| 40 | |
| 41 const int kCornerGroupId = 10; | |
| 42 const int kCornerUpperLeft = 11; | |
| 43 const int kCornerUpperRight = 12; | |
| 44 const int kCornerLowerLeft = 13; | |
| 45 const int kCornerLowerRight = 14; | |
| 46 const int kCornerDefault = 20; | |
| 47 | |
| 48 CornerSelectionMenuModel::CornerSelectionMenuModel(Balloon* balloon) | |
| 49 : ui::SimpleMenuModel(this), | |
| 50 balloon_(balloon) { | |
| 51 AddRadioItem(kCornerDefault, | |
| 52 l10n_util::GetStringUTF16(IDS_NOTIFICATION_POSITION_DEFAULT), | |
| 53 kCornerGroupId); | |
| 54 AddSeparator(ui::NORMAL_SEPARATOR); | |
| 55 AddRadioItem(kCornerUpperLeft, | |
| 56 l10n_util::GetStringUTF16(IDS_NOTIFICATION_POSITION_UPPER_LEFT), | |
| 57 kCornerGroupId); | |
| 58 AddRadioItem(kCornerUpperRight, | |
| 59 l10n_util::GetStringUTF16(IDS_NOTIFICATION_POSITION_UPPER_RIGHT), | |
| 60 kCornerGroupId); | |
| 61 AddRadioItem(kCornerLowerLeft, | |
| 62 l10n_util::GetStringUTF16(IDS_NOTIFICATION_POSITION_LOWER_LEFT), | |
| 63 kCornerGroupId); | |
| 64 AddRadioItem(kCornerLowerRight, | |
| 65 l10n_util::GetStringUTF16(IDS_NOTIFICATION_POSITION_LOWER_RIGHT), | |
| 66 kCornerGroupId); | |
| 67 } | |
| 68 | |
| 69 CornerSelectionMenuModel::~CornerSelectionMenuModel() { | |
| 70 } | |
| 71 | |
| 72 bool CornerSelectionMenuModel::IsCommandIdChecked(int command_id) const { | |
| 73 // TODO(dimich): MessageCenter does not use this preference (yet?) | |
| 74 if (NotificationUIManager::DelegatesToMessageCenter()) | |
| 75 return false; | |
| 76 | |
| 77 NotificationPrefsManager* prefs = | |
| 78 static_cast<BalloonNotificationUIManager*>( | |
| 79 g_browser_process->notification_ui_manager())->prefs_manager(); | |
| 80 | |
| 81 BalloonCollection::PositionPreference current = | |
| 82 prefs->GetPositionPreference(); | |
| 83 | |
| 84 if (command_id == kCornerUpperLeft) | |
| 85 return (current == BalloonCollection::UPPER_LEFT); | |
| 86 else if (command_id == kCornerUpperRight) | |
| 87 return (current == BalloonCollection::UPPER_RIGHT); | |
| 88 else if (command_id == kCornerLowerLeft) | |
| 89 return (current == BalloonCollection::LOWER_LEFT); | |
| 90 else if (command_id == kCornerLowerRight) | |
| 91 return (current == BalloonCollection::LOWER_RIGHT); | |
| 92 else if (command_id == kCornerDefault) | |
| 93 return (current == BalloonCollection::DEFAULT_POSITION); | |
| 94 | |
| 95 NOTREACHED(); | |
| 96 return false; | |
| 97 } | |
| 98 | |
| 99 bool CornerSelectionMenuModel::IsCommandIdEnabled(int command_id) const { | |
| 100 // All the menu options are always enabled. | |
| 101 return true; | |
| 102 } | |
| 103 | |
| 104 bool CornerSelectionMenuModel::GetAcceleratorForCommandId( | |
| 105 int command_id, ui::Accelerator* accelerator) { | |
| 106 // Currently no accelerators. | |
| 107 return false; | |
| 108 } | |
| 109 | |
| 110 void CornerSelectionMenuModel::ExecuteCommand(int command_id, int event_flags) { | |
| 111 // TODO(dimich): MessageCenter does not use this preference (yet?) | |
| 112 if (NotificationUIManager::DelegatesToMessageCenter()) | |
| 113 return; | |
| 114 | |
| 115 NotificationPrefsManager* prefs = | |
| 116 static_cast<BalloonNotificationUIManager*>( | |
| 117 g_browser_process->notification_ui_manager())->prefs_manager(); | |
| 118 | |
| 119 if (command_id == kCornerUpperLeft) | |
| 120 prefs->SetPositionPreference(BalloonCollection::UPPER_LEFT); | |
| 121 else if (command_id == kCornerUpperRight) | |
| 122 prefs->SetPositionPreference(BalloonCollection::UPPER_RIGHT); | |
| 123 else if (command_id == kCornerLowerLeft) | |
| 124 prefs->SetPositionPreference(BalloonCollection::LOWER_LEFT); | |
| 125 else if (command_id == kCornerLowerRight) | |
| 126 prefs->SetPositionPreference(BalloonCollection::LOWER_RIGHT); | |
| 127 else if (command_id == kCornerDefault) | |
| 128 prefs->SetPositionPreference(BalloonCollection::DEFAULT_POSITION); | |
| 129 else | |
| 130 NOTREACHED(); | |
| 131 } | |
| 132 | |
| 133 NotificationOptionsMenuModel::NotificationOptionsMenuModel(Balloon* balloon) | |
| 134 : ui::SimpleMenuModel(this), | |
| 135 balloon_(balloon) { | |
| 136 const Notification& notification = balloon->notification(); | |
| 137 const GURL& origin = notification.origin_url(); | |
| 138 | |
| 139 if (origin.SchemeIs(extensions::kExtensionScheme)) { | |
| 140 ExtensionService* extension_service = | |
| 141 balloon_->profile()->GetExtensionService(); | |
| 142 const extensions::Extension* extension = | |
| 143 extension_service->extensions()->GetExtensionOrAppByURL(origin); | |
| 144 // We get back no extension here when we show the notification after | |
| 145 // the extension has crashed. | |
| 146 if (extension) { | |
| 147 const base::string16 disable_label = l10n_util::GetStringUTF16( | |
| 148 IDS_EXTENSIONS_DISABLE); | |
| 149 AddItem(kToggleExtensionCommand, disable_label); | |
| 150 } | |
| 151 } else if (!notification.display_source().empty()) { | |
| 152 const base::string16 disable_label = l10n_util::GetStringFUTF16( | |
| 153 IDS_NOTIFICATION_BALLOON_REVOKE_MESSAGE, | |
| 154 notification.display_source()); | |
| 155 AddItem(kTogglePermissionCommand, disable_label); | |
| 156 } | |
| 157 | |
| 158 if (!notification.display_source().empty()) { | |
| 159 const base::string16 settings_label = l10n_util::GetStringUTF16( | |
| 160 IDS_NOTIFICATIONS_SETTINGS_BUTTON); | |
| 161 AddItem(kOpenContentSettingsCommand, settings_label); | |
| 162 } | |
| 163 | |
| 164 corner_menu_model_.reset(new CornerSelectionMenuModel(balloon)); | |
| 165 AddSubMenu(kCornerSelectionSubMenu, | |
| 166 l10n_util::GetStringUTF16(IDS_NOTIFICATION_CHOOSE_POSITION), | |
| 167 corner_menu_model_.get()); | |
| 168 } | |
| 169 | |
| 170 NotificationOptionsMenuModel::~NotificationOptionsMenuModel() { | |
| 171 } | |
| 172 | |
| 173 bool NotificationOptionsMenuModel::IsItemForCommandIdDynamic(int command_id) | |
| 174 const { | |
| 175 return command_id == kTogglePermissionCommand || | |
| 176 command_id == kToggleExtensionCommand; | |
| 177 } | |
| 178 | |
| 179 base::string16 NotificationOptionsMenuModel::GetLabelForCommandId( | |
| 180 int command_id) const { | |
| 181 // TODO(tfarina,johnnyg): Remove this code if we decide to close notifications | |
| 182 // after permissions are revoked. | |
| 183 if (command_id == kTogglePermissionCommand || | |
| 184 command_id == kToggleExtensionCommand) { | |
| 185 const Notification& notification = balloon_->notification(); | |
| 186 const GURL& origin = notification.origin_url(); | |
| 187 | |
| 188 DesktopNotificationService* service = | |
| 189 DesktopNotificationServiceFactory::GetForProfile(balloon_->profile()); | |
| 190 if (origin.SchemeIs(extensions::kExtensionScheme)) { | |
| 191 ExtensionService* extension_service = | |
| 192 balloon_->profile()->GetExtensionService(); | |
| 193 const extensions::Extension* extension = | |
| 194 extension_service->extensions()->GetExtensionOrAppByURL(origin); | |
| 195 if (extension) { | |
| 196 return l10n_util::GetStringUTF16( | |
| 197 extension_service->IsExtensionEnabled(extension->id()) ? | |
| 198 IDS_EXTENSIONS_DISABLE : | |
| 199 IDS_EXTENSIONS_ENABLE); | |
| 200 } | |
| 201 } else { | |
| 202 if (service->GetContentSetting(origin) == CONTENT_SETTING_ALLOW) { | |
| 203 return l10n_util::GetStringFUTF16( | |
| 204 IDS_NOTIFICATION_BALLOON_REVOKE_MESSAGE, | |
| 205 notification.display_source()); | |
| 206 } else { | |
| 207 return l10n_util::GetStringFUTF16( | |
| 208 IDS_NOTIFICATION_BALLOON_ENABLE_MESSAGE, | |
| 209 notification.display_source()); | |
| 210 } | |
| 211 } | |
| 212 } else if (command_id == kOpenContentSettingsCommand) { | |
| 213 return l10n_util::GetStringUTF16(IDS_NOTIFICATIONS_SETTINGS_BUTTON); | |
| 214 } | |
| 215 return base::string16(); | |
| 216 } | |
| 217 | |
| 218 bool NotificationOptionsMenuModel::IsCommandIdChecked(int /* command_id */) | |
| 219 const { | |
| 220 // Nothing in the menu is checked. | |
| 221 return false; | |
| 222 } | |
| 223 | |
| 224 bool NotificationOptionsMenuModel::IsCommandIdEnabled(int /* command_id */) | |
| 225 const { | |
| 226 // All the menu options are always enabled. | |
| 227 return true; | |
| 228 } | |
| 229 | |
| 230 bool NotificationOptionsMenuModel::GetAcceleratorForCommandId( | |
| 231 int /* command_id */, ui::Accelerator* /* accelerator */) { | |
| 232 // Currently no accelerators. | |
| 233 return false; | |
| 234 } | |
| 235 | |
| 236 void NotificationOptionsMenuModel::ExecuteCommand(int command_id, | |
| 237 int event_flags) { | |
| 238 DesktopNotificationService* service = | |
| 239 DesktopNotificationServiceFactory::GetForProfile(balloon_->profile()); | |
| 240 ExtensionService* extension_service = | |
| 241 balloon_->profile()->GetExtensionService(); | |
| 242 const GURL& origin = balloon_->notification().origin_url(); | |
| 243 switch (command_id) { | |
| 244 case kTogglePermissionCommand: | |
| 245 if (service->GetContentSetting(origin) == CONTENT_SETTING_ALLOW) | |
| 246 service->DenyPermission(origin); | |
| 247 else | |
| 248 service->GrantPermission(origin); | |
| 249 break; | |
| 250 case kToggleExtensionCommand: { | |
| 251 const extensions::Extension* extension = | |
| 252 extension_service->extensions()->GetExtensionOrAppByURL(origin); | |
| 253 if (extension) { | |
| 254 const std::string& id = extension->id(); | |
| 255 if (extension_service->IsExtensionEnabled(id)) | |
| 256 extension_service->DisableExtension( | |
| 257 id, extensions::Extension::DISABLE_USER_ACTION); | |
| 258 else | |
| 259 extension_service->EnableExtension(id); | |
| 260 } | |
| 261 break; | |
| 262 } | |
| 263 case kOpenContentSettingsCommand: { | |
| 264 chrome::HostDesktopType active_desktop = chrome::GetActiveDesktop(); | |
| 265 Browser* browser = chrome::FindLastActiveWithProfile( | |
| 266 balloon_->profile(), active_desktop); | |
| 267 if (!browser) { | |
| 268 // It is possible that there is no browser window (e.g. when there are | |
| 269 // background pages). | |
| 270 browser = new Browser(Browser::CreateParams(balloon_->profile(), | |
| 271 active_desktop)); | |
| 272 } | |
| 273 chrome::ShowContentSettings(browser, CONTENT_SETTINGS_TYPE_NOTIFICATIONS); | |
| 274 break; | |
| 275 } | |
| 276 default: | |
| 277 NOTREACHED(); | |
| 278 break; | |
| 279 } | |
| 280 } | |
| OLD | NEW |