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