OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef WEBKIT_BROWSER_FILEAPI_FILE_WRITER_DELEGATE_H_ | 5 #include "storage/browser/fileapi/file_writer_delegate.h" |
6 #define WEBKIT_BROWSER_FILEAPI_FILE_WRITER_DELEGATE_H_ | |
7 | |
8 #include "base/files/file.h" | |
9 #include "base/files/file_path.h" | |
10 #include "base/memory/scoped_ptr.h" | |
11 #include "base/memory/weak_ptr.h" | |
12 #include "base/time/time.h" | |
13 #include "net/base/file_stream.h" | |
14 #include "net/base/io_buffer.h" | |
15 #include "net/url_request/url_request.h" | |
16 #include "webkit/browser/storage_browser_export.h" | |
17 | |
18 namespace storage { | |
19 | |
20 class FileStreamWriter; | |
21 | |
22 class STORAGE_EXPORT_PRIVATE FileWriterDelegate | |
23 : public net::URLRequest::Delegate { | |
24 public: | |
25 enum FlushPolicy { | |
26 FLUSH_ON_COMPLETION, | |
27 NO_FLUSH_ON_COMPLETION, | |
28 }; | |
29 | |
30 enum WriteProgressStatus { | |
31 SUCCESS_IO_PENDING, | |
32 SUCCESS_COMPLETED, | |
33 ERROR_WRITE_STARTED, | |
34 ERROR_WRITE_NOT_STARTED, | |
35 }; | |
36 | |
37 typedef base::Callback<void(base::File::Error result, | |
38 int64 bytes, | |
39 WriteProgressStatus write_status)> | |
40 DelegateWriteCallback; | |
41 | |
42 FileWriterDelegate(scoped_ptr<FileStreamWriter> file_writer, | |
43 FlushPolicy flush_policy); | |
44 virtual ~FileWriterDelegate(); | |
45 | |
46 void Start(scoped_ptr<net::URLRequest> request, | |
47 const DelegateWriteCallback& write_callback); | |
48 | |
49 // Cancels the current write operation. This will synchronously or | |
50 // asynchronously call the given write callback (which may result in | |
51 // deleting this). | |
52 void Cancel(); | |
53 | |
54 virtual void OnReceivedRedirect(net::URLRequest* request, | |
55 const net::RedirectInfo& redirect_info, | |
56 bool* defer_redirect) OVERRIDE; | |
57 virtual void OnAuthRequired(net::URLRequest* request, | |
58 net::AuthChallengeInfo* auth_info) OVERRIDE; | |
59 virtual void OnCertificateRequested( | |
60 net::URLRequest* request, | |
61 net::SSLCertRequestInfo* cert_request_info) OVERRIDE; | |
62 virtual void OnSSLCertificateError(net::URLRequest* request, | |
63 const net::SSLInfo& ssl_info, | |
64 bool fatal) OVERRIDE; | |
65 virtual void OnResponseStarted(net::URLRequest* request) OVERRIDE; | |
66 virtual void OnReadCompleted(net::URLRequest* request, | |
67 int bytes_read) OVERRIDE; | |
68 | |
69 private: | |
70 void OnGetFileInfoAndStartRequest( | |
71 scoped_ptr<net::URLRequest> request, | |
72 base::File::Error error, | |
73 const base::File::Info& file_info); | |
74 void Read(); | |
75 void OnDataReceived(int bytes_read); | |
76 void Write(); | |
77 void OnDataWritten(int write_response); | |
78 void OnError(base::File::Error error); | |
79 void OnProgress(int bytes_read, bool done); | |
80 void OnWriteCancelled(int status); | |
81 void MaybeFlushForCompletion(base::File::Error error, | |
82 int bytes_written, | |
83 WriteProgressStatus progress_status); | |
84 void OnFlushed(base::File::Error error, | |
85 int bytes_written, | |
86 WriteProgressStatus progress_status, | |
87 int flush_error); | |
88 | |
89 WriteProgressStatus GetCompletionStatusOnError() const; | |
90 | |
91 DelegateWriteCallback write_callback_; | |
92 scoped_ptr<FileStreamWriter> file_stream_writer_; | |
93 base::Time last_progress_event_time_; | |
94 bool writing_started_; | |
95 FlushPolicy flush_policy_; | |
96 int bytes_written_backlog_; | |
97 int bytes_written_; | |
98 int bytes_read_; | |
99 scoped_refptr<net::IOBufferWithSize> io_buffer_; | |
100 scoped_refptr<net::DrainableIOBuffer> cursor_; | |
101 scoped_ptr<net::URLRequest> request_; | |
102 | |
103 base::WeakPtrFactory<FileWriterDelegate> weak_factory_; | |
104 | |
105 DISALLOW_COPY_AND_ASSIGN(FileWriterDelegate); | |
106 }; | |
107 | |
108 } // namespace storage | |
109 | |
110 #endif // WEBKIT_BROWSER_FILEAPI_FILE_WRITER_DELEGATE_H_ | |
OLD | NEW |