blob: ef247cc8e9a63b686ba069c7a269a82161c77736 [file] [log] [blame]
// 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 "ui/base/cocoa/focus_tracker.h"
#include "base/basictypes.h"
@implementation FocusTracker
- (id)initWithWindow:(NSWindow*)window {
if ((self = [super init])) {
NSResponder* current_focus = [window firstResponder];
// Special case NSTextViews, because they are removed from the
// view hierarchy when their text field does not have focus. If
// an NSTextView is the current first responder, save a pointer to
// its NSTextField delegate instead.
if ([current_focus isKindOfClass:[NSTextView class]]) {
id delegate = [(NSTextView*)current_focus delegate];
if ([delegate isKindOfClass:[NSTextField class]])
current_focus = delegate;
else
current_focus = nil;
}
if ([current_focus isKindOfClass:[NSView class]]) {
NSView* current_focus_view = (NSView*)current_focus;
focusedView_.reset([current_focus_view retain]);
}
}
return self;
}
- (BOOL)restoreFocusInWindow:(NSWindow*)window {
if (!focusedView_.get())
return NO;
if ([focusedView_ window] && [focusedView_ window] == window)
return [window makeFirstResponder:focusedView_.get()];
return NO;
}
@end