[Mac] Hook up the "Remove" button in the collected cookies dialog.

chrome/browser/ui/cocoa/content_settings/collected_cookies_mac.mm

 // Copyright (c) 2012 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/ui/cocoa/content_settings/collected_cookies_mac.h"
 
 #include <vector>
 
 #include "base/mac/bundle_locations.h"
-#import "base/mac/mac_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/prefs/pref_service.h"
 #include "base/strings/sys_string_conversions.h"
-#include "chrome/browser/browsing_data/browsing_data_appcache_helper.h"
-#include "chrome/browser/browsing_data/browsing_data_cookie_helper.h"
-#include "chrome/browser/browsing_data/browsing_data_database_helper.h"
-#include "chrome/browser/browsing_data/browsing_data_file_system_helper.h"
-#include "chrome/browser/browsing_data/browsing_data_indexed_db_helper.h"
-#include "chrome/browser/browsing_data/browsing_data_local_storage_helper.h"
-#include "chrome/browser/browsing_data/browsing_data_server_bound_cert_helper.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/content_settings/cookie_settings.h"
 #include "chrome/browser/content_settings/local_shared_objects_container.h"
 #include "chrome/browser/content_settings/tab_specific_content_settings.h"
 #include "chrome/browser/infobars/infobar_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser_dialogs.h"
 #import "chrome/browser/ui/cocoa/constrained_window/constrained_window_custom_sheet.h"
 #import "chrome/browser/ui/cocoa/content_settings/cookie_details_view_controller.h"
 #import "chrome/browser/ui/cocoa/vertical_gradient_view.h"
 #include "chrome/browser/ui/collected_cookies_infobar_delegate.h"
 #include "chrome/common/pref_names.h"
 #include "content/public/browser/notification_details.h"
 #include "content/public/browser/notification_source.h"
 #include "content/public/browser/web_contents.h"
-#include "content/public/browser/web_contents_view.h"
 #include "grit/generated_resources.h"
 #include "grit/theme_resources.h"
 #include "third_party/apple_sample_code/ImageAndTextCell.h"
 #import "third_party/google_toolbox_for_mac/src/AppKit/GTMNSAnimation+Duration.h"
 #import "third_party/google_toolbox_for_mac/src/AppKit/GTMUILocalizerAndLayoutTweaker.h"
 #include "ui/base/l10n/l10n_util_mac.h"
 #include "ui/base/resource/resource_bundle.h"
 #include "ui/gfx/image/image.h"
 #include "ui/gfx/image/image_skia.h"
 #include "ui/gfx/image/image_skia_util_mac.h"
@@ skipping 56 matching lines @@
   window_->CloseWebContentsModalDialog();
 }
 
 void CollectedCookiesMac::OnConstrainedWindowClosed(
     ConstrainedWindowMac* window) {
   base::MessageLoop::current()->DeleteSoon(FROM_HERE, this);
 }
 
 #pragma mark Window Controller
 
-@interface CollectedCookiesWindowController(Private)
--(void)showInfoBarForDomain:(const base::string16&)domain
-                    setting:(ContentSetting)setting;
--(void)showInfoBarForMultipleDomainsAndSetting:(ContentSetting)setting;
--(void)animateInfoBar;
+@interface CollectedCookiesWindowController (Private)
+- (void)showInfoBarForDomain:(const base::string16&)domain
+                     setting:(ContentSetting)setting;
+- (void)showInfoBarForMultipleDomainsAndSetting:(ContentSetting)setting;
+- (void)animateInfoBar;
 @end
 
 @implementation CollectedCookiesWindowController
 
-@synthesize allowedCookiesButtonsEnabled =
-    allowedCookiesButtonsEnabled_;
-@synthesize blockedCookiesButtonsEnabled =
-    blockedCookiesButtonsEnabled_;
-
 @synthesize allowedTreeController = allowedTreeController_;
 @synthesize blockedTreeController = blockedTreeController_;
 @synthesize allowedOutlineView = allowedOutlineView_;
 @synthesize blockedOutlineView = blockedOutlineView_;
 @synthesize infoBar = infoBar_;
 @synthesize infoBarIcon = infoBarIcon_;
 @synthesize infoBarText = infoBarText_;
 @synthesize tabView = tabView_;
 @synthesize blockedScrollView = blockedScrollView_;
 @synthesize blockedCookiesText = blockedCookiesText_;
 @synthesize cookieDetailsViewPlaceholder = cookieDetailsViewPlaceholder_;
 
