[Payment Request] Selector view edit mode

ios/chrome/browser/ui/payments/shipping_address_selection_coordinator.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/shipping_address_selection_coordinator.h"
 
 #include "base/strings/sys_string_conversions.h"
 #include "components/autofill/core/browser/autofill_profile.h"
 #include "components/payments/core/strings_util.h"
 #include "ios/chrome/browser/payments/payment_request.h"
@@ skipping 16 matching lines @@
 
 @interface ShippingAddressSelectionCoordinator ()
 
 @property(nonatomic, strong) AddressEditCoordinator* addressEditCoordinator;
 
 @property(nonatomic, strong)
     PaymentRequestSelectorViewController* viewController;
 
 @property(nonatomic, strong) ShippingAddressSelectionMediator* mediator;
 
+// Initializes and starts the AddressEditCoordinator. Sets |address| as the
+// address to be edited.
+- (void)startAddressEditCoordinatorWithAddress:
+    (autofill::AutofillProfile*)address;
+
 // Called when the user selects a shipping address. The cell is checked, the
 // UI is locked so that the user can't interact with it, then the delegate is
 // notified. The delay is here to let the user get a visual feedback of the
 // selection before this view disappears.
 - (void)delayedNotifyDelegateOfSelection:
     (autofill::AutofillProfile*)shippingAddress;
 
 @end
 
 @implementation ShippingAddressSelectionCoordinator
@@ skipping 60 matching lines @@
                                              ->shipping_profiles()[index]];
 }
 
 - (void)paymentRequestSelectorViewControllerDidFinish:
     (PaymentRequestSelectorViewController*)controller {
   [self.delegate shippingAddressSelectionCoordinatorDidReturn:self];
 }
 
 - (void)paymentRequestSelectorViewControllerDidSelectAddItem:
     (PaymentRequestSelectorViewController*)controller {
-  self.addressEditCoordinator = [[AddressEditCoordinator alloc]
-      initWithBaseViewController:self.viewController];
-  self.addressEditCoordinator.paymentRequest = self.paymentRequest;
-  self.addressEditCoordinator.delegate = self;
-  [self.addressEditCoordinator start];
+  [self startAddressEditCoordinatorWithAddress:nil];
+}
+
+- (void)paymentRequestSelectorViewController:
+            (PaymentRequestSelectorViewController*)controller
+                            didChangeToState:
+                                (PaymentRequestSelectorState)state {
+  // Update the data source with the new state.
+  self.mediator.state = state;
+
+  [self.viewController loadModel];
+  [self.viewController.collectionView reloadData];
+}
+
+- (void)paymentRequestSelectorViewController:
+            (PaymentRequestSelectorViewController*)controller
+              didSelectItemAtIndexForEditing:(NSUInteger)index {
+  DCHECK(index < self.paymentRequest->shipping_profiles().size());
+  [self
+      startAddressEditCoordinatorWithAddress:self.paymentRequest
+                                                 ->shipping_profiles()[index]];
 }
 
 #pragma mark - AddressEditCoordinatorDelegate
 
 - (void)addressEditCoordinator:(AddressEditCoordinator*)coordinator
        didFinishEditingAddress:(autofill::AutofillProfile*)address {
+  // Update the data source with the new data.
+  [self.mediator createItems];
+
+  [self.viewController loadModel];
+  [self.viewController.collectionView reloadData];
+
   [self.addressEditCoordinator stop];
   self.addressEditCoordinator = nil;
 
-  // Inform |self.delegate| that |address| has been selected.
-  [self.delegate shippingAddressSelectionCoordinator:self
-                            didSelectShippingAddress:address];
+  if (self.mediator.state != PaymentRequestSelectorStateEdit) {
+    // Inform |self.delegate| that |address| has been selected.
+    [self.delegate shippingAddressSelectionCoordinator:self
+                              didSelectShippingAddress:address];
+  }
 }
 
 - (void)addressEditCoordinatorDidCancel:(AddressEditCoordinator*)coordinator {
   [self.addressEditCoordinator stop];
   self.addressEditCoordinator = nil;
 }
 
 #pragma mark - Helper methods
 
+- (void)startAddressEditCoordinatorWithAddress:
+    (autofill::AutofillProfile*)address {
+  self.addressEditCoordinator = [[AddressEditCoordinator alloc]
+      initWithBaseViewController:self.viewController];
+  self.addressEditCoordinator.paymentRequest = self.paymentRequest;
+  self.addressEditCoordinator.address = address;
+  self.addressEditCoordinator.delegate = self;
+  [self.addressEditCoordinator start];
+}
+
 - (void)delayedNotifyDelegateOfSelection:
     (autofill::AutofillProfile*)shippingAddress {
   self.viewController.view.userInteractionEnabled = NO;
   __weak ShippingAddressSelectionCoordinator* weakSelf = self;
   dispatch_after(
       dispatch_time(DISPATCH_TIME_NOW, kDelegateNotificationDelayInNanoSeconds),
       dispatch_get_main_queue(), ^{
         [weakSelf.mediator setState:PaymentRequestSelectorStatePending];
         [weakSelf.viewController loadModel];
         [weakSelf.viewController.collectionView reloadData];
 
         [weakSelf.delegate shippingAddressSelectionCoordinator:weakSelf
                                       didSelectShippingAddress:shippingAddress];
       });
 }
 
 @end