Autofill i18n: Set postal code and state field labels based on the selected country.

chrome/browser/webdata/web_database.cc

 // Copyright (c) 2011 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_database.h"
 
 #include <algorithm>
 #include <limits>
 #include <set>
 #include <string>
 
 #include "app/sql/statement.h"
 #include "app/sql/transaction.h"
 #include "base/string_number_conversions.h"
 #include "base/string_split.h"
 #include "base/string_util.h"
 #include "base/tuple.h"
 #include "base/utf_string_conversions.h"
+#include "chrome/browser/autofill/autofill_country.h"
 #include "chrome/browser/autofill/autofill_profile.h"
 #include "chrome/browser/autofill/autofill_type.h"
 #include "chrome/browser/autofill/credit_card.h"
 #include "chrome/browser/diagnostics/sqlite_diagnostics.h"
 #include "chrome/browser/history/history_database.h"
 #include "chrome/browser/password_manager/encryptor.h"
 #include "chrome/browser/search_engines/template_url.h"
 #include "chrome/browser/webdata/autofill_change.h"
 #include "chrome/common/guid.h"
 #include "chrome/common/notification_service.h"
@@ skipping 87 matching lines @@
 //   first_name
 //   middle_name
 //   last_name
 //   email
 //   company_name
 //   address_line_1
 //   address_line_2
 //   city
 //   state
 //   zipcode
-//   country
+//   country            The country name.  Deprecated, should be removed once
+//                      the stable channel reaches version 11.
+//   country_code
 //   phone
 //   fax
 //   date_modified      The date on which this profile was last modified.
 //                      Added in version 30.
 //
 // credit_cards         This table contains credit card data added by the user
 //                      with the AutoFill dialog.  Most of the columns are
 //                      standard entries in a credit card form.
 //
 //   guid               A guid string to uniquely identify the profile.
