1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
// 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 "base/stl_util.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 "url/gurl.h"
namespace image_writer_api = extensions::api::image_writer_private;
class Profile;
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:
typedef std::string ExtensionId;
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,
const Operation::StartWriteCallback& callback);
// Starts a WriteFromFile operation.
void StartWriteFromFile(const ExtensionId& extension_id,
const base::FilePath& path,
const std::string& device_path,
const Operation::StartWriteCallback& callback);
// Cancels the extensions current operation if any.
void CancelWrite(const ExtensionId& extension_id,
const Operation::CancelWriteCallback& callback);
// Starts a write that removes the partition table.
void DestroyPartitions(const ExtensionId& extension_id,
const std::string& device_path,
const 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);
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,
UnloadedExtensionInfo::Reason reason) override;
Operation* GetOperation(const ExtensionId& extension_id);
void DeleteOperation(const ExtensionId& extension_id);
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_
|