Add a counter to calculate the total size of site data.

chrome/browser/browsing_data/site_data_size_collector.h

+// 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.
+
+#ifndef CHROME_BROWSER_BROWSING_DATA_SITE_DATA_SIZE_COLLECTOR_H_
+#define CHROME_BROWSER_BROWSING_DATA_SITE_DATA_SIZE_COLLECTOR_H_
+
+#include "base/memory/weak_ptr.h"
+#include "chrome/browser/browsing_data/browsing_data_appcache_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_cache_storage_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_channel_id_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_cookie_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_counter.h"
+#include "chrome/browser/browsing_data/browsing_data_database_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_file_system_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_flash_lso_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_indexed_db_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_local_storage_helper.h"
+#include "chrome/browser/browsing_data/browsing_data_service_worker_helper.h"
+#include "chrome/browser/profiles/profile.h"
+#include "content/public/browser/storage_partition.h"
+
+namespace {
+
+typedef std::list<net::CanonicalCookie> CookieList;
+typedef std::list<BrowsingDataDatabaseHelper::DatabaseInfo> DatabaseInfoList;
+typedef std::list<BrowsingDataLocalStorageHelper::LocalStorageInfo>
+    LocalStorageInfoList;
+typedef std::list<content::IndexedDBInfo>
+    IndexedDBInfoList;
+typedef std::list<BrowsingDataFileSystemHelper::FileSystemInfo>
+    FileSystemInfoList;
+typedef net::ChannelIDStore::ChannelIDList ChannelIDList;
+typedef std::list<content::ServiceWorkerUsageInfo> ServiceWorkerUsageInfoList;
+typedef std::list<content::CacheStorageUsageInfo> CacheStorageUsageInfoList;
+typedef std::map<GURL, std::list<content::AppCacheInfo> > AppCacheInfoMap;
+typedef std::vector<std::string> FlashLSODomainList;
+
+}  // namespace
+
+class SiteDataSizeCollector {
+ public:
+  SiteDataSizeCollector(const base::FilePath& default_storage_partition_path,
+                        BrowsingDataCookieHelper* cookie_helper,
+                        BrowsingDataDatabaseHelper* database_helper,
+                        BrowsingDataLocalStorageHelper* local_storage_helper,
+                        BrowsingDataAppCacheHelper* appcache_helper,
+                        BrowsingDataIndexedDBHelper* indexed_db_helper,
+                        BrowsingDataFileSystemHelper* file_system_helper,
+                        BrowsingDataChannelIDHelper* channel_id_helper,
+                        BrowsingDataServiceWorkerHelper* service_worker_helper,
+                        BrowsingDataCacheStorageHelper* cache_storage_helper,
+                        BrowsingDataFlashLSOHelper* flash_lso_helper);
+  virtual ~SiteDataSizeCollector();
+
+  using FetchCallback = base::Callback<void(int64_t)>;
+
+  // Requests to fetch the total storage space used by site data.
+  void Fetch(const FetchCallback& callback);
+
+ private:
+  // Callback methods to be invoked when fetching the data is complete.
+  void OnAppCacheModelInfoLoaded(
+      scoped_refptr<content::AppCacheInfoCollection>);
+  void OnCookiesModelInfoLoaded(const net::CookieList& cookie_list);
+  void OnDatabaseModelInfoLoaded(const DatabaseInfoList& database_info_list);
+  void OnLocalStorageModelInfoLoaded(
+      const LocalStorageInfoList& local_storage_info_list);
+  void OnIndexedDBModelInfoLoaded(
+      const IndexedDBInfoList& indexed_db_info_list);
+  void OnFileSystemModelInfoLoaded(
+      const FileSystemInfoList& file_system_info_list);
+  void OnChannelIDModelInfoLoaded(const ChannelIDList& channel_id_list);
+  void OnServiceWorkerModelInfoLoaded(
+      const ServiceWorkerUsageInfoList& service_worker_info_list);
+  void OnCacheStorageModelInfoLoaded(
+      const CacheStorageUsageInfoList& cache_storage_info_list);
+  void OnFlashLSOInfoLoaded(const FlashLSODomainList& domains);
+
+  // Callback for when the size is fetched from each storage backend.
+  void OnStorageSizeFetched(int64_t size);
+
+  // Path of the default storage partition of this profile.
+  base::FilePath default_storage_partition_path_;
+
+  // Pointers to the helper objects, needed to retreive all the types of locally
+  // stored data.
+  scoped_refptr<BrowsingDataAppCacheHelper> appcache_helper_;
+  scoped_refptr<BrowsingDataCookieHelper> cookie_helper_;
+  scoped_refptr<BrowsingDataDatabaseHelper> database_helper_;
+  scoped_refptr<BrowsingDataLocalStorageHelper> local_storage_helper_;
+  scoped_refptr<BrowsingDataIndexedDBHelper> indexed_db_helper_;
+  scoped_refptr<BrowsingDataFileSystemHelper> file_system_helper_;
+  scoped_refptr<BrowsingDataChannelIDHelper> channel_id_helper_;
+  scoped_refptr<BrowsingDataServiceWorkerHelper> service_worker_helper_;
+  scoped_refptr<BrowsingDataCacheStorageHelper> cache_storage_helper_;
+  scoped_refptr<BrowsingDataFlashLSOHelper> flash_lso_helper_;
+
+  // Callback called when sizes of all site data are fetched and accumulated.
+  FetchCallback fetch_callback_;
+
+  // Keeps track of how many fetch operations are ongoing.
+  int in_flight_operations_;
+
+  // Keeps track of the sum of all fetched size.
+  int64_t total_bytes_;
+
+  base::WeakPtrFactory<SiteDataSizeCollector> weak_ptr_factory_;
+
+  DISALLOW_COPY_AND_ASSIGN(SiteDataSizeCollector);
+};
+
+#endif  // CHROME_BROWSER_BROWSING_DATA_SITE_DATA_SIZE_COLLECTOR_H_