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_