[Mac] Implements context menus for Page Actions. Introduces a reusable subcla...

chrome/browser/cocoa/extensions/browser_actions_controller.mm

 // Copyright (c) 2009 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 "browser_actions_controller.h"
 
 #include <string>
 
 #include "app/gfx/canvas_paint.h"
 #include "base/sys_string_conversions.h"
 #include "chrome/browser/browser.h"
+#include "chrome/browser/cocoa/extensions/extension_action_context_menu.h"
 #include "chrome/browser/cocoa/extensions/extension_popup_controller.h"
 #include "chrome/browser/cocoa/toolbar_button_cell.h"
 #include "chrome/browser/extensions/extension_browser_event_router.h"
 #include "chrome/browser/extensions/extensions_service.h"
 #include "chrome/browser/extensions/extension_tabs_module.h"
 #include "chrome/browser/extensions/image_loading_tracker.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/tab_contents/tab_contents.h"
 #include "chrome/common/notification_observer.h"
 #include "chrome/common/notification_registrar.h"
 #include "skia/ext/skia_utils_mac.h"
 
-static const CGFloat kBrowserActionBadgeOriginYOffset = -4;
+static const CGFloat kBrowserActionBadgeOriginYOffset = 5;
 
 // Since the container is the maximum height of the toolbar, we have to move the
 // buttons up by this amount in order to have them look vertically centered
 // within the toolbar.
 static const CGFloat kBrowserActionOriginYOffset = 5;
 
 // The size of each button on the toolbar.
 static const CGFloat kBrowserActionHeight = 27;
 extern const CGFloat kBrowserActionWidth = 29;
 
 // The padding between browser action buttons.
 extern const CGFloat kBrowserActionButtonPadding = 3;
 
 NSString* const kBrowserActionsChangedNotification = @"BrowserActionsChanged";
 
-@interface BrowserActionBadgeView : NSView {
+@interface BrowserActionCell : ToolbarButtonCell {
  @private
-  // The current tab ID used when drawing the badge.
+  // The current tab ID used when drawing the cell.
   int tabId_;
 
-  // The action we're drawing the badge for. Weak.
+  // The action we're drawing the cell for. Weak.
   ExtensionAction* extensionAction_;
 }
 
 @property(readwrite, nonatomic) int tabId;
 @property(readwrite, nonatomic) ExtensionAction* extensionAction;
 
 @end
 
-@implementation BrowserActionBadgeView
+@implementation BrowserActionCell
 
-- (void)drawRect:(NSRect)dirtyRect {
+- (void)drawWithFrame:(NSRect)cellFrame inView:(NSView*)controlView {
+  [super drawWithFrame:cellFrame inView:controlView];
+
   // CanvasPaint draws its content to the current NSGraphicsContext in its
   // destructor. If anything needs to be drawn afterwards, then enclose this
   // in a nested block.
-  NSRect badgeBounds = [self bounds];
-  badgeBounds.origin.y += kBrowserActionBadgeOriginYOffset;
-  gfx::CanvasPaint canvas(badgeBounds, false);
+  cellFrame.origin.y += kBrowserActionBadgeOriginYOffset;
+  gfx::CanvasPaint canvas(cellFrame, false);
   canvas.set_composite_alpha(true);
-  gfx::Rect boundingRect(NSRectToCGRect(badgeBounds));
+  gfx::Rect boundingRect(NSRectToCGRect(cellFrame));
   extensionAction_->PaintBadge(&canvas, boundingRect, tabId_);
 }
 
 @synthesize tabId = tabId_;
 @synthesize extensionAction = extensionAction_;
 
 @end
 
 class ExtensionImageTrackerBridge;
 
 @interface BrowserActionButton : NSButton {
  @private
   scoped_ptr<ExtensionImageTrackerBridge> imageLoadingBridge_;
 
   scoped_nsobject<NSImage> defaultIcon_;
 
   scoped_nsobject<NSImage> tabSpecificIcon_;
 
-  scoped_nsobject<NSView> badgeView_;
-
   // The extension for this button. Weak.
   Extension* extension_;
 
   // Weak. Owns us.
   BrowserActionsController* controller_;
 }
 
 - (id)initWithExtension:(Extension*)extension
              controller:(BrowserActionsController*)controller
                 xOffset:(int)xOffset;
@@ skipping 60 matching lines @@
   ImageLoadingTracker* tracker_;
 
   // Used for registering to receive notifications and automatic clean up.
   NotificationRegistrar registrar_;
 
   DISALLOW_COPY_AND_ASSIGN(ExtensionImageTrackerBridge);
 };
 
 @implementation BrowserActionButton
 
++ (Class)cellClass {
+  return [BrowserActionCell class];
+}
+
 - (id)initWithExtension:(Extension*)extension
              controller:(BrowserActionsController*)controller
                 xOffset:(int)xOffset {
   NSRect frame = NSMakeRect(xOffset,
                             kBrowserActionOriginYOffset,
                             kBrowserActionWidth,
                             kBrowserActionHeight);
   if ((self = [super initWithFrame:frame])) {
-    ToolbarButtonCell* cell = [[[ToolbarButtonCell alloc] init] autorelease];
+    BrowserActionCell* cell = [[[BrowserActionCell alloc] init] autorelease];
     // [NSButton setCell:] warns to NOT use setCell: other than in the
     // initializer of a control.  However, we are using a basic
     // NSButton whose initializer does not take an NSCell as an
     // object.  To honor the assumed semantics, we do nothing with
     // NSButton between alloc/init and setCell:.
     [self setCell:cell];
+    [cell setTabId:[controller currentTabId]];
+    [cell setExtensionAction:extension->browser_action()];
+
     [self setTitle:@""];
     [self setButtonType:NSMomentaryChangeButton];
     [self setShowsBorderOnlyWhileMouseInside:YES];
 
     [self setTarget:controller];
     [self setAction:@selector(browserActionClicked:)];
 
+    [self setMenu:[[[ExtensionActionContextMenu alloc]
+        initWithExtension:extension] autorelease]];
+
     extension_ = extension;
     controller_ = controller;
     imageLoadingBridge_.reset(new ExtensionImageTrackerBridge(self, extension));
 
-    NSRect badgeFrame = [self bounds];
-    badgeView_.reset([[BrowserActionBadgeView alloc] initWithFrame:badgeFrame]);
-    [badgeView_ setTabId:[controller currentTabId]];
-    [badgeView_ setExtensionAction:extension->browser_action()];
-    [self addSubview:badgeView_];
-
     [self updateState];
   }
 
   return self;
 }
 
 - (void)setDefaultIcon:(NSImage*)image {
   defaultIcon_.reset([image retain]);
 }
 
@@ skipping 14 matching lines @@
   }
 
   SkBitmap image = extension_->browser_action()->GetIcon(tabId);
   if (!image.isNull()) {
     [self setTabSpecificIcon:gfx::SkBitmapToNSImage(image)];
     [self setImage:tabSpecificIcon_];
   } else if (defaultIcon_) {
     [self setImage:defaultIcon_];
   }
 
-  [badgeView_ setTabId:tabId];
+  [[self cell] setTabId:tabId];
 
   [self setNeedsDisplay:YES];
 }
 
 @synthesize extension = extension_;
 
 @end
 
 @interface BrowserActionsController(Private)
 
@@ skipping 221 matching lines @@
     return -1;
 
   return selected_tab->controller().session_id().id();
 }
 
 - (NSButton*)buttonWithIndex:(int)index {
   return [buttonOrder_ objectAtIndex:(NSUInteger)index];
 }
 
 @end