Extend simple_file_system to use SandboxedFileSystemOperation

webkit/tools/test_shell/simple_file_system.h

 // Copyright (c) 2010 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_SYSTEM_H_
 #define WEBKIT_TOOLS_TEST_SHELL_SIMPLE_FILE_SYSTEM_H_
 
 #include <vector>
 #include "base/file_util_proxy.h"
 #include "base/id_map.h"
+#include "base/scoped_temp_dir.h"
+#include "base/weak_ptr.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebFileSystem.h"
-#include "third_party/WebKit/WebKit/chromium/public/WebFileSystemCallbacks.h"
-#include "webkit/fileapi/file_system_operation.h"
+#include "webkit/fileapi/file_system_types.h"
 
-class SimpleFileSystem : public WebKit::WebFileSystem {
+namespace WebKit {
+class WebFileSystemCallbacks;
+class WebFrame;
+}
+
+namespace fileapi {
+class SandboxedFileSystemContext;
+class SandboxedFileSystemOperation;
+}
+
+class SimpleFileSystem
+    : public WebKit::WebFileSystem,
+      public base::SupportsWeakPtr<SimpleFileSystem> {
  public:
-  SimpleFileSystem() {}
+  SimpleFileSystem();
   virtual ~SimpleFileSystem();
 
-  void RemoveCompletedOperation(int request_id);
+  void OpenFileSystem(WebKit::WebFrame* frame,
+                      WebKit::WebFileSystem::Type type,
+                      long long size,
+                      bool create,
+                      WebKit::WebFileSystemCallbacks* callbacks);
 
   // WebKit::WebFileSystem methods.
   virtual void move(const WebKit::WebString& src_path,
                     const WebKit::WebString& dest_path,
                     WebKit::WebFileSystemCallbacks* callbacks);
   virtual void copy(const WebKit::WebString& src_path,
                     const WebKit::WebString& dest_path,
                     WebKit::WebFileSystemCallbacks* callbacks);
   virtual void remove(const WebKit::WebString& path,
                       WebKit::WebFileSystemCallbacks* callbacks);
@@ skipping 11 matching lines @@
                           WebKit::WebFileSystemCallbacks* callbacks);
   virtual void directoryExists(const WebKit::WebString& path,
                                WebKit::WebFileSystemCallbacks* callbacks);
   virtual void readDirectory(const WebKit::WebString& path,
                              WebKit::WebFileSystemCallbacks* callbacks);
   virtual WebKit::WebFileWriter* createFileWriter(
       const WebKit::WebString& path, WebKit::WebFileWriterClient* client);
 
  private:
   // Helpers.
-  fileapi::FileSystemOperation* GetNewOperation(
+  fileapi::SandboxedFileSystemOperation* GetNewOperation(
       WebKit::WebFileSystemCallbacks* callbacks);
 
-  // Keeps ongoing file system operations.
-  typedef IDMap<fileapi::FileSystemOperation, IDMapOwnPointer> OperationsMap;
-  OperationsMap operations_;
+  // A temporary directory for FileSystem API.
+  ScopedTempDir file_system_dir_;
+
+  scoped_ptr<fileapi::SandboxedFileSystemContext> sandboxed_context_;
 
   DISALLOW_COPY_AND_ASSIGN(SimpleFileSystem);
 };
 
 #endif  // WEBKIT_TOOLS_TEST_SHELL_SIMPLE_FILE_SYSTEM_H_