| // 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_cookie_helper.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::StorageUsageInfo> IndexedDBInfoList; |
| typedef std::list<BrowsingDataFileSystemHelper::FileSystemInfo> |
| FileSystemInfoList; |
| typedef std::list<content::StorageUsageInfo> ServiceWorkerUsageInfoList; |
| typedef std::list<content::StorageUsageInfo> CacheStorageUsageInfoList; |
| 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, |
| 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 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<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_ |