[OSX, OmniTheatre] Handle OriginChip click properly.

chrome/browser/ui/cocoa/location_bar/autocomplete_text_field_cell.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/location_bar/autocomplete_text_field_cell.h"
 
 #include "base/logging.h"
 #include "base/mac/foundation_util.h"
 #include "base/mac/mac_logging.h"
 #include "chrome/browser/search/search.h"
@@ skipping 147 matching lines @@
   std::reverse(decorations->begin() + left_count, decorations->end());
   std::reverse(decoration_frames->begin() + left_count,
                decoration_frames->end());
 
   *remaining_frame = frame;
   return left_count;
 }
 
 }  // namespace
 
+@interface AutocompleteTextFieldCell ()
+// Post an OnSetFocus notification to the observer of |controlView|.
+- (void)focusNotificationFor:(NSEvent*)event
+                      ofView:(AutocompleteTextField*)controlView;
+@end
+
 @implementation AutocompleteTextFieldCell
 
 @synthesize isPopupMode = isPopupMode_;
 
 - (CGFloat)topTextFrameOffset {
   return 3.0;
 }
 
 - (CGFloat)bottomTextFrameOffset {
   return 3.0;
@@ skipping 226 matching lines @@
   LocationBarDecoration* decoration =
       [self decorationForEvent:theEvent inRect:cellFrame ofView:controlView];
   if (decoration)
     return decoration->GetMenu();
   return nil;
 }
 
 - (BOOL)mouseDown:(NSEvent*)theEvent
            inRect:(NSRect)cellFrame
            ofView:(AutocompleteTextField*)controlView {
+  // TODO(groby): Factor this into three pieces - find target for event, handle
+  // delayed focus (for any and all events), execute mouseDown for target.
+
+  // Check if this mouseDown was the reason the control became firstResponder.
+  // If not, discard focus event.
+  base::scoped_nsobject<NSEvent> focusEvent(focusEvent_.release());
+  if (![theEvent isEqual:focusEvent])
+    focusEvent.reset();
+
   LocationBarDecoration* decoration =
       [self decorationForEvent:theEvent inRect:cellFrame ofView:controlView];
   if (!decoration || !decoration->AcceptsMousePress())
     return NO;
 
   NSRect decorationRect =
       [self frameForDecoration:decoration inFrame:cellFrame];
 
   // If the decoration is draggable, then initiate a drag if the user
   // drags or holds the mouse down for awhile.
@@ skipping 50 matching lines @@
 
     button->SetButtonState(ButtonDecoration::kButtonStatePressed);
     [controlView setNeedsDisplay:YES];
 
     // Track the mouse until the user releases the button.
     [self trackMouse:theEvent
               inRect:cellFrame
               ofView:controlView
         untilMouseUp:YES];
 
+    // Post delayed focus notification, if necessary.
+    if (focusEvent.get())
+      [self focusNotificationFor:focusEvent ofView:controlView];
+
     // Set the proper state (hover or normal) once the mouse has been released,
     // and call |OnMousePressed| if the button was released while the mouse was
     // within the bounds of the button.
     const NSPoint mouseLocation =
         [[controlView window] mouseLocationOutsideOfEventStream];
     const NSPoint point = [controlView convertPoint:mouseLocation fromView:nil];
     if (NSMouseInRect(point, cellFrame, [controlView isFlipped])) {
       button->SetButtonState(ButtonDecoration::kButtonStateHover);
       [controlView setNeedsDisplay:YES];
       handled = decoration->AsButtonDecoration()->OnMousePressed(
@@ skipping 306 matching lines @@
   [controlView setNeedsDisplay:YES];
   NSTextView* fieldEditor =
       base::mac::ObjCCastStrict<NSTextView>([controlView currentEditor]);
   [fieldEditor updateInsertionPointStateAndRestartTimer:YES];
 }
 
 - (BOOL)showsFirstResponder {
   return [super showsFirstResponder] && !hideFocusState_;
 }
 
+- (void)focusNotificationFor:(NSEvent*)event
+                      ofView:(AutocompleteTextField*)controlView {
+  if ([controlView observer]) {
+    const bool controlDown = ([event modifierFlags] & NSControlKeyMask) != 0;
+    [controlView observer]->OnSetFocus(controlDown);
+  }
+}
+
+- (void)handleFocusEvent:(NSEvent*)event
+                  ofView:(AutocompleteTextField*)controlView {
+  // Only intercept left button click. All other events cause immediate focus.
+  if ([event type] == NSLeftMouseDown) {
+    LocationBarDecoration* decoration =
+        [self decorationForEvent:event
+                          inRect:[controlView bounds]
+                          ofView:controlView];
+    // Only ButtonDecorations need a delayed focus handling.
+    if (decoration && decoration->AsButtonDecoration()) {
+      focusEvent_.reset([event retain]);
+      return;
+    }
+  }
+
+  // Handle event immediately.
+  [self focusNotificationFor:event ofView:controlView];
+}
+
 @end