[Payment Request] Refactors the edit view controller

ios/chrome/browser/ui/payments/credit_card_edit_mediator.mm

 // Copyright 2017 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.
 
 #import "ios/chrome/browser/ui/payments/credit_card_edit_mediator.h"
 
 #include "base/strings/sys_string_conversions.h"
 #include "components/autofill/core/browser/autofill_data_util.h"
 #include "components/autofill/core/browser/autofill_profile.h"
 #include "components/autofill/core/browser/credit_card.h"
@@ skipping 47 matching lines @@
 @synthesize creditCard = _creditCard;
 @synthesize editorFields = _editorFields;
 
 - (instancetype)initWithPaymentRequest:(PaymentRequest*)paymentRequest
                             creditCard:(autofill::CreditCard*)creditCard {
   self = [super init];
   if (self) {
     _paymentRequest = paymentRequest;
     _creditCard = creditCard;
     _editorFields = [self createEditorFields];
+    _state = _creditCard ? EditViewControllerStateEdit
+                         : EditViewControllerStateCreate;
   }
   return self;
 }
 
-- (CollectionViewItem*)serverCardSummaryItem {
-  // Return nil if creating or editing a local card.
-  if (!_creditCard || autofill::IsCreditCardLocal(*_creditCard))
-    return nil;
+- (CollectionViewItem*)headerItem {
+  if (_creditCard && !autofill::IsCreditCardLocal(*_creditCard)) {
+    // Return an item that identifies the server card being edited.
+    PaymentMethodItem* cardSummaryItem = [[PaymentMethodItem alloc] init];
+    cardSummaryItem.methodID =
+        base::SysUTF16ToNSString(_creditCard->NetworkAndLastFourDigits());
+    cardSummaryItem.methodDetail = base::SysUTF16ToNSString(
+        _creditCard->GetRawInfo(autofill::CREDIT_CARD_NAME_FULL));
+    const int issuerNetworkIconID =
+        autofill::data_util::GetPaymentRequestData(_creditCard->network())
+            .icon_resource_id;
+    cardSummaryItem.methodTypeIcon = NativeImage(issuerNetworkIconID);
+    return cardSummaryItem;
+  }
 
-  PaymentMethodItem* cardSummaryItem = [[PaymentMethodItem alloc] init];
-  cardSummaryItem.methodID =
-      base::SysUTF16ToNSString(_creditCard->NetworkAndLastFourDigits());
-  cardSummaryItem.methodDetail = base::SysUTF16ToNSString(
-      _creditCard->GetRawInfo(autofill::CREDIT_CARD_NAME_FULL));
-  const int issuerNetworkIconID =
-      autofill::data_util::GetPaymentRequestData(_creditCard->network())
-          .icon_resource_id;
-  cardSummaryItem.methodTypeIcon = NativeImage(issuerNetworkIconID);
-  return cardSummaryItem;
-}
-
-- (CollectionViewItem*)acceptedPaymentMethodsItem {
-  // Return nil if a server card is being edited.
-  if (_creditCard && !autofill::IsCreditCardLocal(*_creditCard))
-    return nil;
-
+  // Otherwise, return an item that displays a list of payment method type icons
+  // for the accepted payment methods.
   NSMutableArray* issuerNetworkIcons = [NSMutableArray array];
   for (const auto& supportedNetwork :
        _paymentRequest->supported_card_networks()) {
     const std::string issuerNetwork =
         GetIssuerNetworkForBasicCardIssuerNetwork(supportedNetwork);
     const autofill::data_util::PaymentRequestData& cardData =
         GetPaymentRequestData(issuerNetwork);
     UIImage* issuerNetworkIcon =
         ResizeImage(NativeImage(cardData.icon_resource_id),
                     CGSizeMake(kCardIssuerNetworkIconDimension,
                                kCardIssuerNetworkIconDimension),
                     ProjectionMode::kAspectFillNoClipping);
     issuerNetworkIcon.accessibilityLabel =
         l10n_util::GetNSString(cardData.a11y_label_resource_id);
     [issuerNetworkIcons addObject:issuerNetworkIcon];
   }
 
   AcceptedPaymentMethodsItem* acceptedMethodsItem =
       [[AcceptedPaymentMethodsItem alloc] init];
   acceptedMethodsItem.message =
       l10n_util::GetNSString(IDS_PAYMENTS_ACCEPTED_CARDS_LABEL);
   acceptedMethodsItem.methodTypeIcons = issuerNetworkIcons;
   return acceptedMethodsItem;
 }
 
-- (NSString*)billingAddressLabelForProfileWithGUID:(NSString*)profileGUID {
-  DCHECK(profileGUID);
-  autofill::AutofillProfile* profile =
-      autofill::PersonalDataManager::GetProfileFromProfilesByGUID(
-          base::SysNSStringToUTF8(profileGUID),
-          _paymentRequest->billing_profiles());
-  DCHECK(profile);
-  return GetBillingAddressLabelFromAutofillProfile(*profile);
+- (BOOL)shouldHideBackgroundForHeaderItem {
+  // YES if the header item displays the accepted payment method type icons.
+  return !_creditCard || autofill::IsCreditCardLocal(*_creditCard);
 }
 
 - (UIImage*)cardTypeIconFromCardNumber:(NSString*)cardNumber {
   const char* issuerNetwork = autofill::CreditCard::GetCardNetwork(
       base::SysNSStringToUTF16(cardNumber));
   // This should not happen in Payment Request.
   if (issuerNetwork == autofill::kGenericCard)
     return nil;
 
   // Resize and return the card issuer network icon.
   int resourceID = autofill::data_util::GetPaymentRequestData(issuerNetwork)
                        .icon_resource_id;
   CGFloat dimension = kCardIssuerNetworkIconDimension;
   return ResizeImage(NativeImage(resourceID), CGSizeMake(dimension, dimension),
                      ProjectionMode::kAspectFillNoClipping);
 }
 
 - (NSArray<EditorField*>*)editorFields {
   return _editorFields;
 }
 
 #pragma mark - Helper methods
 
+// Returns the billing address label from an autofill profile with the given
+// guid. Returns nil if the profile does not have an address.
+- (NSString*)billingAddressLabelForProfileWithGUID:(NSString*)profileGUID {
+  DCHECK(profileGUID);
+  autofill::AutofillProfile* profile =
+      autofill::PersonalDataManager::GetProfileFromProfilesByGUID(
+          base::SysNSStringToUTF8(profileGUID),
+          _paymentRequest->billing_profiles());
+  DCHECK(profile);
+  return GetBillingAddressLabelFromAutofillProfile(*profile);
+}
+
 - (NSArray<EditorField*>*)createEditorFields {
-  // Server credit cards are not editable.
+  NSString* billingAddressGUID =
+      _creditCard && !_creditCard->billing_address_id().empty()
+          ? base::SysUTF8ToNSString(_creditCard->billing_address_id())
+          : nil;
+  NSString* billingAddressLabel =
+      billingAddressGUID
+          ? [self billingAddressLabelForProfileWithGUID:billingAddressGUID]
+          : nil;
+  EditorField* billingAddressEditorField = [[EditorField alloc]
+      initWithAutofillUIType:AutofillUITypeCreditCardBillingAddress
+                   fieldType:EditorFieldTypeSelector
+                       label:l10n_util::GetNSString(
+                                 IDS_PAYMENTS_BILLING_ADDRESS)
+                       value:billingAddressGUID
+                    required:YES];
+  [billingAddressEditorField setDisplayValue:billingAddressLabel];
+
+  // For server cards, only the billing address can be edited.
   if (_creditCard && !autofill::IsCreditCardLocal(*_creditCard))
-    return @[];
+    return @[ billingAddressEditorField ];
 
   NSString* creditCardNumber =
       _creditCard ? base::SysUTF16ToNSString(_creditCard->number()) : nil;
 
   NSString* creditCardName =
       _creditCard
           ? autofill::GetCreditCardName(
                 *_creditCard, GetApplicationContext()->GetApplicationLocale())
           : nil;
 
   NSString* creditCardExpMonth =
       _creditCard
           ? [NSString stringWithFormat:@"%02d", _creditCard->expiration_month()]
           : nil;
 
   NSString* creditCardExpYear =
       _creditCard
           ? [NSString stringWithFormat:@"%04d", _creditCard->expiration_year()]
           : nil;
 
-  return @[
+  NSMutableArray* editorFields = [[NSMutableArray alloc] init];
+  [editorFields addObjectsFromArray:@[
     [[EditorField alloc]
         initWithAutofillUIType:AutofillUITypeFromAutofillType(
                                    autofill::CREDIT_CARD_NUMBER)
+                     fieldType:EditorFieldTypeTextField
                          label:l10n_util::GetNSString(IDS_PAYMENTS_CARD_NUMBER)
                          value:creditCardNumber
                       required:YES],
     [[EditorField alloc]
         initWithAutofillUIType:AutofillUITypeFromAutofillType(
                                    autofill::CREDIT_CARD_NAME_FULL)
+                     fieldType:EditorFieldTypeTextField
                          label:l10n_util::GetNSString(IDS_PAYMENTS_NAME_ON_CARD)
                          value:creditCardName
                       required:YES],
     [[EditorField alloc]
         initWithAutofillUIType:AutofillUITypeFromAutofillType(
                                    autofill::CREDIT_CARD_EXP_MONTH)
+                     fieldType:EditorFieldTypeTextField
                          label:l10n_util::GetNSString(IDS_PAYMENTS_EXP_MONTH)
                          value:creditCardExpMonth
                       required:YES],
     [[EditorField alloc]
         initWithAutofillUIType:AutofillUITypeFromAutofillType(
                                    autofill::CREDIT_CARD_EXP_4_DIGIT_YEAR)
+                     fieldType:EditorFieldTypeTextField
                          label:l10n_util::GetNSString(IDS_PAYMENTS_EXP_YEAR)
                          value:creditCardExpYear
                       required:YES]
-  ];
+  ]];
+  // The billing address field goes at the end.
+  [editorFields addObject:billingAddressEditorField];
+  return editorFields;
 }
 
 @end