@@ skipping 21 matching lines @@
 
 using base::Time;
 
 namespace {
 
 typedef std::vector<Tuple3<int64, string16, string16> > AutofillElementList;
 
 // Current version number.  Note: when changing the current version number,
 // corresponding changes must happen in the unit tests, and new migration test
 // added.  See |WebDatabaseMigrationTest::kCurrentTestedVersionNumber|.
-const int kCurrentVersionNumber = 32;
-const int kCompatibleVersionNumber = 32;
+const int kCurrentVersionNumber = 33;
+const int kCompatibleVersionNumber = 33;
 
 // ID of the url column in keywords.
 const int kUrlIdPosition = 16;
 
 // Keys used in the meta table.
 const char* kDefaultSearchProviderKey = "Default Search Provider ID";
 const char* kBuiltinKeywordVersion = "Builtin Keyword Version";
 
 // The maximum length allowed for form data.
 const size_t kMaxDataLength = 1024;
@@ skipping 94 matching lines @@
   text = profile.GetFieldText(AutoFillType(ADDRESS_HOME_LINE2));
   s->BindString16(8, LimitDataSize(text));
   text = profile.GetFieldText(AutoFillType(ADDRESS_HOME_CITY));
   s->BindString16(9, LimitDataSize(text));
   text = profile.GetFieldText(AutoFillType(ADDRESS_HOME_STATE));
   s->BindString16(10, LimitDataSize(text));
   text = profile.GetFieldText(AutoFillType(ADDRESS_HOME_ZIP));
   s->BindString16(11, LimitDataSize(text));
   text = profile.GetFieldText(AutoFillType(ADDRESS_HOME_COUNTRY));
   s->BindString16(12, LimitDataSize(text));
+  std::string country_code = profile.CountryCode();
+  s->BindString(13, country_code);
   text = profile.GetFieldText(AutoFillType(PHONE_HOME_WHOLE_NUMBER));
-  s->BindString16(13, LimitDataSize(text));
+  s->BindString16(14, LimitDataSize(text));
   text = profile.GetFieldText(AutoFillType(PHONE_FAX_WHOLE_NUMBER));
-  s->BindString16(14, LimitDataSize(text));
-  s->BindInt64(15, Time::Now().ToTimeT());
+  s->BindString16(15, LimitDataSize(text));
+  s->BindInt64(16, Time::Now().ToTimeT());
 }
 
 AutoFillProfile* AutoFillProfileFromStatement(const sql::Statement& s) {
   AutoFillProfile* profile = new AutoFillProfile;
   profile->set_guid(s.ColumnString(0));
   DCHECK(guid::IsValidGUID(profile->guid()));
   // TODO(dhollowa): remove label from |autofill_profiles| table.
   // Column 1 is label.
 
   profile->SetInfo(AutoFillType(NAME_FIRST),
                    s.ColumnString16(2));
   profile->SetInfo(AutoFillType(NAME_MIDDLE),
                    s.ColumnString16(3));
   profile->SetInfo(AutoFillType(NAME_LAST),
                    s.ColumnString16(4));
   profile->SetInfo(AutoFillType(EMAIL_ADDRESS),
                    s.ColumnString16(5));
   profile->SetInfo(AutoFillType(COMPANY_NAME),
                    s.ColumnString16(6));
   profile->SetInfo(AutoFillType(ADDRESS_HOME_LINE1),
                    s.ColumnString16(7));
   profile->SetInfo(AutoFillType(ADDRESS_HOME_LINE2),
                    s.ColumnString16(8));
   profile->SetInfo(AutoFillType(ADDRESS_HOME_CITY),
                    s.ColumnString16(9));
   profile->SetInfo(AutoFillType(ADDRESS_HOME_STATE),
                    s.ColumnString16(10));
   profile->SetInfo(AutoFillType(ADDRESS_HOME_ZIP),
                    s.ColumnString16(11));
-  profile->SetInfo(AutoFillType(ADDRESS_HOME_COUNTRY),
-                   s.ColumnString16(12));
+  // Intentionally skip column 12, which stores the localized country name.
+  profile->SetCountryCode(s.ColumnString(13));
   profile->SetInfo(AutoFillType(PHONE_HOME_WHOLE_NUMBER),
-                   s.ColumnString16(13));
+                   s.ColumnString16(14));
   profile->SetInfo(AutoFillType(PHONE_FAX_WHOLE_NUMBER),
-                   s.ColumnString16(14));
-  // Intentionally skip column 15, which stores the profile's modification date.
+                   s.ColumnString16(15));
+  // Intentionally skip column 16, which stores the profile's modification date.
 
   return profile;
 }
 
 void BindCreditCardToStatement(const CreditCard& credit_card,
                                sql::Statement* s) {
   DCHECK(guid::IsValidGUID(credit_card.guid()));
   s->BindString(0, credit_card.guid());
   s->BindString16(1, credit_card.Label());
 
@@ skipping 392 matching lines @@
                      "middle_name VARCHAR, "
                      "last_name VARCHAR, "
                      "email VARCHAR, "
                      "company_name VARCHAR, "
                      "address_line_1 VARCHAR, "
                      "address_line_2 VARCHAR, "
                      "city VARCHAR, "
                      "state VARCHAR, "
                      "zipcode VARCHAR, "
                      "country VARCHAR, "
+                     "country_code VARCHAR, "
                      "phone VARCHAR, "
                      "fax VARCHAR, "
                      "date_modified INTEGER NOT NULL DEFAULT 0)")) {
       NOTREACHED();
       return false;
     }
     if (!db_.Execute("CREATE INDEX autofill_profiles_label_index "
                      "ON autofill_profiles (label)")) {
       NOTREACHED();
       return false;
@@ skipping 855 matching lines @@
   if (s.Step())
     return RemoveFormElementForID(s.ColumnInt64(0));
   return false;
 }
 
 bool WebDatabase::AddAutoFillProfile(const AutoFillProfile& profile) {
   sql::Statement s(db_.GetUniqueStatement(
       "INSERT INTO autofill_profiles"
       "(guid, label, first_name, middle_name, last_name, email,"
       " company_name, address_line_1, address_line_2, city, state, zipcode,"
-      " country, phone, fax, date_modified)"
-      "VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)"));
+      " country, country_code, phone, fax, date_modified)"
+      "VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?,?)"));
   if (!s) {
     NOTREACHED() << "Statement prepare failed";
     return false;
   }
 
   BindAutoFillProfileToStatement(profile, &s);
 
   if (!s.Run()) {
     NOTREACHED();
     return false;
   }
 
   return s.Succeeded();
 }
 
 bool WebDatabase::GetAutoFillProfile(const std::string& guid,
                                      AutoFillProfile** profile) {
   DCHECK(guid::IsValidGUID(guid));
   DCHECK(profile);
   sql::Statement s(db_.GetUniqueStatement(
       "SELECT guid, label, first_name, middle_name, last_name, email, "
       "company_name, address_line_1, address_line_2, city, state, zipcode, "
-      "country, phone, fax, date_modified "
+      "country, country_code, phone, fax, date_modified "
       "FROM autofill_profiles "
       "WHERE guid = ?"));
   if (!s) {
     NOTREACHED() << "Statement prepare failed";
     return false;
   }
 
   s.BindString(0, guid);
   if (!s.Step())
     return false;
 
   *profile = AutoFillProfileFromStatement(s);
 
   return s.Succeeded();
 }
 
 bool WebDatabase::GetAutoFillProfiles(
     std::vector<AutoFillProfile*>* profiles) {
   DCHECK(profiles);
   profiles->clear();
 
   sql::Statement s(db_.GetUniqueStatement(
       "SELECT guid, label, first_name, middle_name, last_name, email, "
       "company_name, address_line_1, address_line_2, city, state, zipcode, "
-      "country, phone, fax, date_modified "
+      "country, country_code, phone, fax, date_modified "
       "FROM autofill_profiles"));
   if (!s) {
     NOTREACHED() << "Statement prepare failed";
     return false;
   }
 
   while (s.Step())
     profiles->push_back(AutoFillProfileFromStatement(s));
 
   return s.Succeeded();
 }
 
 bool WebDatabase::UpdateAutoFillProfile(const AutoFillProfile& profile) {
   DCHECK(guid::IsValidGUID(profile.guid()));
 
   AutoFillProfile* tmp_profile = NULL;
   if (!GetAutoFillProfile(profile.guid(), &tmp_profile))
     return false;
 
   // Preserve appropriate modification dates by not updating unchanged profiles.
   scoped_ptr<AutoFillProfile> old_profile(tmp_profile);
   if (*old_profile == profile)
     return true;
 
   sql::Statement s(db_.GetUniqueStatement(
       "UPDATE autofill_profiles "
       "SET guid=?, label=?, first_name=?, middle_name=?, last_name=?, "
       "    email=?, company_name=?, address_line_1=?, address_line_2=?, "
-      "    city=?, state=?, zipcode=?, country=?, phone=?, fax=?, "
-      "    date_modified=? "
+      "    city=?, state=?, zipcode=?, country=?, country_code=?, phone=?, "
+      "    fax=?, date_modified=? "
       "WHERE guid=?"));
   if (!s) {
     NOTREACHED() << "Statement prepare failed";
     return false;
   }
 
   BindAutoFillProfileToStatement(profile, &s);
-  s.BindString(16, profile.guid());
+  s.BindString(17, profile.guid());
   bool result = s.Run();
   DCHECK_GT(db_.GetLastChangeCount(), 0);
   return result;
 }
 
 bool WebDatabase::RemoveAutoFillProfile(const std::string& guid) {
   DCHECK(guid::IsValidGUID(guid));
   sql::Statement s(db_.GetUniqueStatement(
       "DELETE FROM autofill_profiles WHERE guid = ?"));
   if (!s) {
@@ skipping 816 matching lines @@
           return sql::INIT_FAILURE;
         }
       }
 
       meta_table_.SetVersionNumber(32);
       meta_table_.SetCompatibleVersionNumber(
           std::min(32, kCompatibleVersionNumber));
 
       // FALL THROUGH
 
