Home | History | Annotate | Download | only in apps 
      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