[Payment Request] Selector view edit mode

ios/chrome/browser/ui/payments/billing_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/billing_address_selection_mediator.h"
 
 #include "base/logging.h"
 #include "components/autofill/core/browser/autofill_profile.h"
@@ skipping 20 matching lines @@
 
 // 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 selected billing address, if any.
 @property(nonatomic, assign) autofill::AutofillProfile* selectedBillingProfile;
 
 // The selectable items to display in the collection.
-@property(nonatomic, strong) NSArray<AutofillProfileItem*>* items;
+@property(nonatomic, strong) NSMutableArray<AutofillProfileItem*>* items;
 
 @end
 
 @implementation BillingAddressSelectionMediator
 
 @synthesize state = _state;
 @synthesize selectedItemIndex = _selectedItemIndex;
 @synthesize paymentRequest = _paymentRequest;
 @synthesize selectedBillingProfile = _selectedBillingProfile;
 @synthesize items = _items;
 
 - (instancetype)initWithPaymentRequest:(PaymentRequest*)paymentRequest
                 selectedBillingProfile:
                     (autofill::AutofillProfile*)selectedBillingProfile {
   self = [super init];
   if (self) {
     DCHECK(paymentRequest);
     _paymentRequest = paymentRequest;
     _selectedBillingProfile = selectedBillingProfile;
     _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_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*>& billingProfiles =
       _paymentRequest->billing_profiles();
 
-  NSMutableArray<AutofillProfileItem*>* items =
-      [NSMutableArray arrayWithCapacity:billingProfiles.size()];
+  _items = [NSMutableArray arrayWithCapacity:billingProfiles.size()];
   for (size_t index = 0; index < billingProfiles.size(); ++index) {
     autofill::AutofillProfile* billingProfile = billingProfiles[index];
     DCHECK(billingProfile);
     AutofillProfileItem* item = [[AutofillProfileItem alloc] init];
     item.name = GetNameLabelFromAutofillProfile(*billingProfile);
     item.address = GetBillingAddressLabelFromAutofillProfile(*billingProfile);
     item.phoneNumber = GetPhoneNumberLabelFromAutofillProfile(*billingProfile);
     if (self.selectedBillingProfile == billingProfile)
       _selectedItemIndex = index;
 
-    [items addObject:item];
+    [_items addObject:item];
   }
-
-  return items;
 }
 
 @end