[Payment Request] Address edit view controller (Part 2)

ios/chrome/browser/ui/payments/payment_request_edit_view_controller.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.
 
 #import "ios/chrome/browser/ui/payments/payment_request_edit_view_controller.h"
 
 #include "base/logging.h"
 #import "base/mac/foundation_util.h"
 #include "base/strings/sys_string_conversions.h"
 #include "components/strings/grit/components_strings.h"
@@ skipping 50 matching lines @@
 typedef NS_ENUM(NSInteger, ItemType) {
   ItemTypeHeader = kItemTypeEnumZero,
   ItemTypeFooter,
   ItemTypeTextField,      // This is a repeated item type.
   ItemTypeSelectorField,  // This is a repeated item type.
   ItemTypeErrorMessage,   // This is a repeated item type.
 };
 
 }  // namespace
 
-@interface PaymentRequestEditViewController ()<
-    AutofillEditAccessoryDelegate,
-    UITextFieldDelegate> {
+@interface PaymentRequestEditViewController ()<AutofillEditAccessoryDelegate,
+                                               UITextFieldDelegate,
+                                               UIPickerViewDataSource,
+                                               UIPickerViewDelegate> {
   // The currently focused cell. May be nil.
   __weak AutofillEditCell* _currentEditingCell;
 
   AutofillEditAccessoryView* _accessoryView;
 }
 
+// The map of autofill types to the fields definitions for the editor.
+@property(nonatomic, strong)
+    NSMutableDictionary<NSNumber*, EditorField*>* fieldsMap;
+
 // The list of field definitions for the editor.
 @property(nonatomic, strong) NSArray<EditorField*>* fields;
 
+// The map of autofill types to lists of UIPickerView options.
+@property(nonatomic, strong)
+    NSMutableDictionary<NSNumber*, NSArray<NSString*>*>* options;
+
+// The map of autofill types to UIPickerView views.
+@property(nonatomic, strong)
+    NSMutableDictionary<NSNumber*, UIPickerView*>* pickerViews;
+
 // Returns the indexPath for the same row as that of |indexPath| in a section
 // with the given offset relative to that of |indexPath|. May return nil.
 - (NSIndexPath*)indexPathWithSectionOffset:(NSInteger)offset
                                   fromPath:(NSIndexPath*)indexPath;
 
 // Returns the text field with the given offset relative to the currently
 // focused text field. May return nil.
 - (AutofillEditCell*)nextTextFieldWithOffset:(NSInteger)offset;
 
 // Enables or disables the accessory view's previous and next buttons depending
 // on whether there is a text field before and after the currently focused text
 // field.
 - (void)updateAccessoryViewButtonsStates;
 
 // Adds an error message item in the section |sectionIdentifier| if
 // |errorMessage| is non-empty. Otherwise removes such an item if one exists.
 - (void)addOrRemoveErrorMessage:(NSString*)errorMessage
         inSectionWithIdentifier:(NSInteger)sectionIdentifier;
 
 @end
 
 @implementation PaymentRequestEditViewController
 
 @synthesize dataSource = _dataSource;
 @synthesize delegate = _delegate;
 @synthesize validatorDelegate = _validatorDelegate;
+@synthesize fieldsMap = _fieldsMap;
 @synthesize fields = _fields;
+@synthesize options = _options;
+@synthesize pickerViews = _pickerViews;
 
 - (instancetype)initWithStyle:(CollectionViewControllerStyle)style {
   self = [super initWithStyle:style];
   if (self) {
     _accessoryView = [[AutofillEditAccessoryView alloc] initWithDelegate:self];
+    _options = [[NSMutableDictionary alloc] init];
+    _pickerViews = [[NSMutableDictionary alloc] init];
   }
   return self;
 }
 
 - (void)viewDidAppear:(BOOL)animated {
   [super viewDidAppear:animated];
   [[NSNotificationCenter defaultCenter]
       addObserver:self
          selector:@selector(keyboardDidShow)
              name:UIKeyboardDidShowNotification
            object:nil];
 }
 
 - (void)viewWillDisappear:(BOOL)animated {
   [super viewWillDisappear:animated];
   [[NSNotificationCenter defaultCenter]
       removeObserver:self
                 name:UIKeyboardDidShowNotification
               object:nil];
 }
 
 #pragma mark - CollectionViewController methods
 
 - (void)loadModel {
   [super loadModel];
   CollectionViewModel* model = self.collectionViewModel;
 
+  [self.pickerViews removeAllObjects];
+
   CollectionViewItem* headerItem = [_dataSource headerItem];
   if (headerItem) {
     [headerItem setType:ItemTypeHeader];
     [model addSectionWithIdentifier:SectionIdentifierHeader];
     [model addItem:headerItem toSectionWithIdentifier:SectionIdentifierHeader];
   }
 
   // Iterate over the fields and add the respective sections and items.
-  int sectionIdentifier = static_cast<int>(SectionIdentifierFirstField);
-  for (EditorField* field in self.fields) {
+  [self.fields enumerateObjectsUsingBlock:^(EditorField* field,
+                                            NSUInteger index, BOOL* stop) {
+    NSInteger sectionIdentifier = SectionIdentifierFirstField + index;
     [model addSectionWithIdentifier:sectionIdentifier];
     switch (field.fieldType) {
       case EditorFieldTypeTextField: {
         AutofillEditItem* item =
             [[AutofillEditItem alloc] initWithType:ItemTypeTextField];
         item.textFieldName = field.label;
-        item.textFieldEnabled = YES;
+        item.textFieldEnabled = field.enabled;
         item.textFieldValue = field.value;
         item.required = field.isRequired;
         item.autofillUIType = field.autofillUIType;
-        [model addItem:item
-            toSectionWithIdentifier:static_cast<NSInteger>(sectionIdentifier)];
+        [model addItem:item toSectionWithIdentifier:sectionIdentifier];
         field.item = item;
+
         break;
       }
       case EditorFieldTypeSelector: {
         PaymentsSelectorEditItem* item = [[PaymentsSelectorEditItem alloc]
             initWithType:ItemTypeSelectorField];
         item.name = field.label;
         item.value = field.displayValue;
         item.required = field.isRequired;
         item.autofillUIType = field.autofillUIType;
         item.accessoryType = MDCCollectionViewCellAccessoryDisclosureIndicator;
-        [model addItem:item
-            toSectionWithIdentifier:static_cast<NSInteger>(sectionIdentifier)];
+        [model addItem:item toSectionWithIdentifier:sectionIdentifier];
         field.item = item;
         break;
       }
       default:
         NOTREACHED();
     }
 
-    field.sectionIdentifier = static_cast<NSInteger>(sectionIdentifier);
-    ++sectionIdentifier;
-  }
+    field.sectionIdentifier = sectionIdentifier;
+  }];
 
   [self loadFooterItems];
 }
 
 - (void)viewDidLoad {
   [super viewDidLoad];
 
   self.collectionView.accessibilityIdentifier =
       kPaymentRequestEditCollectionViewAccessibilityID;
 
   // Customize collection view settings.
   self.styler.cellStyle = MDCCollectionViewCellStyleCard;
   self.styler.separatorInset =
       UIEdgeInsetsMake(0, kSeparatorEdgeInset, 0, kSeparatorEdgeInset);
 }
 
 #pragma mark - PaymentRequestEditConsumer
 
 - (void)setEditorFields:(NSArray<EditorField*>*)fields {
   self.fields = fields;
+  self.fieldsMap = [[NSMutableDictionary alloc] initWithCapacity:fields.count];
+  // Iterate over the fields and populate the map.
+  [self.fields enumerateObjectsUsingBlock:^(EditorField* field,
+                                            NSUInteger index, BOOL* stop) {
+    NSNumber* key = [NSNumber numberWithInt:field.autofillUIType];
+    [self.fieldsMap setObject:field forKey:key];
+  }];
+}
+
+- (void)setOptions:(NSArray<NSString*>*)options
+    forEditorField:(EditorField*)field {
+  DCHECK(field.fieldType == EditorFieldTypeTextField);
+  AutofillEditItem* item =
+      base::mac::ObjCCastStrict<AutofillEditItem>(field.item);
+
+  // Enable the previously disabled text field and reset its value.
+  item.textFieldEnabled = YES;
+  item.textFieldValue = nil;
+
+  // Cache the options if there are any and set the text field's UIPickerView.
+  if (options.count) {
+    NSNumber* key = [NSNumber numberWithInt:field.autofillUIType];
+    [self.options setObject:options forKey:key];
+
+    UIPickerView* pickerView = [[UIPickerView alloc] initWithFrame:CGRectZero];
+    pickerView.delegate = self;
+    pickerView.dataSource = self;
+    [self.pickerViews setObject:pickerView forKey:key];
+    item.inputView = pickerView;
+  }
+
+  // Reload the item.
+  NSIndexPath* indexPath =
+      [self.collectionViewModel indexPathForItemType:ItemTypeTextField
+                                   sectionIdentifier:field.sectionIdentifier];
+  [self.collectionView reloadItemsAtIndexPaths:@[ indexPath ]];
 }
 
 #pragma mark - UITextFieldDelegate
 
 - (void)textFieldDidBeginEditing:(UITextField*)textField {
   _currentEditingCell = AutofillEditCellForTextField(textField);
   [textField setInputAccessoryView:_accessoryView];
   [self updateAccessoryViewButtonsStates];
 }
 
 - (void)textFieldDidEndEditing:(UITextField*)textField {
   DCHECK(_currentEditingCell == AutofillEditCellForTextField(textField));
 
-  // Validate the text field.
   CollectionViewModel* model = self.collectionViewModel;
 
   NSIndexPath* indexPath = [self indexPathForCurrentTextField];
   AutofillEditItem* item = base::mac::ObjCCastStrict<AutofillEditItem>(
       [model itemAtIndexPath:indexPath]);
 
-  // Create a dummy EditorField for validation only.
-  EditorField* fieldForValidation =
-      [[EditorField alloc] initWithAutofillUIType:item.autofillUIType
-                                        fieldType:EditorFieldTypeTextField
-                                            label:nil
-                                            value:textField.text
-                                         required:item.required];
+  // Find and validate the respective editor field.
+  NSNumber* key = [NSNumber numberWithInt:item.autofillUIType];
+  EditorField* field = self.fieldsMap[key];
+  DCHECK(field);
+  field.value = textField.text;
   NSString* errorMessage =
       [_validatorDelegate paymentRequestEditViewController:self
-                                             validateField:fieldForValidation];
+                                             validateField:field];
   NSInteger sectionIdentifier =
       [model sectionIdentifierForSection:[indexPath section]];
   [self addOrRemoveErrorMessage:errorMessage
         inSectionWithIdentifier:sectionIdentifier];
 
   [textField setInputAccessoryView:nil];
   _currentEditingCell = nil;
 }
 
 - (BOOL)textFieldShouldReturn:(UITextField*)textField {
@@ skipping 18 matching lines @@
 - (void)previousPressed {
   AutofillEditCell* previousCell = [self nextTextFieldWithOffset:-1];
   if (previousCell)
     [previousCell.textField becomeFirstResponder];
 }
 
 - (void)closePressed {
   [[_currentEditingCell textField] resignFirstResponder];
 }
 
+#pragma mark - UIPickerViewDataSource methods
+
+- (NSInteger)numberOfComponentsInPickerView:(UIPickerView*)thePickerView {
+  return 1;
+}
+
+- (NSInteger)pickerView:(UIPickerView*)thePickerView
+    numberOfRowsInComponent:(NSInteger)component {
+  NSArray<NSNumber*>* indices =
+      [self.pickerViews allKeysForObject:thePickerView];
+  DCHECK(indices.count == 1);
+  NSArray<NSString*>* options = self.options[indices[0]];
+  return options.count;
+}
+
+#pragma mark - UIPickerViewDelegate methods
+
+- (NSString*)pickerView:(UIPickerView*)thePickerView
+            titleForRow:(NSInteger)row
+           forComponent:(NSInteger)component {
+  NSArray<NSNumber*>* indices =
+      [self.pickerViews allKeysForObject:thePickerView];
+  DCHECK(indices.count == 1);
+  NSArray<NSString*>* options = self.options[indices[0]];
+  DCHECK(row < static_cast<NSInteger>(options.count));
+  return options[row];
+}
+
+- (void)pickerView:(UIPickerView*)thePickerView
+      didSelectRow:(NSInteger)row
+       inComponent:(NSInteger)component {
+  DCHECK(_currentEditingCell);
+  _currentEditingCell.textField.text =
+      [self pickerView:thePickerView titleForRow:row forComponent:component];
+}
+
 #pragma mark - UICollectionViewDataSource
 
 - (UICollectionViewCell*)collectionView:(UICollectionView*)collectionView
                  cellForItemAtIndexPath:(NSIndexPath*)indexPath {
   UICollectionViewCell* cell =
       [super collectionView:collectionView cellForItemAtIndexPath:indexPath];
 
   CollectionViewItem* item =
       [self.collectionViewModel itemAtIndexPath:indexPath];
   switch (item.type) {
@@ skipping 40 matching lines @@
   [super collectionView:collectionView didSelectItemAtIndexPath:indexPath];
 
   // Every field has its own section. Find out which field is selected using
   // the section of |indexPath|. Adjust the index if a header section is
   // present before the editor fields.
   NSInteger index = indexPath.section;
   if ([self.collectionViewModel
           hasSectionForSectionIdentifier:SectionIdentifierHeader])
     index--;
   DCHECK(index >= 0 && index < static_cast<NSInteger>(self.fields.count));
-  [_delegate
-      paymentRequestEditViewController:self
-                        didSelectField:[self.fields objectAtIndex:index]];
+  EditorField* field = [self.fields objectAtIndex:index];
+
+  // If a selector field is selected, blur the focused text field.
+  if (field.fieldType == EditorFieldTypeSelector)
+    [[_currentEditingCell textField] resignFirstResponder];
+
+  [_delegate paymentRequestEditViewController:self didSelectField:field];
 }
 
 #pragma mark MDCCollectionViewStylingDelegate
 
 - (CGFloat)collectionView:(UICollectionView*)collectionView
     cellHeightAtIndexPath:(NSIndexPath*)indexPath {
   CollectionViewItem* item =
       [self.collectionViewModel itemAtIndexPath:indexPath];
   switch (item.type) {
     case ItemTypeHeader:
@@ skipping 165 matching lines @@
 
 - (NSIndexPath*)indexPathForCurrentTextField {
   DCHECK(_currentEditingCell);
   NSIndexPath* indexPath =
       [[self collectionView] indexPathForCell:_currentEditingCell];
   DCHECK(indexPath);
   return indexPath;
 }
 
 @end