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 "apps/app_shim/app_shim_handler_mac.h"
   6
   7 #include <map>
   8
   9 #include "apps/shell_window_registry.h"
  10 #include "base/bind.h"
  11 #include "base/logging.h"
  12 #include "base/memory/singleton.h"
  13 #include "base/message_loop/message_loop.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 TerminateIfNoShellWindows() {
  26   bool shell_windows_left =
  27       apps::ShellWindowRegistry::IsShellWindowRegisteredInAnyProfile(0);
  28   if (!shell_windows_left && !AppListService::Get()->IsAppListVisible())
  29     chrome::AttemptExit();
  30 }
  31
  32 class AppShimHandlerRegistry : public content::NotificationObserver {
  33  public:
  34   static AppShimHandlerRegistry* GetInstance() {
  35     return Singleton<AppShimHandlerRegistry,
  36                      LeakySingletonTraits<AppShimHandlerRegistry> >::get();
  37   }
  38
  39   AppShimHandler* GetForAppMode(const std::string& app_mode_id) const {
  40     HandlerMap::const_iterator it = handlers_.find(app_mode_id);
  41     if (it != handlers_.end())
  42       return it->second;
  43
  44     return default_handler_;
  45   }
  46
  47   bool SetForAppMode(const std::string& app_mode_id, AppShimHandler* handler) {
  48     bool inserted_or_removed = handler ?
  49         handlers_.insert(HandlerMap::value_type(app_mode_id, handler)).second :
  50         handlers_.erase(app_mode_id) == 1;
  51     DCHECK(inserted_or_removed);
  52     return inserted_or_removed;
  53   }
  54
  55   void SetDefaultHandler(AppShimHandler* handler) {
  56     DCHECK_NE(default_handler_ == NULL, handler == NULL);
  57     default_handler_ = handler;
  58   }
  59
  60   void MaybeTerminate() {
  61     if (!browser_opened_ever_) {
  62       // Post this to give ShellWindows a chance to remove themselves from the
  63       // registry.
  64       base::MessageLoop::current()->PostTask(
  65           FROM_HERE,
  66           base::Bind(&TerminateIfNoShellWindows));
  67     }
  68   }
  69
  70  private:
  71   friend struct DefaultSingletonTraits<AppShimHandlerRegistry>;
  72   typedef std::map<std::string, AppShimHandler*> HandlerMap;
  73
  74   AppShimHandlerRegistry()
  75       : default_handler_(NULL),
  76         browser_opened_ever_(false) {
  77     registrar_.Add(
  78         this, chrome::NOTIFICATION_BROWSER_OPENED,
  79         content::NotificationService::AllBrowserContextsAndSources());
  80   }
  81
  82   virtual ~AppShimHandlerRegistry() {}
  83
  84   // content::NotificationObserver override:
  85   virtual void Observe(
  86       int type,
  87       const content::NotificationSource& source,
  88       const content::NotificationDetails& details) OVERRIDE {
  89     DCHECK_EQ(chrome::NOTIFICATION_BROWSER_OPENED, type);
  90     registrar_.Remove(
  91         this, chrome::NOTIFICATION_BROWSER_OPENED,
  92         content::NotificationService::AllBrowserContextsAndSources());
  93     browser_opened_ever_ = true;
  94   }
  95
  96   HandlerMap handlers_;
  97   AppShimHandler* default_handler_;
  98   content::NotificationRegistrar registrar_;
  99   bool browser_opened_ever_;
 100
 101   DISALLOW_COPY_AND_ASSIGN(AppShimHandlerRegistry);
 102 };
 103
 104 }  // namespace
 105
 106 // static
 107 void AppShimHandler::RegisterHandler(const std::string& app_mode_id,
 108                                      AppShimHandler* handler) {
 109   DCHECK(handler);
 110   AppShimHandlerRegistry::GetInstance()->SetForAppMode(app_mode_id, handler);
 111 }
 112
 113 // static
 114 void AppShimHandler::RemoveHandler(const std::string& app_mode_id) {
 115   AppShimHandlerRegistry::GetInstance()->SetForAppMode(app_mode_id, NULL);
 116 }
 117
 118 // static
 119 AppShimHandler* AppShimHandler::GetForAppMode(const std::string& app_mode_id) {
 120   return AppShimHandlerRegistry::GetInstance()->GetForAppMode(app_mode_id);
 121 }
 122
 123 // static
 124 void AppShimHandler::SetDefaultHandler(AppShimHandler* handler) {
 125   AppShimHandlerRegistry::GetInstance()->SetDefaultHandler(handler);
 126 }
 127
 128 // static
 129 void AppShimHandler::MaybeTerminate() {
 130   AppShimHandlerRegistry::GetInstance()->MaybeTerminate();
 131 }
 132
 133 }  // namespace apps