AutoFill Profiles dialog handles multi-select and delete on Mac

chrome/browser/autofill/autofill_dialog_controller_mac.mm

 // Copyright (c) 2010 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 "chrome/browser/autofill/autofill_dialog_controller_mac.h"
 #include "app/l10n_util.h"
 #include "app/resource_bundle.h"
 #include "base/mac_util.h"
 #include "base/sys_string_conversions.h"
 #include "chrome/browser/browser.h"
@@ skipping 173 matching lines @@
                         creditCards:personal_data_manager_->credit_cards()];
 }
 
 }  // namespace AutoFillDialogControllerInternal
 
 @implementation AutoFillDialogController
 
 @synthesize autoFillEnabled = autoFillEnabled_;
 @synthesize auxiliaryEnabled = auxiliaryEnabled_;
 @synthesize itemIsSelected = itemIsSelected_;
+@synthesize multipleSelected = multipleSelected_;
 
 + (void)showAutoFillDialogWithObserver:(AutoFillDialogObserver*)observer
                profile:(Profile*)profile
        importedProfile:(AutoFillProfile*) importedProfile
     importedCreditCard:(CreditCard*) importedCreditCard {
   AutoFillDialogController* controller =
       [AutoFillDialogController controllerWithObserver:observer
                                          profile:profile
                                  importedProfile:importedProfile
                               importedCreditCard:importedCreditCard];
@@ skipping 145 matching lines @@
 
     // Update the selection to the newly added item.
     NSInteger row = [self rowFromCreditCardIndex:creditCards_.size() - 1];
     [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:row]
             byExtendingSelection:NO];
   }
   [sheet orderOut:self];
   creditCardSheetController.reset(nil);
 }
 
-// Deletes selected item, either address or credit card depending on the item
-// selected.
+// Deletes selected items; either addresses, credit cards, or a mixture of the
+// two depending on the items selected.
 - (IBAction)deleteSelection:(id)sender {
+  NSIndexSet* selectedRows = [tableView_ selectedRowIndexes];
   NSInteger selectedRow = [tableView_ selectedRow];
-  if ([self isProfileRow:selectedRow]) {
-    profiles_.erase(profiles_.begin() + [self profileIndexFromRow:selectedRow]);
+  NSInteger rowCount = [tableView_ numberOfRows];
 
-    // Select the previous row if possible, else current row, else deselect all.
-    if ([self tableView:tableView_ shouldSelectRow:selectedRow-1]) {
-      [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:selectedRow-1]
-              byExtendingSelection:NO];
-    } else if ([self tableView:tableView_ shouldSelectRow:selectedRow]) {
-      [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:selectedRow]
-              byExtendingSelection:NO];
-    } else {
-      [tableView_ selectRowIndexes:[NSIndexSet indexSet]
-              byExtendingSelection:NO];
+  // Loop through from last to first deleting selected items as we go.
+  for (NSInteger i = rowCount-1; i>=0; --i) {

[Nico, 2010/07/20 03:18:20]

You want:

    for (NSUInteger i = [selection lastIndex];
       i != NSNotFound;
       i = [selection indexLessThanIndex:i]) {

+    if (![selectedRows containsIndex:i])
+      continue;
+
+    // We keep track of the "top most" selection in the list so we know where
+    // to to set new selection below.
+    if (i < selectedRow)
+      selectedRow = i;
+
+    if ([self isProfileRow:i]) {
+      profiles_.erase(
+          profiles_.begin() + [self profileIndexFromRow:i]);
+    } else if ([self isCreditCardRow:i]) {
+      creditCards_.erase(
+          creditCards_.begin() + [self creditCardIndexFromRow:i]);
     }
-    UpdateProfileLabels(&profiles_);
-    [tableView_ reloadData];
-  } else if ([self isCreditCardRow:selectedRow]) {
-    creditCards_.erase(
-        creditCards_.begin() + [self creditCardIndexFromRow:selectedRow]);
+  }
 
-    // Select the previous row if possible, else current row, else deselect all.
-    if ([self tableView:tableView_ shouldSelectRow:selectedRow-1]) {
-      [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:selectedRow-1]
-              byExtendingSelection:NO];
-    } else if ([self tableView:tableView_ shouldSelectRow:selectedRow]) {
-      [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:selectedRow]
-              byExtendingSelection:NO];
-    } else {
-      [tableView_ selectRowIndexes:[NSIndexSet indexSet]
-              byExtendingSelection:NO];
-    }
-    [tableView_ reloadData];
+  // Select the previous row if possible, else current row, else deselect all.
+  if ([self tableView:tableView_ shouldSelectRow:selectedRow-1]) {
+    [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:selectedRow-1]
+            byExtendingSelection:NO];
+  } else if ([self tableView:tableView_ shouldSelectRow:selectedRow]) {
+    [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:selectedRow]
+            byExtendingSelection:NO];
+  } else {
+    [tableView_ selectRowIndexes:[NSIndexSet indexSet] byExtendingSelection:NO];
   }
