| Index: extensions/browser/api/power/power_api.h
|
| diff --git a/extensions/browser/api/power/power_api.h b/extensions/browser/api/power/power_api.h
|
| index 228e517b12eb9739480992af441367f7f2656efd..3260f83fe663390a20b060ea92b071c0c428a3fe 100644
|
| --- a/extensions/browser/api/power/power_api.h
|
| +++ b/extensions/browser/api/power/power_api.h
|
| @@ -5,7 +5,20 @@
|
| #ifndef EXTENSIONS_BROWSER_API_POWER_POWER_API_H_
|
| #define EXTENSIONS_BROWSER_API_POWER_POWER_API_H_
|
|
|
| +#include <map>
|
| +#include <string>
|
| +
|
| +#include "base/callback.h"
|
| +#include "base/memory/scoped_ptr.h"
|
| +#include "content/public/browser/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 {
|
|
|
| @@ -33,6 +46,77 @@ class PowerReleaseKeepAwakeFunction : public SyncExtensionFunction {
|
| bool RunSync() 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<scoped_ptr<content::PowerSaveBlocker>(
|
| + content::PowerSaveBlocker::PowerSaveBlockerType,
|
| + content::PowerSaveBlocker::Reason,
|
| + const std::string&)> 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,
|
| + core_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(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_;
|
| +
|
| + scoped_ptr<content::PowerSaveBlocker> power_save_blocker_;
|
| +
|
| + // Current level used by |power_save_blocker_|. Meaningless if
|
| + // |power_save_blocker_| is NULL.
|
| + core_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, core_api::power::Level> ExtensionLevelMap;
|
| + ExtensionLevelMap extension_levels_;
|
| +
|
| + DISALLOW_COPY_AND_ASSIGN(PowerApi);
|
| +};
|
| +
|
| } // namespace extensions
|
|
|
| #endif // EXTENSIONS_BROWSER_API_POWER_POWER_API_H_
|
|
|