support CC expiration dates in imperative autocomplete dialog.

chrome/browser/ui/autofill/autofill_dialog_controller.cc

 // 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/ui/autofill/autofill_dialog_controller.h"
 
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/autofill/autofill_manager.h"
 #include "chrome/browser/autofill/personal_data_manager.h"
 #include "chrome/browser/autofill/personal_data_manager_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/autofill/autofill_dialog_view.h"
 #include "chrome/common/form_data.h"
 #include "content/public/browser/web_contents.h"
 
 namespace autofill {
 
 namespace {
 
+// Returns true if |input| should be shown when |field| has been requeted.
+bool InputMatchesField(const DetailInput& input,
+                       const AutofillField& field) {
+  // If any credit card expiration info is asked for, show both month and year
+  // inputs.
+  if (field.type() == CREDIT_CARD_EXP_4_DIGIT_YEAR ||
+      field.type() == CREDIT_CARD_EXP_2_DIGIT_YEAR ||
+      field.type() == CREDIT_CARD_EXP_DATE_4_DIGIT_YEAR ||
+      field.type() == CREDIT_CARD_EXP_DATE_2_DIGIT_YEAR ||
+      field.type() == CREDIT_CARD_EXP_MONTH) {
+    return input.type == CREDIT_CARD_EXP_4_DIGIT_YEAR ||
+           input.type == CREDIT_CARD_EXP_MONTH;
+  }
+
+  return input.type == field.type();
+}
+
 // Returns true if |input| should be used for a site-requested |field|. If
 // non-empty, |section_suffix| overrides the section specified by |input|.
 bool DetailInputMatchesFieldWithSection(const std::string& section_suffix,
                                         const DetailInput& input,
                                         const AutofillField& field) {
   bool right_section = section_suffix.empty() ||
       EndsWith(field.section(), section_suffix, false);
-  return input.type == field.type() && right_section;
+  return InputMatchesField(input, field) && right_section;
 }
 
 // Returns true if |input| should be used for a site-requested |field|.
 bool DetailInputMatchesField(const DetailInput& input,
                              const AutofillField& field) {
   std::string section_suffix = input.section_suffix ? input.section_suffix : "";
   return DetailInputMatchesFieldWithSection(section_suffix, input, field);
 }
 
 // Looks through |input_template| for the types in |requested_data|. Appends
@@ skipping 86 matching lines @@
   }
 
   int row_id = 0;
 
   const DetailInput kEmailInputs[] = {
     { ++row_id, EMAIL_ADDRESS, "Email address" },
   };
 
   const DetailInput kCCInputs[] = {
     { ++row_id, CREDIT_CARD_NUMBER, "Card number" },
-    { ++row_id, CREDIT_CARD_EXP_2_DIGIT_YEAR, "Expiration MM/YY" },
+    { ++row_id, CREDIT_CARD_EXP_MONTH },
+    {   row_id, CREDIT_CARD_EXP_4_DIGIT_YEAR },
     {   row_id, CREDIT_CARD_VERIFICATION_CODE, "CVC" },
     { ++row_id, CREDIT_CARD_NAME, "Cardholder name" },
   };
 
   const DetailInput kBillingInputs[] = {
     { ++row_id, ADDRESS_HOME_LINE1, "Street address", "billing" },
     { ++row_id, ADDRESS_HOME_LINE2, "Street address (optional)", "billing" },
     { ++row_id, ADDRESS_HOME_CITY, "City", "billing" },
     { ++row_id, ADDRESS_HOME_STATE, "State", "billing" },
     {   row_id, ADDRESS_HOME_ZIP, "ZIP code", "billing", 0.5 },
@@ skipping 108 matching lines @@
     case SECTION_BILLING:
       return requested_billing_fields_;
     case SECTION_SHIPPING:
       return requested_shipping_fields_;
   }
 
   NOTREACHED();
   return requested_shipping_fields_;
 }
 
