| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef EXTENSIONS_BROWSER_API_POWER_POWER_API_H_ |
| #define EXTENSIONS_BROWSER_API_POWER_POWER_API_H_ |
| |
| #include <map> |
| #include <memory> |
| #include <string> |
| |
| #include "base/callback.h" |
| #include "base/macros.h" |
| #include "device/power_save_blocker/power_save_blocker.h" |
| #include "extensions/browser/browser_context_keyed_api_factory.h" |
| #include "extensions/browser/extension_function.h" |
| #include "extensions/browser/extension_registry_observer.h" |
| #include "extensions/common/api/power.h" |
| |
| namespace content { |
| class BrowserContext; |
| } |
| |
| namespace extensions { |
| |
| // Implementation of the chrome.power.requestKeepAwake API. |
| class PowerRequestKeepAwakeFunction : public UIThreadExtensionFunction { |
| public: |
| DECLARE_EXTENSION_FUNCTION("power.requestKeepAwake", POWER_REQUESTKEEPAWAKE) |
| |
| protected: |
| ~PowerRequestKeepAwakeFunction() override {} |
| |
| // ExtensionFunction: |
| ResponseAction Run() override; |
| }; |
| |
| // Implementation of the chrome.power.releaseKeepAwake API. |
| class PowerReleaseKeepAwakeFunction : public UIThreadExtensionFunction { |
| public: |
| DECLARE_EXTENSION_FUNCTION("power.releaseKeepAwake", POWER_RELEASEKEEPAWAKE) |
| |
| protected: |
| ~PowerReleaseKeepAwakeFunction() override {} |
| |
| // ExtensionFunction: |
| ResponseAction Run() override; |
| }; |
| |
| // Handles calls made via the chrome.power API. There is a separate instance of |
| // this class for each profile, as requests are tracked by extension ID, but a |
| // regular and incognito profile will share the same instance. |
| class PowerAPI : public BrowserContextKeyedAPI, |
| public extensions::ExtensionRegistryObserver { |
| public: |
| typedef base::Callback<std::unique_ptr<device::PowerSaveBlocker>( |
| device::PowerSaveBlocker::PowerSaveBlockerType, |
| device::PowerSaveBlocker::Reason, |
| const std::string&, |
| scoped_refptr<base::SequencedTaskRunner> ui_task_runner, |
| scoped_refptr<base::SingleThreadTaskRunner> file_task_runner)> |
| CreateBlockerFunction; |
| |
| static PowerAPI* Get(content::BrowserContext* context); |
| |
| // BrowserContextKeyedAPI implementation. |
| static BrowserContextKeyedAPIFactory<PowerAPI>* GetFactoryInstance(); |
| |
| // Adds an extension lock at |level| for |extension_id|, replacing the |
| // extension's existing lock, if any. |
| void AddRequest(const std::string& extension_id, api::power::Level level); |
| |
| // Removes an extension lock for an extension. Calling this for an |
| // extension id without a lock will do nothing. |
| void RemoveRequest(const std::string& extension_id); |
| |
| // Replaces the function that will be called to create PowerSaveBlocker |
| // objects. Passing an empty callback will revert to the default. |
| void SetCreateBlockerFunctionForTesting( |
| const CreateBlockerFunction& function); |
| |
| // Overridden from extensions::ExtensionRegistryObserver. |
| void OnExtensionUnloaded(content::BrowserContext* browser_context, |
| const Extension* extension, |
| UnloadedExtensionInfo::Reason reason) override; |
| |
| private: |
| friend class BrowserContextKeyedAPIFactory<PowerAPI>; |
| |
| explicit PowerAPI(content::BrowserContext* context); |
| ~PowerAPI() override; |
| |
| // Updates |power_save_blocker_| and |current_level_| after iterating |
| // over |extension_levels_|. |
| void UpdatePowerSaveBlocker(); |
| |
| // BrowserContextKeyedAPI implementation. |
| static const char* service_name() { return "PowerAPI"; } |
| static const bool kServiceRedirectedInIncognito = true; |
| static const bool kServiceIsCreatedWithBrowserContext = false; |
| void Shutdown() override; |
| |
| content::BrowserContext* browser_context_; |
| |
| // Function that should be called to create PowerSaveBlocker objects. |
| // Tests can change this to record what would've been done instead of |
| // actually changing the system power-saving settings. |
| CreateBlockerFunction create_blocker_function_; |
| |
| std::unique_ptr<device::PowerSaveBlocker> power_save_blocker_; |
| |
| // Current level used by |power_save_blocker_|. Meaningless if |
| // |power_save_blocker_| is NULL. |
| api::power::Level current_level_; |
| |
| // Map from extension ID to the corresponding level for each extension |
| // that has an outstanding request. |
| typedef std::map<std::string, api::power::Level> ExtensionLevelMap; |
| ExtensionLevelMap extension_levels_; |
| |
| DISALLOW_COPY_AND_ASSIGN(PowerAPI); |
| }; |
| |
| } // namespace extensions |
| |
| #endif // EXTENSIONS_BROWSER_API_POWER_POWER_API_H_ |