Change InfoBar-related notifications to be sourced from a TabContentsWrapper, not a TabContents. ...

chrome/browser/ui/cocoa/infobars/infobar_container_controller.mm

 // Copyright (c) 2011 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 "base/logging.h"
 #include "base/mac/mac_util.h"
 #include "chrome/browser/tab_contents/confirm_infobar_delegate.h"
 #import "chrome/browser/ui/cocoa/animatable_view.h"
 #include "chrome/browser/ui/cocoa/infobars/infobar.h"
 #import "chrome/browser/ui/cocoa/infobars/infobar_container_controller.h"
@@ skipping 12 matching lines @@
       : controller_(controller) {
   }
 
  private:
   // NotificationObserver implementation
   void Observe(NotificationType type,
                const NotificationSource& source,
                const NotificationDetails& details) {
     switch (type.value) {
       case NotificationType::TAB_CONTENTS_INFOBAR_ADDED:
-        [controller_ addInfoBar:Details<InfoBarDelegate>(details).ptr()
-                        animate:YES];
+        [controller_ addInfoBar:Details<InfoBar>(details).ptr()
+                        animate:YES
+              deleteImmediately:NO];
         break;
-      case NotificationType::TAB_CONTENTS_INFOBAR_REMOVED:
+
+      case NotificationType::TAB_CONTENTS_INFOBAR_REMOVED: {
+        typedef std::pair<InfoBarDelegate*, bool> RemoveDetails;
+        RemoveDetails* remove_details = Details<RemoveDetails>(details).ptr();
         [controller_
-          closeInfoBarsForDelegate:Details<InfoBarDelegate>(details).ptr()
-                           animate:YES];
-        break;
-      case NotificationType::TAB_CONTENTS_INFOBAR_REPLACED: {
-        typedef std::pair<InfoBarDelegate*, InfoBarDelegate*>
-            InfoBarDelegatePair;
-        InfoBarDelegatePair* delegates =
-            Details<InfoBarDelegatePair>(details).ptr();
-        [controller_
-         replaceInfoBarsForDelegate:delegates->first with:delegates->second];
+            closeInfoBarsForDelegate:remove_details->first
+                             animate:(remove_details->second ? YES : NO)];
         break;
       }
+
+      case NotificationType::TAB_CONTENTS_INFOBAR_REPLACED: {
+        typedef std::pair<InfoBarDelegate*, InfoBar*> ReplaceDetails;
+        ReplaceDetails* replace_details =
+            Details<ReplaceDetails>(details).ptr();
+        [controller_ closeInfoBarsForDelegate:replace_details->first
+                                      animate:NO];
+        [controller_ addInfoBar:replace_details->second
+                        animate:NO
+              deleteImmediately:NO];
+        break;
+      }
+
       default:
         NOTREACHED();  // we don't ask for anything else!
         break;
     }
 
     [controller_ positionInfoBarsAndRedraw];
   }
 
   InfoBarContainerController* controller_;  // weak, owns us.
 };
@@ skipping 60 matching lines @@
   [self positionInfoBarsAndRedraw];
 }
 
 - (void)changeTabContents:(TabContentsWrapper*)contents {
   registrar_.RemoveAll();
   [self removeAllInfoBars];
 
   currentTabContents_ = contents;
   if (currentTabContents_) {
     for (size_t i = 0; i < currentTabContents_->infobar_count(); ++i) {
-      [self addInfoBar:currentTabContents_->GetInfoBarDelegateAt(i)
-               animate:NO];
+      InfoBar* infobar = currentTabContents_->GetInfoBarDelegateAt(i)->
+          CreateInfoBar(currentTabContents_);
+      [self addInfoBar:infobar animate:NO deleteImmediately:YES];
     }
 
-    Source<TabContents> source(currentTabContents_->tab_contents());
+    Source<TabContentsWrapper> source(currentTabContents_);
     registrar_.Add(infoBarObserver_.get(),
                    NotificationType::TAB_CONTENTS_INFOBAR_ADDED, source);
     registrar_.Add(infoBarObserver_.get(),
                    NotificationType::TAB_CONTENTS_INFOBAR_REMOVED, source);
     registrar_.Add(infoBarObserver_.get(),
                    NotificationType::TAB_CONTENTS_INFOBAR_REPLACED, source);
   }
 
   [self positionInfoBarsAndRedraw];
 }
@@ skipping 27 matching lines @@
 
 @implementation InfoBarContainerController (PrivateMethods)
 
 - (CGFloat)desiredHeight {
   CGFloat height = 0;
   for (InfoBarController* controller in infobarControllers_.get())
     height += NSHeight([[controller view] frame]);
   return height;
 }
 
-- (void)addInfoBar:(InfoBarDelegate*)delegate animate:(BOOL)animate {
-  scoped_ptr<InfoBar> infobar(delegate->CreateInfoBar(currentTabContents_));
+- (void)addInfoBar:(InfoBar*)infobar
+    animate:(BOOL)animate
+    deleteImmediately:(BOOL)deleteImmediately {
   InfoBarController* controller = infobar->controller();
   [controller setContainerController:self];
   [[controller animatableView] setResizeDelegate:self];
   [[self view] addSubview:[controller view]];
   [infobarControllers_ addObject:[controller autorelease]];
 
   if (animate)
     [controller animateOpen];
   else
     [controller open];
+
+  if (deleteImmediately)
+    delete infobar;
+  else
+    MessageLoop::current()->DeleteSoon(FROM_HERE, infobar);
 }
 
 - (void)closeInfoBarsForDelegate:(InfoBarDelegate*)delegate
                          animate:(BOOL)animate {
   for (InfoBarController* controller in
        [NSArray arrayWithArray:infobarControllers_.get()]) {
     if ([controller delegate] == delegate) {
       [controller infobarWillClose];
       if (animate)
         [controller animateClosed];
       else
         [controller close];
     }
   }
 }
 
-- (void)replaceInfoBarsForDelegate:(InfoBarDelegate*)old_delegate
-                              with:(InfoBarDelegate*)new_delegate {
-  [self closeInfoBarsForDelegate:old_delegate animate:NO];
-  [self addInfoBar:new_delegate animate:NO];
-}
-
 - (void)removeAllInfoBars {
   // stopAnimation can remove the infobar from infobarControllers_ if it was in
   // the midst of closing, so copy the array so mutations won't cause problems.
   for (InfoBarController* controller in
        [NSArray arrayWithArray:infobarControllers_.get()]) {
     [[controller animatableView] stopAnimation];
     [[controller view] removeFromSuperview];
   }
   [infobarControllers_ removeAllObjects];
   [closingInfoBars_ removeAllObjects];
@@ skipping 16 matching lines @@
     frame.size.width = NSWidth(containerBounds);
     [view setFrame:frame];
 
     minY += NSHeight(frame);
   }
 
   [resizeDelegate_ resizeView:[self view] newHeight:[self desiredHeight]];
 }
 
 @end