Adding methods that will be used by the quota management UI....

webkit/database/databases_table.cc

 // Copyright (c) 2009 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 "webkit/database/databases_table.h"
 
 #include "app/sql/connection.h"
 #include "app/sql/statement.h"
 #include "base/logging.h"
 #include "base/utf_string_conversions.h"
@@ skipping 94 matching lines @@
       SQL_FROM_HERE, "DELETE FROM Databases WHERE origin = ? AND name = ?"));
   if (delete_statement.is_valid() &&
       delete_statement.BindString(0, UTF16ToUTF8(origin_identifier)) &&
       delete_statement.BindString(1, UTF16ToUTF8(database_name))) {
     return (delete_statement.Run() && db_->GetLastChangeCount());
   }
 
   return false;
 }
 
+bool DatabasesTable::GetAllOrigins(std::vector<string16>* origins) {
+  sql::Statement statement(db_->GetCachedStatement(
+      SQL_FROM_HERE, "SELECT DISTINCT origin FROM Databases ORDER BY origin"));
+  if (statement.is_valid()) {
+    while (statement.Step())
+      origins->push_back(UTF8ToUTF16(statement.ColumnString(0)));
+    return statement.Succeeded();
+  }
+
+  return false;
+}
+
 bool DatabasesTable::GetAllDatabaseDetailsForOrigin(
     const string16& origin_identifier,
     std::vector<DatabaseDetails>* details_vector) {
   sql::Statement statement(db_->GetCachedStatement(
       SQL_FROM_HERE, "SELECT name, description, estimated_size "
-                     "FROM Databases WHERE origin = ? ORDER BY origin, name"));
+                     "FROM Databases WHERE origin = ? ORDER BY name"));
   if (statement.is_valid() &&
       statement.BindString(0, UTF16ToUTF8(origin_identifier))) {
     while (statement.Step()) {
       DatabaseDetails details;
       details.origin_identifier = origin_identifier;
       details.database_name = UTF8ToUTF16(statement.ColumnString(0));
       details.description = UTF8ToUTF16(statement.ColumnString(1));
       details.estimated_size = statement.ColumnInt64(2);
       details_vector->push_back(details);
     }
     return statement.Succeeded();
   }
 
   return false;
 }
 
+bool DatabasesTable::DeleteOrigin(const string16& origin_identifier) {
+  sql::Statement delete_statement(db_->GetCachedStatement(
+      SQL_FROM_HERE, "DELETE FROM Databases WHERE origin = ?"));
+  if (delete_statement.is_valid() &&
+      delete_statement.BindString(0, UTF16ToUTF8(origin_identifier))) {
+    return (delete_statement.Run() && db_->GetLastChangeCount());
+  }
+
+  return false;
+}
+
 }  // namespace webkit_database