1 // Copyright 2014 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/cocoa/apps/quit_with_apps_controller_mac.h" 6 7 #include "base/command_line.h" 8 #include "base/i18n/number_formatting.h" 9 #include "base/prefs/pref_registry_simple.h" 10 #include "base/prefs/pref_service.h" 11 #include "base/strings/sys_string_conversions.h" 12 #include "base/strings/utf_string_conversions.h" 13 #include "chrome/browser/apps/app_window_registry_util.h" 14 #include "chrome/browser/browser_process.h" 15 #include "chrome/browser/notifications/notification.h" 16 #include "chrome/browser/notifications/notification_ui_manager.h" 17 #include "chrome/browser/profiles/profile_manager.h" 18 #include "chrome/browser/ui/browser_iterator.h" 19 #include "chrome/common/chrome_switches.h" 20 #include "chrome/common/pref_names.h" 21 #include "chrome/grit/chromium_strings.h" 22 #include "chrome/grit/generated_resources.h" 23 #include "chrome/grit/google_chrome_strings.h" 24 #include "extensions/browser/app_window/app_window.h" 25 #include "extensions/browser/app_window/native_app_window.h" 26 #include "extensions/common/extension.h" 27 #include "grit/chrome_unscaled_resources.h" 28 #include "ui/base/l10n/l10n_util.h" 29 #include "ui/base/l10n/l10n_util_mac.h" 30 #include "ui/base/resource/resource_bundle.h" 31 32 const char kQuitWithAppsOriginUrl[] = "chrome://quit-with-apps"; 33 const int kQuitAllAppsButtonIndex = 0; 34 const int kDontShowAgainButtonIndex = 1; 35 36 const char QuitWithAppsController::kQuitWithAppsNotificationID[] = 37 "quit-with-apps"; 38 39 QuitWithAppsController::QuitWithAppsController() 40 : suppress_for_session_(false) { 41 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); 42 43 // There is only ever one notification to replace, so use the same replace_id 44 // each time. 45 base::string16 replace_id = base::UTF8ToUTF16(id()); 46 47 message_center::ButtonInfo quit_apps_button_info( 48 l10n_util::GetStringUTF16(IDS_QUIT_WITH_APPS_QUIT_LABEL)); 49 message_center::ButtonInfo suppression_button_info( 50 l10n_util::GetStringUTF16(IDS_QUIT_WITH_APPS_SUPPRESSION_LABEL)); 51 message_center::RichNotificationData rich_notification_data; 52 rich_notification_data.buttons.push_back(quit_apps_button_info); 53 rich_notification_data.buttons.push_back(suppression_button_info); 54 55 notification_.reset(new Notification( 56 message_center::NOTIFICATION_TYPE_SIMPLE, 57 GURL(kQuitWithAppsOriginUrl), 58 l10n_util::GetStringUTF16(IDS_QUIT_WITH_APPS_TITLE), 59 l10n_util::GetStringUTF16(IDS_QUIT_WITH_APPS_EXPLANATION), 60 ui::ResourceBundle::GetSharedInstance().GetImageNamed( 61 IDR_PRODUCT_LOGO_128), 62 blink::WebTextDirectionDefault, 63 message_center::NotifierId(message_center::NotifierId::SYSTEM_COMPONENT, 64 kQuitWithAppsNotificationID), 65 l10n_util::GetStringUTF16(IDS_QUIT_WITH_APPS_NOTIFICATION_DISPLAY_SOURCE), 66 replace_id, 67 rich_notification_data, 68 this)); 69 } 70 71 QuitWithAppsController::~QuitWithAppsController() {} 72 73 void QuitWithAppsController::Display() {} 74 75 void QuitWithAppsController::Error() {} 76 77 void QuitWithAppsController::Close(bool by_user) { 78 if (by_user) 79 suppress_for_session_ = true; 80 } 81 82 void QuitWithAppsController::Click() { 83 g_browser_process->notification_ui_manager()->CancelById(id()); 84 } 85 86 void QuitWithAppsController::ButtonClick(int button_index) { 87 g_browser_process->notification_ui_manager()->CancelById(id()); 88 if (button_index == kQuitAllAppsButtonIndex) { 89 AppWindowRegistryUtil::CloseAllAppWindows(); 90 } else if (button_index == kDontShowAgainButtonIndex) { 91 g_browser_process->local_state()->SetBoolean( 92 prefs::kNotifyWhenAppsKeepChromeAlive, false); 93 } 94 } 95 96 content::WebContents* QuitWithAppsController::GetWebContents() const { 97 return NULL; 98 } 99 100 std::string QuitWithAppsController::id() const { 101 return kQuitWithAppsNotificationID; 102 } 103 104 bool QuitWithAppsController::ShouldQuit() { 105 DCHECK_CURRENTLY_ON(content::BrowserThread::UI); 106 107 // Quit immediately if this is a test. 108 if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kTestType) && 109 !CommandLine::ForCurrentProcess()->HasSwitch( 110 switches::kAppsKeepChromeAliveInTests)) { 111 return true; 112 } 113 114 // Quit immediately if there are no windows or the confirmation has been 115 // suppressed. 116 if (!AppWindowRegistryUtil::IsAppWindowRegisteredInAnyProfile(0)) 117 return true; 118 119 // If there are browser windows, and this notification has been suppressed for 120 // this session or permanently, then just return false to prevent Chrome from 121 // quitting. If there are no browser windows, always show the notification. 122 bool suppress_always = !g_browser_process->local_state()->GetBoolean( 123 prefs::kNotifyWhenAppsKeepChromeAlive); 124 if (!chrome::BrowserIterator().done() && 125 (suppress_for_session_ || suppress_always)) { 126 return false; 127 } 128 129 ProfileManager* profile_manager = g_browser_process->profile_manager(); 130 DCHECK(profile_manager); 131 132 std::vector<Profile*> profiles(profile_manager->GetLoadedProfiles()); 133 DCHECK(profiles.size()); 134 135 // Delete any existing notification to ensure this one is shown. 136 g_browser_process->notification_ui_manager()->CancelById(id()); 137 g_browser_process->notification_ui_manager()->Add(*notification_, 138 profiles[0]); 139 140 // Always return false, the notification UI can be used to quit all apps which 141 // will cause Chrome to quit. 142 return false; 143 } 144 145 // static 146 void QuitWithAppsController::RegisterPrefs(PrefRegistrySimple* registry) { 147 registry->RegisterBooleanPref(prefs::kNotifyWhenAppsKeepChromeAlive, true); 148 } 149