+@synthesize allowedCookiesButtonsEnabled =
+    allowedCookiesButtonsEnabled_;
+@synthesize blockedCookiesButtonsEnabled =
+    blockedCookiesButtonsEnabled_;
+@synthesize deleteCookiesButtonEnabled = deleteCookiesButtonEnabled_;
+
 - (id)initWithWebContents:(content::WebContents*)webContents
       collectedCookiesMac:(CollectedCookiesMac*)collectedCookiesMac {
   DCHECK(webContents);
 
   NSString* nibpath =
       [base::mac::FrameworkBundle() pathForResource:@"CollectedCookies"
                                              ofType:@"nib"];
   if ((self = [super initWithWindowNibPath:nibpath owner:self])) {
     webContents_ = webContents;
     collectedCookiesMac_ = collectedCookiesMac;
@@ skipping 100 matching lines @@
     lastDomain = host_node->GetTitle();
   }
   if (multipleDomainsChanged)
     [self showInfoBarForMultipleDomainsAndSetting:setting];
   else
     [self showInfoBarForDomain:lastDomain setting:setting];
   contentSettingsChanged_ = YES;
 }
 
 - (IBAction)allowOrigin:(id)sender {
-  [self    addException:CONTENT_SETTING_ALLOW
+  [self addException:CONTENT_SETTING_ALLOW
       forTreeController:blockedTreeController_];
 }
 
 - (IBAction)allowForSessionFromOrigin:(id)sender {
-  [self    addException:CONTENT_SETTING_SESSION_ONLY
+  [self addException:CONTENT_SETTING_SESSION_ONLY
       forTreeController:blockedTreeController_];
 }
 
 - (IBAction)blockOrigin:(id)sender {
-  [self    addException:CONTENT_SETTING_BLOCK
+  [self addException:CONTENT_SETTING_BLOCK
       forTreeController:allowedTreeController_];
 }
 
+- (IBAction)deleteSelected:(id)sender {
+  NSArray* nodes = [allowedTreeController_ selectedNodes];
+  for (NSTreeNode* cocoaTreeNode in nodes) {
+    CookieTreeNode* cookieNode = static_cast<CookieTreeNode*>(
+        [[cocoaTreeNode representedObject] treeNode]);
+    allowedTreeModel_->DeleteCookieNode(cookieNode);
+  }
+}
+
 - (CocoaCookieTreeNode*)cocoaAllowedTreeModel {
-  return cocoaAllowedTreeModel_.get();
-}
-- (void)setCocoaAllowedTreeModel:(CocoaCookieTreeNode*)model {
-  cocoaAllowedTreeModel_.reset([model retain]);
+  return allowedControllerBridge_->cocoa_model();
 }
 
 - (CookiesTreeModel*)allowedTreeModel {
   return allowedTreeModel_.get();
 }
 
 - (CocoaCookieTreeNode*)cocoaBlockedTreeModel {
-  return cocoaBlockedTreeModel_.get();
-}
-- (void)setCocoaBlockedTreeModel:(CocoaCookieTreeNode*)model {
-  cocoaBlockedTreeModel_.reset([model retain]);
+  return blockedControllerBridge_->cocoa_model();
 }
 
 - (CookiesTreeModel*)blockedTreeModel {
   return blockedTreeModel_.get();
 }
 
 - (void)outlineView:(NSOutlineView*)outlineView
     willDisplayCell:(id)cell
      forTableColumn:(NSTableColumn*)tableColumn
                item:(id)item {
   CocoaCookieTreeNode* node = [item representedObject];
   int index;
   if (outlineView == allowedOutlineView_)
     index = allowedTreeModel_->GetIconIndex([node treeNode]);
   else
     index = blockedTreeModel_->GetIconIndex([node treeNode]);
   NSImage* icon = nil;
   if (index >= 0)
     icon = [icons_ objectAtIndex:index];
   else
     icon = [icons_ lastObject];
   DCHECK([cell isKindOfClass:[ImageAndTextCell class]]);
   [static_cast<ImageAndTextCell*>(cell) setImage:icon];
 }
 
 - (void)outlineViewSelectionDidChange:(NSNotification*)notif {
   BOOL isAllowedOutlineView;
   if ([notif object] == allowedOutlineView_) {
     isAllowedOutlineView = YES;
+    [self setDeleteCookiesButtonEnabled:YES];
   } else if ([notif object] == blockedOutlineView_) {
     isAllowedOutlineView = NO;
+    [self setDeleteCookiesButtonEnabled:NO];
   } else {
     NOTREACHED();
     return;
   }
   NSTreeController* controller =
       isAllowedOutlineView ? allowedTreeController_ : blockedTreeController_;
 
   NSArray* nodes = [controller selectedNodes];
   for (NSTreeNode* treeNode in nodes) {
     CocoaCookieTreeNode* node = [treeNode representedObject];
@@ skipping 43 matching lines @@
     [icons_ addObject:gfx::NSImageFromImageSkia(*it)];
   }
 
   // Default icon will be the last item in the array.
   ResourceBundle& rb = ResourceBundle::GetSharedInstance();
   // TODO(rsesek): Rename this resource now that it's in multiple places.
   [icons_ addObject:
       rb.GetNativeImageNamed(IDR_BOOKMARK_BAR_FOLDER).ToNSImage()];
 
   // Create the Cocoa model.
-  CookieTreeNode* root =
-      static_cast<CookieTreeNode*>(allowedTreeModel_->GetRoot());
-  base::scoped_nsobject<CocoaCookieTreeNode> model(
-      [[CocoaCookieTreeNode alloc] initWithNode:root]);
-  [self setCocoaAllowedTreeModel:model.get()];  // Takes ownership.
-  root = static_cast<CookieTreeNode*>(blockedTreeModel_->GetRoot());
-  model.reset(
-      [[CocoaCookieTreeNode alloc] initWithNode:root]);
-  [self setCocoaBlockedTreeModel:model.get()];  // Takes ownership.
+  allowedControllerBridge_.reset(
+      new CookiesTreeControllerBridge(allowedTreeModel_.get()));
+  blockedControllerBridge_.reset(
+      new CookiesTreeControllerBridge(blockedTreeModel_.get()));
 }
 
--(void)showInfoBarForMultipleDomainsAndSetting:(ContentSetting)setting {
+- (void)showInfoBarForMultipleDomainsAndSetting:(ContentSetting)setting {
   NSString* label;
   switch (setting) {
     case CONTENT_SETTING_BLOCK:
       label = l10n_util::GetNSString(
           IDS_COLLECTED_COOKIES_MULTIPLE_BLOCK_RULES_CREATED);
       break;
 
     case CONTENT_SETTING_ALLOW:
       label = l10n_util::GetNSString(
           IDS_COLLECTED_COOKIES_MULTIPLE_ALLOW_RULES_CREATED);
       break;
 
     case CONTENT_SETTING_SESSION_ONLY:
       label = l10n_util::GetNSString(
           IDS_COLLECTED_COOKIES_MULTIPLE_SESSION_RULES_CREATED);
       break;
 
     default:
       NOTREACHED();
       label = [[[NSString alloc] init] autorelease];
   }
   [infoBarText_ setStringValue:label];
   [self animateInfoBar];
 }
 
--(void)showInfoBarForDomain:(const base::string16&)domain
-                    setting:(ContentSetting)setting {
+- (void)showInfoBarForDomain:(const base::string16&)domain
+                     setting:(ContentSetting)setting {
   NSString* label;
   switch (setting) {
     case CONTENT_SETTING_BLOCK:
       label = l10n_util::GetNSStringF(
           IDS_COLLECTED_COOKIES_BLOCK_RULE_CREATED,
           domain);
       break;
 
     case CONTENT_SETTING_ALLOW:
       label = l10n_util::GetNSStringF(
           IDS_COLLECTED_COOKIES_ALLOW_RULE_CREATED,
           domain);
       break;
 
     case CONTENT_SETTING_SESSION_ONLY:
       label = l10n_util::GetNSStringF(
           IDS_COLLECTED_COOKIES_SESSION_RULE_CREATED,
           domain);
       break;
 
     default:
       NOTREACHED();
       label = [[[NSString alloc] init] autorelease];
   }
   [infoBarText_ setStringValue:label];
   [self animateInfoBar];
 }
 
--(void)animateInfoBar {
+- (void)animateInfoBar {
   if (infoBarVisible_)
     return;
 
   infoBarVisible_ = YES;
 
-  NSMutableArray* animations = [NSMutableArray arrayWithCapacity:3];
-
   NSWindow* sheet = [self window];
   NSRect sheetFrame = [sheet frame];
   NSRect infoBarFrame = [infoBar_ frame];
   NSRect tabViewFrame = [tabView_ frame];
 
   // Calculate the end position of the info bar and set it to its start
   // position.
   infoBarFrame.origin.y = NSHeight(sheetFrame);
   infoBarFrame.size.width = NSWidth(sheetFrame);
   [infoBar_ setFrame:infoBarFrame];
   [[[self window] contentView] addSubview:infoBar_];
 
   // Calculate the new position of the sheet.
   sheetFrame.origin.y -= NSHeight(infoBarFrame);
   sheetFrame.size.height += NSHeight(infoBarFrame);
 
-  // Slide the infobar in.
-  [animations addObject:
-      [NSDictionary dictionaryWithObjectsAndKeys:
-          infoBar_, NSViewAnimationTargetKey,
-          [NSValue valueWithRect:infoBarFrame],
-              NSViewAnimationEndFrameKey,
-          nil]];
-  // Make sure the tab view ends up in the right position.
-  [animations addObject:
-      [NSDictionary dictionaryWithObjectsAndKeys:
-          tabView_, NSViewAnimationTargetKey,
-          [NSValue valueWithRect:tabViewFrame],
-              NSViewAnimationEndFrameKey,
-          nil]];
+  NSArray* animations = @[
+    // Slide the infobar in.
+    @{
+      NSViewAnimationTargetKey : infoBar_,
+      NSViewAnimationEndFrameKey : [NSValue valueWithRect:infoBarFrame]
+    },
+    // Make sure the tab view ends up in the right position.
+    @{
+      NSViewAnimationTargetKey : tabView_,
+      NSViewAnimationEndFrameKey : [NSValue valueWithRect:tabViewFrame]
+    },
+    // Grow the sheet.
+    @{
+      NSViewAnimationTargetKey : sheet,
+      NSViewAnimationEndFrameKey : [NSValue valueWithRect:sheetFrame]
+    }
+  ];
 
-  // Grow the sheet.
-  [animations addObject:
-      [NSDictionary dictionaryWithObjectsAndKeys:
-          sheet, NSViewAnimationTargetKey,
-          [NSValue valueWithRect:sheetFrame],
-              NSViewAnimationEndFrameKey,
-          nil]];
   [animation_ setViewAnimations:animations];
   // The default duration is 0.5s, which actually feels slow in here, so speed
   // it up a bit.
-  [animation_ gtm_setDuration:0.2
-                    eventMask:NSLeftMouseUpMask];
+  [animation_ gtm_setDuration:0.2 eventMask:NSLeftMouseUpMask];
   [animation_ startAnimation];
 }
 
-- (void)         tabView:(NSTabView*)tabView
+- (void)tabView:(NSTabView*)tabView
     didSelectTabViewItem:(NSTabViewItem*)tabViewItem {
   NSTreeController* treeController = nil;
   switch ([tabView indexOfTabViewItem:tabViewItem]) {
     case kAllowedCookiesTabIndex:
       treeController = allowedTreeController_;
       break;
     case kBlockedCookiesTabIndex:
       treeController = blockedTreeController_;
       break;
     default:
       NOTREACHED();
       return;
   }
   [detailsViewController_ configureBindingsForTreeController:treeController];
 }
 
 @end