src/chrome/browser/webdata/web_intents_table.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/webdata/web_intents_table.h"
   6
   7 #include <string>
   8
   9 #include "base/i18n/case_conversion.h"
  10 #include "base/logging.h"
  11 #include "base/strings/string_util.h"
  12 #include "base/strings/utf_string_conversions.h"
  13 #include "components/webdata/common/web_database.h"
  14 #include "net/base/mime_util.h"
  15 #include "sql/statement.h"
  16 #include "third_party/sqlite/sqlite3.h"
  17 #include "url/gurl.h"
  18
  19 namespace {
  20
  21 WebDatabaseTable::TypeKey GetKey() {
  22   // We just need a unique constant. Use the address of a static that
  23   // COMDAT folding won't touch in an optimizing linker.
  24   static int table_key = 0;
  25   return reinterpret_cast<void*>(&table_key);
  26 }
  27
  28 }  // namespace
  29
  30 WebIntentsTable::WebIntentsTable() {
  31 }
  32
  33 WebIntentsTable::~WebIntentsTable() {
  34 }
  35
  36 WebIntentsTable* WebIntentsTable::FromWebDatabase(WebDatabase* db) {
  37   return static_cast<WebIntentsTable*>(db->GetTable(GetKey()));
  38 }
  39
  40 WebDatabaseTable::TypeKey WebIntentsTable::GetTypeKey() const {
  41   return GetKey();
  42 }
  43
  44 bool WebIntentsTable::CreateTablesIfNecessary() {
  45   if (!db_->DoesTableExist("web_intents")) {
  46     if (!db_->Execute("CREATE TABLE web_intents ("
  47                       " service_url LONGVARCHAR,"
  48                       " action VARCHAR,"
  49                       " type VARCHAR,"
  50                       " title LONGVARCHAR,"
  51                       " disposition VARCHAR,"
  52                       " scheme VARCHAR,"
  53                       " UNIQUE (service_url, action, scheme, type))")) {
  54       return false;
  55     }
  56     if (!db_->Execute("CREATE INDEX IF NOT EXISTS web_intents_index"
  57                       " ON web_intents (action)"))
  58       return false;
  59     if (!db_->Execute("CREATE INDEX IF NOT EXISTS web_intents_index"
  60                       " ON web_intents (scheme)"))
  61       return false;
  62   }
  63
  64   if (!db_->DoesTableExist("web_intents_defaults")) {
  65     if (!db_->Execute("CREATE TABLE web_intents_defaults ("
  66                       " action VARCHAR,"
  67                       " type VARCHAR,"
  68                       " url_pattern LONGVARCHAR,"
  69                       " user_date INTEGER,"
  70                       " suppression INTEGER,"
  71                       " service_url LONGVARCHAR,"
  72                       " scheme VARCHAR,"
  73                       " UNIQUE (action, scheme, type, url_pattern))")) {
  74       return false;
  75     }
  76     if (!db_->Execute("CREATE INDEX IF NOT EXISTS web_intents_default_index"
  77                       " ON web_intents_defaults (action)"))
  78       return false;
  79
  80     if (!db_->Execute("CREATE INDEX IF NOT EXISTS web_intents_default_index"
  81                       " ON web_intents_defaults (scheme)"))
  82       return false;
  83   }
  84
  85   return true;
  86 }
  87
  88 // TODO(jhawkins): Figure out Sync story.
  89 bool WebIntentsTable::IsSyncable() {
  90   return false;
  91 }
  92
  93 bool WebIntentsTable::MigrateToVersion(int version,
  94                                        bool* update_compatible_version) {
  95   if (version == 46) {
  96     *update_compatible_version = true;
  97     return MigrateToVersion46AddSchemeColumn();
  98   }
  99
 100   return true;
 101 }
 102
 103 // Updates the table by way of renaming the old tables, rerunning
 104 // the Init method, then selecting old values into the new tables.
 105 bool WebIntentsTable::MigrateToVersion46AddSchemeColumn() {
 106   // If the old table doesn't exist, there's nothing to migrate.
 107   if (!db_->DoesTableExist("web_intents"))
 108     return true;
 109
 110   if (!db_->Execute("ALTER TABLE web_intents RENAME TO old_web_intents"))
 111     return false;
 112
 113   if (!db_->Execute("ALTER TABLE web_intents_defaults"
 114                     " RENAME TO old_web_intents_defaults"))
 115     return false;
 116
 117   if (!CreateTablesIfNecessary())
 118     return false;
 119
 120   int error = db_->ExecuteAndReturnErrorCode(
 121       "INSERT INTO web_intents"
 122       " (service_url, action, type, title, disposition)"
 123       " SELECT "
 124       " service_url, action, type, title, disposition"
 125       " FROM old_web_intents");
 126
 127   if (error != SQLITE_OK) {
 128     DLOG(WARNING) << "Could not copy old intent data to upgraded table."
 129         << db_->GetErrorMessage();
 130   }
 131
 132
 133   error = db_->ExecuteAndReturnErrorCode(
 134       "INSERT INTO web_intents_defaults"
 135       " (service_url, action, type, url_pattern, user_date, suppression)"
 136       " SELECT "
 137       " service_url, action, type, url_pattern, user_date, suppression"
 138       " FROM old_web_intents_defaults");
 139
 140   if (error != SQLITE_OK) {
 141     DLOG(WARNING) << "Could not copy old intent defaults to upgraded table."
 142         << db_->GetErrorMessage();
 143   }
 144
 145   if (!db_->Execute("DROP table old_web_intents")) {
 146     LOG(WARNING) << "Could not drop backup web_intents table.";
 147     return false;
 148   }
 149
 150   if (!db_->Execute("DROP table old_web_intents_defaults")) {
 151     DLOG(WARNING) << "Could not drop backup web_intents_defaults table.";
 152     return false;
 153   }
 154
 155   return true;
 156 }