1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "chrome/browser/browsing_data/browsing_data_local_storage_helper.h"
6
7 #include "base/bind.h"
8 #include "chrome/browser/browsing_data/browsing_data_helper.h"
9 #include "chrome/browser/profiles/profile.h"
10 #include "content/public/browser/browser_thread.h"
11 #include "content/public/browser/local_storage_usage_info.h"
12 #include "content/public/browser/storage_partition.h"
13
14 using content::BrowserContext;
15 using content::BrowserThread;
16 using content::DOMStorageContext;
17
LocalStorageInfo(const GURL & origin_url,int64 size,base::Time last_modified)18 BrowsingDataLocalStorageHelper::LocalStorageInfo::LocalStorageInfo(
19 const GURL& origin_url, int64 size, base::Time last_modified)
20 : origin_url(origin_url), size(size), last_modified(last_modified) {}
21
~LocalStorageInfo()22 BrowsingDataLocalStorageHelper::LocalStorageInfo::~LocalStorageInfo() {}
23
BrowsingDataLocalStorageHelper(Profile * profile)24 BrowsingDataLocalStorageHelper::BrowsingDataLocalStorageHelper(
25 Profile* profile)
26 : dom_storage_context_(
27 BrowserContext::GetDefaultStoragePartition(profile)->
28 GetDOMStorageContext()),
29 is_fetching_(false) {
30 DCHECK(dom_storage_context_);
31 }
32
~BrowsingDataLocalStorageHelper()33 BrowsingDataLocalStorageHelper::~BrowsingDataLocalStorageHelper() {
34 }
35
StartFetching(const base::Callback<void (const std::list<LocalStorageInfo> &)> & callback)36 void BrowsingDataLocalStorageHelper::StartFetching(
37 const base::Callback<void(const std::list<LocalStorageInfo>&)>& callback) {
38 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
39 DCHECK(!is_fetching_);
40 DCHECK_EQ(false, callback.is_null());
41
42 is_fetching_ = true;
43 completion_callback_ = callback;
44 dom_storage_context_->GetLocalStorageUsage(
45 base::Bind(
46 &BrowsingDataLocalStorageHelper::GetUsageInfoCallback, this));
47 }
48
DeleteOrigin(const GURL & origin)49 void BrowsingDataLocalStorageHelper::DeleteOrigin(const GURL& origin) {
50 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
51 dom_storage_context_->DeleteLocalStorage(origin);
52 }
53
GetUsageInfoCallback(const std::vector<content::LocalStorageUsageInfo> & infos)54 void BrowsingDataLocalStorageHelper::GetUsageInfoCallback(
55 const std::vector<content::LocalStorageUsageInfo>& infos) {
56 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
57
58 for (size_t i = 0; i < infos.size(); ++i) {
59 // Non-websafe state is not considered browsing data.
60 const content::LocalStorageUsageInfo& info = infos[i];
61 if (BrowsingDataHelper::HasWebScheme(info.origin)) {
62 local_storage_info_.push_back(
63 LocalStorageInfo(info.origin, info.data_size, info.last_modified));
64 }
65 }
66
67 BrowserThread::PostTask(
68 BrowserThread::UI, FROM_HERE,
69 base::Bind(&BrowsingDataLocalStorageHelper::CallCompletionCallback,
70 this));
71 }
72
CallCompletionCallback()73 void BrowsingDataLocalStorageHelper::CallCompletionCallback() {
74 DCHECK(is_fetching_);
75 completion_callback_.Run(local_storage_info_);
76 completion_callback_.Reset();
77 is_fetching_ = false;
78 }
79
80 //---------------------------------------------------------
81
CannedBrowsingDataLocalStorageHelper(Profile * profile)82 CannedBrowsingDataLocalStorageHelper::CannedBrowsingDataLocalStorageHelper(
83 Profile* profile)
84 : BrowsingDataLocalStorageHelper(profile),
85 profile_(profile) {
86 }
87
88 CannedBrowsingDataLocalStorageHelper*
Clone()89 CannedBrowsingDataLocalStorageHelper::Clone() {
90 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
91 CannedBrowsingDataLocalStorageHelper* clone =
92 new CannedBrowsingDataLocalStorageHelper(profile_);
93
94 clone->pending_local_storage_info_ = pending_local_storage_info_;
95 return clone;
96 }
97
AddLocalStorage(const GURL & origin)98 void CannedBrowsingDataLocalStorageHelper::AddLocalStorage(
99 const GURL& origin) {
100 if (BrowsingDataHelper::HasWebScheme(origin))
101 pending_local_storage_info_.insert(origin);
102 }
103
Reset()104 void CannedBrowsingDataLocalStorageHelper::Reset() {
105 pending_local_storage_info_.clear();
106 }
107
empty() const108 bool CannedBrowsingDataLocalStorageHelper::empty() const {
109 return pending_local_storage_info_.empty();
110 }
111
GetLocalStorageCount() const112 size_t CannedBrowsingDataLocalStorageHelper::GetLocalStorageCount() const {
113 return pending_local_storage_info_.size();
114 }
115
116 const std::set<GURL>&
GetLocalStorageInfo() const117 CannedBrowsingDataLocalStorageHelper::GetLocalStorageInfo() const {
118 return pending_local_storage_info_;
119 }
120
StartFetching(const base::Callback<void (const std::list<LocalStorageInfo> &)> & callback)121 void CannedBrowsingDataLocalStorageHelper::StartFetching(
122 const base::Callback<void(const std::list<LocalStorageInfo>&)>& callback) {
123 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
124 DCHECK(!callback.is_null());
125
126 std::list<LocalStorageInfo> result;
127 for (std::set<GURL>::iterator iter = pending_local_storage_info_.begin();
128 iter != pending_local_storage_info_.end(); ++iter) {
129 result.push_back(
130 LocalStorageInfo(*iter, 0, base::Time()));
131 }
132
133 BrowserThread::PostTask(
134 BrowserThread::UI, FROM_HERE, base::Bind(callback, result));
135 }
136
DeleteOrigin(const GURL & origin)137 void CannedBrowsingDataLocalStorageHelper::DeleteOrigin(const GURL& origin) {
138 pending_local_storage_info_.erase(origin);
139 BrowsingDataLocalStorageHelper::DeleteOrigin(origin);
140 }
141
~CannedBrowsingDataLocalStorageHelper()142 CannedBrowsingDataLocalStorageHelper::~CannedBrowsingDataLocalStorageHelper() {}
143