Merge 121330 - Modify schema to include defaulting information.

chrome/browser/webdata/web_intents_table.cc

Index: chrome/browser/webdata/web_intents_table.cc
===================================================================
--- chrome/browser/webdata/web_intents_table.cc	(revision 122525)
+++ chrome/browser/webdata/web_intents_table.cc	(working copy)
@@ -1,12 +1,17 @@
-// 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;
@@ -48,22 +53,40 @@
 }
 
 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))")) {
-    return false;
+  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 web_intents_index ON web_intents (action)"))
+  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;
+  }
+
   return true;
 }
 
@@ -141,3 +164,83 @@
 
   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();
+}