Add checkbox to card unmasking prompt

chrome/browser/ui/autofill/card_unmask_prompt_controller_impl.cc

 // Copyright 2014 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/card_unmask_prompt_controller_impl.h"
 
 #include "base/bind.h"
+#include "base/prefs/pref_service.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/autofill/risk_util.h"
 #include "chrome/browser/ui/autofill/card_unmask_prompt_view.h"
 #include "chrome/grit/generated_resources.h"
+#include "components/autofill/core/common/autofill_pref_names.h"
+#include "components/user_prefs/user_prefs.h"
+#include "content/public/browser/web_contents.h"
 #include "grit/theme_resources.h"
 #include "ui/base/l10n/l10n_util.h"
 
 namespace autofill {
 
 CardUnmaskPromptControllerImpl::CardUnmaskPromptControllerImpl(
     content::WebContents* web_contents)
     : web_contents_(web_contents),
       card_unmask_view_(nullptr),
       weak_pointer_factory_(this) {
@@ skipping 23 matching lines @@
 }
 
 void CardUnmaskPromptControllerImpl::OnUnmaskDialogClosed() {
   card_unmask_view_ = nullptr;
   delegate_->OnUnmaskPromptClosed();
 }
 
 void CardUnmaskPromptControllerImpl::OnUnmaskResponse(
     const base::string16& cvc,
     const base::string16& exp_month,
-    const base::string16& exp_year) {
+    const base::string16& exp_year,
+    bool should_store_pan) {
   card_unmask_view_->DisableAndWaitForVerification();
 
   DCHECK(!cvc.empty());
   pending_response_.cvc = cvc;
   pending_response_.exp_month = exp_month;
   pending_response_.exp_year = exp_year;
+  pending_response_.should_store_pan = should_store_pan;
+  // Remember the last choice the user made (on this device).
+  user_prefs::UserPrefs::Get(web_contents_->GetBrowserContext())->SetBoolean(
+      prefs::kAutofillWalletImportStorageCheckboxState, should_store_pan);
+
   if (!pending_response_.risk_data.empty())
     delegate_->OnUnmaskResponse(pending_response_);
 }
 
 content::WebContents* CardUnmaskPromptControllerImpl::GetWebContents() {
   return web_contents_;
 }
 
 base::string16 CardUnmaskPromptControllerImpl::GetWindowTitle() const {
   // TODO(estade): i18n.
@@ skipping 22 matching lines @@
 
 int CardUnmaskPromptControllerImpl::GetCvcImageRid() const {
   return card_.type() == kAmericanExpressCard ? IDR_CREDIT_CARD_CVC_HINT_AMEX
                                               : IDR_CREDIT_CARD_CVC_HINT;
 }
 
 bool CardUnmaskPromptControllerImpl::ShouldRequestExpirationDate() const {
   return card_.GetServerStatus() == CreditCard::EXPIRED;
 }
 
+bool CardUnmaskPromptControllerImpl::GetStoreLocallyStartState() const {
+  // TODO(estade): Don't even offer to save on Linux? Offer to save but
+  // default to false?
+  PrefService* prefs =
+      user_prefs::UserPrefs::Get(web_contents_->GetBrowserContext());
+  return prefs->GetBoolean(prefs::kAutofillWalletImportStorageCheckboxState);
+}
+
 bool CardUnmaskPromptControllerImpl::InputTextIsValid(
     const base::string16& input_text) const {
   base::string16 trimmed_text;
   base::TrimWhitespace(input_text, base::TRIM_ALL, &trimmed_text);
   size_t input_size = card_.type() == kAmericanExpressCard ? 4 : 3;
   return trimmed_text.size() == input_size &&
          base::ContainsOnlyChars(trimmed_text,
                                  base::ASCIIToUTF16("0123456789"));
 }
 
 void CardUnmaskPromptControllerImpl::LoadRiskFingerprint() {
   LoadRiskData(
       0, web_contents_,
       base::Bind(&CardUnmaskPromptControllerImpl::OnDidLoadRiskFingerprint,
                  weak_pointer_factory_.GetWeakPtr()));
 }
 
 void CardUnmaskPromptControllerImpl::OnDidLoadRiskFingerprint(
     const std::string& risk_data) {
   pending_response_.risk_data = risk_data;
   if (!pending_response_.cvc.empty())
     delegate_->OnUnmaskResponse(pending_response_);
 }
 
 }  // namespace autofill