Count site data size for important sites

chrome/browser/ui/webui/settings/settings_clear_browsing_data_handler.cc

 // Copyright 2015 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/settings/settings_clear_browsing_data_handler.h"
 
 #include <stddef.h>
 #include <vector>
 
 #include "base/feature_list.h"
 #include "base/macros.h"
 #include "base/memory/ptr_util.h"
 #include "base/metrics/histogram_macros.h"
 #include "base/metrics/sparse_histogram.h"
 #include "base/values.h"
 #include "chrome/browser/browsing_data/browsing_data_counter_factory.h"
 #include "chrome/browser/browsing_data/browsing_data_counter_utils.h"
 #include "chrome/browser/browsing_data/browsing_data_helper.h"
 #include "chrome/browser/browsing_data/browsing_data_important_sites_util.h"
 #include "chrome/browser/browsing_data/browsing_data_remover_factory.h"
 #include "chrome/browser/browsing_data/chrome_browsing_data_remover_delegate.h"
+#include "chrome/browser/engagement/important_sites_usage_counter.h"
 #include "chrome/browser/engagement/important_sites_util.h"
 #include "chrome/browser/history/web_history_service_factory.h"
 #include "chrome/browser/sync/profile_sync_service_factory.h"
 #include "chrome/common/channel_info.h"
 #include "chrome/common/chrome_features.h"
 #include "chrome/common/pref_names.h"
 #include "components/browsing_data/core/history_notice_utils.h"
 #include "components/browsing_data/core/pref_names.h"
 #include "components/prefs/pref_service.h"
 #include "content/public/browser/browsing_data_filter_builder.h"
