| // Copyright (c) 2011 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #ifndef WEBKIT_TOOLS_TEST_SHELL_SIMPLE_FILE_WRITER_H_ |
| #define WEBKIT_TOOLS_TEST_SHELL_SIMPLE_FILE_WRITER_H_ |
| |
| #include "base/memory/ref_counted.h" |
| #include "base/memory/weak_ptr.h" |
| #include "webkit/fileapi/webfilewriter_base.h" |
| |
| namespace net { |
| class URLRequestContext; |
| } // namespace net |
| |
| namespace fileapi { |
| class FileSystemContext; |
| } |
| |
| // An implementation of WebFileWriter for use in test_shell and DRT. |
| class SimpleFileWriter : public fileapi::WebFileWriterBase, |
| public base::SupportsWeakPtr<SimpleFileWriter> { |
| public: |
| SimpleFileWriter( |
| const GURL& path, |
| WebKit::WebFileWriterClient* client, |
| fileapi::FileSystemContext* file_system_context); |
| virtual ~SimpleFileWriter(); |
| |
| // Called by SimpleResourceLoaderBridge when the context is |
| // created and destroyed. |
| static void InitializeOnIOThread(net::URLRequestContext* request_context) { |
| request_context_ = request_context; |
| } |
| static void CleanupOnIOThread() { |
| request_context_ = NULL; |
| } |
| |
| protected: |
| // WebFileWriterBase overrides |
| virtual void DoTruncate(const GURL& path, int64 offset); |
| virtual void DoWrite(const GURL& path, const GURL& blob_url, |
| int64 offset); |
| virtual void DoCancel(); |
| |
| private: |
| class IOThreadProxy; |
| scoped_refptr<IOThreadProxy> io_thread_proxy_; |
| static net::URLRequestContext* request_context_; |
| }; |
| |
| #endif // WEBKIT_TOOLS_TEST_SHELL_SIMPLE_FILE_WRITER_H_ |