| // Copyright (c) 2012 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/browsing_data/browsing_data_local_storage_helper.h" |
| |
| #include <utility> |
| #include <vector> |
| |
| #include "base/bind.h" |
| #include "base/location.h" |
| #include "base/task/post_task.h" |
| #include "chrome/browser/browsing_data/browsing_data_helper.h" |
| #include "chrome/browser/profiles/profile.h" |
| #include "content/public/browser/browser_task_traits.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/storage_partition.h" |
| #include "content/public/browser/storage_usage_info.h" |
| #include "url/gurl.h" |
| #include "url/origin.h" |
| #include "url/url_constants.h" |
| |
| using content::BrowserContext; |
| using content::BrowserThread; |
| using content::DOMStorageContext; |
| |
| namespace { |
| |
| // Only websafe state is considered browsing data. |
| bool HasStorageScheme(const GURL& origin_url) { |
| return BrowsingDataHelper::HasWebScheme(origin_url); |
| } |
| |
| void GetUsageInfoCallback( |
| BrowsingDataLocalStorageHelper::FetchCallback callback, |
| const std::vector<content::StorageUsageInfo>& infos) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| DCHECK(!callback.is_null()); |
| |
| std::list<content::StorageUsageInfo> result; |
| for (const content::StorageUsageInfo& info : infos) { |
| if (!HasStorageScheme(info.origin.GetURL())) |
| continue; |
| result.push_back(info); |
| } |
| |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), result)); |
| } |
| |
| } // namespace |
| |
| BrowsingDataLocalStorageHelper::BrowsingDataLocalStorageHelper(Profile* profile) |
| : dom_storage_context_(BrowserContext::GetDefaultStoragePartition(profile) |
| ->GetDOMStorageContext()) { |
| DCHECK(dom_storage_context_); |
| } |
| |
| BrowsingDataLocalStorageHelper::~BrowsingDataLocalStorageHelper() { |
| } |
| |
| void BrowsingDataLocalStorageHelper::StartFetching(FetchCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| DCHECK(!callback.is_null()); |
| dom_storage_context_->GetLocalStorageUsage( |
| base::BindOnce(&GetUsageInfoCallback, std::move(callback))); |
| } |
| |
| void BrowsingDataLocalStorageHelper::DeleteOrigin(const url::Origin& origin, |
| base::OnceClosure callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| dom_storage_context_->DeleteLocalStorage(origin, std::move(callback)); |
| } |
| |
| //--------------------------------------------------------- |
| |
| CannedBrowsingDataLocalStorageHelper::CannedBrowsingDataLocalStorageHelper( |
| Profile* profile) |
| : BrowsingDataLocalStorageHelper(profile) { |
| } |
| |
| void CannedBrowsingDataLocalStorageHelper::AddLocalStorage( |
| const GURL& origin_url) { |
| if (!HasStorageScheme(origin_url)) |
| return; |
| pending_local_storage_info_.insert(origin_url); |
| url::Origin origin = url::Origin::Create(origin_url); |
| } |
| |
| void CannedBrowsingDataLocalStorageHelper::Reset() { |
| pending_local_storage_info_.clear(); |
| } |
| |
| bool CannedBrowsingDataLocalStorageHelper::empty() const { |
| return pending_local_storage_info_.empty(); |
| } |
| |
| size_t CannedBrowsingDataLocalStorageHelper::GetLocalStorageCount() const { |
| return pending_local_storage_info_.size(); |
| } |
| |
| const std::set<GURL>& |
| CannedBrowsingDataLocalStorageHelper::GetLocalStorageInfo() const { |
| return pending_local_storage_info_; |
| } |
| |
| void CannedBrowsingDataLocalStorageHelper::StartFetching( |
| FetchCallback callback) { |
| DCHECK_CURRENTLY_ON(BrowserThread::UI); |
| DCHECK(!callback.is_null()); |
| |
| std::list<content::StorageUsageInfo> result; |
| for (const GURL& url : pending_local_storage_info_) |
| result.push_back( |
| content::StorageUsageInfo(url::Origin::Create(url), 0, base::Time())); |
| |
| base::PostTaskWithTraits(FROM_HERE, {BrowserThread::UI}, |
| base::BindOnce(std::move(callback), result)); |
| } |
| |
| void CannedBrowsingDataLocalStorageHelper::DeleteOrigin( |
| const url::Origin& origin, |
| base::OnceClosure callback) { |
| pending_local_storage_info_.erase(origin.GetURL()); |
| BrowsingDataLocalStorageHelper::DeleteOrigin(origin, std::move(callback)); |
| } |
| |
| CannedBrowsingDataLocalStorageHelper::~CannedBrowsingDataLocalStorageHelper() {} |