drive: Add information about in-flight operations to chrome:drive-internals

chrome/browser/ui/webui/chromeos/drive_internals_ui.cc

 // Copyright (c) 2012 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/ui/webui/chromeos/drive_internals_ui.h"
 
 #include "base/bind.h"
 #include "base/file_util.h"
 #include "base/format_macros.h"
 #include "base/stringprintf.h"
 #include "base/memory/weak_ptr.h"
 #include "base/path_service.h"
 #include "base/sys_info.h"
 #include "chrome/browser/chromeos/gdata/auth_service.h"
 #include "chrome/browser/chromeos/gdata/drive.pb.h"
 #include "chrome/browser/chromeos/gdata/drive_cache.h"
 #include "chrome/browser/chromeos/gdata/drive_service_interface.h"
 #include "chrome/browser/chromeos/gdata/gdata_file_system_interface.h"
 #include "chrome/browser/chromeos/gdata/gdata_system_service.h"
 #include "chrome/browser/chromeos/gdata/gdata_util.h"
+#include "chrome/browser/chromeos/gdata/operation_registry.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/webui/chrome_web_ui_data_source.h"
 #include "chrome/common/url_constants.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/web_ui.h"
 #include "content/public/browser/web_ui_message_handler.h"
 #include "grit/browser_resources.h"
 
 using content::BrowserThread;
 
@@ skipping 158 matching lines @@
       const std::vector<std::string>& resource_ids);
 
   // Called when GetCacheEntryOnUIThread() is complete.
   void OnGetCacheEntry(const std::string& resource_id,
                        bool success,
                        const gdata::DriveCacheEntry& cache_entry);
 
   // Called when GetFreeDiskSpace() is complete.
   void OnGetFreeDiskSpace(base::DictionaryValue* local_storage_summary);
 
+  // Updates the summary about in-flight operations.
+  void UpdateInFlightOperations(
+      const gdata::DriveServiceInterface* drive_service);
+
   // The number of pending ReadDirectoryByPath() calls.
   int num_pending_reads_;
   base::WeakPtrFactory<DriveInternalsWebUIHandler> weak_ptr_factory_;
   DISALLOW_COPY_AND_ASSIGN(DriveInternalsWebUIHandler);
 };
 
 void DriveInternalsWebUIHandler::RegisterMessages() {
   web_ui()->RegisterMessageCallback(
       "pageLoaded",
       base::Bind(&DriveInternalsWebUIHandler::OnPageLoaded,
@@ skipping 17 matching lines @@
   DCHECK(drive_service);
 
   // Update the auth status section.
   base::DictionaryValue auth_status;
   auth_status.SetBoolean("has-refresh-token",
                          drive_service->HasRefreshToken());
   auth_status.SetBoolean("has-access-token",
                          drive_service->HasAccessToken());
   web_ui()->CallJavascriptFunction("updateAuthStatus", auth_status);
 
+  // Update information about in-flight operations.
+  UpdateInFlightOperations(drive_service);

[satorux1, 2012/08/22 20:20:02]

as mentioned in .js file, I think we should update the section every second. To
do that, we should fetch the latest status from JavaScript side. I guess we
don't have to initiate the update of this section from C++ side.

[Haruki Sato, 2012/08/24 19:06:11]

Done.

+
   // Start updating the GCache contents section.
   Profile* profile = Profile::FromWebUI(web_ui());
   const FilePath root_path =
       gdata::DriveCache::GetCacheRootPath(profile);
   base::ListValue* gcache_contents = new ListValue;
   base::DictionaryValue* gcache_summary = new DictionaryValue;
   BrowserThread::PostBlockingPoolTaskAndReply(
       FROM_HERE,
       base::Bind(&GetGCacheContents,
                  root_path,
@@ skipping 123 matching lines @@
 
 void DriveInternalsWebUIHandler::OnGetFreeDiskSpace(
     base::DictionaryValue* local_storage_summary) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   DCHECK(local_storage_summary);
 
   web_ui()->CallJavascriptFunction(
       "updateLocalStorageUsage", *local_storage_summary);
 }
 
+void DriveInternalsWebUIHandler::UpdateInFlightOperations(
+    const gdata::DriveServiceInterface* drive_service) {
+  std::vector<gdata::OperationRegistry::ProgressStatus>
+      progress_status_list = drive_service->operation_registry()->
+      GetProgressStatusList();
+
+  base::ListValue in_flight_operations;
+  for (std::vector<gdata::OperationRegistry::ProgressStatus>::iterator i
+           = progress_status_list.begin();
+       i != progress_status_list.end(); ++i) {
+    base::DictionaryValue* status = new DictionaryValue;
+    status->SetInteger("operation_id", i->operation_id);
+    status->SetString(
+        "operation_type",
+        gdata::OperationRegistry::OperationTypeToString(i->operation_type));
+    status->SetString("file_path", i->file_path.AsUTF8Unsafe());
+    status->SetString(
+        "transfer_state",
+        gdata::OperationRegistry::OperationTransferStateToString(
+            i->transfer_state));
+    status->SetString("start_time",
+                      gdata::util::FormatTimeAsStringLocaltime(i->start_time));
+    status->SetDouble("progress_current", i->progress_current);
+    status->SetDouble("progress_total", i->progress_total);
+    in_flight_operations.Append(status);
+  }
+  web_ui()->CallJavascriptFunction("updateInFlightOperations",
+                                   in_flight_operations);
+}
+
 }  // namespace
 
 DriveInternalsUI::DriveInternalsUI(content::WebUI* web_ui)
     : WebUIController(web_ui) {
   web_ui->AddMessageHandler(new DriveInternalsWebUIHandler());
 
   ChromeWebUIDataSource* source =
       new ChromeWebUIDataSource(chrome::kChromeUIDriveInternalsHost);
   source->add_resource_path("drive_internals.css", IDR_DRIVE_INTERNALS_CSS);
   source->add_resource_path("drive_internals.js", IDR_DRIVE_INTERNALS_JS);
   source->set_default_resource(IDR_DRIVE_INTERNALS_HTML);
 
   Profile* profile = Profile::FromWebUI(web_ui);
   ChromeURLDataManager::AddDataSource(profile, source);
 }
 
 }  // namespace chromeos