storage/browser/fileapi/file_writer_delegate.h

   1 // Copyright (c) 2012 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.
   4
   5 #ifndef STORAGE_BROWSER_FILEAPI_FILE_WRITER_DELEGATE_H_
   6 #define STORAGE_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 "storage/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  // STORAGE_BROWSER_FILEAPI_FILE_WRITER_DELEGATE_H_