+ui::ComboboxModel* AutofillDialogController::ComboboxModelForAutofillType(
+    AutofillFieldType type) {
+  switch (type) {
+    case CREDIT_CARD_EXP_MONTH:
+      if (!cc_exp_month_combobox_model_.get())
+        cc_exp_month_combobox_model_.reset(new MonthComboboxModel());
+      return cc_exp_month_combobox_model_.get();
+
+    case CREDIT_CARD_EXP_4_DIGIT_YEAR:
+      if (!cc_exp_year_combobox_model_.get())
+        cc_exp_year_combobox_model_.reset(new YearComboboxModel());
+      return cc_exp_year_combobox_model_.get();
+
+    default:
+      return NULL;
+  }
+}
+
 ui::ComboboxModel* AutofillDialogController::ComboboxModelForSection(
     DialogSection section) {
   return SuggestionsModelForSection(section);
 }
 
 void AutofillDialogController::ViewClosed(DialogAction action) {
   if (action == ACTION_SUBMIT) {
     FillOutputForSection(SECTION_EMAIL);
     FillOutputForSection(SECTION_CC);
     FillOutputForSection(SECTION_BILLING);
     if (view_->UseBillingForShipping()) {
       FillOutputForSectionWithComparator(
           SECTION_BILLING,
           base::Bind(DetailInputMatchesFieldWithSection, "shipping"));
     } else {
       FillOutputForSection(SECTION_SHIPPING);
     }
     // TODO(estade): pass the result along to the page.
   }
 
   callback_.Run(&form_structure_);
   delete this;
 }
 
 void AutofillDialogController::GenerateComboboxModels() {
   PersonalDataManager* manager =
       PersonalDataManagerFactory::GetForProfile(profile_);
   const std::vector<CreditCard*>& cards = manager->credit_cards();
   for (size_t i = 0; i < cards.size(); ++i) {
-    suggested_cc_.AddItem(cards[i]->Label(), cards[i]->guid());
+    suggested_cc_.AddItem(cards[i]->guid(), cards[i]->Label());
   }
-  suggested_cc_.AddItem(ASCIIToUTF16("Enter new card"), "");
+  suggested_cc_.AddItem("", ASCIIToUTF16("Enter new card"));
 
   const std::vector<AutofillProfile*>& profiles = manager->profiles();
   for (size_t i = 0; i < profiles.size(); ++i) {
     string16 email = profiles[i]->GetCanonicalizedInfo(EMAIL_ADDRESS);
     if (!email.empty())
-      suggested_email_.AddItem(email, profiles[i]->guid());
-    suggested_billing_.AddItem(profiles[i]->Label(), profiles[i]->guid());
-    suggested_shipping_.AddItem(profiles[i]->Label(), profiles[i]->guid());
+      suggested_email_.AddItem(profiles[i]->guid(), email);
+    suggested_billing_.AddItem(profiles[i]->guid(), profiles[i]->Label());
+    suggested_shipping_.AddItem(profiles[i]->guid(), profiles[i]->Label());
   }
-  suggested_billing_.AddItem(ASCIIToUTF16("Enter new billing"), "");
-  suggested_email_.AddItem(ASCIIToUTF16("Enter new email"), "");
-  suggested_shipping_.AddItem(ASCIIToUTF16("Enter new shipping"), "");
+  suggested_billing_.AddItem("", ASCIIToUTF16("Enter new billing"));
+  suggested_email_.AddItem("", ASCIIToUTF16("Enter new email"));
+  suggested_shipping_.AddItem("", ASCIIToUTF16("Enter new shipping"));
 }
 
 void AutofillDialogController::PopulateInputsWithGuesses() {
   PersonalDataManager* manager =
       PersonalDataManagerFactory::GetForProfile(profile_);
 
   DetailInputs* profile_inputs[] = { &requested_email_fields_,
                                      &requested_billing_fields_ };
   const std::vector<AutofillProfile*>& profiles = manager->profiles();
   FormGroup* best_profile =
@@ skipping 56 matching lines @@
       }
     }
   }
 }
 
 void AutofillDialogController::FillOutputForSection(DialogSection section) {
   FillOutputForSectionWithComparator(section,
                                      base::Bind(DetailInputMatchesField));
 }
 
-AutofillDialogController::SuggestionsComboboxModel* AutofillDialogController::
-    SuggestionsModelForSection(DialogSection section) {
+SuggestionsComboboxModel* AutofillDialogController::SuggestionsModelForSection(
+    DialogSection section) {
   switch (section) {
     case SECTION_EMAIL:
       return &suggested_email_;
     case SECTION_CC:
       return &suggested_cc_;
     case SECTION_BILLING:
       return &suggested_billing_;
     case SECTION_SHIPPING:
       return &suggested_shipping_;
   }
 
   NOTREACHED();
   return NULL;
 }
 
-// SuggestionsComboboxModel ----------------------------------------------------
-
-AutofillDialogController::SuggestionsComboboxModel::SuggestionsComboboxModel() {
-}
-
-AutofillDialogController::SuggestionsComboboxModel::
-    ~SuggestionsComboboxModel() {}
-
-void AutofillDialogController::SuggestionsComboboxModel::AddItem(
-    const string16& item, const std::string& key) {
-  items_.push_back(std::make_pair(key, item));
-}
-
-std::string AutofillDialogController::SuggestionsComboboxModel::GetItemKeyAt(
-    int index) {
-  return items_[index].first;
-}
-
-int AutofillDialogController::SuggestionsComboboxModel::GetItemCount() const {
-  return items_.size();
-}
-
-string16 AutofillDialogController::SuggestionsComboboxModel::GetItemAt(
-    int index) {
-  return items_[index].second;
-}
-
 }  // namespace autofill
-