+    case 32:
+      // Add |country_code| column to |autofill_profiles| table.
+      // Note that we need to check for the column's existence due to the fact
+      // that for a version 22 database the |autofill_profiles| table gets
+      // created fresh with |InitAutoFillProfilesTable|.
+      if (!db_.DoesColumnExist("autofill_profiles", "country_code")) {
+        if (!db_.Execute("ALTER TABLE autofill_profiles ADD COLUMN "
+                         "country_code VARCHAR")) {
+          LOG(WARNING) << "Unable to update web database to version 33.";
+          NOTREACHED();
+          return sql::INIT_FAILURE;
+        }
+
+        // Set all the |country_code| fields to match existing |country| values.
+        {
+          sql::Statement s(db_.GetUniqueStatement("SELECT guid, country "
+                                                  "FROM autofill_profiles"));
+
+          if (!s) {
+            LOG(WARNING) << "Unable to update web database to version 33.";
+            NOTREACHED();
+            return sql::INIT_FAILURE;
+          }
+
+          while (s.Step()) {
+            sql::Statement update_s(
+                db_.GetUniqueStatement("UPDATE autofill_profiles "
+                                       "SET country_code=? WHERE guid=?"));
+            if (!update_s) {
+              LOG(WARNING) << "Unable to update web database to version 33.";
+              NOTREACHED();
+              return sql::INIT_FAILURE;
+            }
+            string16 country = s.ColumnString16(1);
+            update_s.BindString(0, AutoFillCountry::GetCountryCode(country));
+            update_s.BindString(1, s.ColumnString(0));
+
+            if (!update_s.Run()) {
+              LOG(WARNING) << "Unable to update web database to version 33.";
+              NOTREACHED();
+              return sql::INIT_FAILURE;
+            }
+          }
+        }
+      }
+
+      meta_table_.SetVersionNumber(33);
+      meta_table_.SetCompatibleVersionNumber(
+          std::min(33, kCompatibleVersionNumber));
+
+      // FALL THROUGH
+
     // Add successive versions here.  Each should set the version number and
     // compatible version number as appropriate, then fall through to the next
     // case.
 
     case kCurrentVersionNumber:
       // No migration needed.
       return sql::INIT_OK;
   }
 }