| OLD | NEW |
| (Empty) |
| 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 } | |
| OLD | NEW |