[SystemInfo API] Implement storage.getAvailableCapacity API

chrome/browser/extensions/api/system_storage/system_storage_api.cc

 // Copyright 2013 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/api/system_storage/system_storage_api.h"
 
 namespace extensions {
 
 using api::system_storage::StorageUnitInfo;
 namespace EjectDevice = api::system_storage::EjectDevice;
+namespace GetAvailableCapacity = api::system_storage::GetAvailableCapacity;
 
 SystemStorageGetInfoFunction::SystemStorageGetInfoFunction() {
 }
 
 SystemStorageGetInfoFunction::~SystemStorageGetInfoFunction() {
 }
 
 bool SystemStorageGetInfoFunction::RunImpl() {
   StorageInfoProvider::Get()->StartQueryInfo(
       base::Bind(&SystemStorageGetInfoFunction::OnGetStorageInfoCompleted,
@@ skipping 64 matching lines @@
       break;
     case StorageMonitor::EJECT_FAILURE:
       result = api::system_storage::EJECT_DEVICE_RESULT_CODE_FAILURE;
   }
 
   SetResult(new base::StringValue(
       api::system_storage::ToString(result)));
   SendResponse(true);
 }
 
+SystemStorageGetAvailableCapacityFunction::
+    SystemStorageGetAvailableCapacityFunction() {
+}
+
+SystemStorageGetAvailableCapacityFunction::
+    ~SystemStorageGetAvailableCapacityFunction() {
+}
+
+bool SystemStorageGetAvailableCapacityFunction::RunImpl() {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+
+  scoped_ptr<GetAvailableCapacity::Params> params(
+      GetAvailableCapacity::Params::Create(*args_));
+  EXTENSION_FUNCTION_VALIDATE(params.get());
+
+  StorageMonitor::GetInstance()->EnsureInitialized(base::Bind(
+      &SystemStorageGetAvailableCapacityFunction::OnStorageMonitorInit,
+      this,
+      params->id));
+  return true;
+}
+
+void SystemStorageGetAvailableCapacityFunction::OnStorageMonitorInit(
+    const std::string& transient_id) {
+  content::BrowserThread::PostTaskAndReplyWithResult(
+      content::BrowserThread::FILE,
+      FROM_HERE,
+      base::Bind(
+          &StorageInfoProvider::GetStorageFreeSpaceFromTransientIdOnFileThread,
+          StorageInfoProvider::Get(), transient_id),
+      base::Bind(
+          &SystemStorageGetAvailableCapacityFunction::OnQueryCompleted,
+          this, transient_id));
+}
+
+void SystemStorageGetAvailableCapacityFunction::OnQueryCompleted(
+    const std::string& transient_id, double available_capacity) {
+  bool success = available_capacity >= 0;
+  if (success) {
+    api::system_storage::StorageAvailableCapacityInfo result;
+    result.id = transient_id;
+    result.available_capacity = available_capacity;
+    SetResult(result.ToValue().release());
+  } else {
+    SetError("Error occurred when querying available capacity.");
+  }
+  SendResponse(success);
+}
+
 }  // namespace extensions