Add truncate and cancel for FileWriter; write and tests will come in later CL...

webkit/fileapi/file_system_operation.cc

 // 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.
 
 #include "webkit/fileapi/file_system_operation.h"
 
 #include "base/time.h"
 #include "googleurl/src/gurl.h"
 #include "webkit/fileapi/file_system_callback_dispatcher.h"
 
 namespace fileapi {
 
 FileSystemOperation::FileSystemOperation(
     FileSystemCallbackDispatcher* dispatcher,
     scoped_refptr<base::MessageLoopProxy> proxy)
     : proxy_(proxy),
       dispatcher_(dispatcher),
-      callback_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)) {
+      callback_factory_(ALLOW_THIS_IN_INITIALIZER_LIST(this)),
+      cancel_operation_(NULL) {
   DCHECK(dispatcher);
 #ifndef NDEBUG
-  operation_pending_ = false;
+  pending_operation_ = kOperationNone;
 #endif
 }
 
 FileSystemOperation::~FileSystemOperation() {
 }
 
 void FileSystemOperation::CreateFile(const FilePath& path,
                                      bool exclusive) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationCreateFile;
 #endif
 
   base::FileUtilProxy::CreateOrOpen(
     proxy_, path, base::PLATFORM_FILE_CREATE | base::PLATFORM_FILE_READ,
     callback_factory_.NewCallback(
         exclusive ? &FileSystemOperation::DidCreateFileExclusive
                   : &FileSystemOperation::DidCreateFileNonExclusive));
 }
 
 void FileSystemOperation::CreateDirectory(const FilePath& path,
                                           bool exclusive,
                                           bool recursive) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationCreateDirectory;
 #endif
 
   base::FileUtilProxy::CreateDirectory(
       proxy_, path, exclusive, recursive, callback_factory_.NewCallback(
           &FileSystemOperation::DidFinishFileOperation));
 }
 
 void FileSystemOperation::Copy(const FilePath& src_path,
                                const FilePath& dest_path) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationCopy;
 #endif
 
   base::FileUtilProxy::Copy(proxy_, src_path, dest_path,
       callback_factory_.NewCallback(
           &FileSystemOperation::DidFinishFileOperation));
 }
 
 void FileSystemOperation::Move(const FilePath& src_path,
                                const FilePath& dest_path) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationMove;
 #endif
 
   base::FileUtilProxy::Move(proxy_, src_path, dest_path,
       callback_factory_.NewCallback(
           &FileSystemOperation::DidFinishFileOperation));
 }
 
 void FileSystemOperation::DirectoryExists(const FilePath& path) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationDirectoryExists;
 #endif
 
   base::FileUtilProxy::GetFileInfo(proxy_, path, callback_factory_.NewCallback(
       &FileSystemOperation::DidDirectoryExists));
 }
 
 void FileSystemOperation::FileExists(const FilePath& path) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationFileExists;
 #endif
 
   base::FileUtilProxy::GetFileInfo(proxy_, path, callback_factory_.NewCallback(
       &FileSystemOperation::DidFileExists));
 }
 
 void FileSystemOperation::GetMetadata(const FilePath& path) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationGetMetadata;
 #endif
 
   base::FileUtilProxy::GetFileInfo(proxy_, path, callback_factory_.NewCallback(
       &FileSystemOperation::DidGetMetadata));
 }
 
 void FileSystemOperation::ReadDirectory(const FilePath& path) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationReadDirectory;
 #endif
 
   base::FileUtilProxy::ReadDirectory(proxy_, path,
       callback_factory_.NewCallback(
           &FileSystemOperation::DidReadDirectory));
 }
 
 void FileSystemOperation::Remove(const FilePath& path) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationRemove;
 #endif
 
   base::FileUtilProxy::Delete(proxy_, path, callback_factory_.NewCallback(
       &FileSystemOperation::DidFinishFileOperation));
 }
 
 void FileSystemOperation::Write(
-    const FilePath&,
-    const GURL&,
-    int64) {
+    const FilePath& path,
+    const GURL& blob_url,
+    int64 offset) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationWrite;
 #endif
   NOTREACHED();
-  // TODO(ericu):
-  // Set up a loop that, via multiple callback invocations, reads from a
-  // URLRequest wrapping blob_url, writes the bytes to the file, reports
-  // progress events no more frequently than some set rate, and periodically
-  // checks to see if it's been cancelled.
 }
 
 void FileSystemOperation::Truncate(const FilePath& path, int64 length) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationTruncate;
 #endif
