Adding Size() to AbstractFileEnumerator

webkit/fileapi/file_system_file_util.cc

 // 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.
 
 #include "webkit/fileapi/file_system_file_util.h"
 
 #include <stack>
 #include <vector>
 
 #include "base/file_util_proxy.h"
@@ skipping 502 matching lines @@
     : public FileSystemFileUtil::AbstractFileEnumerator {
  public:
   FileSystemFileEnumerator(const FilePath& root_path,
                            bool recursive,
                            file_util::FileEnumerator::FileType file_type)
     : file_enum_(root_path, recursive, file_type) {
   }
 
   ~FileSystemFileEnumerator() {}
 
-  virtual FilePath Next();
-  virtual bool IsDirectory();
+  virtual FilePath Next() OVERRIDE;
+  virtual int64 Size() OVERRIDE;
+  virtual bool IsDirectory() OVERRIDE;
 
  private:
   file_util::FileEnumerator file_enum_;
+  file_util::FileEnumerator::FindInfo file_util_info_;
 };
 
 FilePath FileSystemFileEnumerator::Next() {
-  return file_enum_.Next();
+  FilePath rv = file_enum_.Next();
+  if (!rv.empty())
+    file_enum_.GetFindInfo(&file_util_info_);
+  return rv;
+}
+
+int64 FileSystemFileEnumerator::Size() {
+  return file_util::FileEnumerator::GetFilesize(file_util_info_);
 }
 
 bool FileSystemFileEnumerator::IsDirectory() {
-  file_util::FileEnumerator::FindInfo file_util_info;
-  file_enum_.GetFindInfo(&file_util_info);
-  return file_util::FileEnumerator::IsDirectory(file_util_info);
+  return file_util::FileEnumerator::IsDirectory(file_util_info_);
 }
 
 FileSystemFileUtil::AbstractFileEnumerator*
 FileSystemFileUtil::CreateFileEnumerator(
     FileSystemOperationContext* unused,
     const FilePath& root_path) {
   return new FileSystemFileEnumerator(
       root_path, true, static_cast<file_util::FileEnumerator::FileType>(
       file_util::FileEnumerator::FILES |
       file_util::FileEnumerator::DIRECTORIES));
 }
 
 }  // namespace fileapi