allow wallet items to appear in requestAutocomplete UI

chrome/browser/ui/autofill/autofill_dialog_controller_impl.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_impl.h"
 
 #include <string>
 
 #include "base/logging.h"
+#include "base/string_number_conversions.h"
 #include "base/string_split.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/autofill/autofill_country.h"
 #include "chrome/browser/autofill/autofill_manager.h"
 #include "chrome/browser/autofill/autofill_type.h"
 #include "chrome/browser/autofill/personal_data_manager.h"
 #include "chrome/browser/autofill/personal_data_manager_factory.h"
 #include "chrome/browser/autofill/validation.h"
 #include "chrome/browser/autofill/wallet/full_wallet.h"
 #include "chrome/browser/autofill/wallet/wallet_items.h"
 #include "chrome/browser/autofill/wallet/wallet_service_url.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/autofill/autofill_dialog_view.h"
+#include "chrome/browser/ui/autofill/data_model_wrapper.h"
 #include "chrome/common/form_data.h"
 #include "content/public/browser/navigation_controller.h"
 #include "content/public/browser/navigation_details.h"
 #include "content/public/browser/navigation_entry.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/notification_types.h"
 #include "content/public/browser/web_contents.h"
 #include "content/public/common/url_constants.h"
 #include "grit/chromium_strings.h"
 #include "grit/generated_resources.h"
@@ skipping 337 matching lines @@
     return string16();
 
   SuggestionsMenuModel* model = SuggestionsMenuModelForSection(section);
   std::string item_key = model->GetItemKeyAt(model->checked_item());
   if (item_key.empty())
     return string16();
 
   if (section == SECTION_EMAIL)
     return model->GetLabelAt(model->checked_item());
 
