Count site data size for important sites

chrome/browser/engagement/important_sites_usage_counter.h

+// Copyright 2017 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.
+
+#ifndef CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_USAGE_COUNTER_H_
+#define CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_USAGE_COUNTER_H_
+
+#include <string>
+#include <vector>
+
+#include "base/callback.h"
+#include "base/macros.h"
+#include "base/sequenced_task_runner_helpers.h"
+#include "chrome/browser/engagement/important_sites_util.h"
+#include "content/public/browser/local_storage_usage_info.h"
+#include "storage/browser/quota/quota_manager.h"
+
+namespace content {
+class DOMStorageContext;
+}
+
+namespace storage {
+class QuotaManager;
+}
+
+// A helper class for important storage that retrieves the localstorage and
+// quota usage for each domain in |ImportantDomainInfo|, populates
+// |ImportantDomainInfo::usage| in the |sites| entries and return the result via
+// |callback|.
+class ImportantSitesUsageCounter {
+ public:
+  using UsageCallback = base::OnceCallback<void(
+      std::vector<ImportantSitesUtil::ImportantDomainInfo> sites)>;
+
+  // Populates the ImportantDomainInfo::usage field of each site with the amount
+  // of localstorage and quota bytes used. |callback| is asynchronously invoked
+  // on the UI thread with the |sites| vector (populated with usage) as an
+  // argument.
+  static void GetUsage(
+      std::vector<ImportantSitesUtil::ImportantDomainInfo> sites,
+      storage::QuotaManager* quota_manager,
+      content::DOMStorageContext* dom_storage_context,
+      UsageCallback callback);
+
+ private:
+  friend class base::DeleteHelper<ImportantSitesUsageCounter>;
+
+  ImportantSitesUsageCounter(
+      std::vector<ImportantSitesUtil::ImportantDomainInfo> sites,
+      storage::QuotaManager* quota_manager,
+      content::DOMStorageContext* dom_storage_context,
+      UsageCallback callback);
+  ~ImportantSitesUsageCounter();
+
+  void RunAndDestroySelfWhenFinished();
+
+  void GetQuotaUsageOnIOThread();
+
+  void ReceiveQuotaUsageOnIOThread(
+      const std::vector<storage::UsageInfo>& usage_infos);
+
+  void ReceiveQuotaUsage(const std::vector<storage::UsageInfo>& usage_infos);
+
+  void ReceiveLocalStorageUsage(
+      const std::vector<content::LocalStorageUsageInfo>& storage_infos);
+
+  // Look up the corresponding ImportantDomainInfo for |url| and increase its
+  // usage by |size|.
+  void IncrementUsage(const std::string& domain, int64_t size);
+
+  void Done();
+
+  UsageCallback callback_;
+  std::vector<ImportantSitesUtil::ImportantDomainInfo> sites_;
+  storage::QuotaManager* quota_manager_;
+  content::DOMStorageContext* dom_storage_context_;
+  int tasks_;
+
+  DISALLOW_IMPLICIT_CONSTRUCTORS(ImportantSitesUsageCounter);
+};
+
+#endif  // CHROME_BROWSER_ENGAGEMENT_IMPORTANT_SITES_USAGE_COUNTER_H_