Merge 121330 - Modify schema to include defaulting information.

chrome/browser/webdata/web_intents_table.cc

-// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// 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/webdata/web_intents_table.h"
+#include <string>
 
+#include "base/i18n/case_conversion.h"
 #include "base/logging.h"
+#include "base/string_util.h"
 #include "base/utf_string_conversions.h"
+#include "chrome/browser/intents/default_web_intent_service.h"
 #include "googleurl/src/gurl.h"
+#include "net/base/mime_util.h"
 #include "sql/statement.h"
 
 using webkit_glue::WebIntentServiceData;
 
 namespace {
 
 bool ExtractIntents(sql::Statement* s,
                     std::vector<WebIntentServiceData>* services) {
   DCHECK(s);
   if (!s->is_valid())
@@ skipping 21 matching lines @@
 
 WebIntentsTable::WebIntentsTable(sql::Connection* db,
                                  sql::MetaTable* meta_table)
     : WebDatabaseTable(db, meta_table) {
 }
 
 WebIntentsTable::~WebIntentsTable() {
 }
 
 bool WebIntentsTable::Init() {
-  if (db_->DoesTableExist("web_intents"))
-    return true;
+  if (!db_->DoesTableExist("web_intents")) {
+    if (!db_->Execute("CREATE TABLE web_intents ("
+                      "service_url LONGVARCHAR,"
+                      "action VARCHAR,"
+                      "type VARCHAR,"
+                      "title LONGVARCHAR,"
+                      "disposition VARCHAR,"
+                      "UNIQUE (service_url, action, type))")) {
+      return false;
+    }
+  }
 
-  if (!db_->Execute("CREATE TABLE web_intents ("
-                    "service_url LONGVARCHAR,"
-                    "action VARCHAR,"
-                    "type VARCHAR,"
-                    "title VARCHAR,"
-                    "disposition VARCHAR,"
-                    "UNIQUE (service_url, action, type))")) {
+  if (!db_->DoesTableExist("web_intents_defaults")) {
+    if (!db_->Execute("CREATE TABLE web_intents_defaults ("
+                      "action VARCHAR,"
+                      "type VARCHAR,"
+                      "url_pattern LONGVARCHAR,"
+                      "user_date INTEGER,"
+                      "suppression INTEGER,"
+                      "service_url LONGVARCHAR,"
+                      "UNIQUE (action, type, url_pattern))")) {
+      return false;
+    }
+  }
+
+  if (!db_->Execute("CREATE INDEX IF NOT EXISTS web_intents_index "
+                    "ON web_intents (action)"))
+    return false;
+
+  if (!db_->Execute("CREATE INDEX IF NOT EXISTS web_intents_default_index "
+                    "ON web_intents_defaults (action)")) {
     return false;
   }
 
-  if (!db_->Execute("CREATE INDEX web_intents_index ON web_intents (action)"))
-    return false;
-
   return true;
 }
 
 // TODO(jhawkins): Figure out Sync story.
 bool WebIntentsTable::IsSyncable() {
   return false;
 }
 
 bool WebIntentsTable::GetWebIntentServices(
     const string16& action,
@@ skipping 57 matching lines @@
     const WebIntentServiceData& service) {
   sql::Statement s(db_->GetUniqueStatement(
       "DELETE FROM web_intents "
       "WHERE service_url = ? AND action = ? AND type = ?"));
   s.BindString(0, service.service_url.spec());
   s.BindString16(1, service.action);
   s.BindString16(2, service.type);
 
   return s.Run();
 }
+
+bool WebIntentsTable::GetDefaultServices(
+    const string16& action,
+    std::vector<DefaultWebIntentService>* default_services) {
+  sql::Statement s(db_->GetUniqueStatement(
+      "SELECT action, type, url_pattern, user_date, suppression, "
+      "service_url FROM web_intents_defaults "
+      "WHERE action=?"));
+  s.BindString16(0, action);
+
+  while (s.Step()) {
+    DefaultWebIntentService entry;
+    entry.action = s.ColumnString16(0);
+    entry.type = s.ColumnString16(1);
+    if (entry.url_pattern.Parse(s.ColumnString(2)) !=
+        URLPattern::PARSE_SUCCESS) {
+      return false;
+    }
+    entry.user_date = s.ColumnInt(3);
+    entry.suppression = s.ColumnInt(4);
+    entry.service_url = s.ColumnString(5);
+
+    default_services->push_back(entry);
+  }
+
+  return s.Succeeded();
+}
+
+bool WebIntentsTable::GetAllDefaultServices(
+    std::vector<DefaultWebIntentService>* default_services) {
+  sql::Statement s(db_->GetUniqueStatement(
+      "SELECT action, type, url_pattern, user_date, suppression, "
+      "service_url FROM web_intents_defaults"));
+
+  while (s.Step()) {
+    DefaultWebIntentService entry;
+    entry.action = s.ColumnString16(0);
+    entry.type = s.ColumnString16(1);
+    if (entry.url_pattern.Parse(s.ColumnString(2)) !=
+        URLPattern::PARSE_SUCCESS) {
+      return false;
+    }
+    entry.user_date = s.ColumnInt(3);
+    entry.suppression = s.ColumnInt(4);
+    entry.service_url = s.ColumnString(5);
+
+    default_services->push_back(entry);
+  }
+
+  return s.Succeeded();
+
+}
+
+bool WebIntentsTable::SetDefaultService(
+    const DefaultWebIntentService& default_service) {
+  sql::Statement s(db_->GetUniqueStatement(
+      "INSERT OR REPLACE INTO web_intents_defaults "
+      "(action, type, url_pattern, user_date, suppression, service_url) "
+      "VALUES (?, ?, ?, ?, ?, ?)"));
+  s.BindString16(0, default_service.action);
+  s.BindString16(1, default_service.type);
+  s.BindString(2, default_service.url_pattern.GetAsString());
+  s.BindInt(3, default_service.user_date);
+  s.BindInt(4, default_service.suppression);
+  s.BindString(5, default_service.service_url);
+
+  return s.Run();
+}
+
+bool WebIntentsTable::RemoveDefaultService(
+    const DefaultWebIntentService& default_service) {
+  sql::Statement s(db_->GetUniqueStatement(
+      "DELETE FROM web_intents_defaults "
+      "WHERE action = ? AND type = ? AND url_pattern = ?"));
+  s.BindString16(0, default_service.action);
+  s.BindString16(1, default_service.type);
+  s.BindString(2, default_service.url_pattern.GetAsString());
+
+  return s.Run();
+}