src/chrome/browser/notifications/notification_options_menu_model.cc

   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/extensions/extension.h"
  29 #include "chrome/common/url_constants.h"
  30 #include "content/public/browser/web_contents_delegate.h"
  31 #include "extensions/common/constants.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 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 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 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 string16 NotificationOptionsMenuModel::GetLabelForCommandId(int command_id)
 180     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 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, or for a chrome frame process on windows).
 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 }