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 3// found in the LICENSE file. 4 5#ifndef CHROME_BROWSER_CHROMEOS_DRIVE_FILEAPI_WEBKIT_FILE_STREAM_WRITER_IMPL_H_ 6#define CHROME_BROWSER_CHROMEOS_DRIVE_FILEAPI_WEBKIT_FILE_STREAM_WRITER_IMPL_H_ 7 8#include "base/basictypes.h" 9#include "base/callback.h" 10#include "base/files/file.h" 11#include "base/files/file_path.h" 12#include "base/memory/scoped_ptr.h" 13#include "base/memory/weak_ptr.h" 14#include "storage/browser/fileapi/file_stream_writer.h" 15 16namespace base { 17class TaskRunner; 18} // namespace base 19 20namespace net { 21class IOBuffer; 22} // namespace net 23 24namespace drive { 25 26class FileSystemInterface; 27 28namespace internal { 29 30// The implementation of storage::FileStreamWriter for the Drive File System. 31class WebkitFileStreamWriterImpl : public storage::FileStreamWriter { 32 public: 33 // Callback to return the FileSystemInterface instance. This is an 34 // injecting point for testing. 35 // Note that the callback will be copied between threads (IO and UI), and 36 // will be called on UI thread. 37 typedef base::Callback<FileSystemInterface*()> FileSystemGetter; 38 39 // Creates a writer for a file at |file_path| on FileSystem returned by 40 // |file_system_getter| that starts writing from |offset|. 41 // When invalid parameters are set, the first call to Write() method fails. 42 // Uses |file_task_runner| for local file operations. 43 WebkitFileStreamWriterImpl(const FileSystemGetter& file_system_getter, 44 base::TaskRunner* file_task_runner, 45 const base::FilePath& file_path, 46 int64 offset); 47 virtual ~WebkitFileStreamWriterImpl(); 48 49 // FileWriter override. 50 virtual int Write(net::IOBuffer* buf, int buf_len, 51 const net::CompletionCallback& callback) OVERRIDE; 52 virtual int Cancel(const net::CompletionCallback& callback) OVERRIDE; 53 virtual int Flush(const net::CompletionCallback& callback) OVERRIDE; 54 55 private: 56 // Part of Write(). Called after CreateWritableSnapshotFile is completed. 57 void WriteAfterCreateWritableSnapshotFile( 58 net::IOBuffer* buf, 59 int buf_len, 60 base::File::Error open_result, 61 const base::FilePath& local_path, 62 const base::Closure& close_callback_on_ui_thread); 63 64 FileSystemGetter file_system_getter_; 65 scoped_refptr<base::TaskRunner> file_task_runner_; 66 const base::FilePath file_path_; 67 const int64 offset_; 68 69 scoped_ptr<storage::FileStreamWriter> local_file_writer_; 70 base::Closure close_callback_on_ui_thread_; 71 net::CompletionCallback pending_write_callback_; 72 net::CompletionCallback pending_cancel_callback_; 73 74 // Note: This should remain the last member so it'll be destroyed and 75 // invalidate the weak pointers before any other members are destroyed. 76 base::WeakPtrFactory<WebkitFileStreamWriterImpl> weak_ptr_factory_; 77 78 DISALLOW_COPY_AND_ASSIGN(WebkitFileStreamWriterImpl); 79}; 80 81} // namespace internal 82} // namespace drive 83 84#endif // CHROME_BROWSER_CHROMEOS_DRIVE_FILEAPI_WEBKIT_FILE_STREAM_WRITER_IMPL_H_ 85