Update error badge on wrench menu button

chrome/browser/ui/cocoa/toolbar/toolbar_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.
 
 #import "chrome/browser/ui/cocoa/toolbar/toolbar_controller.h"
 
 #include <algorithm>
 
 #include "base/mac/mac_util.h"
 #include "base/memory/singleton.h"
@@ skipping 63 matching lines @@
 const CGFloat kMinimumLocationBarWidth = 100.0;
 
 // The duration of any animation that occurs within the toolbar in seconds.
 const CGFloat kAnimationDuration = 0.2;
 
 // The amount of left padding that the wrench menu should have.
 const CGFloat kWrenchMenuLeftPadding = 3.0;
 
 }  // namespace
 
-@interface ToolbarController(Private)
+@interface ToolbarController()
+@property(assign, nonatomic) Browser* browser;
 - (void)addAccessibilityDescriptions;
 - (void)initCommandStatus:(CommandUpdater*)commands;
 - (void)prefChanged:(std::string*)prefName;
 - (BackgroundGradientView*)backgroundGradientView;
 - (void)toolbarFrameChanged;
 - (void)pinLocationBarToLeftOfBrowserActionsContainerAndAnimate:(BOOL)animate;
 - (void)maintainMinimumLocationBarWidth;
 - (void)adjustBrowserActionsContainerForNewWindow:(NSNotification*)notification;
 - (void)browserActionsContainerDragged:(NSNotification*)notification;
 - (void)browserActionsContainerDragFinished:(NSNotification*)notification;
@@ skipping 26 matching lines @@
 
 // A class registered for C++ notifications. This is used to detect changes in
 // preferences and upgrade available notifications. Bridges the notification
 // back to the ToolbarController.
 class NotificationBridge : public NotificationObserver {
  public:
   explicit NotificationBridge(ToolbarController* controller)
       : controller_(controller) {
     registrar_.Add(this, chrome::NOTIFICATION_UPGRADE_RECOMMENDED,
                    NotificationService::AllSources());
+    registrar_.Add(this, chrome::NOTIFICATION_GLOBAL_ERRORS_CHANGED,
+                   Source<Profile>([controller browser]->profile()));
   }
 
   // Overridden from NotificationObserver:
   virtual void Observe(int type,
                        const NotificationSource& source,
                        const NotificationDetails& details) {
     switch (type) {
       case chrome::NOTIFICATION_PREF_CHANGED:
         [controller_ prefChanged:Details<std::string>(details).ptr()];
         break;
       case chrome::NOTIFICATION_UPGRADE_RECOMMENDED:
+      case chrome::NOTIFICATION_GLOBAL_ERRORS_CHANGED:
         [controller_ badgeWrenchMenuIfNeeded];
         break;
       default:
         NOTREACHED();
     }
   }
 
  private:
   ToolbarController* controller_;  // weak, owns us
 
   NotificationRegistrar registrar_;
 };
 
 }  // namespace ToolbarControllerInternal
 
 @implementation ToolbarController
 
+@synthesize browser = browser_;
+
 - (id)initWithModel:(ToolbarModel*)model
            commands:(CommandUpdater*)commands
             profile:(Profile*)profile
             browser:(Browser*)browser
      resizeDelegate:(id<ViewResizer>)resizeDelegate
        nibFileNamed:(NSString*)nibName {
   DCHECK(model && commands && profile && [nibName length]);
   if ((self = [super initWithNibName:nibName
                               bundle:base::mac::MainAppBundle()])) {
     toolbarModel_ = model;
@@ skipping 636 matching lines @@
 - (void)hideDropURLsIndicatorInView:(NSView*)view {
   // Do nothing.
 }
 
 // (URLDropTargetController protocol)
 - (BOOL)isUnsupportedDropData:(id<NSDraggingInfo>)info {
   return drag_util::IsUnsupportedDropData(profile_, info);
 }
 
 @end