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 Singleton<AppShimHandlerRegistry,
  39                      LeakySingletonTraits<AppShimHandlerRegistry> >::get();
  40   }
  41
  42   AppShimHandler* GetForAppMode(const std::string& app_mode_id) const {
  43     HandlerMap::const_iterator it = handlers_.find(app_mode_id);
  44     if (it != handlers_.end())
  45       return it->second;
  46
  47     return default_handler_;
  48   }
  49
  50   bool SetForAppMode(const std::string& app_mode_id, AppShimHandler* handler) {
  51     bool inserted_or_removed = handler ?
  52         handlers_.insert(HandlerMap::value_type(app_mode_id, handler)).second :
  53         handlers_.erase(app_mode_id) == 1;
  54     DCHECK(inserted_or_removed);
  55     return inserted_or_removed;
  56   }
  57
  58   void SetDefaultHandler(AppShimHandler* handler) {
  59     DCHECK_NE(default_handler_ == NULL, handler == NULL);
  60     default_handler_ = handler;
  61   }
  62
  63   void MaybeTerminate() {
  64     if (!browser_session_running_) {
  65       // Post this to give AppWindows a chance to remove themselves from the
  66       // registry.
  67       base::MessageLoop::current()->PostTask(
  68           FROM_HERE, base::Bind(&TerminateIfNoAppWindows));
  69     }
  70   }
  71
  72   bool ShouldRestoreSession() {
  73     return !browser_session_running_;
  74   }
  75
  76  private:
  77   friend struct DefaultSingletonTraits<AppShimHandlerRegistry>;
  78   typedef std::map<std::string, AppShimHandler*> HandlerMap;
  79
  80   AppShimHandlerRegistry()
  81       : default_handler_(NULL),
  82         browser_session_running_(false) {
  83     registrar_.Add(
  84         this, chrome::NOTIFICATION_BROWSER_OPENED,
  85         content::NotificationService::AllBrowserContextsAndSources());
  86     registrar_.Add(
  87         this, chrome::NOTIFICATION_CLOSE_ALL_BROWSERS_REQUEST,
  88         content::NotificationService::AllBrowserContextsAndSources());
  89     registrar_.Add(
  90         this, chrome::NOTIFICATION_BROWSER_CLOSE_CANCELLED,
  91         content::NotificationService::AllBrowserContextsAndSources());
  92   }
  93
  94   ~AppShimHandlerRegistry() override {}
  95
  96   // content::NotificationObserver override:
  97   void Observe(int type,
  98                const content::NotificationSource& source,
  99                const content::NotificationDetails& details) override {
 100     switch (type) {
 101       case chrome::NOTIFICATION_BROWSER_OPENED:
 102       case chrome::NOTIFICATION_BROWSER_CLOSE_CANCELLED:
 103         browser_session_running_ = true;
 104         break;
 105       case chrome::NOTIFICATION_CLOSE_ALL_BROWSERS_REQUEST:
 106         browser_session_running_ = false;
 107         break;
 108       default:
 109         NOTREACHED();
 110     }
 111   }
 112
 113   HandlerMap handlers_;
 114   AppShimHandler* default_handler_;
 115   content::NotificationRegistrar registrar_;
 116   bool browser_session_running_;
 117
 118   DISALLOW_COPY_AND_ASSIGN(AppShimHandlerRegistry);
 119 };
 120
 121 }  // namespace
 122
 123 // static
 124 void AppShimHandler::RegisterHandler(const std::string& app_mode_id,
 125                                      AppShimHandler* handler) {
 126   DCHECK(handler);
 127   AppShimHandlerRegistry::GetInstance()->SetForAppMode(app_mode_id, handler);
 128 }
 129
 130 // static
 131 void AppShimHandler::RemoveHandler(const std::string& app_mode_id) {
 132   AppShimHandlerRegistry::GetInstance()->SetForAppMode(app_mode_id, NULL);
 133 }
 134
 135 // static
 136 AppShimHandler* AppShimHandler::GetForAppMode(const std::string& app_mode_id) {
 137   return AppShimHandlerRegistry::GetInstance()->GetForAppMode(app_mode_id);
 138 }
 139
 140 // static
 141 void AppShimHandler::SetDefaultHandler(AppShimHandler* handler) {
 142   AppShimHandlerRegistry::GetInstance()->SetDefaultHandler(handler);
 143 }
 144
 145 // static
 146 void AppShimHandler::MaybeTerminate() {
 147   AppShimHandlerRegistry::GetInstance()->MaybeTerminate();
 148 }
 149
 150 // static
 151 bool AppShimHandler::ShouldRestoreSession() {
 152   return AppShimHandlerRegistry::GetInstance()->ShouldRestoreSession();
 153 }
 154
 155 }  // namespace apps