+#include "content/public/browser/storage_partition.h"
 #include "content/public/browser/web_ui.h"
 #include "ui/base/text/bytes_formatting.h"
 
 using ImportantReason = ImportantSitesUtil::ImportantReason;
 
 namespace {
 
 const int kMaxTimesHistoryNoticeShown = 1;
 
-const int kMaxImportantSites = 10;
-
 // TODO(msramek): Get the list of deletion preferences from the JS side.
 const char* kCounterPrefs[] = {
   browsing_data::prefs::kDeleteBrowsingHistory,
   browsing_data::prefs::kDeleteCache,
   browsing_data::prefs::kDeleteDownloadHistory,
   browsing_data::prefs::kDeleteFormData,
   browsing_data::prefs::kDeleteHostedAppsData,
   browsing_data::prefs::kDeleteMediaLicenses,
   browsing_data::prefs::kDeletePasswords,
 };
@@ skipping 260 matching lines @@
       "History.ClearBrowsingData.ShownHistoryNoticeAfterClearing", show_notice);
 
   ResolveJavascriptCallback(base::Value(webui_callback_id),
                             base::Value(show_notice));
   task_observer_.reset();
 }
 
 void ClearBrowsingDataHandler::HandleFetchImportantSites(
     const base::ListValue* args) {
   AllowJavascript();
-  const base::Value* callback_id;
-  CHECK(args->Get(0, &callback_id));
+  std::string callback_id;
+  CHECK(args->GetString(0, &callback_id));
 
   bool important_sites_flag_enabled =
       base::FeatureList::IsEnabled(features::kImportantSitesInCbd);
   Profile* profile = profile_->GetOriginalProfile();
   bool important_sites_dialog_disabled =
       ImportantSitesUtil::IsDialogDisabled(profile);
+
+  if (!important_sites_flag_enabled || important_sites_dialog_disabled) {
+    auto result = base::MakeUnique<base::DictionaryValue>();
+    result->SetBoolean(kFlagEnabledField, important_sites_flag_enabled);
+    result->Set(kImportantSitesField, base::MakeUnique<base::ListValue>());
+    ResolveJavascriptCallback(base::Value(callback_id), *result);
+    return;
+  }
+
+  std::vector<ImportantSitesUtil::ImportantDomainInfo> important_sites =
+      ImportantSitesUtil::GetImportantRegisterableDomains(
+          profile, ImportantSitesUtil::kMaxImportantSites);
+  content::StoragePartition* partition =
+      content::BrowserContext::GetDefaultStoragePartition(profile);
+  storage::QuotaManager* quota_manager = partition->GetQuotaManager();
+  content::DOMStorageContext* dom_storage = partition->GetDOMStorageContext();
+
+  ImportantSitesUsageCounter::GetUsage(
+      std::move(important_sites), quota_manager, dom_storage,
+      base::Bind(&ClearBrowsingDataHandler::OnFetchImportantSitesFinished,
+                 base::Unretained(this), callback_id));

[Bernhard Bauer, 2017/05/04 14:16:45]

What if this object is destroyed before the callback runs?

[dullweber, 2017/05/05 08:35:28]

I assumed that it worked because the BrowsingDataRemover and
BrowsingDataCounters are also passing a Unretained pointer but Martin just
explained that they work differently. I will pass a weak_ptr instead.

+}
+
+void ClearBrowsingDataHandler::OnFetchImportantSitesFinished(
+    const std::string& callback_id,
+    std::vector<ImportantSitesUtil::ImportantDomainInfo> important_sites) {
   auto important_sites_list = base::MakeUnique<base::ListValue>();
 
-  if (important_sites_flag_enabled && !important_sites_dialog_disabled) {
-    std::vector<ImportantSitesUtil::ImportantDomainInfo> important_sites =
-        ImportantSitesUtil::GetImportantRegisterableDomains(profile,
-                                                            kMaxImportantSites);
-    for (const auto& info : important_sites) {
-      auto entry = base::MakeUnique<base::DictionaryValue>();
-      entry->SetString(kRegisterableDomainField, info.registerable_domain);
-      // The |reason_bitfield| is only passed to Javascript to be logged
-      // from |HandleClearBrowsingData|.
-      entry->SetInteger(kReasonBitField, info.reason_bitfield);
-      entry->SetString(kExampleOriginField, info.example_origin.spec());
-      // Initially all sites are selected for deletion.
-      entry->SetBoolean(kIsCheckedField, true);
-      // TODO(dullweber): Get size.
-      entry->SetString(kStorageSizeField, ui::FormatBytes(0));
-      bool has_notifications =
-          (info.reason_bitfield & (1 << ImportantReason::NOTIFICATIONS)) != 0;
-      entry->SetBoolean(kHasNotificationsField, has_notifications);
-      important_sites_list->Append(std::move(entry));
-    }
+  for (const auto& info : important_sites) {
+    auto entry = base::MakeUnique<base::DictionaryValue>();
+    entry->SetString(kRegisterableDomainField, info.registerable_domain);
+    // The |reason_bitfield| is only passed to Javascript to be logged
+    // from |HandleClearBrowsingData|.
+    entry->SetInteger(kReasonBitField, info.reason_bitfield);
+    entry->SetString(kExampleOriginField, info.example_origin.spec());
+    // Initially all sites are selected for deletion.
+    entry->SetBoolean(kIsCheckedField, true);
+    entry->SetString(kStorageSizeField, ui::FormatBytes(info.usage));
+    bool has_notifications =
+        (info.reason_bitfield & (1 << ImportantReason::NOTIFICATIONS)) != 0;
+    entry->SetBoolean(kHasNotificationsField, has_notifications);
+    important_sites_list->Append(std::move(entry));
   }
 
-  std::unique_ptr<base::DictionaryValue> result(new base::DictionaryValue());
-  result->SetBoolean(kFlagEnabledField, important_sites_flag_enabled);
+  auto result = base::MakeUnique<base::DictionaryValue>();

[Bernhard Bauer, 2017/05/04 14:16:45]

You could even stack-allocate the DictionaryValue.

[dullweber, 2017/05/05 08:35:28]

Done.

+  result->SetBoolean(kFlagEnabledField, true);
   result->Set(kImportantSitesField, std::move(important_sites_list));
-
-  ResolveJavascriptCallback(*callback_id, *result);
+  ResolveJavascriptCallback(base::Value(callback_id), *result);
 }
 
 void ClearBrowsingDataHandler::HandleInitialize(const base::ListValue* args) {
   AllowJavascript();
   const base::Value* callback_id;
   CHECK(args->Get(0, &callback_id));
 
   // Needed because WebUI doesn't handle renderer crashes. See crbug.com/610450.
   task_observer_.reset();
 
@@ skipping 66 matching lines @@
 
 void ClearBrowsingDataHandler::UpdateCounterText(
     std::unique_ptr<browsing_data::BrowsingDataCounter::Result> result) {
   CallJavascriptFunction(
       "cr.webUIListenerCallback", base::Value("update-counter-text"),
       base::Value(result->source()->GetPrefName()),
       base::Value(GetChromeCounterTextFromResult(result.get())));
 }
 
 }  // namespace settings