chrome/browser/ui/autofill/country_combobox_model.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/ui/autofill/country_combobox_model.h"
   6
   7 #include "base/logging.h"
   8 #include "base/strings/utf_string_conversions.h"
   9 #include "chrome/browser/browser_process.h"
  10 #include "components/autofill/core/browser/autofill_country.h"
  11 #include "components/autofill/core/browser/personal_data_manager.h"
  12 #include "ui/base/l10n/l10n_util_collator.h"
  13 #include "ui/base/models/combobox_model_observer.h"
  14
  15 namespace autofill {
  16
  17 CountryComboboxModel::CountryComboboxModel(const PersonalDataManager& manager)
  18     : default_index_(0) {
  19   // Insert the default country at the top as well as in the ordered list.
  20   const std::string& app_locale = g_browser_process->GetApplicationLocale();
  21   std::string default_country_code =
  22       manager.GetDefaultCountryCodeForNewAddress();
  23   DCHECK(!default_country_code.empty());
  24
  25   countries_.push_back(new AutofillCountry(default_country_code, app_locale));
  26   // The separator item.
  27   countries_.push_back(NULL);
  28
  29   // The sorted list of countries.
  30   std::vector<std::string> country_codes;
  31   AutofillCountry::GetAvailableCountries(&country_codes);
  32   std::vector<AutofillCountry*> sorted_countries;
  33
  34   for (std::vector<std::string>::iterator iter = country_codes.begin();
  35        iter != country_codes.end(); ++iter) {
  36     sorted_countries.push_back(new AutofillCountry(*iter, app_locale));
  37   }
  38
  39   l10n_util::SortStringsUsingMethod(app_locale,
  40                                     &sorted_countries,
  41                                     &AutofillCountry::name);
  42   countries_.insert(countries_.end(),
  43                     sorted_countries.begin(),
  44                     sorted_countries.end());
  45 }
  46
  47 CountryComboboxModel::~CountryComboboxModel() {}
  48
  49 int CountryComboboxModel::GetItemCount() const {
  50   return countries_.size();
  51 }
  52
  53 base::string16 CountryComboboxModel::GetItemAt(int index) {
  54   AutofillCountry* country = countries_[index];
  55   if (country)
  56     return countries_[index]->name();
  57
  58   // The separator item. Implemented for platforms that don't yet support
  59   // IsItemSeparatorAt().
  60   return base::ASCIIToUTF16("---");
  61 }
  62
  63 bool CountryComboboxModel::IsItemSeparatorAt(int index) {
  64   return !countries_[index];
  65 }
  66
  67 int CountryComboboxModel::GetDefaultIndex() const {
  68   return default_index_;
  69 }
  70
  71 void CountryComboboxModel::AddObserver(ui::ComboboxModelObserver* observer) {
  72   observers_.AddObserver(observer);
  73 }
  74
  75 void CountryComboboxModel::RemoveObserver(ui::ComboboxModelObserver* observer) {
  76   observers_.RemoveObserver(observer);
  77 }
  78
  79 void CountryComboboxModel::ResetDefault() {
  80   SetDefaultIndex(0);
  81 }
  82
  83 void CountryComboboxModel::SetDefaultCountry(const std::string& country_code) {
  84   DCHECK_EQ(2U, country_code.length());
  85
  86   for (size_t i = 0; i < countries_.size(); ++i) {
  87     if (countries_[i] && countries_[i]->country_code() == country_code) {
  88       SetDefaultIndex(i);
  89       return;
  90     }
  91   }
  92
  93   NOTREACHED();
  94 }
  95
  96 std::string CountryComboboxModel::GetDefaultCountryCode() const {
  97   return countries_[default_index_]->country_code();
  98 }
  99
 100 void CountryComboboxModel::SetDefaultIndex(int index) {
 101   DCHECK_GE(index, 0);
 102   DCHECK_LT(index, static_cast<int>(countries_.size()));
 103   DCHECK(!IsItemSeparatorAt(index));
 104
 105   if (index == default_index_)
 106     return;
 107
 108   default_index_ = index;
 109   FOR_EACH_OBSERVER(ui::ComboboxModelObserver, observers_,
 110                     OnComboboxModelChanged(this));
 111 }
 112
 113 }  // namespace autofill