Make a platform-independent ToolbarActionViewController

chrome/browser/ui/cocoa/extensions/browser_action_button.h

 // 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.
 
 #ifndef CHROME_BROWSER_UI_COCOA_EXTENSIONS_BROWSER_ACTION_BUTTON_H_
 #define CHROME_BROWSER_UI_COCOA_EXTENSIONS_BROWSER_ACTION_BUTTON_H_
 
 #import <Cocoa/Cocoa.h>
 
+#include <string>
+
 #import "base/mac/scoped_nsobject.h"
 #include "base/memory/scoped_ptr.h"
 #import "chrome/browser/ui/cocoa/image_button_cell.h"
 
 class Browser;
-class ExtensionAction;
+@class BrowserActionsController;
 @class ExtensionActionContextMenuController;
-class ExtensionActionIconFactoryBridge;
+class ToolbarActionViewController;
+class ToolbarActionViewDelegateBridge;
 
-namespace extensions {
-class Extension;
+namespace content {
+class WebContents;
 }
 
 // Fired on each drag event while the user is moving the button.
 extern NSString* const kBrowserActionButtonDraggingNotification;
 // Fired when the user drops the button.
 extern NSString* const kBrowserActionButtonDragEndNotification;
 
 @interface BrowserActionButton : NSButton<NSMenuDelegate> {
  @private
-  // Bridge to proxy Chrome notifications to the Obj-C class as well as load the
-  // extension's icon.
-  scoped_ptr<ExtensionActionIconFactoryBridge> iconFactoryBridge_;
-
   // Used to move the button and query whether a button is currently animating.
   base::scoped_nsobject<NSViewAnimation> moveAnimation_;
 
-  // The extension for this button. Weak.
-  const extensions::Extension* extension_;
+  // The bridge between the view controller and this object.
+  scoped_ptr<ToolbarActionViewDelegateBridge> viewControllerDelegate_;
 
-  // The ID of the active tab.
-  int tabId_;
+  // The controller that handles most non-view logic.
+  scoped_ptr<ToolbarActionViewController> viewController_;
+
+  // The current tab's web contents.
+  content::WebContents* webContents_;
 
   // Whether the button is currently being dragged.
   BOOL isBeingDragged_;
 
   // Drag events could be intercepted by other buttons, so to make sure that
   // this is the only button moving if it ends up being dragged. This is set to
   // YES upon |mouseDown:|.
   BOOL dragCouldStart_;
 
   // The point where the mouse down event occurred. Used to prevent a drag from
   // starting until it moves at least kMinimumDragDistance.
   NSPoint dragStartPoint_;
 
   base::scoped_nsobject<
       ExtensionActionContextMenuController> contextMenuController_;
 }
 
 - (id)initWithFrame:(NSRect)frame
-          extension:(const extensions::Extension*)extension
-            browser:(Browser*)browser
-              tabId:(int)tabId;
+     viewController:(scoped_ptr<ToolbarActionViewController>)viewController
+        webContents:(content::WebContents*)webContents
+         controller:(BrowserActionsController*)controller
+     menuController:(ExtensionActionContextMenuController*)menuController;
 
 - (void)setFrame:(NSRect)frameRect animate:(BOOL)animate;
 
 - (void)updateState;
 
 - (BOOL)isAnimating;
 
+- (ToolbarActionViewController*)viewController;
+
 // Returns a pointer to an autoreleased NSImage with the badge, shadow and
 // cell image drawn into it.
 - (NSImage*)compositedImage;
 
 @property(readonly, nonatomic) BOOL isBeingDragged;
-@property(readonly, nonatomic) const extensions::Extension* extension;
-@property(readwrite, nonatomic) int tabId;
+@property(readwrite, nonatomic) content::WebContents* webContents;
 
 @end
 
 @interface BrowserActionCell : ImageButtonCell {
  @private
-  // The current tab ID used when drawing the cell.
-  int tabId_;
+  // The current tab's web contents used when drawing the cell.
+  content::WebContents* webContents_;
 
-  // The action we're drawing the cell for. Weak.
-  ExtensionAction* extensionAction_;
+  // The view controller for the parent button. Weak.
+  ToolbarActionViewController* viewController_;
 }
 
-@property(readwrite, nonatomic) int tabId;
-@property(readwrite, nonatomic) ExtensionAction* extensionAction;
+@property(readwrite, nonatomic) content::WebContents* webContents;
+@property(readwrite, nonatomic) ToolbarActionViewController* viewController;
 
 @end
 
 #endif  // CHROME_BROWSER_UI_COCOA_EXTENSIONS_BROWSER_ACTION_BUTTON_H_