[Payment Request] Selector view edit mode

ios/chrome/browser/ui/payments/shipping_address_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/shipping_address_selection_mediator.h"
 
 #include "components/autofill/core/browser/autofill_profile.h"
 #include "components/strings/grit/components_strings.h"
@@ skipping 16 matching lines @@
 }  // namespace
 
 @interface ShippingAddressSelectionMediator ()
 
 // 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<AutofillProfileItem*>* items;
+@property(nonatomic, strong) NSMutableArray<AutofillProfileItem*>* items;
 
 @end
 
 @implementation ShippingAddressSelectionMediator
 
 @synthesize headerText = _headerText;
 @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 {
   if (!self.headerText.length)
     return nil;
 
   PaymentsTextItem* headerItem = [[PaymentsTextItem alloc] init];
   headerItem.text = self.headerText;
   if (self.state == PaymentRequestSelectorStateError)
     headerItem.image = NativeImage(IDR_IOS_PAYMENTS_WARNING);
   return headerItem;
 }
 
 - (NSArray<CollectionViewItem*>*)selectableItems {
   return self.items;
 }
 
 - (CollectionViewItem*)addButtonItem {
   PaymentsTextItem* addButtonItem = [[PaymentsTextItem alloc] init];
   addButtonItem.text = l10n_util::GetNSString(IDS_PAYMENTS_ADD_ADDRESS);
   addButtonItem.image = NativeImage(IDR_IOS_PAYMENTS_ADD);
   return addButtonItem;
 }
 
-#pragma mark - Helper methods
+#pragma mark - Public methods
 
-- (NSArray<AutofillProfileItem*>*)createItems {
+- (void)loadItems {
   const std::vector<autofill::AutofillProfile*>& shippingProfiles =
       _paymentRequest->shipping_profiles();
-  NSMutableArray<AutofillProfileItem*>* items =
-      [NSMutableArray arrayWithCapacity:shippingProfiles.size()];
+  _items = [NSMutableArray arrayWithCapacity:shippingProfiles.size()];
   for (size_t index = 0; index < shippingProfiles.size(); ++index) {
     autofill::AutofillProfile* shippingAddress = shippingProfiles[index];
     DCHECK(shippingAddress);
     AutofillProfileItem* item = [[AutofillProfileItem alloc] init];
     item.name = GetNameLabelFromAutofillProfile(*shippingAddress);
     item.address = GetShippingAddressLabelFromAutofillProfile(*shippingAddress);
     item.phoneNumber = GetPhoneNumberLabelFromAutofillProfile(*shippingAddress);
     if (_paymentRequest->selected_shipping_profile() == shippingAddress)
       _selectedItemIndex = index;
 
-    [items addObject:item];
+    [_items addObject:item];
   }
-  return items;
 }
 
 @end