[Payment Request] Selector view edit mode

ios/chrome/browser/ui/payments/payment_method_selection_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.
 
 #include <vector>
 
 #import "ios/chrome/browser/ui/payments/payment_method_selection_mediator.h"
 
 #include "base/strings/sys_string_conversions.h"
 #include "components/autofill/core/browser/autofill_data_util.h"
@@ skipping 18 matching lines @@
 }  // namespace
 
 @interface PaymentMethodSelectionMediator ()
 
 // The PaymentRequest object owning an instance of web::PaymentRequest as
 // provided by the page invoking the Payment Request API. This is a weak
 // pointer and should outlive this class.
 @property(nonatomic, assign) PaymentRequest* paymentRequest;
 
 // The selectable items to display in the collection.
-@property(nonatomic, strong) NSArray<PaymentMethodItem*>* items;
+@property(nonatomic, strong) NSMutableArray<PaymentMethodItem*>* items;
 
 @end
 
 @implementation PaymentMethodSelectionMediator
 
 @synthesize state = _state;
 @synthesize selectedItemIndex = _selectedItemIndex;
 @synthesize paymentRequest = _paymentRequest;
 @synthesize items = _items;
 
 - (instancetype)initWithPaymentRequest:(PaymentRequest*)paymentRequest {
   self = [super init];
   if (self) {
     _paymentRequest = paymentRequest;
     _selectedItemIndex = NSUIntegerMax;
-    _items = [self createItems];
+    [self loadItems];
   }
   return self;
 }
 
 #pragma mark - PaymentRequestSelectorViewControllerDataSource
 
+- (BOOL)allowsEditMode {
+  return YES;
+}
+
 - (CollectionViewItem*)headerItem {
   return nil;
 }
 
 - (NSArray<CollectionViewItem*>*)selectableItems {
   return self.items;
 }
 
 - (CollectionViewItem*)addButtonItem {
   PaymentsTextItem* addButtonItem = [[PaymentsTextItem alloc] init];
   addButtonItem.text = l10n_util::GetNSString(IDS_PAYMENTS_ADD_CARD);
   addButtonItem.image = NativeImage(IDR_IOS_PAYMENTS_ADD);
   return addButtonItem;
 }
 
-#pragma mark - Helper methods
+#pragma mark - Public methods
 
-- (NSArray<PaymentMethodItem*>*)createItems {
+- (void)loadItems {
   const std::vector<autofill::CreditCard*>& paymentMethods =
       _paymentRequest->credit_cards();
-  NSMutableArray<PaymentMethodItem*>* items =
-      [NSMutableArray arrayWithCapacity:paymentMethods.size()];
+  _items = [NSMutableArray arrayWithCapacity:paymentMethods.size()];
   for (size_t index = 0; index < paymentMethods.size(); ++index) {
     autofill::CreditCard* paymentMethod = paymentMethods[index];
     DCHECK(paymentMethod);
     PaymentMethodItem* item = [[PaymentMethodItem alloc] init];
     item.methodID =
         base::SysUTF16ToNSString(paymentMethod->NetworkAndLastFourDigits());
     item.methodDetail = base::SysUTF16ToNSString(
         paymentMethod->GetRawInfo(autofill::CREDIT_CARD_NAME_FULL));
 
     autofill::AutofillProfile* billingAddress =
         autofill::PersonalDataManager::GetProfileFromProfilesByGUID(
             paymentMethod->billing_address_id(),
             _paymentRequest->billing_profiles());
     if (billingAddress) {
       item.methodAddress =
           GetBillingAddressLabelFromAutofillProfile(*billingAddress);
     }
 
     int methodTypeIconID =
         autofill::data_util::GetPaymentRequestData(paymentMethod->network())
             .icon_resource_id;
     item.methodTypeIcon = NativeImage(methodTypeIconID);
 
     item.reserveRoomForAccessoryType = YES;
     if (_paymentRequest->selected_credit_card() == paymentMethod)
       _selectedItemIndex = index;
 
-    [items addObject:item];
+    [_items addObject:item];
   }
-  return items;
 }
 
 @end