-  // TODO(ericu):
-  NOTREACHED();
+  base::FileUtilProxy::Truncate(proxy_, path, length,
+      callback_factory_.NewCallback(
+          &FileSystemOperation::DidFinishFileOperation));
 }
 
 void FileSystemOperation::TouchFile(const FilePath& path,
                                     const base::Time& last_access_time,
                                     const base::Time& last_modified_time) {
 #ifndef NDEBUG
-  DCHECK(!operation_pending_);
-  operation_pending_ = true;
+  DCHECK(kOperationNone == pending_operation_);
+  pending_operation_ = kOperationTouchFile;
 #endif
 
   base::FileUtilProxy::Touch(
       proxy_, path, last_access_time, last_modified_time,
       callback_factory_.NewCallback(&FileSystemOperation::DidTouchFile));
 }
 
-void FileSystemOperation::Cancel() {
+// We can only get here on a write or truncate that's not yet completed.
+// We don't support cancelling any other operation at this time.
+void FileSystemOperation::Cancel(FileSystemOperation* cancel_operation) {
 #ifndef NDEBUG
-  DCHECK(operation_pending_);
+  DCHECK(kOperationTruncate == pending_operation_);
+  // FIXME(ericu): Cancelling for writes coming soon.
 #endif
-  NOTREACHED();
-  // TODO(ericu):
-  // Make sure this was done on a FileSystemOperation used for a Write.
-  // Then set a flag that ensures that the Write loop will exit without
-  // reporting any more progress, with a failure notification.
+  // We're cancelling a truncate operation, but we can't actually stop it
+  // since it's been proxied to another thread.  We need to save the
+  // cancel_operation so that when the truncate returns, it can see that it's
+  // been cancelled, report it, and report that the cancel has succeeded.
+  cancel_operation_ = cancel_operation;
 }
 
 void FileSystemOperation::DidCreateFileExclusive(
     base::PlatformFileError rv,
     base::PassPlatformFile file,
     bool created) {
   DidFinishFileOperation(rv);
 }
 
 void FileSystemOperation::DidCreateFileNonExclusive(
     base::PlatformFileError rv,
     base::PassPlatformFile file,
     bool created) {
   // Suppress the already exists error and report success.
   if (rv == base::PLATFORM_FILE_OK ||
       rv == base::PLATFORM_FILE_ERROR_EXISTS)
     dispatcher_->DidSucceed();
   else
     dispatcher_->DidFail(rv);
 }
 
 void FileSystemOperation::DidFinishFileOperation(
     base::PlatformFileError rv) {
-  if (rv == base::PLATFORM_FILE_OK)
+  if (cancel_operation_) {
+#ifndef NDEBUG
+    DCHECK(kOperationTruncate == pending_operation_);
+#endif
+    FileSystemOperation *cancel_op = cancel_operation_;
+    // This call deletes us, so we have to extract cancel_op first.
+    dispatcher_->DidFail(base::PLATFORM_FILE_ERROR_ABORT);
+    cancel_op->dispatcher_->DidSucceed();
+  } else if (rv == base::PLATFORM_FILE_OK) {
     dispatcher_->DidSucceed();
-  else
+  } else {
     dispatcher_->DidFail(rv);
+  }
 }
 
 void FileSystemOperation::DidDirectoryExists(
     base::PlatformFileError rv,
     const base::PlatformFileInfo& file_info) {
   if (rv == base::PLATFORM_FILE_OK) {
     if (file_info.is_directory)
       dispatcher_->DidSucceed();
     else
       dispatcher_->DidFail(base::PLATFORM_FILE_ERROR_FAILED);
@@ skipping 29 matching lines @@
     dispatcher_->DidReadDirectory(entries, false /* has_more */);
   else
     dispatcher_->DidFail(rv);
 }
 
 void FileSystemOperation::DidWrite(
     base::PlatformFileError rv,
     int64 bytes,
     bool complete) {
   if (rv == base::PLATFORM_FILE_OK)
-    /* dispatcher_->DidWrite(bytes, complete) TODO(ericu): Coming soon. */ {}
+    dispatcher_->DidWrite(bytes, complete);
   else
     dispatcher_->DidFail(rv);
 }
 
 void FileSystemOperation::DidTouchFile(base::PlatformFileError rv) {
   if (rv == base::PLATFORM_FILE_OK)
     dispatcher_->DidSucceed();
   else
     dispatcher_->DidFail(rv);
 }
 
 }  // namespace fileapi