+
+  UpdateProfileLabels(&profiles_);
+  [tableView_ reloadData];
 }
 
 // Edits the selected item, either address or credit card depending on the item
 // selected.
 - (IBAction)editSelection:(id)sender {
   NSInteger selectedRow = [tableView_ selectedRow];
   if ([self isProfileRow:selectedRow]) {
     if (!addressSheetController.get()) {
       int i = [self profileIndexFromRow:selectedRow];
 
@@ skipping 129 matching lines @@
           creditCards_[
               [self creditCardIndexFromRow:row]].PreviewSummary());
     }
 
     return @"";
   }
 
   return @"";
 }
 
-// We implement this delegate method to update our |itemIsSelected| property.
+// We implement this delegate method to update our |itemIsSelected| and
+// |multipleSelected| properties.
 // The "Edit..." and "Remove" buttons' enabled state depends on having a
-// valid selection in the table.
+// valid selection in the table.  The "Edit..." button depends on having
+// exactly one item selected.
 - (void)tableViewSelectionDidChange:(NSNotification *)aNotification {
   if ([tableView_ selectedRow] >= 0)
     [self setItemIsSelected:YES];
   else
     [self setItemIsSelected:NO];
+
+  [self setMultipleSelected:([[tableView_ selectedRowIndexes] count] > 1UL)];
 }
 
 - (NSInteger)numberOfRowsInTableView:(NSTableView *)tableView {
   if (tableView == tableView_) {
     // 1 section header, the profiles, 1 section header, the credit cards.
     return 1 + profiles_.size() + 1 + creditCards_.size();
   }
 
   return 0;
 }
@@ skipping 84 matching lines @@
                                    [self rowFromProfileIndex:i]]
           byExtendingSelection:NO];
 }
 
 - (void)selectCreditCardAtIndex:(size_t)i {
   [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:
                                    [self rowFromCreditCardIndex:i]]
           byExtendingSelection:NO];
 }
 
+- (void)addSelectedAddressAtIndex:(size_t)i {
+  [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:
+                                   [self rowFromProfileIndex:i]]
+          byExtendingSelection:YES];
+}
+
+- (void)addSelectedCreditCardAtIndex:(size_t)i {
+  [tableView_ selectRowIndexes:[NSIndexSet indexSetWithIndex:
+                                   [self rowFromCreditCardIndex:i]]
+          byExtendingSelection:YES];
+}
+
+- (BOOL)editButtonEnabled {
+  return [editButton_ isEnabled];
+}
+
 @end
 
 @implementation AutoFillDialogController (PrivateMethods)
 
 // Run application modal.
 - (void)runModalDialog {
   // Use stored window geometry if it exists.
   if (g_browser_process && g_browser_process->local_state()) {
     sizeSaver_.reset([[WindowSizeAutosaver alloc]
         initWithWindow:[self window]
@@ skipping 121 matching lines @@
   if (!image) {
     image = rb.GetNSImageNamed(IDR_INPUT_GOOD);
     DCHECK(image);
     return image;
   }
 
   return nil;
 }
 
 @end