1 // Copyright 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.
5 #ifndef CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_OPERATION_MANAGER_H_
6 #define CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_OPERATION_MANAGER_H_
10 #include "base/bind.h"
11 #include "base/callback.h"
12 #include "base/memory/weak_ptr.h"
13 #include "base/stl_util.h"
14 #include "base/timer/timer.h"
15 #include "chrome/browser/extensions/api/image_writer_private/image_writer_private_api.h"
16 #include "chrome/browser/extensions/api/image_writer_private/operation.h"
17 #include "chrome/common/extensions/api/image_writer_private.h"
18 #include "content/public/browser/notification_observer.h"
19 #include "content/public/browser/notification_registrar.h"
20 #include "extensions/browser/browser_context_keyed_api_factory.h"
23 namespace image_writer_api = extensions::api::image_writer_private;
31 namespace extensions {
32 namespace image_writer {
36 // Manages image writer operations for the current profile. Including clean-up
37 // and message routing.
38 class OperationManager : public BrowserContextKeyedAPI,
39 public content::NotificationObserver,
40 public base::SupportsWeakPtr<OperationManager> {
42 typedef std::string ExtensionId;
44 explicit OperationManager(content::BrowserContext* context);
45 virtual ~OperationManager();
47 virtual void Shutdown() OVERRIDE;
49 // Starts a WriteFromUrl operation.
50 void StartWriteFromUrl(const ExtensionId& extension_id,
52 const std::string& hash,
53 const std::string& device_path,
54 const Operation::StartWriteCallback& callback);
56 // Starts a WriteFromFile operation.
57 void StartWriteFromFile(const ExtensionId& extension_id,
58 const base::FilePath& path,
59 const std::string& device_path,
60 const Operation::StartWriteCallback& callback);
62 // Cancels the extensions current operation if any.
63 void CancelWrite(const ExtensionId& extension_id,
64 const Operation::CancelWriteCallback& callback);
66 // Starts a write that removes the partition table.
67 void DestroyPartitions(const ExtensionId& extension_id,
68 const std::string& device_path,
69 const Operation::StartWriteCallback& callback);
71 // Callback for progress events.
72 virtual void OnProgress(const ExtensionId& extension_id,
73 image_writer_api::Stage stage,
75 // Callback for completion events.
76 virtual void OnComplete(const ExtensionId& extension_id);
78 // Callback for error events.
79 virtual void OnError(const ExtensionId& extension_id,
80 image_writer_api::Stage stage,
82 const std::string& error_message);
84 // BrowserContextKeyedAPI
85 static BrowserContextKeyedAPIFactory<OperationManager>* GetFactoryInstance();
86 static OperationManager* Get(content::BrowserContext* context);
88 Profile* profile() { return profile_; }
92 static const char* service_name() {
93 return "OperationManager";
96 // NotificationObserver
97 virtual void Observe(int type,
98 const content::NotificationSource& source,
99 const content::NotificationDetails& details) OVERRIDE;
101 Operation* GetOperation(const ExtensionId& extension_id);
102 void DeleteOperation(const ExtensionId& extension_id);
104 friend class BrowserContextKeyedAPIFactory<OperationManager>;
105 typedef std::map<ExtensionId, scoped_refptr<Operation> > OperationMap;
108 OperationMap operations_;
109 content::NotificationRegistrar registrar_;
111 base::WeakPtrFactory<OperationManager> weak_factory_;
113 DISALLOW_COPY_AND_ASSIGN(OperationManager);
116 } // namespace image_writer
117 } // namespace extensions
119 #endif // CHROME_BROWSER_EXTENSIONS_API_IMAGE_WRITER_PRIVATE_OPERATION_MANAGER_H_