Multi-account AccountChooser for interactive autocomplete.

chrome/browser/ui/autofill/autofill_dialog_models.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_models.h"
 
 #include "base/bind.h"
 #include "base/prefs/pref_service.h"
 #include "base/stringprintf.h"
 #include "base/strings/string_number_conversions.h"
@@ skipping 80 matching lines @@
   return false;
 }
 
 void SuggestionsMenuModel::ExecuteCommand(int command_id, int event_flags) {
   checked_item_ = command_id;
   delegate_->SuggestionItemSelected(*this);
 }
 
 // AccountChooserModel ---------------------------------------------------------
 
-const int AccountChooserModel::kWalletItemId = 0;
+const int AccountChooserModel::kContentAreaWalletItemId = 0;

[Evan Stade, 2013/03/29 23:08:45]

I do not understand this variable name or its purpose.

[aruslan, 2013/03/30 00:07:47]

Done.

 const int AccountChooserModel::kAutofillItemId = 1;
+const int AccountChooserModel::kFirstAccountItemId = 2;
 
 AccountChooserModelDelegate::~AccountChooserModelDelegate() {}
 
 AccountChooserModel::AccountChooserModel(
     AccountChooserModelDelegate* delegate,
     PrefService* prefs)
     : ALLOW_THIS_IN_INITIALIZER_LIST(ui::SimpleMenuModel(this)),
       account_delegate_(delegate),
       prefs_(prefs),
-      checked_item_(kWalletItemId),
+      checked_item_(kContentAreaWalletItemId),
       had_wallet_error_(false) {
   pref_change_registrar_.Init(prefs);
   pref_change_registrar_.Add(
       prefs::kAutofillDialogPayWithoutWallet,
       base::Bind(&AccountChooserModel::PrefChanged, base::Unretained(this)));
-
-  // TODO(estade): proper strings and l10n.
-  AddCheckItem(kWalletItemId, ASCIIToUTF16("Google Wallet"));
-  SetIcon(
-      kWalletItemId,
-      ui::ResourceBundle::GetSharedInstance().GetImageNamed(IDR_WALLET_ICON));
-  AddCheckItemWithStringId(kAutofillItemId,
-                           IDS_AUTOFILL_DIALOG_PAY_WITHOUT_WALLET);
   UpdateCheckmarkFromPref();
+  ReconstructMenuItems();
 }
 
 AccountChooserModel::~AccountChooserModel() {
 }
 
+void AccountChooserModel::ForceWalletAccountSelected() {
+  if (checked_item_ == kContentAreaWalletItemId) return;

[Evan Stade, 2013/03/29 23:08:45]

please put return on a new line.

[aruslan, 2013/03/30 00:07:47]

Done.

+  checked_item_ = kContentAreaWalletItemId;
+  account_delegate_->AccountChoiceChanged();
+}
+
+bool AccountChooserModel::HasAccountsToChoose() const {
+  return !had_wallet_error_ &&
+         !(current_username_.empty() && available_accounts_.empty());
+}
+
+void AccountChooserModel::SetAvailableAccounts(
+    const std::vector<std::string>& accounts) {
+  available_accounts_ = accounts;
+  ReconstructMenuItems();
+}
+
+void AccountChooserModel::SetCurrentlySignedInAccount(
+    const std::string& account) {
+  current_username_ = account;
+  ReconstructMenuItems();
+}
+
+std::string AccountChooserModel::GetCurrentlySignedInAccount() const {
+  return current_username_;
+}
+
+void AccountChooserModel::ResetCurrentlySignedInAccount() {
+  current_username_.clear();
+  ReconstructMenuItems();
+}
+
 bool AccountChooserModel::IsCommandIdChecked(int command_id) const {
   return command_id == checked_item_;
 }
 
 bool AccountChooserModel::IsCommandIdEnabled(int command_id) const {
-  if (command_id == kWalletItemId && had_wallet_error_)
+  // Currently, _any_ (non-sign-in) error disables _all_ Wallet accounts.
+  if (command_id != kAutofillItemId && had_wallet_error_)
     return false;
 
   return true;
 }
 
 bool AccountChooserModel::GetAcceleratorForCommandId(
     int command_id,
     ui::Accelerator* accelerator) {
   return false;
 }
 
 void AccountChooserModel::ExecuteCommand(int command_id, int event_flags) {
   if (checked_item_ == command_id)
     return;
 
   checked_item_ = command_id;
   account_delegate_->AccountChoiceChanged();
 }
 
 void AccountChooserModel::SetHadWalletError() {
+  // Any non-sign-in error disables all Wallet accounts.
   had_wallet_error_ = true;
   checked_item_ = kAutofillItemId;
+  ResetCurrentlySignedInAccount();
   account_delegate_->AccountChoiceChanged();
 }
 
 void AccountChooserModel::SetHadWalletSigninError() {
   checked_item_ = kAutofillItemId;
+  ResetCurrentlySignedInAccount();
   account_delegate_->AccountChoiceChanged();
 }
 
 bool AccountChooserModel::WalletIsSelected() const {
-  return checked_item_ == kWalletItemId;
+  return checked_item_ != kAutofillItemId;
+}
+
+bool AccountChooserModel::IsCurrentlySignedInAccountSelected() const {
+  return checked_item_ == kContentAreaWalletItemId;
 }
 
 void AccountChooserModel::PrefChanged(const std::string& pref) {
   DCHECK(pref == prefs::kAutofillDialogPayWithoutWallet);
   UpdateCheckmarkFromPref();
   account_delegate_->AccountChoiceChanged();
 }
 
 void AccountChooserModel::UpdateCheckmarkFromPref() {
   if (prefs_->GetBoolean(prefs::kAutofillDialogPayWithoutWallet))
     checked_item_ = kAutofillItemId;
   else
-    checked_item_ = kWalletItemId;
+    checked_item_ = kContentAreaWalletItemId;
+}
+
+void AccountChooserModel::ReconstructMenuItems() {
+  // If menu has to be reconstructed, either the currently sign-in account
+  // should be selected, or the autofill data should be used.
+  DCHECK_GE(kAutofillItemId, checked_item_);
+  Clear();
+  const gfx::Image& wallet_icon =
+      ui::ResourceBundle::GetSharedInstance().GetImageNamed(IDR_WALLET_ICON);
+
+  if (!current_username_.empty()) {
+    AddCheckItem(kContentAreaWalletItemId, ASCIIToUTF16(current_username_));
+    SetIcon(GetIndexOfCommandId(kContentAreaWalletItemId), wallet_icon);
+  }
+
+  for (size_t i = 0; i < available_accounts_.size(); ++i) {
+    if (available_accounts_[i] != current_username_) {
+      AddCheckItem(kFirstAccountItemId + i,
+                   ASCIIToUTF16(available_accounts_[i]));
+      SetIcon(GetIndexOfCommandId(kFirstAccountItemId + i), wallet_icon);
+    }
+  }
+
+  AddCheckItemWithStringId(kAutofillItemId,
+                           IDS_AUTOFILL_DIALOG_PAY_WITHOUT_WALLET);
 }
 
 // MonthComboboxModel ----------------------------------------------------------
 
 MonthComboboxModel::MonthComboboxModel() {}
 
 MonthComboboxModel::~MonthComboboxModel() {}
 
 int MonthComboboxModel::GetItemCount() const {
   // 12 months plus the empty entry.
@@ skipping 26 matching lines @@
 }
 
 string16 YearComboboxModel::GetItemAt(int index) {
   if (index == 0)
     return string16();
 
   return base::IntToString16(this_year_ + index - 1);
 }
 
 }  // autofill