[Payment Request] Moves PaymentRequestViewController to :payments_ui

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

 // Copyright 2016 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/payment_request_view_controller.h"
 
 #include "base/mac/foundation_util.h"
 
-#include "base/strings/sys_string_conversions.h"
-#include "base/strings/utf_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"
-#include "components/autofill/core/browser/field_types.h"
-#include "components/autofill/core/browser/personal_data_manager.h"
-#include "components/payments/core/currency_formatter.h"
-#include "components/payments/core/strings_util.h"
 #include "components/strings/grit/components_strings.h"
-#include "ios/chrome/browser/payments/payment_request.h"
-#import "ios/chrome/browser/payments/payment_request_util.h"
 #import "ios/chrome/browser/ui/autofill/cells/status_item.h"
 #import "ios/chrome/browser/ui/collection_view/cells/MDCCollectionViewCell+Chrome.h"
 #import "ios/chrome/browser/ui/collection_view/cells/collection_view_detail_item.h"
 #import "ios/chrome/browser/ui/collection_view/cells/collection_view_footer_item.h"
+#import "ios/chrome/browser/ui/collection_view/cells/collection_view_item+collection_view_controller.h"
 #import "ios/chrome/browser/ui/collection_view/cells/collection_view_item.h"
 #import "ios/chrome/browser/ui/collection_view/collection_view_model.h"
 #import "ios/chrome/browser/ui/colors/MDCPalette+CrAdditions.h"
-#import "ios/chrome/browser/ui/payments/cells/autofill_profile_item.h"
 #import "ios/chrome/browser/ui/payments/cells/page_info_item.h"
-#import "ios/chrome/browser/ui/payments/cells/payment_method_item.h"
-#import "ios/chrome/browser/ui/payments/cells/payments_text_item.h"
 #import "ios/chrome/browser/ui/payments/cells/price_item.h"
 #import "ios/chrome/browser/ui/payments/payment_request_view_controller_actions.h"
 #include "ios/chrome/browser/ui/rtl_geometry.h"
 #include "ios/chrome/browser/ui/uikit_ui_util.h"
 #include "ios/chrome/grit/ios_strings.h"
 #import "ios/third_party/material_components_ios/src/components/Buttons/src/MaterialButtons.h"
-#import "ios/third_party/material_components_ios/src/components/CollectionCells/src/MaterialCollectionCells.h"
 #import "ios/third_party/material_components_ios/src/components/Typography/src/MaterialTypography.h"
-#include "ios/web/public/payments/payment_request.h"
 #include "ui/base/l10n/l10n_util.h"
-#include "ui/base/resource/resource_bundle.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
-namespace {
-using ::payment_request_util::GetNameLabelFromAutofillProfile;
-using ::payment_request_util::GetShippingAddressLabelFromAutofillProfile;
-using ::payment_request_util::GetPhoneNumberLabelFromAutofillProfile;
-using ::payment_request_util::GetEmailLabelFromAutofillProfile;
-using ::payment_request_util::GetShippingSectionTitle;
-using ::payments::GetShippingOptionSectionString;
-using ::payments::GetShippingAddressSectionString;
-
-// String used as the "URL" to take the user to the settings page for card and
-// address options. Needs to be URL-like; otherwise, the link will not appear
-// as a link in the UI (see setLabelLinkURL: in CollectionViewFooterCell).
-const char kSettingsURL[] = "settings://card-and-address";
-
-const CGFloat kFooterCellHorizontalPadding = 16;
-
-}  // namespace
-
 NSString* const kPaymentRequestCollectionViewID =
     @"kPaymentRequestCollectionViewID";
 
 namespace {
+const CGFloat kFooterCellHorizontalPadding = 16;
 
 const CGFloat kButtonEdgeInset = 9;
 const CGFloat kSeparatorEdgeInset = 14;
 
 typedef NS_ENUM(NSInteger, SectionIdentifier) {
   SectionIdentifierSummary = kSectionIdentifierEnumZero,
   SectionIdentifierShipping,
   SectionIdentifierPayment,
   SectionIdentifierContactInfo,
   SectionIdentifierFooter,
 };
 
 typedef NS_ENUM(NSInteger, ItemType) {
   ItemTypeSummaryPageInfo = kItemTypeEnumZero,
   ItemTypeSpinner,
   ItemTypeSummaryTotal,
   ItemTypeShippingTitle,
   ItemTypeShippingAddress,
-  ItemTypeAddShippingAddress,
   ItemTypeShippingOption,
-  ItemTypeSelectShippingOption,
-  ItemTypePaymentTitle,
+  ItemTypePaymentHeader,
   ItemTypePaymentMethod,
-  ItemTypeAddPaymentMethod,
-  ItemTypeContactInfoTitle,
+  ItemTypeContactInfoHeader,
   ItemTypeContactInfo,
-  ItemTypeAddContactInfo,
   ItemTypeFooterText,
 };
 
 }  // namespace
 
 @interface PaymentRequestViewController ()<
