| 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 |