apps/app_load_service.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_load_service.h"
   6
   7 #include "apps/app_load_service_factory.h"
   8 #include "apps/app_restore_service.h"
   9 #include "apps/launcher.h"
  10 #include "chrome/browser/extensions/extension_service.h"
  11 #include "chrome/browser/extensions/unpacked_installer.h"
  12 #include "chrome/browser/profiles/profile.h"
  13 #include "content/public/browser/notification_details.h"
  14 #include "content/public/browser/notification_service.h"
  15 #include "content/public/browser/notification_types.h"
  16 #include "extensions/browser/app_window/app_window_registry.h"
  17 #include "extensions/browser/extension_host.h"
  18 #include "extensions/browser/extension_prefs.h"
  19 #include "extensions/browser/extension_registry.h"
  20 #include "extensions/browser/extension_system.h"
  21 #include "extensions/browser/notification_types.h"
  22 #include "extensions/common/extension.h"
  23
  24 using extensions::Extension;
  25 using extensions::ExtensionPrefs;
  26 using extensions::ExtensionSystem;
  27
  28 namespace apps {
  29
  30 AppLoadService::PostReloadAction::PostReloadAction()
  31     : action_type(LAUNCH_FOR_RELOAD),
  32       command_line(base::CommandLine::NO_PROGRAM) {
  33 }
  34
  35 AppLoadService::AppLoadService(Profile* profile)
  36     : profile_(profile) {
  37   registrar_.Add(this,
  38                  extensions::NOTIFICATION_EXTENSION_HOST_DID_STOP_LOADING,
  39                  content::NotificationService::AllSources());
  40   extensions::ExtensionRegistry::Get(profile_)->AddObserver(this);
  41 }
  42
  43 AppLoadService::~AppLoadService() {
  44   extensions::ExtensionRegistry::Get(profile_)->RemoveObserver(this);
  45 }
  46
  47 void AppLoadService::RestartApplication(const std::string& extension_id) {
  48   post_reload_actions_[extension_id].action_type = RESTART;
  49   ExtensionService* service = extensions::ExtensionSystem::Get(profile_)->
  50       extension_service();
  51   DCHECK(service);
  52   service->ReloadExtension(extension_id);
  53 }
  54
  55 void AppLoadService::RestartApplicationIfRunning(
  56     const std::string& extension_id) {
  57   if (apps::AppRestoreService::Get(profile_)->IsAppRestorable(extension_id))
  58     RestartApplication(extension_id);
  59 }
  60
  61 bool AppLoadService::LoadAndLaunch(const base::FilePath& extension_path,
  62                                    const base::CommandLine& command_line,
  63                                    const base::FilePath& current_dir) {
  64   ExtensionService* extension_service =
  65       ExtensionSystem::Get(profile_)->extension_service();
  66   std::string extension_id;
  67   if (!extensions::UnpackedInstaller::Create(extension_service)->
  68           LoadFromCommandLine(base::FilePath(extension_path), &extension_id)) {
  69     return false;
  70   }
  71
  72   // Schedule the app to be launched once loaded.
  73   PostReloadAction& action = post_reload_actions_[extension_id];
  74   action.action_type = LAUNCH_FOR_LOAD_AND_LAUNCH;
  75   action.command_line = command_line;
  76   action.current_dir = current_dir;
  77   return true;
  78 }
  79
  80 // static
  81 AppLoadService* AppLoadService::Get(Profile* profile) {
  82   return apps::AppLoadServiceFactory::GetForProfile(profile);
  83 }
  84
  85 void AppLoadService::Observe(int type,
  86                              const content::NotificationSource& source,
  87                              const content::NotificationDetails& details) {
  88   DCHECK_EQ(type, extensions::NOTIFICATION_EXTENSION_HOST_DID_STOP_LOADING);
  89   extensions::ExtensionHost* host =
  90       content::Details<extensions::ExtensionHost>(details).ptr();
  91   const Extension* extension = host->extension();
  92   // It is possible for an extension to be unloaded before it stops loading.
  93   if (!extension)
  94     return;
  95   std::map<std::string, PostReloadAction>::iterator it =
  96       post_reload_actions_.find(extension->id());
  97   if (it == post_reload_actions_.end())
  98     return;
  99
 100   switch (it->second.action_type) {
 101     case LAUNCH_FOR_RELOAD:
 102       LaunchPlatformApp(profile_, extension, extensions::SOURCE_RELOAD);
 103       break;
 104     case RESTART:
 105       RestartPlatformApp(profile_, extension);
 106       break;
 107     case LAUNCH_FOR_LOAD_AND_LAUNCH:
 108       LaunchPlatformAppWithCommandLine(profile_,
 109                                        extension,
 110                                        it->second.command_line,
 111                                        it->second.current_dir,
 112                                        extensions::SOURCE_LOAD_AND_LAUNCH);
 113       break;
 114     default:
 115       NOTREACHED();
 116   }
 117
 118   post_reload_actions_.erase(it);
 119 }
 120
 121 void AppLoadService::OnExtensionUnloaded(
 122     content::BrowserContext* browser_context,
 123     const Extension* extension,
 124     extensions::UnloadedExtensionInfo::Reason reason) {
 125   if (!extension->is_platform_app())
 126     return;
 127
 128   extensions::ExtensionPrefs* extension_prefs =
 129       extensions::ExtensionPrefs::Get(browser_context);
 130   if (WasUnloadedForReload(extension->id(), reason) &&
 131       extension_prefs->IsActive(extension->id()) &&
 132       !HasPostReloadAction(extension->id())) {
 133     post_reload_actions_[extension->id()].action_type = LAUNCH_FOR_RELOAD;
 134   }
 135 }
 136
 137 bool AppLoadService::WasUnloadedForReload(
 138     const extensions::ExtensionId& extension_id,
 139     const extensions::UnloadedExtensionInfo::Reason reason) {
 140   if (reason == extensions::UnloadedExtensionInfo::REASON_DISABLE) {
 141     ExtensionPrefs* prefs = ExtensionPrefs::Get(profile_);
 142     return (prefs->GetDisableReasons(extension_id) &
 143             Extension::DISABLE_RELOAD) != 0;
 144   }
 145   return false;
 146 }
 147
 148 bool AppLoadService::HasPostReloadAction(const std::string& extension_id) {
 149   return post_reload_actions_.find(extension_id) != post_reload_actions_.end();
 150 }
 151
 152 }  // namespace apps