| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 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/extensions/api/power/power_api_manager.h" | |
| 6 | |
| 7 #include "base/bind.h" | |
| 8 #include "base/lazy_instance.h" | |
| 9 #include "chrome/browser/chrome_notification_types.h" | |
| 10 #include "content/public/browser/notification_service.h" | |
| 11 #include "extensions/browser/extension_registry.h" | |
| 12 #include "extensions/common/extension.h" | |
| 13 | |
| 14 namespace extensions { | |
| 15 | |
| 16 namespace { | |
| 17 | |
| 18 const char kPowerSaveBlockerReason[] = "extension"; | |
| 19 | |
| 20 content::PowerSaveBlocker::PowerSaveBlockerType | |
| 21 LevelToPowerSaveBlockerType(api::power::Level level) { | |
| 22 switch (level) { | |
| 23 case api::power::LEVEL_SYSTEM: | |
| 24 return content::PowerSaveBlocker::kPowerSaveBlockPreventAppSuspension; | |
| 25 case api::power::LEVEL_DISPLAY: // fallthrough | |
| 26 case api::power::LEVEL_NONE: | |
| 27 return content::PowerSaveBlocker::kPowerSaveBlockPreventDisplaySleep; | |
| 28 } | |
| 29 NOTREACHED() << "Unhandled level " << level; | |
| 30 return content::PowerSaveBlocker::kPowerSaveBlockPreventDisplaySleep; | |
| 31 } | |
| 32 | |
| 33 base::LazyInstance<BrowserContextKeyedAPIFactory<PowerApiManager> > g_factory = | |
| 34 LAZY_INSTANCE_INITIALIZER; | |
| 35 | |
| 36 } // namespace | |
| 37 | |
| 38 // static | |
| 39 PowerApiManager* PowerApiManager::Get(content::BrowserContext* context) { | |
| 40 return BrowserContextKeyedAPIFactory<PowerApiManager>::Get(context); | |
| 41 } | |
| 42 | |
| 43 // static | |
| 44 BrowserContextKeyedAPIFactory<PowerApiManager>* | |
| 45 PowerApiManager::GetFactoryInstance() { | |
| 46 return g_factory.Pointer(); | |
| 47 } | |
| 48 | |
| 49 void PowerApiManager::AddRequest(const std::string& extension_id, | |
| 50 api::power::Level level) { | |
| 51 extension_levels_[extension_id] = level; | |
| 52 UpdatePowerSaveBlocker(); | |
| 53 } | |
| 54 | |
| 55 void PowerApiManager::RemoveRequest(const std::string& extension_id) { | |
| 56 extension_levels_.erase(extension_id); | |
| 57 UpdatePowerSaveBlocker(); | |
| 58 } | |
| 59 | |
| 60 void PowerApiManager::SetCreateBlockerFunctionForTesting( | |
| 61 CreateBlockerFunction function) { | |
| 62 create_blocker_function_ = !function.is_null() ? function : | |
| 63 base::Bind(&content::PowerSaveBlocker::Create); | |
| 64 } | |
| 65 | |
| 66 void PowerApiManager::Observe(int type, | |
| 67 const content::NotificationSource& source, | |
| 68 const content::NotificationDetails& details) { | |
| 69 DCHECK_EQ(type, chrome::NOTIFICATION_APP_TERMINATING); | |
| 70 power_save_blocker_.reset(); | |
| 71 } | |
| 72 | |
| 73 void PowerApiManager::OnExtensionUnloaded( | |
| 74 content::BrowserContext* browser_context, | |
| 75 const Extension* extension, | |
| 76 UnloadedExtensionInfo::Reason reason) { | |
| 77 RemoveRequest(extension->id()); | |
| 78 UpdatePowerSaveBlocker(); | |
| 79 } | |
| 80 | |
| 81 PowerApiManager::PowerApiManager(content::BrowserContext* context) | |
| 82 : browser_context_(context), | |
| 83 create_blocker_function_(base::Bind(&content::PowerSaveBlocker::Create)), | |
| 84 current_level_(api::power::LEVEL_SYSTEM) { | |
| 85 ExtensionRegistry::Get(browser_context_)->AddObserver(this); | |
| 86 registrar_.Add(this, chrome::NOTIFICATION_APP_TERMINATING, | |
| 87 content::NotificationService::AllSources()); | |
| 88 } | |
| 89 | |
| 90 PowerApiManager::~PowerApiManager() {} | |
| 91 | |
| 92 void PowerApiManager::UpdatePowerSaveBlocker() { | |
| 93 if (extension_levels_.empty()) { | |
| 94 power_save_blocker_.reset(); | |
| 95 return; | |
| 96 } | |
| 97 | |
| 98 api::power::Level new_level = api::power::LEVEL_SYSTEM; | |
| 99 for (ExtensionLevelMap::const_iterator it = extension_levels_.begin(); | |
| 100 it != extension_levels_.end(); ++it) { | |
| 101 if (it->second == api::power::LEVEL_DISPLAY) | |
| 102 new_level = it->second; | |
| 103 } | |
| 104 | |
| 105 // If the level changed and we need to create a new blocker, do a swap | |
| 106 // to ensure that there isn't a brief period where power management is | |
| 107 // unblocked. | |
| 108 if (!power_save_blocker_ || new_level != current_level_) { | |
| 109 content::PowerSaveBlocker::PowerSaveBlockerType type = | |
| 110 LevelToPowerSaveBlockerType(new_level); | |
| 111 scoped_ptr<content::PowerSaveBlocker> new_blocker( | |
| 112 create_blocker_function_.Run(type, kPowerSaveBlockerReason)); | |
| 113 power_save_blocker_.swap(new_blocker); | |
| 114 current_level_ = new_level; | |
| 115 } | |
| 116 } | |
| 117 | |
| 118 void PowerApiManager::Shutdown() { | |
| 119 // Unregister here rather than in the d'tor; otherwise this call will recreate | |
| 120 // the already-deleted ExtensionRegistry. | |
| 121 ExtensionRegistry::Get(browser_context_)->RemoveObserver(this); | |
| 122 } | |
| 123 | |
| 124 } // namespace extensions | |
| OLD | NEW |