Implement download link in chrome://indexeddb-internals/

content/browser/indexed_db/indexed_db_context_impl.cc

 // 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 "content/browser/indexed_db/indexed_db_context_impl.h"
 
 #include <vector>
 
 #include "base/bind.h"
 #include "base/command_line.h"
@@ skipping 112 matching lines @@
     origins.push_back(*iter);
   }
   return origins;
 }
 
 std::vector<IndexedDBInfo> IndexedDBContextImpl::GetAllOriginsInfo() {
   std::vector<GURL> origins = GetAllOrigins();
   std::vector<IndexedDBInfo> result;
   for (std::vector<GURL>::const_iterator iter = origins.begin();
        iter != origins.end(); ++iter) {
-    const GURL& origin = *iter;
+    const GURL& origin_url = *iter;
 
-    string16 origin_id = DatabaseUtil::GetOriginIdentifier(origin);
-    base::FilePath idb_directory = GetIndexedDBFilePath(origin_id);
-    result.push_back(IndexedDBInfo(origin,
-                                   GetOriginDiskUsage(origin),
-                                   GetOriginLastModified(origin),
+    base::FilePath idb_directory = GetFilePath(origin_url);
+    result.push_back(IndexedDBInfo(origin_url,
+                                   GetOriginDiskUsage(origin_url),
+                                   GetOriginLastModified(origin_url),
                                    idb_directory));
   }
   return result;
 }
 
 int64 IndexedDBContextImpl::GetOriginDiskUsage(const GURL& origin_url) {
-  if (data_path_.empty() || !IsInOriginSet(origin_url))
+  if (data_path_.empty() || !HasOrigin(origin_url))
     return 0;
   EnsureDiskUsageCacheInitialized(origin_url);
   return origin_size_map_[origin_url];
 }
 
 base::Time IndexedDBContextImpl::GetOriginLastModified(const GURL& origin_url) {
-  if (data_path_.empty() || !IsInOriginSet(origin_url))
+  if (data_path_.empty() || !HasOrigin(origin_url))
     return base::Time();
-  string16 origin_id = DatabaseUtil::GetOriginIdentifier(origin_url);
-  base::FilePath idb_directory = GetIndexedDBFilePath(origin_id);
+  base::FilePath idb_directory = GetFilePath(origin_url);
   base::PlatformFileInfo file_info;
   if (!file_util::GetFileInfo(idb_directory, &file_info))
     return base::Time();
   return file_info.last_modified;
 }
 
 void IndexedDBContextImpl::DeleteForOrigin(const GURL& origin_url) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::WEBKIT_DEPRECATED));
-  if (data_path_.empty() || !IsInOriginSet(origin_url))
+  ForceClose(origin_url);
+  if (data_path_.empty() || !HasOrigin(origin_url))
+    return;
+
+  base::FilePath idb_directory = GetFilePath(origin_url);
+  EnsureDiskUsageCacheInitialized(origin_url);
+  const bool recursive = true;
+  bool deleted = file_util::Delete(idb_directory, recursive);
+
+  QueryDiskAndUpdateQuotaUsage(origin_url);
+  if (deleted) {
+    RemoveFromOriginSet(origin_url);
+    origin_size_map_.erase(origin_url);
+    space_available_map_.erase(origin_url);
+  }
+}
+
+void IndexedDBContextImpl::ForceClose(const GURL& origin_url) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::WEBKIT_DEPRECATED));
+  if (data_path_.empty() || !HasOrigin(origin_url))
     return;
 
   if (connections_.find(origin_url) != connections_.end()) {
     ConnectionSet& connections = connections_[origin_url];
     ConnectionSet::iterator it = connections.begin();
     while (it != connections.end()) {
       // Remove before closing so callbacks don't double-erase
       WebKit::WebIDBDatabase* db = *it;
       connections.erase(it++);
       db->forceClose();
     }
-    DCHECK(connections_[origin_url].size() == 0);
+    DCHECK_EQ(connections_[origin_url].size(), 0UL);
     connections_.erase(origin_url);
   }
-
-  string16 origin_id = DatabaseUtil::GetOriginIdentifier(origin_url);
-  base::FilePath idb_directory = GetIndexedDBFilePath(origin_id);
-  EnsureDiskUsageCacheInitialized(origin_url);
-  const bool recursive = true;
-  bool deleted = file_util::Delete(idb_directory, recursive);
-
-  QueryDiskAndUpdateQuotaUsage(origin_url);
-  if (deleted) {
-    RemoveFromOriginSet(origin_url);
-    origin_size_map_.erase(origin_url);
-    space_available_map_.erase(origin_url);
-  }
 }
 
+base::FilePath IndexedDBContextImpl::GetFilePath(const GURL& origin_url) {
+  string16 origin_id = DatabaseUtil::GetOriginIdentifier(origin_url);
+  return GetIndexedDBFilePath(origin_id);
+}
+
 base::FilePath IndexedDBContextImpl::GetFilePathForTesting(
   const string16& origin_id) const {
   return GetIndexedDBFilePath(origin_id);
 }
 
 void IndexedDBContextImpl::ConnectionOpened(const GURL& origin_url,
                                             WebIDBDatabase* connection) {
-  DCHECK(connections_[origin_url].count(connection) == 0);
+  DCHECK_EQ(connections_[origin_url].count(connection), 0UL);
   if (quota_manager_proxy()) {
     quota_manager_proxy()->NotifyStorageAccessed(
         quota::QuotaClient::kIndexedDatabase, origin_url,
         quota::kStorageTypeTemporary);
   }
   connections_[origin_url].insert(connection);
   if (AddToOriginSet(origin_url)) {
     // A newly created db, notify the quota system.
     QueryDiskAndUpdateQuotaUsage(origin_url);
   } else {
@@ skipping 164 matching lines @@
     std::vector<GURL> origins;
     GetAllOriginsAndPaths(data_path_, &origins, NULL);
     for (std::vector<GURL>::const_iterator iter = origins.begin();
          iter != origins.end(); ++iter) {
       origin_set_->insert(*iter);
     }
   }
   return origin_set_.get();
 }
 
+bool IndexedDBContextImpl::HasOrigin(const GURL& origin_url) {
+  std::set<GURL>* set = GetOriginSet();
+  return set->find(origin_url) != set->end();
+}
+
 void IndexedDBContextImpl::ResetCaches() {
   origin_set_.reset();
   origin_size_map_.clear();
   space_available_map_.clear();
 }
 
 }  // namespace content