Storage manager: Show the disk usage of other users.

chrome/browser/ui/webui/options/chromeos/storage_manager_handler.cc

 // Copyright 2016 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/options/chromeos/storage_manager_handler.h"
 
+#include <algorithm>
+#include <numeric>
 #include <string>
 
 #include "base/files/file_util.h"
 #include "base/sys_info.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/chromeos/arc/arc_auth_service.h"
 #include "chrome/browser/chromeos/drive/file_system_util.h"
 #include "chrome/browser/chromeos/file_manager/path_util.h"
 #include "chrome/browser/platform_util.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/grit/generated_resources.h"
+#include "chromeos/cryptohome/homedir_methods.h"
 #include "components/browsing_data/storage_partition_http_cache_data_remover.h"
 #include "components/drive/chromeos/file_system_interface.h"
+#include "components/user_manager/user_manager.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/browser_thread.h"
 #include "ui/base/l10n/l10n_util.h"
 #include "ui/base/text/bytes_formatting.h"
 
 namespace chromeos {
 namespace options {
 namespace {
 
 void GetSizeStatOnBlockingPool(const base::FilePath& mount_path,
@@ skipping 45 matching lines @@
   localized_strings->SetString(
       "storageSubitemLabelDownloads", l10n_util::GetStringUTF16(
           IDS_OPTIONS_SETTINGS_STORAGE_SUBITEM_LABEL_DOWNLOADS));
   localized_strings->SetString(
       "storageSubitemLabelDriveCache", l10n_util::GetStringUTF16(
           IDS_OPTIONS_SETTINGS_STORAGE_SUBITEM_LABEL_DRIVE_CACHE));
   localized_strings->SetString(
       "storageSubitemLabelBrowsingData", l10n_util::GetStringUTF16(
           IDS_OPTIONS_SETTINGS_STORAGE_SUBITEM_LABEL_BROWSING_DATA));
   localized_strings->SetString(
+      "storageSubitemLabelOtherUsers", l10n_util::GetStringUTF16(
+          IDS_OPTIONS_SETTINGS_STORAGE_SUBITEM_LABEL_OTHER_USERS));
+  localized_strings->SetString(
       "storageSubitemLabelArc", l10n_util::GetStringUTF16(
           IDS_OPTIONS_SETTINGS_STORAGE_SUBITEM_LABEL_ARC));
   localized_strings->SetString(
       "storageSizeCalculating", l10n_util::GetStringUTF16(
           IDS_OPTIONS_SETTINGS_STORAGE_SIZE_CALCULATING));
   localized_strings->SetString(
       "storageClearDriveCacheDialogTitle", l10n_util::GetStringUTF16(
           IDS_OPTIONS_SETTINGS_STORAGE_CLEAR_DRIVE_CACHE_DIALOG_TITLE));
   localized_strings->SetString(
       "storageClearDriveCacheDescription", l10n_util::GetStringUTF16(
@@ skipping 45 matching lines @@
       base::Bind(&StorageManagerHandler::HandleClearDriveCache,
                  base::Unretained(this)));
 }
 
 void StorageManagerHandler::HandleUpdateStorageInfo(
     const base::ListValue* unused_args) {
   UpdateSizeStat();
   UpdateDownloadsSize();
   UpdateDriveCacheSize();
   UpdateBrowsingDataSize();
+  UpdateOtherUsersSize();
   UpdateArcSize();
 }
 
 void StorageManagerHandler::HandleOpenDownloads(
     const base::ListValue* unused_args) {
   Profile* const profile = Profile::FromWebUI(web_ui());
   const base::FilePath downloads_path =
       file_manager::util::GetDownloadsFolderForProfile(profile);
   platform_util::OpenItem(
        profile,
@@ skipping 109 matching lines @@
     browser_site_data_size_ = size;
   else
     browser_cache_size_ = size;
 
   web_ui()->CallJavascriptFunctionUnsafe(
      "options.StorageManager.setBrowsingDataSize",
      base::StringValue(ui::FormatBytes(static_cast<int64_t>(
          browser_cache_size_ + browser_site_data_size_))));
 }
 
+void StorageManagerHandler::UpdateOtherUsersSize() {
+  other_users_.clear();
+  user_sizes_.clear();
+  const user_manager::UserList& users =
+      user_manager::UserManager::Get()->GetUsers();
+  for (const auto& user : users) {
+    if (user->is_active())
+      continue;
+    other_users_.push_back(user);
+    cryptohome::HomedirMethods::GetInstance()->GetAccountDiskUsage(
+        cryptohome::Identification(user->GetAccountId()),
+        base::Bind(&StorageManagerHandler::OnGetOtherUserSize,
+                   weak_ptr_factory_.GetWeakPtr()));
+  }
+}
+
+void StorageManagerHandler::OnGetOtherUserSize(bool success, int64_t size) {
+  user_sizes_.push_back(success ? size : -1);
+  if (user_sizes_.size() == other_users_.size()) {
+    base::StringValue other_users_size(l10n_util::GetStringUTF16(
+        IDS_OPTIONS_SETTINGS_STORAGE_SIZE_UNKNOWN));
+    // If all the requests succeed, shows the total bytes in the UI.
+    if (std::count(user_sizes_.begin(), user_sizes_.end(), -1) == 0) {
+      other_users_size = base::StringValue(ui::FormatBytes(
+          std::accumulate(user_sizes_.begin(), user_sizes_.end(), 0LL)));
+    }
+    web_ui()->CallJavascriptFunctionUnsafe(
+       "options.StorageManager.setOtherUsersSize", other_users_size);
+  }
+}
+
 void StorageManagerHandler::UpdateArcSize() {
   Profile* const profile = Profile::FromWebUI(web_ui());
   if (!arc::ArcAuthService::IsAllowedForProfile(profile) ||
       arc::ArcAuthService::IsOptInVerificationDisabled() ||
       !arc::ArcAuthService::Get()->IsArcEnabled()) {
     return;
   }
 
   // Shows the item "Android apps and cache" and start calculating size.
   web_ui()->CallJavascriptFunctionUnsafe(
@@ skipping 24 matching lines @@
   web_ui()->CallJavascriptFunctionUnsafe("options.StorageManager.setArcSize",
                                          arc_size);
 }
 
 void StorageManagerHandler::OnClearDriveCacheDone(bool success) {
   UpdateDriveCacheSize();
 }
 
 }  // namespace options
 }  // namespace chromeos