chrome/browser/apps/app_shim/app_shim_handler_mac.cc

   1 // Copyright 2013 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/apps/app_shim/app_shim_handler_mac.h"
   6
   7 #include <map>
   8
   9 #include "base/bind.h"
  10 #include "base/logging.h"
  11 #include "base/memory/singleton.h"
  12 #include "base/message_loop/message_loop.h"
  13 #include "chrome/browser/apps/app_window_registry_util.h"
  14 #include "chrome/browser/chrome_notification_types.h"
  15 #include "chrome/browser/lifetime/application_lifetime.h"
  16 #include "chrome/browser/ui/app_list/app_list_service.h"
  17 #include "content/public/browser/notification_observer.h"
  18 #include "content/public/browser/notification_registrar.h"
  19 #include "content/public/browser/notification_service.h"
  20
  21 namespace apps {
  22
  23 namespace {
  24
  25 void TerminateIfNoAppWindows() {
  26   bool app_windows_left =
  27       AppWindowRegistryUtil::IsAppWindowVisibleInAnyProfile(0);
  28   if (!app_windows_left &&
  29       !AppListService::Get(chrome::HOST_DESKTOP_TYPE_NATIVE)
  30            ->IsAppListVisible()) {
  31     chrome::AttemptExit();
  32   }
  33 }
  34
  35 class AppShimHandlerRegistry : public content::NotificationObserver {
  36  public:
  37   static AppShimHandlerRegistry* GetInstance() {
  38     return base::Singleton<
  39         AppShimHandlerRegistry,
  40         base::LeakySingletonTraits<AppShimHandlerRegistry>>::get();
  41   }
  42
  43   AppShimHandler* GetForAppMode(const std::string& app_mode_id) const {
  44     HandlerMap::const_iterator it = handlers_.find(app_mode_id);
  45     if (it != handlers_.end())
  46       return it->second;
  47
  48     return default_handler_;
  49   }
  50
  51   bool SetForAppMode(const std::string& app_mode_id, AppShimHandler* handler) {
  52     bool inserted_or_removed = handler ?
  53         handlers_.insert(HandlerMap::value_type(app_mode_id, handler)).second :
  54         handlers_.erase(app_mode_id) == 1;
  55     DCHECK(inserted_or_removed);
  56     return inserted_or_removed;
  57   }
  58
  59   void SetDefaultHandler(AppShimHandler* handler) {
  60     DCHECK_NE(default_handler_ == NULL, handler == NULL);
  61     default_handler_ = handler;
  62   }
  63
  64   void MaybeTerminate() {
  65     if (!browser_session_running_) {
  66       // Post this to give AppWindows a chance to remove themselves from the
  67       // registry.
  68       base::MessageLoop::current()->PostTask(
  69           FROM_HERE, base::Bind(&TerminateIfNoAppWindows));
  70     }
  71   }
  72
  73   bool ShouldRestoreSession() {
  74     return !browser_session_running_;
  75   }
  76
  77  private:
  78   friend struct base::DefaultSingletonTraits<AppShimHandlerRegistry>;
  79   typedef std::map<std::string, AppShimHandler*> HandlerMap;
  80
  81   AppShimHandlerRegistry()
  82       : default_handler_(NULL),
  83         browser_session_running_(false) {
  84     registrar_.Add(
  85         this, chrome::NOTIFICATION_BROWSER_OPENED,
  86         content::NotificationService::AllBrowserContextsAndSources());
  87     registrar_.Add(
  88         this, chrome::NOTIFICATION_CLOSE_ALL_BROWSERS_REQUEST,
  89         content::NotificationService::AllBrowserContextsAndSources());
  90     registrar_.Add(
  91         this, chrome::NOTIFICATION_BROWSER_CLOSE_CANCELLED,
  92         content::NotificationService::AllBrowserContextsAndSources());
  93   }
  94
  95   ~AppShimHandlerRegistry() override {}
  96
  97   // content::NotificationObserver override:
  98   void Observe(int type,
  99                const content::NotificationSource& source,
 100                const content::NotificationDetails& details) override {
 101     switch (type) {
 102       case chrome::NOTIFICATION_BROWSER_OPENED:
 103       case chrome::NOTIFICATION_BROWSER_CLOSE_CANCELLED:
 104         browser_session_running_ = true;
 105         break;
 106       case chrome::NOTIFICATION_CLOSE_ALL_BROWSERS_REQUEST:
 107         browser_session_running_ = false;
 108         break;
 109       default:
 110         NOTREACHED();
 111     }
 112   }
 113
 114   HandlerMap handlers_;
 115   AppShimHandler* default_handler_;
 116   content::NotificationRegistrar registrar_;
 117   bool browser_session_running_;
 118
 119   DISALLOW_COPY_AND_ASSIGN(AppShimHandlerRegistry);
 120 };
 121
 122 }  // namespace
 123
 124 // static
 125 void AppShimHandler::RegisterHandler(const std::string& app_mode_id,
 126                                      AppShimHandler* handler) {
 127   DCHECK(handler);
 128   AppShimHandlerRegistry::GetInstance()->SetForAppMode(app_mode_id, handler);
 129 }
 130
 131 // static
 132 void AppShimHandler::RemoveHandler(const std::string& app_mode_id) {
 133   AppShimHandlerRegistry::GetInstance()->SetForAppMode(app_mode_id, NULL);
 134 }
 135
 136 // static
 137 AppShimHandler* AppShimHandler::GetForAppMode(const std::string& app_mode_id) {
 138   return AppShimHandlerRegistry::GetInstance()->GetForAppMode(app_mode_id);
 139 }
 140
 141 // static
 142 void AppShimHandler::SetDefaultHandler(AppShimHandler* handler) {
 143   AppShimHandlerRegistry::GetInstance()->SetDefaultHandler(handler);
 144 }
 145
 146 // static
 147 void AppShimHandler::MaybeTerminate() {
 148   AppShimHandlerRegistry::GetInstance()->MaybeTerminate();
 149 }
 150
 151 // static
 152 bool AppShimHandler::ShouldRestoreSession() {
 153   return AppShimHandlerRegistry::GetInstance()->ShouldRestoreSession();
 154 }
 155
 156 }  // namespace apps