-  if (section == SECTION_CC) {
+  scoped_ptr<DataModelWrapper> wrapper = CreateWrapper(section);
+  return wrapper->GetDisplayText();
+}
+
+scoped_ptr<DataModelWrapper> AutofillDialogControllerImpl::CreateWrapper(
+    DialogSection section) {
+  SuggestionsMenuModel* model = SuggestionsMenuModelForSection(section);
+  std::string item_key = model->GetItemKeyAt(model->checked_item());
+  scoped_ptr<DataModelWrapper> wrapper;
+  if (item_key.empty())
+    return wrapper.Pass();
+
+  if (CanPayWithWallet()) {
+    int index;
+    bool success = !base::StringToInt(item_key, &index);

[Dan Beam, 2013/02/15 02:11:45]

shouldn't this be without the "!" ?

[Evan Stade, 2013/02/20 00:03:39]

yes

+    DCHECK(success);
+
+    if (section == SECTION_CC) {
+      wrapper.reset(
+          new WalletInstrumentWrapper(wallet_items_->instruments()[index]));
+    } else {
+      wrapper.reset(
+          new WalletAddressWrapper(wallet_items_->addresses()[index]));
+    }
+  } else if (section == SECTION_CC) {
     CreditCard* card = GetManager()->GetCreditCardByGUID(item_key);
-    return card->TypeAndLastFourDigits();
+    DCHECK(card);
+    wrapper.reset(new AutofillCreditCardWrapper(card));
+  } else {
+    // Calculate the variant by looking at how many items come from the same
+    // FormGroup. TODO(estade): add a test for this.
+    size_t variant = 0;
+    for (int i = model->checked_item() - 1; i >= 0; --i) {
+      if (model->GetItemKeyAt(i) == item_key)
+        variant++;
+      else
+        break;
+    }
+
+    AutofillProfile* profile = GetManager()->GetProfileByGUID(item_key);
+    DCHECK(profile);
+    wrapper.reset(new AutofillDataModelWrapper(profile, variant));
   }
 
-  const std::string app_locale = AutofillCountry::ApplicationLocale();
-  AutofillProfile* profile = GetManager()->GetProfileByGUID(item_key);
-  string16 comma = ASCIIToUTF16(", ");
-  string16 label = profile->GetInfo(NAME_FULL, app_locale) +
-      comma + profile->GetInfo(ADDRESS_HOME_LINE1, app_locale);
-  string16 address2 = profile->GetInfo(ADDRESS_HOME_LINE2, app_locale);
-  if (!address2.empty())
-    label += comma + address2;
-  label += ASCIIToUTF16("\n") +
-      profile->GetInfo(ADDRESS_HOME_CITY, app_locale) + comma +
-      profile->GetInfo(ADDRESS_HOME_STATE, app_locale) + ASCIIToUTF16(" ") +
-      profile->GetInfo(ADDRESS_HOME_ZIP, app_locale);
-  return label;
+  return wrapper.Pass();
 }
 
 gfx::Image AutofillDialogControllerImpl::SuggestionIconForSection(
     DialogSection section) {
   if (section != SECTION_CC)
     return gfx::Image();
 
-  std::string item_key =
-      suggested_cc_.GetItemKeyAt(suggested_cc_.checked_item());
-  if (item_key.empty())
+  scoped_ptr<DataModelWrapper> model = CreateWrapper(section);
+  if (!model.get())
     return gfx::Image();
-  ui::ResourceBundle& rb = ui::ResourceBundle::GetSharedInstance();
-  CreditCard* card = GetManager()->GetCreditCardByGUID(item_key);
-  return rb.GetImageNamed(card->IconResourceId());
+
+  return model->GetIcon();
 }
 
 void AutofillDialogControllerImpl::EditClickedForSection(
     DialogSection section) {
   SuggestionsMenuModel* model = SuggestionsMenuModelForSection(section);
   DetailInputs* inputs = MutableRequestedFieldsForSection(section);
 
   if (section == SECTION_EMAIL) {
     // TODO(estade): shouldn't need to make this check.
     if (inputs->empty())
       return;
 
     (*inputs)[0].autofilled_value = model->GetLabelAt(model->checked_item());
   } else {
-    std::string guid = model->GetItemKeyAt(model->checked_item());
-    DCHECK(!guid.empty());
-
-    FormGroup* form_group = section == SECTION_CC ?
-        static_cast<FormGroup*>(GetManager()->GetCreditCardByGUID(guid)) :
-        static_cast<FormGroup*>(GetManager()->GetProfileByGUID(guid));
-    DCHECK(form_group);
-    FillInputFromFormGroup(form_group, inputs);
+    scoped_ptr<DataModelWrapper> model = CreateWrapper(section);
+    model->FillInputs(inputs);
   }
 
   section_editing_state_[section] = true;
   view_->UpdateSection(section);
 }
 
 bool AutofillDialogControllerImpl::InputIsValid(AutofillFieldType type,
                                                 const string16& value) {
   switch (type) {
     case EMAIL_ADDRESS:
@@ skipping 268 matching lines @@
   WalletRequestCompleted(true);
 }
 
 void AutofillDialogControllerImpl::OnDidGetWalletItems(
     scoped_ptr<wallet::WalletItems> wallet_items) {
   bool items_changed = !wallet_items_ || *wallet_items != *wallet_items_;
   wallet_items_ = wallet_items.Pass();
   WalletRequestCompleted(true);
 
   if (items_changed) {
+    GenerateSuggestionsModels();
+    view_->ModelChanged();
     view_->UpdateAccountChooser();
     view_->UpdateNotificationArea();
   }
 }
 
 void AutofillDialogControllerImpl::OnDidSaveAddress(
     const std::string& address_id) {
   NOTIMPLEMENTED() << " address_id=" << address_id;
   WalletRequestCompleted(true);
 }
@@ skipping 90 matching lines @@
   }
 
   wallet_client_.GetWalletItems(this);
   refresh_wallet_items_queued_ = false;
 }
 
 void AutofillDialogControllerImpl::WalletRequestCompleted(bool success) {
   if (!success) {
     had_wallet_error_ = true;
     wallet_items_.reset();
+    GenerateSuggestionsModels();
+    view_->ModelChanged();
     view_->UpdateAccountChooser();
     view_->UpdateNotificationArea();
     return;
   }
 
   if (refresh_wallet_items_queued_)
     ScheduleRefreshWalletItems();
 }
 
 void AutofillDialogControllerImpl::GenerateSuggestionsModels() {
   suggested_cc_.Reset();
   suggested_billing_.Reset();
   suggested_email_.Reset();
   suggested_shipping_.Reset();
 
-  PersonalDataManager* manager = GetManager();
-  const std::vector<CreditCard*>& cards = manager->credit_cards();
-  for (size_t i = 0; i < cards.size(); ++i) {
-    suggested_cc_.AddKeyedItem(cards[i]->guid(), cards[i]->Label());
-  }
-
-  const std::vector<AutofillProfile*>& profiles = manager->GetProfiles();
-  const std::string app_locale = AutofillCountry::ApplicationLocale();
-  for (size_t i = 0; i < profiles.size(); ++i) {
-    if (!IsCompleteProfile(*profiles[i]))
-      continue;
-
-    // Add all email addresses.
-    std::vector<string16> values;
-    profiles[i]->GetMultiInfo(EMAIL_ADDRESS, app_locale, &values);
-    for (size_t j = 0; j < values.size(); ++j) {
-      if (!values[j].empty())
-        suggested_email_.AddKeyedItem(profiles[i]->guid(), values[j]);
+  if (CanPayWithWallet()) {
+    if (wallet_items_.get()) {
+      // TODO(estade): seems we need to hardcode the email address.
+      // TODO(estade): CC and billing need to be combined into one section,
+      // and suggestions added here.
+      const std::vector<wallet::Address*>& addresses =
+          wallet_items_->addresses();
+      for (size_t i = 0; i < addresses.size(); ++i) {
+        suggested_billing_.AddKeyedItem(base::IntToString(i),
+                                        addresses[i]->DisplayName());
+        suggested_shipping_.AddKeyedItem(base::IntToString(i),
+                                         addresses[i]->DisplayName());
+      }
+    }
+  } else {
+    PersonalDataManager* manager = GetManager();
+    const std::vector<CreditCard*>& cards = manager->credit_cards();
+    for (size_t i = 0; i < cards.size(); ++i) {
+      suggested_cc_.AddKeyedItem(cards[i]->guid(), cards[i]->Label());
     }
 
-    // Don't add variants for addresses: the email variants are handled above,
-    // name is part of credit card and we'll just ignore phone number variants.
-    suggested_billing_.AddKeyedItem(profiles[i]->guid(), profiles[i]->Label());
-    suggested_shipping_.AddKeyedItem(profiles[i]->guid(), profiles[i]->Label());
+    const std::vector<AutofillProfile*>& profiles = manager->GetProfiles();
+    const std::string app_locale = AutofillCountry::ApplicationLocale();
+    for (size_t i = 0; i < profiles.size(); ++i) {
+      if (!IsCompleteProfile(*profiles[i]))
+        continue;
+
+      // Add all email addresses.
+      std::vector<string16> values;
+      profiles[i]->GetMultiInfo(EMAIL_ADDRESS, app_locale, &values);
+      for (size_t j = 0; j < values.size(); ++j) {
+        if (!values[j].empty())
+          suggested_email_.AddKeyedItem(profiles[i]->guid(), values[j]);
+      }
+
+      // Don't add variants for addresses: the email variants are handled above,
+      // name is part of credit card and we'll just ignore phone number
+      // variants.
+      suggested_billing_.AddKeyedItem(profiles[i]->guid(),
+                                      profiles[i]->Label());
+      suggested_shipping_.AddKeyedItem(profiles[i]->guid(),
+                                       profiles[i]->Label());
+    }
   }
 
   suggested_email_.AddKeyedItem(
       std::string(),
       l10n_util::GetStringUTF16(IDS_AUTOFILL_DIALOG_ADD_EMAIL_ADDRESS));
   suggested_cc_.AddKeyedItem(
       std::string(),
       l10n_util::GetStringUTF16(IDS_AUTOFILL_DIALOG_ADD_CREDIT_CARD));
   suggested_billing_.AddKeyedItem(
       std::string(),
@@ skipping 13 matching lines @@
     }
   }
 
   return true;
 }
 
 void AutofillDialogControllerImpl::FillOutputForSectionWithComparator(
     DialogSection section,
     const InputFieldComparator& compare) {
   SuggestionsMenuModel* model = SuggestionsMenuModelForSection(section);
-  std::string guid = model->GetItemKeyAt(model->checked_item());
-  PersonalDataManager* manager = GetManager();
-  if (!guid.empty() && !section_editing_state_[section]) {
-    FormGroup* form_group = section == SECTION_CC ?
-        static_cast<FormGroup*>(manager->GetCreditCardByGUID(guid)) :
-        static_cast<FormGroup*>(manager->GetProfileByGUID(guid));
-    DCHECK(form_group);
-
-    // Calculate the variant by looking at how many items come from the same
-    // FormGroup. TODO(estade): add a test for this.
-    size_t variant = 0;
-    for (int i = model->checked_item() - 1; i >= 0; --i) {
-      if (model->GetItemKeyAt(i) == guid)
-        variant++;
-      else
-        break;
-    }
-
-    FillFormStructureForSection(*form_group, variant, section, compare);
+  std::string item_key = model->GetItemKeyAt(model->checked_item());
+  if (!item_key.empty() && !section_editing_state_[section]) {
+    scoped_ptr<DataModelWrapper> model = CreateWrapper(section);
+    // Only fill in data that is associated with this section.
+    const DetailInputs& inputs = RequestedFieldsForSection(section);
+    model->FillFormStructure(inputs, compare, &form_structure_);
 
     // CVC needs special-casing because the CreditCard class doesn't store
     // or handle them.
     if (section == SECTION_CC)
       SetCvcResult(view_->GetCvc());
   } else {
     // The user manually input data.
+    PersonalDataManager* manager = GetManager();
     DetailOutputMap output;
     view_->GetUserInput(section, &output);
 
     // Save the info as new or edited data, then fill it into |form_structure_|.
     if (section == SECTION_CC) {
       CreditCard card;
       FillFormGroupFromOutputs(output, &card);
       if (view_->SaveDetailsLocally())
         manager->SaveImportedCreditCard(card);
       FillFormStructureForSection(card, 0, section, compare);
@@ skipping 92 matching lines @@
     DialogSection section) {
   return const_cast<DetailInputs*>(&RequestedFieldsForSection(section));
 }
 
 void AutofillDialogControllerImpl::HidePopup() {
   if (popup_controller_)
     popup_controller_->Hide();
 }
 
 }  // namespace autofill