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
32const char kQuitWithAppsOriginUrl[] = "chrome://quit-with-apps";
33const int kQuitAllAppsButtonIndex = 0;
34const int kDontShowAgainButtonIndex = 1;
35
36const char QuitWithAppsController::kQuitWithAppsNotificationID[] =
37    "quit-with-apps";
38
39QuitWithAppsController::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
71QuitWithAppsController::~QuitWithAppsController() {}
72
73void QuitWithAppsController::Display() {}
74
75void QuitWithAppsController::Error() {}
76
77void QuitWithAppsController::Close(bool by_user) {
78  if (by_user)
79    suppress_for_session_ = true;
80}
81
82void QuitWithAppsController::Click() {
83  g_browser_process->notification_ui_manager()->CancelById(id());
84}
85
86void 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
96content::WebContents* QuitWithAppsController::GetWebContents() const {
97  return NULL;
98}
99
100std::string QuitWithAppsController::id() const {
101  return kQuitWithAppsNotificationID;
102}
103
104bool 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
146void QuitWithAppsController::RegisterPrefs(PrefRegistrySimple* registry) {
147  registry->RegisterBooleanPref(prefs::kNotifyWhenAppsKeepChromeAlive, true);
148}
149