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