chrome/browser/browsing_data/browsing_data_indexed_db_helper.cc

   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_indexed_db_helper.h"
   6
   7 #include <vector>
   8
   9 #include "base/bind.h"
  10 #include "base/location.h"
  11 #include "base/time/time.h"
  12 #include "chrome/browser/browsing_data/browsing_data_helper.h"
  13 #include "content/public/browser/browser_thread.h"
  14 #include "content/public/browser/indexed_db_context.h"
  15
  16 using content::BrowserThread;
  17 using content::IndexedDBContext;
  18 using content::IndexedDBInfo;
  19
  20 BrowsingDataIndexedDBHelper::BrowsingDataIndexedDBHelper(
  21     IndexedDBContext* indexed_db_context)
  22     : indexed_db_context_(indexed_db_context) {
  23   DCHECK(indexed_db_context_.get());
  24 }
  25
  26 BrowsingDataIndexedDBHelper::~BrowsingDataIndexedDBHelper() {
  27 }
  28
  29 void BrowsingDataIndexedDBHelper::StartFetching(const FetchCallback& callback) {
  30   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  31   DCHECK(!callback.is_null());
  32   indexed_db_context_->TaskRunner()->PostTask(
  33       FROM_HERE,
  34       base::Bind(
  35           &BrowsingDataIndexedDBHelper::FetchIndexedDBInfoInIndexedDBThread,
  36           this, callback));
  37 }
  38
  39 void BrowsingDataIndexedDBHelper::DeleteIndexedDB(const GURL& origin) {
  40   DCHECK_CURRENTLY_ON(BrowserThread::UI);
  41   indexed_db_context_->TaskRunner()->PostTask(
  42       FROM_HERE,
  43       base::Bind(
  44           &BrowsingDataIndexedDBHelper::DeleteIndexedDBInIndexedDBThread,
  45           this,
  46           origin));
  47 }
  48
  49 void BrowsingDataIndexedDBHelper::FetchIndexedDBInfoInIndexedDBThread(
  50     const FetchCallback& callback) {
  51   DCHECK(indexed_db_context_->TaskRunner()->RunsTasksOnCurrentThread());
  52   DCHECK(!callback.is_null());
  53   std::vector<IndexedDBInfo> origins = indexed_db_context_->GetAllOriginsInfo();
  54   std::list<content::IndexedDBInfo> result;
  55   for (const IndexedDBInfo& origin : origins) {
  56     if (!BrowsingDataHelper::HasWebScheme(origin.origin))
  57       continue;  // Non-websafe state is not considered browsing data.
  58     result.push_back(origin);
  59   }
  60   BrowserThread::PostTask(BrowserThread::UI, FROM_HERE,
  61                           base::Bind(callback, result));
  62 }
  63
  64 void BrowsingDataIndexedDBHelper::DeleteIndexedDBInIndexedDBThread(
  65     const GURL& origin) {
  66   DCHECK(indexed_db_context_->TaskRunner()->RunsTasksOnCurrentThread());
  67   indexed_db_context_->DeleteForOrigin(origin);
  68 }
  69
  70 CannedBrowsingDataIndexedDBHelper::
  71 PendingIndexedDBInfo::PendingIndexedDBInfo(const GURL& origin,
  72                                            const base::string16& name)
  73     : origin(origin),
  74       name(name) {
  75 }
  76
  77 CannedBrowsingDataIndexedDBHelper::
  78 PendingIndexedDBInfo::~PendingIndexedDBInfo() {
  79 }
  80
  81 bool CannedBrowsingDataIndexedDBHelper::PendingIndexedDBInfo::operator<(
  82     const PendingIndexedDBInfo& other) const {
  83   if (origin == other.origin)
  84     return name < other.name;
  85   return origin < other.origin;
  86 }
  87
  88 CannedBrowsingDataIndexedDBHelper::CannedBrowsingDataIndexedDBHelper(
  89     content::IndexedDBContext* context)
  90     : BrowsingDataIndexedDBHelper(context) {
  91 }
  92
  93 CannedBrowsingDataIndexedDBHelper::~CannedBrowsingDataIndexedDBHelper() {}
  94
  95 void CannedBrowsingDataIndexedDBHelper::AddIndexedDB(
  96     const GURL& origin, const base::string16& name) {
  97   if (!BrowsingDataHelper::HasWebScheme(origin))
  98     return;  // Non-websafe state is not considered browsing data.
  99
 100   pending_indexed_db_info_.insert(PendingIndexedDBInfo(origin, name));
 101 }
 102
 103 void CannedBrowsingDataIndexedDBHelper::Reset() {
 104   pending_indexed_db_info_.clear();
 105 }
 106
 107 bool CannedBrowsingDataIndexedDBHelper::empty() const {
 108   return pending_indexed_db_info_.empty();
 109 }
 110
 111 size_t CannedBrowsingDataIndexedDBHelper::GetIndexedDBCount() const {
 112   return pending_indexed_db_info_.size();
 113 }
 114
 115 const std::set<CannedBrowsingDataIndexedDBHelper::PendingIndexedDBInfo>&
 116 CannedBrowsingDataIndexedDBHelper::GetIndexedDBInfo() const  {
 117   return pending_indexed_db_info_;
 118 }
 119
 120 void CannedBrowsingDataIndexedDBHelper::StartFetching(
 121     const FetchCallback& callback) {
 122   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 123   DCHECK(!callback.is_null());
 124
 125   std::list<IndexedDBInfo> result;
 126   for (const PendingIndexedDBInfo& pending_info : pending_indexed_db_info_) {
 127     IndexedDBInfo info(pending_info.origin, 0, base::Time(), 0);
 128     result.push_back(info);
 129   }
 130
 131   BrowserThread::PostTask(
 132       BrowserThread::UI, FROM_HERE, base::Bind(callback, result));
 133 }
 134
 135 void CannedBrowsingDataIndexedDBHelper::DeleteIndexedDB(
 136     const GURL& origin) {
 137   for (std::set<PendingIndexedDBInfo>::iterator it =
 138            pending_indexed_db_info_.begin();
 139        it != pending_indexed_db_info_.end(); ) {
 140     if (it->origin == origin)
 141       pending_indexed_db_info_.erase(it++);
 142     else
 143       ++it;
 144   }
 145   BrowsingDataIndexedDBHelper::DeleteIndexedDB(origin);
 146 }