| // Copyright 2013 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 CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_OPERATION_MANAGER_H_ |
| #define CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_OPERATION_MANAGER_H_ |
| |
| #include <map> |
| #include <string> |
| #include "base/bind.h" |
| #include "base/callback.h" |
| #include "base/macros.h" |
| #include "base/memory/weak_ptr.h" |
| #include "base/scoped_observer.h" |
| #include "chrome/browser/extensions/api/image_writer_private/image_writer_private_api.h" |
| #include "chrome/browser/extensions/api/image_writer_private/operation.h" |
| #include "chrome/common/extensions/api/image_writer_private.h" |
| #include "content/public/browser/notification_observer.h" |
| #include "content/public/browser/notification_registrar.h" |
| #include "extensions/browser/browser_context_keyed_api_factory.h" |
| #include "extensions/browser/extension_registry_observer.h" |
| #include "extensions/common/extension_id.h" |
| #include "url/gurl.h" |
| |
| namespace image_writer_api = extensions::api::image_writer_private; |
| |
| namespace content { |
| class BrowserContext; |
| } |
| |
| namespace extensions { |
| class ExtensionRegistry; |
| |
| namespace image_writer { |
| |
| class Operation; |
| |
| // Manages image writer operations for the current profile. Including clean-up |
| // and message routing. |
| class OperationManager : public BrowserContextKeyedAPI, |
| public content::NotificationObserver, |
| public extensions::ExtensionRegistryObserver, |
| public base::SupportsWeakPtr<OperationManager> { |
| public: |
| explicit OperationManager(content::BrowserContext* context); |
| ~OperationManager() override; |
| |
| void Shutdown() override; |
| |
| // Starts a WriteFromUrl operation. |
| void StartWriteFromUrl(const ExtensionId& extension_id, |
| GURL url, |
| const std::string& hash, |
| const std::string& device_path, |
| Operation::StartWriteCallback callback); |
| |
| // Starts a WriteFromFile operation. |
| void StartWriteFromFile(const ExtensionId& extension_id, |
| const base::FilePath& path, |
| const std::string& device_path, |
| Operation::StartWriteCallback callback); |
| |
| // Cancels the extensions current operation if any. |
| void CancelWrite(const ExtensionId& extension_id, |
| Operation::CancelWriteCallback callback); |
| |
| // Starts a write that removes the partition table. |
| void DestroyPartitions(const ExtensionId& extension_id, |
| const std::string& device_path, |
| Operation::StartWriteCallback callback); |
| |
| // Callback for progress events. |
| virtual void OnProgress(const ExtensionId& extension_id, |
| image_writer_api::Stage stage, |
| int progress); |
| // Callback for completion events. |
| virtual void OnComplete(const ExtensionId& extension_id); |
| |
| // Callback for error events. |
| virtual void OnError(const ExtensionId& extension_id, |
| image_writer_api::Stage stage, |
| int progress, |
| const std::string& error_message); |
| |
| // BrowserContextKeyedAPI |
| static BrowserContextKeyedAPIFactory<OperationManager>* GetFactoryInstance(); |
| static OperationManager* Get(content::BrowserContext* context); |
| |
| protected: |
| // Overridden in test. |
| virtual std::unique_ptr<service_manager::Connector> CreateConnector(); |
| |
| private: |
| static const char* service_name() { |
| return "OperationManager"; |
| } |
| |
| // NotificationObserver implementation. |
| void Observe(int type, |
| const content::NotificationSource& source, |
| const content::NotificationDetails& details) override; |
| |
| // ExtensionRegistryObserver implementation. |
| void OnExtensionUnloaded(content::BrowserContext* browser_context, |
| const Extension* extension, |
| UnloadedExtensionReason reason) override; |
| |
| Operation* GetOperation(const ExtensionId& extension_id); |
| void DeleteOperation(const ExtensionId& extension_id); |
| |
| // Accessor to the associated profile download folder |
| base::FilePath GetAssociatedDownloadFolder(); |
| |
| friend class BrowserContextKeyedAPIFactory<OperationManager>; |
| typedef std::map<ExtensionId, scoped_refptr<Operation> > OperationMap; |
| |
| content::BrowserContext* browser_context_; |
| OperationMap operations_; |
| content::NotificationRegistrar registrar_; |
| |
| // Listen to extension unloaded notification. |
| ScopedObserver<ExtensionRegistry, ExtensionRegistryObserver> |
| extension_registry_observer_; |
| |
| base::WeakPtrFactory<OperationManager> weak_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(OperationManager); |
| }; |
| |
| } // namespace image_writer |
| } // namespace extensions |
| |
| #endif // CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_OPERATION_MANAGER_H_ |