Storage manager: Add browsing data.

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 <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 "components/browsing_data/storage_partition_http_cache_data_remover.h"
 #include "components/drive/chromeos/file_system_interface.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,
                                int64_t* total_size,
                                int64_t* available_size) {
   int64_t size = base::SysInfo::AmountOfTotalDiskSpace(mount_path);
   if (size >= 0)
     *total_size = size;
   size = base::SysInfo::AmountOfFreeDiskSpace(mount_path);
   if (size >= 0)
     *available_size = size;
 }
 
 // Threshold to show a message indicating space is critically low (512 MB).
 const int64_t kSpaceCriticallyLowBytes = 512 * 1024 * 1024;
 
 // Threshold to show a message indicating space is low (1 GB).
 const int64_t kSpaceLowBytes = 1 * 1024 * 1024 * 1024;
 
 }  // namespace
 
-StorageManagerHandler::StorageManagerHandler() : weak_ptr_factory_(this) {
+StorageManagerHandler::StorageManagerHandler()
+    : browser_cache_size_(0),
+      browser_site_data_size_(0),
+      weak_ptr_factory_(this) {
 }
 
 StorageManagerHandler::~StorageManagerHandler() {
 }
 
 void StorageManagerHandler::GetLocalizedValues(
     base::DictionaryValue* localized_strings) {
   DCHECK(localized_strings);
 
   RegisterTitle(localized_strings, "storageManagerPage",
@@ skipping 10 matching lines @@
   localized_strings->SetString(
       "storageItemLabelAvailable", l10n_util::GetStringUTF16(
           IDS_OPTIONS_SETTINGS_STORAGE_ITEM_LABEL_AVAILABLE));
   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(
       "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(
       "storageClearDriveCacheDescription", l10n_util::GetStringUTF16(
           IDS_OPTIONS_SETTINGS_STORAGE_CLEAR_DRIVE_CACHE_DESCRIPTION));
   localized_strings->SetString(
       "storageDeleteAllButtonTitle", l10n_util::GetStringUTF16(
@@ skipping 41 matching lines @@
       "clearDriveCache",
       base::Bind(&StorageManagerHandler::HandleClearDriveCache,
                  base::Unretained(this)));
 }
 
 void StorageManagerHandler::HandleUpdateStorageInfo(
     const base::ListValue* unused_args) {
   UpdateSizeStat();
   UpdateDownloadsSize();
   UpdateDriveCacheSize();
+  UpdateBrowsingDataSize();
   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 82 matching lines @@
       base::Bind(&StorageManagerHandler::OnGetDriveCacheSize,
                  weak_ptr_factory_.GetWeakPtr()));
 }
 
 void StorageManagerHandler::OnGetDriveCacheSize(int64_t size) {
   web_ui()->CallJavascriptFunctionUnsafe(
       "options.StorageManager.setDriveCacheSize",
       base::StringValue(ui::FormatBytes(size)));
 }
 
+void StorageManagerHandler::UpdateBrowsingDataSize() {
+  Profile* const profile = Profile::FromWebUI(web_ui());
+  // Fetch the size of http cache in browsing data.
+  // StoragePartitionHttpCacheDataRemover deletes itself when it is done.
+  browsing_data::StoragePartitionHttpCacheDataRemover::CreateForRange(
+      content::BrowserContext::GetDefaultStoragePartition(profile),
+      base::Time(),
+      base::Time::Max())->Count(
+          base::Bind(&StorageManagerHandler::OnGetBrowsingDataSize,
+                     weak_ptr_factory_.GetWeakPtr(), false));
+
+  // TODO(fukino): fetch the size of site data in browsing data.
+}
+
+void StorageManagerHandler::OnGetBrowsingDataSize(bool is_site_data,
+                                                  int64_t size) {
+  if (is_site_data)
+    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::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