Adding support to retrieve remaining and total space on disk/file shelf.

chrome/browser/extensions/extension_file_browser_private_api.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 "chrome/browser/extensions/extension_file_browser_private_api.h"
 
 #include "base/base64.h"
 #include "base/command_line.h"
 #include "base/json/json_writer.h"
 #include "base/logging.h"
@@ skipping 1259 matching lines @@
        it != mount_points.end();
        ++it) {
     mounts->Append(MountPointToValue(profile_, it->second));
   }
 #endif
 
   SendResponse(true);
   return true;
 }
 
+GetSizeStatsFunction::GetSizeStatsFunction() {
+}
+
+GetSizeStatsFunction::~GetSizeStatsFunction() {
+}
+
+void GetSizeStatsFunction::CallGetSizeStatsOnFileThread(
+    const char* mount_path) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+
+  size_t total_size_kb = 0;
+  size_t remaining_size_kb = 0;
+#ifdef OS_CHROMEOS
+  chromeos::CrosLibrary::Get()->GetMountLibrary()->GetSizeStatsOnFileThread(
+      mount_path, &total_size_kb, &remaining_size_kb);
+#endif

[rginda, 2011/09/06 21:38:58]

Is this supposed to fall through to the PostTask, or should it return without
posting?

[tonibarzic, 2011/09/06 21:45:32]

it's supposed to fall through.. 

+
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      NewRunnableMethod(this,
+          &GetSizeStatsFunction::GetSizeStatsCallbackOnUIThread,
+          mount_path, total_size_kb, remaining_size_kb));
+}
+
+void GetSizeStatsFunction::GetSizeStatsCallbackOnUIThread(
+    const char* mount_path, size_t total_size_kb, size_t remaining_size_kb) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+
+  base::DictionaryValue* sizes = new base::DictionaryValue();
+  result_.reset(sizes);
+
+  sizes->SetInteger("totalSizeKB", total_size_kb);
+  sizes->SetInteger("remainingSizeKB", remaining_size_kb);
+
+  SendResponse(true);
+}
+
+bool GetSizeStatsFunction::RunImpl() {
+  if (args_->GetSize() != 1) {
+    return false;
+  }
+
+  std::string mount_url;
+  if (!args_->GetString(0, &mount_url))
+    return false;
+
+  UrlList mount_paths;
+  mount_paths.push_back(GURL(mount_url));
+
+  BrowserThread::PostTask(
+      BrowserThread::FILE, FROM_HERE,
+      NewRunnableMethod(this,
+          &GetSizeStatsFunction::GetLocalPathsOnFileThread,
+          mount_paths, reinterpret_cast<void*>(NULL)));
+  return true;
+}
+
+void GetSizeStatsFunction::GetLocalPathsResponseOnUIThread(
+    const FilePathList& files, void* context) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+
+  if (files.size() != 1) {
+    SendResponse(false);
+    return;
+  }
+
+  BrowserThread::PostTask(
+      BrowserThread::FILE, FROM_HERE,
+      NewRunnableMethod(this,
+          &GetSizeStatsFunction::CallGetSizeStatsOnFileThread,
+          files[0].value().c_str()));
+  return;
+}
+
 FormatDeviceFunction::FormatDeviceFunction() {
 }
 
 FormatDeviceFunction::~FormatDeviceFunction() {
 }
 
 bool FormatDeviceFunction::RunImpl() {
   if (args_->GetSize() != 1) {
     return false;
   }
@@ skipping 244 matching lines @@
   // TODO(serya): Create a new string in .grd file for this one in M13.
   dict->SetString("PREVIEW_IMAGE",
       l10n_util::GetStringUTF16(IDS_CERT_MANAGER_VIEW_CERT_BUTTON));
   dict->SetString("PLAY_MEDIA",
       l10n_util::GetStringUTF16(IDS_CONTENT_CONTEXT_PLAY));
   if (CommandLine::ForCurrentProcess()->HasSwitch(switches::kEnableArchives))
     dict->SetString("ENABLE_ARCHIVES", "true");
 
   return true;
 }