+    CollectionViewFooterLinkDelegate,
     PaymentRequestViewControllerActions> {
   UIBarButtonItem* _cancelButton;
   MDCButton* _payButton;
-
-  // The PaymentRequest object having a copy of web::PaymentRequest as provided
-  // by the page invoking the Payment Request API. This is a weak pointer and
-  // should outlive this class.
-  PaymentRequest* _paymentRequest;
-
-  __weak PriceItem* _paymentSummaryItem;
-  __weak AutofillProfileItem* _selectedShippingAddressItem;
-  __weak PaymentsTextItem* _selectedShippingOptionItem;
-  __weak PaymentMethodItem* _selectedPaymentMethodItem;
-  __weak AutofillProfileItem* _selectedContactInfoItem;
 }
 
 @end
 
 @implementation PaymentRequestViewController
 
 @synthesize pageFavicon = _pageFavicon;
 @synthesize pageTitle = _pageTitle;
 @synthesize pageHost = _pageHost;
 @synthesize connectionSecure = _connectionSecure;
 @synthesize pending = _pending;
 @synthesize delegate = _delegate;
-@synthesize showPaymentDataSource = _showPaymentDataSource;
 @synthesize dataSource = _dataSource;
 
-- (instancetype)initWithPaymentRequest:(PaymentRequest*)paymentRequest {
-  DCHECK(paymentRequest);
+- (instancetype)init {
   if ((self = [super initWithStyle:CollectionViewControllerStyleAppBar])) {
     [self setTitle:l10n_util::GetNSString(IDS_PAYMENTS_TITLE)];
 
     // Set up leading (cancel) button.
     _cancelButton = [[UIBarButtonItem alloc]
         initWithTitle:l10n_util::GetNSString(IDS_CANCEL)
                 style:UIBarButtonItemStylePlain
                target:nil
                action:@selector(onCancel)];
     [_cancelButton setTitleTextAttributes:@{
       NSForegroundColorAttributeName : [UIColor lightGrayColor]
     }
                                  forState:UIControlStateDisabled];
     [_cancelButton
         setAccessibilityLabel:l10n_util::GetNSString(IDS_ACCNAME_CANCEL)];
     [self navigationItem].leftBarButtonItem = _cancelButton;
 
     // Set up trailing (pay) button.
     _payButton = [[MDCButton alloc] init];
     [_payButton setTitle:l10n_util::GetNSString(IDS_PAYMENTS_PAY_BUTTON)
                 forState:UIControlStateNormal];
     [_payButton setCustomTitleColor:[UIColor whiteColor]];
     [_payButton setInkColor:[UIColor colorWithWhite:1 alpha:0.2]];
     [_payButton addTarget:self
                    action:@selector(onConfirm)
          forControlEvents:UIControlEventTouchUpInside];
     [_payButton sizeToFit];
-    [_payButton setEnabled:(paymentRequest->selected_credit_card() != nil)];
     [_payButton setAutoresizingMask:UIViewAutoresizingFlexibleTrailingMargin() |
                                     UIViewAutoresizingFlexibleTopMargin |
                                     UIViewAutoresizingFlexibleBottomMargin];
 
     // The navigation bar will set the rightBarButtonItem's height to the full
     // height of the bar. We don't want that for the button so we use a UIView
     // here to contain the button instead and the button is vertically centered
     // inside the full bar height.
     UIView* buttonView = [[UIView alloc] initWithFrame:CGRectZero];
     [buttonView addSubview:_payButton];
     // Navigation bar button items are aligned with the trailing edge of the
     // screen. Make the enclosing view larger here. The pay button will be
     // aligned with the leading edge of the enclosing view leaving an inset on
     // the trailing edge.
     CGRect buttonViewBounds = buttonView.bounds;
     buttonViewBounds.size.width =
         [_payButton frame].size.width + kButtonEdgeInset;
     buttonView.bounds = buttonViewBounds;
 
     UIBarButtonItem* payButtonItem =
         [[UIBarButtonItem alloc] initWithCustomView:buttonView];
     [self navigationItem].rightBarButtonItem = payButtonItem;
-
-    _paymentRequest = paymentRequest;
-
-    // By default, data source is shown.
-    _showPaymentDataSource = TRUE;
   }
   return self;
 }
 
 - (void)onCancel {
   [_delegate paymentRequestViewControllerDidCancel:self];
 }
 
 - (void)onConfirm {
   [_delegate paymentRequestViewControllerDidConfirm:self];
 }
 
+#pragma mark - Setters
+
+- (void)setDataSource:(id<PaymentRequestViewControllerDataSource>)dataSource {
+  _dataSource = dataSource;
+  [_payButton setEnabled:[_dataSource canPay]];
+}
+
 #pragma mark - CollectionViewController methods
 
 - (void)loadModel {
   [super loadModel];
   CollectionViewModel* model = self.collectionViewModel;
 
   // Summary section.
   [model addSectionWithIdentifier:SectionIdentifierSummary];
 
   PageInfoItem* pageInfo =
       [[PageInfoItem alloc] initWithType:ItemTypeSummaryPageInfo];
   pageInfo.pageFavicon = _pageFavicon;
   pageInfo.pageTitle = _pageTitle;
   pageInfo.pageHost = _pageHost;
   pageInfo.connectionSecure = _connectionSecure;
   [model setHeader:pageInfo forSectionWithIdentifier:SectionIdentifierSummary];
 
   if (_pending) {
     [_payButton setEnabled:NO];
     [_cancelButton setEnabled:NO];
 
     StatusItem* statusItem = [[StatusItem alloc] initWithType:ItemTypeSpinner];
     statusItem.text = l10n_util::GetNSString(IDS_PAYMENTS_PROCESSING_MESSAGE);
     [model addItem:statusItem toSectionWithIdentifier:SectionIdentifierSummary];
     return;
   }
 
-  PriceItem* paymentSummaryItem =
-      [[PriceItem alloc] initWithType:ItemTypeSummaryTotal];
-  _paymentSummaryItem = paymentSummaryItem;
-  [self fillPaymentSummaryItem:paymentSummaryItem
-               withPaymentItem:_paymentRequest->payment_details().total
-         withTotalValueChanged:NO];
-  if (!_paymentRequest->payment_details().display_items.empty()) {
-    paymentSummaryItem.accessoryType =
-        MDCCollectionViewCellAccessoryDisclosureIndicator;
-    paymentSummaryItem.accessibilityTraits |= UIAccessibilityTraitButton;
-  }
-  [model addItem:paymentSummaryItem
-      toSectionWithIdentifier:SectionIdentifierSummary];
+  [self addPaymentSummaryItem];
 
   // Shipping section.
-  [model addSectionWithIdentifier:SectionIdentifierShipping];
+  if ([_dataSource requestShipping]) {
+    [model addSectionWithIdentifier:SectionIdentifierShipping];
 
-  PaymentsTextItem* shippingTitle =
-      [[PaymentsTextItem alloc] initWithType:ItemTypeShippingTitle];
-  shippingTitle.text =
-      GetShippingSectionTitle(_paymentRequest->shipping_type());
-  [model setHeader:shippingTitle
-      forSectionWithIdentifier:SectionIdentifierShipping];
+    CollectionViewItem* shippingSectionHeaderItem =
+        [_dataSource shippingSectionHeaderItem];
+    [shippingSectionHeaderItem setType:ItemTypeShippingTitle];
+    [model setHeader:shippingSectionHeaderItem
+        forSectionWithIdentifier:SectionIdentifierShipping];
 
-  CollectionViewItem* shippingAddressItem = nil;
-  if (_paymentRequest->selected_shipping_profile()) {
-    AutofillProfileItem* selectedShippingAddressItem =
-        [[AutofillProfileItem alloc] initWithType:ItemTypeShippingAddress];
-    shippingAddressItem = selectedShippingAddressItem;
-    _selectedShippingAddressItem = selectedShippingAddressItem;
-    [self fillShippingAddressItem:selectedShippingAddressItem
-              withAutofillProfile:_paymentRequest->selected_shipping_profile()];
-    selectedShippingAddressItem.accessoryType =
-        MDCCollectionViewCellAccessoryDisclosureIndicator;
-    selectedShippingAddressItem.accessibilityTraits |=
-        UIAccessibilityTraitButton;
-  } else {
-    CollectionViewDetailItem* addAddressItem = [[CollectionViewDetailItem alloc]
-        initWithType:ItemTypeAddShippingAddress];
-    shippingAddressItem = addAddressItem;
-    addAddressItem.text = SysUTF16ToNSString(
-        GetShippingAddressSectionString(_paymentRequest->shipping_type()));
-    addAddressItem.detailText = [l10n_util::GetNSString(IDS_ADD)
-        uppercaseStringWithLocale:[NSLocale currentLocale]];
-    addAddressItem.accessibilityTraits |= UIAccessibilityTraitButton;
+    [self populateShippingSection];
   }
-  [model addItem:shippingAddressItem
-      toSectionWithIdentifier:SectionIdentifierShipping];
-
-  CollectionViewItem* shippingOptionItem = nil;
-  if (_paymentRequest->selected_shipping_option()) {
-    PaymentsTextItem* selectedShippingOptionItem =
-        [[PaymentsTextItem alloc] initWithType:ItemTypeShippingOption];
-    shippingOptionItem = selectedShippingOptionItem;
-
-    _selectedShippingOptionItem = selectedShippingOptionItem;
-    [self fillShippingOptionItem:selectedShippingOptionItem
-                      withOption:_paymentRequest->selected_shipping_option()];
-    selectedShippingOptionItem.accessoryType =
-        MDCCollectionViewCellAccessoryDisclosureIndicator;
-    selectedShippingOptionItem.accessibilityTraits |=
-        UIAccessibilityTraitButton;
-  } else {
-    CollectionViewDetailItem* selectShippingOptionItem =
-        [[CollectionViewDetailItem alloc]
-            initWithType:ItemTypeSelectShippingOption];
-    shippingOptionItem = selectShippingOptionItem;
-    selectShippingOptionItem.text = base::SysUTF16ToNSString(
-        GetShippingOptionSectionString(_paymentRequest->shipping_type()));
-    selectShippingOptionItem.accessoryType =
-        MDCCollectionViewCellAccessoryDisclosureIndicator;
-    selectShippingOptionItem.accessibilityTraits |= UIAccessibilityTraitButton;
-  }
-  [model addItem:shippingOptionItem
-      toSectionWithIdentifier:SectionIdentifierShipping];
 
   // Payment method section.
   [model addSectionWithIdentifier:SectionIdentifierPayment];
-
-  CollectionViewItem* paymentMethodItem = nil;
-  if (_paymentRequest->selected_credit_card()) {
-    PaymentsTextItem* paymentTitle =
-        [[PaymentsTextItem alloc] initWithType:ItemTypePaymentTitle];
-    paymentTitle.text =
-        l10n_util::GetNSString(IDS_PAYMENT_REQUEST_PAYMENT_METHOD_SECTION_NAME);
-    [model setHeader:paymentTitle
-        forSectionWithIdentifier:SectionIdentifierPayment];
-
-    PaymentMethodItem* selectedPaymentMethodItem =
-        [[PaymentMethodItem alloc] initWithType:ItemTypePaymentMethod];
-    paymentMethodItem = selectedPaymentMethodItem;
-    _selectedPaymentMethodItem = selectedPaymentMethodItem;
-    [self fillPaymentMethodItem:selectedPaymentMethodItem
-                 withCreditCard:_paymentRequest->selected_credit_card()];
-    selectedPaymentMethodItem.accessoryType =
-        MDCCollectionViewCellAccessoryDisclosureIndicator;
-    selectedPaymentMethodItem.accessibilityTraits |= UIAccessibilityTraitButton;
-  } else {
-    CollectionViewDetailItem* addPaymentMethodItem = [
-        [CollectionViewDetailItem alloc] initWithType:ItemTypeAddPaymentMethod];
-    paymentMethodItem = addPaymentMethodItem;
-    addPaymentMethodItem.text =
-        l10n_util::GetNSString(IDS_PAYMENT_REQUEST_PAYMENT_METHOD_SECTION_NAME);
-    addPaymentMethodItem.detailText = [l10n_util::GetNSString(IDS_ADD)
-        uppercaseStringWithLocale:[NSLocale currentLocale]];
-    addPaymentMethodItem.accessibilityTraits |= UIAccessibilityTraitButton;
-  }
-  [model addItem:paymentMethodItem
-      toSectionWithIdentifier:SectionIdentifierPayment];
+  [self populatePaymentMethodSection];
 
   // Contact Info section.
-  [model addSectionWithIdentifier:SectionIdentifierContactInfo];
-
-  CollectionViewItem* contactInfoItem = nil;
-  if (_paymentRequest->selected_contact_profile()) {
-    PaymentsTextItem* contactInfoTitle =
-        [[PaymentsTextItem alloc] initWithType:ItemTypeContactInfoTitle];
-    contactInfoTitle.text =
-        l10n_util::GetNSString(IDS_PAYMENTS_CONTACT_DETAILS_LABEL);
-    [model setHeader:contactInfoTitle
-        forSectionWithIdentifier:SectionIdentifierContactInfo];
-
-    AutofillProfileItem* selectedContactInfoItem =
-        [[AutofillProfileItem alloc] initWithType:ItemTypeContactInfo];
-    contactInfoItem = selectedContactInfoItem;
-    _selectedContactInfoItem = selectedContactInfoItem;
-    [self fillContactInfoItem:selectedContactInfoItem
-          withAutofillProfile:_paymentRequest->selected_contact_profile()];
-    selectedContactInfoItem.accessoryType =
-        MDCCollectionViewCellAccessoryDisclosureIndicator;
-
-  } else {
-    CollectionViewDetailItem* addContactInfoItem =
-        [[CollectionViewDetailItem alloc] initWithType:ItemTypeAddContactInfo];
-    contactInfoItem = addContactInfoItem;
-    addContactInfoItem.text =
-        l10n_util::GetNSString(IDS_PAYMENTS_CONTACT_DETAILS_LABEL);
-    addContactInfoItem.detailText = [l10n_util::GetNSString(IDS_ADD)
-        uppercaseStringWithLocale:[NSLocale currentLocale]];
-    addContactInfoItem.accessibilityTraits |= UIAccessibilityTraitButton;
+  if ([_dataSource requestContactInfo]) {
+    [model addSectionWithIdentifier:SectionIdentifierContactInfo];
+    [self populateContactInfoSection];
   }
-  [model addItem:contactInfoItem
-      toSectionWithIdentifier:SectionIdentifierContactInfo];
 
   // Footer Text section.
   [model addSectionWithIdentifier:SectionIdentifierFooter];
-  CollectionViewFooterItem* footer =
-      [[CollectionViewFooterItem alloc] initWithType:ItemTypeFooterText];
-  if (!_showPaymentDataSource) {
-    footer.text =
-        l10n_util::GetNSString(IDS_PAYMENTS_CARD_AND_ADDRESS_SETTINGS);
-  } else if ([[_dataSource authenticatedAccountName] length]) {
-    const base::string16 accountName =
-        base::SysNSStringToUTF16([_dataSource authenticatedAccountName]);
-    const std::string formattedString = l10n_util::GetStringFUTF8(
-        IDS_PAYMENTS_CARD_AND_ADDRESS_SETTINGS_SIGNED_IN, accountName);
-    footer.text = base::SysUTF8ToNSString(formattedString);
-  } else {
-    footer.text = l10n_util::GetNSString(
-        IDS_PAYMENTS_CARD_AND_ADDRESS_SETTINGS_SIGNED_OUT);
-  }
-  footer.linkURL = GURL(kSettingsURL);
-  footer.linkDelegate = self;
-  [model addItem:footer toSectionWithIdentifier:SectionIdentifierFooter];
+
+  CollectionViewFooterItem* footerItem = [_dataSource footerItem];
+  [footerItem setType:ItemTypeFooterText];
+  footerItem.linkDelegate = self;
+  [model addItem:footerItem toSectionWithIdentifier:SectionIdentifierFooter];
 }
 
 - (void)viewDidLoad {
   [super viewDidLoad];
   self.collectionView.accessibilityIdentifier = kPaymentRequestCollectionViewID;
 
   // Customize collection view settings.
   self.styler.cellStyle = MDCCollectionViewCellStyleCard;
   self.styler.separatorInset =
       UIEdgeInsetsMake(0, kSeparatorEdgeInset, 0, kSeparatorEdgeInset);
 }
 
-- (void)updatePaymentSummaryWithTotalValueChanged:(BOOL)totalValueChanged {
-  [self fillPaymentSummaryItem:_paymentSummaryItem
-               withPaymentItem:_paymentRequest->payment_details().total
-         withTotalValueChanged:totalValueChanged];
+- (void)updatePaymentSummaryItem {
+  CollectionViewModel* model = self.collectionViewModel;
+
+  [model removeItemWithType:ItemTypeSummaryTotal
+      fromSectionWithIdentifier:SectionIdentifierSummary];
+
+  [self addPaymentSummaryItem];
+
+  // Reload the item.
   NSIndexPath* indexPath =
-      [self.collectionViewModel indexPathForItem:_paymentSummaryItem];
+      [model indexPathForItemType:ItemTypeSummaryTotal
+                sectionIdentifier:SectionIdentifierSummary];
   [self.collectionView reloadItemsAtIndexPaths:@[ indexPath ]];
 }
 
-- (void)updateSelectedShippingAddressUI {
-  [self fillShippingAddressItem:_selectedShippingAddressItem
-            withAutofillProfile:_paymentRequest->selected_shipping_profile()];
-  NSIndexPath* indexPath =
-      [self.collectionViewModel indexPathForItem:_selectedShippingAddressItem];
-  [self.collectionView reloadItemsAtIndexPaths:@[ indexPath ]];
+- (void)updateShippingSection {
+  CollectionViewModel* model = self.collectionViewModel;
+
+  [model removeItemWithType:ItemTypeShippingAddress
+      fromSectionWithIdentifier:SectionIdentifierShipping];
+
+  if ([model hasItemForItemType:ItemTypeShippingOption
+              sectionIdentifier:SectionIdentifierShipping]) {
+    [model removeItemWithType:ItemTypeShippingOption
+        fromSectionWithIdentifier:SectionIdentifierShipping];
+  }
+
+  [self populateShippingSection];
+
+  // Reload the section.
+  NSInteger sectionIndex =
+      [model sectionForSectionIdentifier:SectionIdentifierShipping];
+  [self.collectionView
+      reloadSections:[NSIndexSet indexSetWithIndex:sectionIndex]];
+
+  // Update the pay button.
+  [_payButton setEnabled:[_dataSource canPay]];
 }
 
-- (void)updateSelectedShippingOptionUI {
-  [self fillShippingOptionItem:_selectedShippingOptionItem
-                    withOption:_paymentRequest->selected_shipping_option()];
-  NSIndexPath* indexPath =
-      [self.collectionViewModel indexPathForItem:_selectedShippingOptionItem];
-  [self.collectionView reloadItemsAtIndexPaths:@[ indexPath ]];
+- (void)updatePaymentMethodSection {
+  CollectionViewModel* model = self.collectionViewModel;
+
+  [model removeItemWithType:ItemTypePaymentMethod
+      fromSectionWithIdentifier:SectionIdentifierPayment];
+
+  [self populatePaymentMethodSection];
+
+  // Reload the section.
+  NSInteger sectionIndex =
+      [model sectionForSectionIdentifier:SectionIdentifierPayment];
+  [self.collectionView
+      reloadSections:[NSIndexSet indexSetWithIndex:sectionIndex]];
+
+  // Update the pay button.
+  [_payButton setEnabled:[_dataSource canPay]];
 }
 
-- (void)updateSelectedPaymentMethodUI {
-  [self fillPaymentMethodItem:_selectedPaymentMethodItem
-               withCreditCard:_paymentRequest->selected_credit_card()];
-  NSIndexPath* indexPath =
-      [self.collectionViewModel indexPathForItem:_selectedPaymentMethodItem];
-  [self.collectionView reloadItemsAtIndexPaths:@[ indexPath ]];
-}
+- (void)updateContactInfoSection {
+  CollectionViewModel* model = self.collectionViewModel;
 
-- (void)updateSelectedContactInfoUI {
-  [self fillContactInfoItem:_selectedContactInfoItem
-        withAutofillProfile:_paymentRequest->selected_contact_profile()];
-  NSIndexPath* indexPath =
-      [self.collectionViewModel indexPathForItem:_selectedContactInfoItem];
-  [self.collectionView reloadItemsAtIndexPaths:@[ indexPath ]];
-}
+  [model removeItemWithType:ItemTypeContactInfo
+      fromSectionWithIdentifier:SectionIdentifierContactInfo];
 
-#pragma mark - Helper methods
+  [self populateContactInfoSection];
 
-- (void)fillPaymentSummaryItem:(PriceItem*)item
-               withPaymentItem:(web::PaymentItem)paymentItem
-         withTotalValueChanged:(BOOL)totalValueChanged {
-  item.item =
-      base::SysUTF16ToNSString(_paymentRequest->payment_details().total.label);
-  payments::CurrencyFormatter* currencyFormatter =
-      _paymentRequest->GetOrCreateCurrencyFormatter();
-  item.price = SysUTF16ToNSString(l10n_util::GetStringFUTF16(
-      IDS_PAYMENT_REQUEST_ORDER_SUMMARY_SHEET_TOTAL_FORMAT,
-      base::UTF8ToUTF16(currencyFormatter->formatted_currency_code()),
-      currencyFormatter->Format(base::UTF16ToASCII(paymentItem.amount.value))));
-  item.notification = totalValueChanged
-                          ? l10n_util::GetNSString(IDS_PAYMENTS_UPDATED_LABEL)
-                          : nil;
-}
+  // Reload the section.
+  NSInteger sectionIndex =
+      [model sectionForSectionIdentifier:SectionIdentifierContactInfo];
+  [self.collectionView
+      reloadSections:[NSIndexSet indexSetWithIndex:sectionIndex]];
 
-- (void)fillShippingAddressItem:(AutofillProfileItem*)item
-            withAutofillProfile:(autofill::AutofillProfile*)profile {
-  DCHECK(profile);
-  item.name = GetNameLabelFromAutofillProfile(*profile);
-  item.address = GetShippingAddressLabelFromAutofillProfile(*profile);
-  item.phoneNumber = GetPhoneNumberLabelFromAutofillProfile(*profile);
-}
-
-- (void)fillShippingOptionItem:(PaymentsTextItem*)item
-                    withOption:(web::PaymentShippingOption*)option {
-  item.text = base::SysUTF16ToNSString(option->label);
-  payments::CurrencyFormatter* currencyFormatter =
-      _paymentRequest->GetOrCreateCurrencyFormatter();
-  item.detailText = SysUTF16ToNSString(
-      currencyFormatter->Format(base::UTF16ToASCII(option->amount.value)));
-}
-
-- (void)fillPaymentMethodItem:(PaymentMethodItem*)item
-               withCreditCard:(autofill::CreditCard*)creditCard {
-  item.methodID =
-      base::SysUTF16ToNSString(creditCard->NetworkAndLastFourDigits());
-  item.methodDetail = base::SysUTF16ToNSString(
-      creditCard->GetRawInfo(autofill::CREDIT_CARD_NAME_FULL));
-  int issuerNetworkIconID =
-      autofill::data_util::GetPaymentRequestData(creditCard->network())
-          .icon_resource_id;
-  item.methodTypeIcon = NativeImage(issuerNetworkIconID);
-}
-
-- (void)fillContactInfoItem:(AutofillProfileItem*)item
-        withAutofillProfile:(autofill::AutofillProfile*)profile {
-  DCHECK(profile);
-  item.name = GetNameLabelFromAutofillProfile(*profile);
-  item.phoneNumber = GetPhoneNumberLabelFromAutofillProfile(*profile);
-  item.email = GetEmailLabelFromAutofillProfile(*profile);
+  // Update the pay button.
+  [_payButton setEnabled:[_dataSource canPay]];
 }
 
 #pragma mark - CollectionViewFooterLinkDelegate
 
 - (void)cell:(CollectionViewFooterCell*)cell didTapLinkURL:(GURL)url {
-  DCHECK_EQ(url, GURL(kSettingsURL)) << "Unknown URL tapped";
   [_delegate paymentRequestViewControllerDidSelectSettings:self];
 }
 
-#pragma mark UICollectionViewDataSource
+#pragma mark - UICollectionViewDataSource
 
 - (UICollectionViewCell*)collectionView:(UICollectionView*)collectionView
                  cellForItemAtIndexPath:(nonnull NSIndexPath*)indexPath {
   UICollectionViewCell* cell =
       [super collectionView:collectionView cellForItemAtIndexPath:indexPath];
 
   NSInteger itemType =
       [self.collectionViewModel itemTypeForIndexPath:indexPath];
   switch (itemType) {
-    case ItemTypeAddShippingAddress: {
-      CollectionViewDetailCell* detailCell =
-          base::mac::ObjCCastStrict<CollectionViewDetailCell>(cell);
-      detailCell.detailTextLabel.font = [MDCTypography body2Font];
-      detailCell.detailTextLabel.textColor =
-          [[MDCPalette cr_bluePalette] tint700];
+    case ItemTypeShippingAddress:
+    case ItemTypePaymentMethod:
+    case ItemTypeShippingOption:
+    case ItemTypeContactInfo: {
+      if ([cell isKindOfClass:[CollectionViewDetailCell class]]) {
+        CollectionViewDetailCell* detailCell =
+            base::mac::ObjCCastStrict<CollectionViewDetailCell>(cell);
+        detailCell.detailTextLabel.font = [MDCTypography body2Font];
+        detailCell.detailTextLabel.textColor =
+            [[MDCPalette cr_bluePalette] tint700];
+      }
       break;
     }
     case ItemTypeFooterText: {
       CollectionViewFooterCell* footerCell =
           base::mac::ObjCCastStrict<CollectionViewFooterCell>(cell);
       footerCell.textLabel.font = [MDCTypography body2Font];
       footerCell.textLabel.textColor = [[MDCPalette greyPalette] tint600];
       footerCell.textLabel.shadowColor = nil;  // No shadow.
       footerCell.horizontalPadding = kFooterCellHorizontalPadding;
       break;
     }
     default:
       break;
   }
   return cell;
 }
 
-#pragma mark UICollectionViewDelegate
+#pragma mark - UICollectionViewDelegate
 
 - (void)collectionView:(UICollectionView*)collectionView
     didSelectItemAtIndexPath:(NSIndexPath*)indexPath {
   [super collectionView:collectionView didSelectItemAtIndexPath:indexPath];
 
   NSInteger itemType =
       [self.collectionViewModel itemTypeForIndexPath:indexPath];
   switch (itemType) {
     case ItemTypeSummaryTotal:
-      if (!_paymentRequest->payment_details().display_items.empty())
         [_delegate
             paymentRequestViewControllerDidSelectPaymentSummaryItem:self];
       break;
     case ItemTypeShippingAddress:
-    case ItemTypeAddShippingAddress:
       [_delegate paymentRequestViewControllerDidSelectShippingAddressItem:self];
       break;
     case ItemTypeShippingOption:
-    case ItemTypeSelectShippingOption:
       [_delegate paymentRequestViewControllerDidSelectShippingOptionItem:self];
       break;
     case ItemTypePaymentMethod:
-    case ItemTypeAddPaymentMethod:
       [_delegate paymentRequestViewControllerDidSelectPaymentMethodItem:self];
       break;
     case ItemTypeContactInfo:
-    case ItemTypeAddContactInfo:
       [_delegate paymentRequestViewControllerDidSelectContactInfoItem:self];
       break;
     case ItemTypeFooterText:
       // Selecting the footer item should not trigger an action, unless the
       // link was clicked, which will call didTapLinkURL:.
       break;
     default:
       NOTREACHED();
       break;
   }
 }
 
-#pragma mark MDCCollectionViewStylingDelegate
+#pragma mark - MDCCollectionViewStylingDelegate
 
 - (CGFloat)collectionView:(UICollectionView*)collectionView
     cellHeightAtIndexPath:(NSIndexPath*)indexPath {
   CollectionViewItem* item =
       [self.collectionViewModel itemAtIndexPath:indexPath];
   switch (item.type) {
     case ItemTypeSpinner:
     case ItemTypeShippingAddress:
+    case ItemTypeShippingOption:
     case ItemTypePaymentMethod:
     case ItemTypeContactInfo:
     case ItemTypeFooterText:
       return [MDCCollectionViewCell
           cr_preferredHeightForWidth:CGRectGetWidth(collectionView.bounds)
                              forItem:item];
-    case ItemTypeShippingOption:
-      return MDCCellDefaultTwoLineHeight;
     case ItemTypeSummaryPageInfo:
     case ItemTypeSummaryTotal:
-    case ItemTypeShippingTitle:
-    case ItemTypeAddShippingAddress:
-    case ItemTypeSelectShippingOption:
-    case ItemTypePaymentTitle:
-    case ItemTypeAddPaymentMethod:
-    case ItemTypeContactInfoTitle:
-    case ItemTypeAddContactInfo:
       return MDCCellDefaultOneLineHeight;
     default:
       NOTREACHED();
       return MDCCellDefaultOneLineHeight;
   }
 }
 
 - (BOOL)collectionView:(UICollectionView*)collectionView
     hidesInkViewAtIndexPath:(NSIndexPath*)indexPath {
   NSInteger type = [self.collectionViewModel itemTypeForIndexPath:indexPath];
   // If there are no payment items to display, there is no effect from touching
   // the total so there should not be an ink ripple. The footer should also not
   // have a ripple.
-  if ((type == ItemTypeSummaryTotal &&
-       _paymentRequest->payment_details().display_items.empty()) ||
+  if ((type == ItemTypeSummaryTotal && ![_dataSource hasPaymentItems]) ||
       (type == ItemTypeFooterText)) {
     return YES;
   } else {
     return NO;
   }
 }
 
 - (BOOL)collectionView:(UICollectionView*)collectionView
     shouldHideItemBackgroundAtIndexPath:(NSIndexPath*)indexPath {
   // No background on the footer text item.
   NSInteger sectionIdentifier =
       [self.collectionViewModel sectionIdentifierForSection:indexPath.section];
   return sectionIdentifier == SectionIdentifierFooter ? YES : NO;
 }
 
+#pragma mark - Helper methods
+
+- (void)addPaymentSummaryItem {
+  CollectionViewItem* item = [_dataSource paymentSummaryItem];
+  [item setType:ItemTypeSummaryTotal];
+  if ([_dataSource hasPaymentItems])
+    item.accessibilityTraits |= UIAccessibilityTraitButton;
+  [self.collectionViewModel addItem:item
+            toSectionWithIdentifier:SectionIdentifierSummary];
+}
+
+- (void)populateShippingSection {
+  CollectionViewModel* model = self.collectionViewModel;
+
+  CollectionViewItem* shippingAddressItem = [_dataSource shippingAddressItem];
+  [shippingAddressItem setType:ItemTypeShippingAddress];
+  shippingAddressItem.accessibilityTraits |= UIAccessibilityTraitButton;
+  [model addItem:shippingAddressItem
+      toSectionWithIdentifier:SectionIdentifierShipping];
+
+  if ([_dataSource canShip]) {
+    CollectionViewItem* shippingOptionItem = [_dataSource shippingOptionItem];
+    [shippingOptionItem setType:ItemTypeShippingOption];
+    shippingOptionItem.accessibilityTraits |= UIAccessibilityTraitButton;
+    [model addItem:shippingOptionItem
+        toSectionWithIdentifier:SectionIdentifierShipping];
+  }
+}
+
+- (void)populatePaymentMethodSection {
+  CollectionViewModel* model = self.collectionViewModel;
+
+  CollectionViewItem* paymentMethodSectionHeaderItem =
+      [_dataSource paymentMethodSectionHeaderItem];
+  if (paymentMethodSectionHeaderItem) {
+    [paymentMethodSectionHeaderItem setType:ItemTypePaymentHeader];
+    [model setHeader:paymentMethodSectionHeaderItem
+        forSectionWithIdentifier:SectionIdentifierPayment];
+  }
+
+  CollectionViewItem* paymentMethodItem = [_dataSource paymentMethodItem];
+  [paymentMethodItem setType:ItemTypePaymentMethod];
+  paymentMethodItem.accessibilityTraits |= UIAccessibilityTraitButton;
+  [model addItem:paymentMethodItem
+      toSectionWithIdentifier:SectionIdentifierPayment];
+}
+
+- (void)populateContactInfoSection {
+  CollectionViewModel* model = self.collectionViewModel;
+
+  CollectionViewItem* contactInfoSectionHeaderItem =
+      [_dataSource contactInfoSectionHeaderItem];
+  if (contactInfoSectionHeaderItem) {
+    [contactInfoSectionHeaderItem setType:ItemTypeContactInfoHeader];
+    [model setHeader:contactInfoSectionHeaderItem
+        forSectionWithIdentifier:SectionIdentifierContactInfo];
+  }
+
+  CollectionViewItem* contactInfoItem = [_dataSource contactInfoItem];
+  [contactInfoItem setType:ItemTypeContactInfo];
+  contactInfoItem.accessibilityTraits |= UIAccessibilityTraitButton;
+  [model addItem:contactInfoItem
+      toSectionWithIdentifier:SectionIdentifierContactInfo];
+}
+
 @end