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
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
|
// Copyright (c) 2012 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_CHROMEOS_GDATA_GDATA_UPLOADER_H_
#define CHROME_BROWSER_CHROMEOS_GDATA_GDATA_UPLOADER_H_
#include <map>
#include <set>
#include <string>
#include "base/basictypes.h"
#include "base/memory/weak_ptr.h"
#include "base/platform_file.h"
#include "chrome/browser/chromeos/gdata/gdata_errorcode.h"
#include "chrome/browser/chromeos/gdata/gdata_params.h"
#include "chrome/browser/chromeos/gdata/gdata_upload_file_info.h"
#include "googleurl/src/gurl.h"
namespace content {
class DownloadItem;
}
namespace gdata {
class GDataFileSystem;
class GDataDirectoryProto;
class DocumentsServiceInterface;
class GDataUploaderInterface {
public:
~GDataUploaderInterface() {}
// Uploads a new file specified by |upload_file_info|. Transfers ownership.
// Returns the upload_id.
//
// WARNING: This is not mockable by gmock because it takes scoped_ptr<>.
// See "Announcing scoped_ptr<>::Pass(). The latest in pointer ownership
// technology!" thread on chromium-dev.
virtual int UploadNewFile(scoped_ptr<UploadFileInfo> upload_file_info) = 0;
// Uploads an existing file (a file that already exists on Drive)
// specified by |local_file_path|. The existing file on Drive will be
// updated. Returns the upload_id. |callback| is run upon completion or
// failure.
virtual int UploadExistingFile(
const GURL& upload_location,
const FilePath& gdata_file_path,
const FilePath& local_file_path,
int64 file_size,
const std::string& content_type,
const UploadFileInfo::UploadCompletionCallback& callback) = 0;
// Updates attributes of streaming upload.
virtual void UpdateUpload(int upload_id,
content::DownloadItem* download) = 0;
// Returns the count of bytes confirmed as uploaded so far.
virtual int64 GetUploadedBytes(int upload_id) const = 0;
};
class GDataUploader : public GDataUploaderInterface {
public:
explicit GDataUploader(DocumentsServiceInterface* documents_service);
virtual ~GDataUploader();
// Sets the file system. This must be called before calling other member
// functions.
//
// TODO(satorux): The dependency to GDataFileSystem should be
// eliminated. http://crbug.com/133860
void set_file_system(GDataFileSystem* file_system) {
file_system_ = file_system;
}
// GDataUploaderInterface overrides.
virtual int UploadNewFile(
scoped_ptr<UploadFileInfo> upload_file_info) OVERRIDE;
virtual int UploadExistingFile(
const GURL& upload_location,
const FilePath& gdata_file_path,
const FilePath& local_file_path,
int64 file_size,
const std::string& content_type,
const UploadFileInfo::UploadCompletionCallback& callback) OVERRIDE;
virtual void UpdateUpload(
int upload_id, content::DownloadItem* download) OVERRIDE;
virtual int64 GetUploadedBytes(int upload_id) const OVERRIDE;
private:
// Lookup UploadFileInfo* in pending_uploads_.
UploadFileInfo* GetUploadFileInfo(int upload_id) const;
// Open the file.
void OpenFile(UploadFileInfo* upload_file_info);
// net::FileStream::Open completion callback. The result of the file
// open operation is passed as |result|.
void OpenCompletionCallback(int upload_id, int result);
// DocumentsService callback for InitiateUpload.
void OnUploadLocationReceived(int upload_id,
GDataErrorCode code,
const GURL& upload_location);
// Uploads the next chunk of data from the file.
void UploadNextChunk(UploadFileInfo* upload_file_info);
// net::FileStream::Read completion callback.
void ReadCompletionCallback(int upload_id,
int bytes_to_read,
int bytes_read);
// Calls DocumentsService's ResumeUpload with the current upload info.
void ResumeUpload(int upload_id);
// DocumentsService callback for ResumeUpload.
void OnResumeUploadResponseReceived(int upload_id,
const ResumeUploadResponse& response,
scoped_ptr<DocumentEntry> entry);
// Callback for handling results of ReadDirectoryByPath() initiated by
// OpenCompletionCallback(). This callback reads the directory entry to
// determine the upload path.
void OnReadDirectoryByPath(int upload_id,
base::PlatformFileError error,
bool hide_hosted_documents,
scoped_ptr<GDataDirectoryProto> dir_proto);
// Initiate the upload.
void InitiateUpload(UploadFileInfo* uploader_file_info);
// Handle failed uploads.
void UploadFailed(scoped_ptr<UploadFileInfo> upload_file_info,
base::PlatformFileError error);
// Removes |upload_id| from UploadFileInfoMap |pending_uploads_|.
// Note that this does not delete the UploadFileInfo object itself,
// because it may still be in use by an asynchronous function.
void RemoveUpload(int upload_id);
// Starts uploading a file with |upload_file_info|. Returns a new upload
// ID assigned to |upload_file_info|.
int StartUploadFile(scoped_ptr<UploadFileInfo> upload_file_info);
// Pointers to GDataFileSystem and DocumentsServiceInterface objects owned by
// GDataSystemService. The lifetime of these two objects is guaranteed to
// exceed that of the GDataUploader instance.
GDataFileSystem* file_system_;
DocumentsServiceInterface* documents_service_;
int next_upload_id_; // id counter.
typedef std::map<int, UploadFileInfo*> UploadFileInfoMap;
UploadFileInfoMap pending_uploads_;
// Factory for various callbacks.
base::WeakPtrFactory<GDataUploader> uploader_factory_;
DISALLOW_COPY_AND_ASSIGN(GDataUploader);
};
} // namespace gdata
#endif // CHROME_BROWSER_CHROMEOS_GDATA_GDATA_UPLOADER_H_
|
