OLD | NEW |
| (Empty) |
1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #import "chrome/browser/ui/cocoa/search_engine_dialog_controller.h" | |
6 | |
7 #include <algorithm> | |
8 | |
9 #include "base/bind.h" | |
10 #include "base/mac/bundle_locations.h" | |
11 #include "base/mac/mac_util.h" | |
12 #include "base/message_loop.h" | |
13 #include "base/sys_string_conversions.h" | |
14 #include "base/time.h" | |
15 #include "chrome/browser/profiles/profile.h" | |
16 #include "chrome/browser/search_engines/template_url.h" | |
17 #include "chrome/browser/search_engines/template_url_service.h" | |
18 #include "chrome/browser/search_engines/template_url_service_factory.h" | |
19 #include "chrome/browser/search_engines/template_url_service_observer.h" | |
20 #include "grit/generated_resources.h" | |
21 #include "grit/theme_resources.h" | |
22 #import "third_party/GTM/AppKit/GTMUILocalizerAndLayoutTweaker.h" | |
23 #include "ui/base/l10n/l10n_util_mac.h" | |
24 #include "ui/base/resource/resource_bundle.h" | |
25 #include "ui/gfx/image/image.h" | |
26 | |
27 // Horizontal spacing between search engine choices. | |
28 const int kSearchEngineSpacing = 20; | |
29 | |
30 // Vertical spacing between the search engine logo and the button underneath. | |
31 const int kLogoButtonSpacing = 10; | |
32 | |
33 // Width of a label used in place of a logo. | |
34 const int kLogoLabelWidth = 170; | |
35 | |
36 // Height of a label used in place of a logo. | |
37 const int kLogoLabelHeight = 25; | |
38 | |
39 @interface SearchEngineDialogController (Private) | |
40 - (void)onTemplateURLServiceChanged; | |
41 - (void)buildSearchEngineView; | |
42 - (NSView*)viewForSearchEngine:(const TemplateURL*)engine | |
43 atIndex:(size_t)index; | |
44 - (IBAction)searchEngineSelected:(id)sender; | |
45 @end | |
46 | |
47 class SearchEngineDialogControllerBridge : | |
48 public base::RefCounted<SearchEngineDialogControllerBridge>, | |
49 public TemplateURLServiceObserver { | |
50 public: | |
51 SearchEngineDialogControllerBridge(SearchEngineDialogController* controller); | |
52 | |
53 // TemplateURLServiceObserver | |
54 virtual void OnTemplateURLServiceChanged(); | |
55 | |
56 private: | |
57 SearchEngineDialogController* controller_; | |
58 }; | |
59 | |
60 SearchEngineDialogControllerBridge::SearchEngineDialogControllerBridge( | |
61 SearchEngineDialogController* controller) : controller_(controller) { | |
62 } | |
63 | |
64 void SearchEngineDialogControllerBridge::OnTemplateURLServiceChanged() { | |
65 [controller_ onTemplateURLServiceChanged]; | |
66 MessageLoop::current()->QuitNow(); | |
67 } | |
68 | |
69 @implementation SearchEngineDialogController | |
70 | |
71 @synthesize profile = profile_; | |
72 @synthesize randomize = randomize_; | |
73 | |
74 - (id)init { | |
75 NSString* nibpath = | |
76 [base::mac::FrameworkBundle() pathForResource:@"SearchEngineDialog" | |
77 ofType:@"nib"]; | |
78 self = [super initWithWindowNibPath:nibpath owner:self]; | |
79 if (self != nil) { | |
80 bridge_ = new SearchEngineDialogControllerBridge(self); | |
81 } | |
82 return self; | |
83 } | |
84 | |
85 - (void)dealloc { | |
86 [super dealloc]; | |
87 } | |
88 | |
89 - (IBAction)showWindow:(id)sender { | |
90 searchEnginesModel_ = TemplateURLServiceFactory::GetForProfile(profile_); | |
91 searchEnginesModel_->AddObserver(bridge_.get()); | |
92 | |
93 if (searchEnginesModel_->loaded()) { | |
94 MessageLoop::current()->PostTask(FROM_HERE, | |
95 base::Bind( | |
96 &SearchEngineDialogControllerBridge::OnTemplateURLServiceChanged, | |
97 bridge_.get())); | |
98 } else { | |
99 searchEnginesModel_->Load(); | |
100 } | |
101 MessageLoop::current()->Run(); | |
102 } | |
103 | |
104 - (void)onTemplateURLServiceChanged { | |
105 searchEnginesModel_->RemoveObserver(bridge_.get()); | |
106 | |
107 // Add the search engines in the search_engines_model_ to the buttons list. | |
108 // The first three will always be from prepopulated data. | |
109 std::vector<const TemplateURL*> templateUrls = | |
110 searchEnginesModel_->GetTemplateURLs(); | |
111 | |
112 // If we have fewer than two search engines, end the search engine dialog | |
113 // immediately, leaving the imported default search engine setting intact. | |
114 if (templateUrls.size() < 2) { | |
115 return; | |
116 } | |
117 | |
118 NSWindow* win = [self window]; | |
119 | |
120 [win setBackgroundColor:[NSColor whiteColor]]; | |
121 | |
122 NSImage* headerImage = ResourceBundle::GetSharedInstance(). | |
123 GetNativeImageNamed(IDR_SEARCH_ENGINE_DIALOG_TOP); | |
124 [headerImageView_ setImage:headerImage]; | |
125 | |
126 // Is the user's default search engine included in the first three | |
127 // prepopulated set? If not, we need to expand the dialog to include a fourth | |
128 // engine. | |
129 const TemplateURL* defaultSearchEngine = | |
130 searchEnginesModel_->GetDefaultSearchProvider(); | |
131 | |
132 std::vector<const TemplateURL*>::iterator engineIter = | |
133 templateUrls.begin(); | |
134 for (int i = 0; engineIter != templateUrls.end(); ++i, ++engineIter) { | |
135 if (i < 3) { | |
136 choices_.push_back(*engineIter); | |
137 } else { | |
138 if (*engineIter == defaultSearchEngine) | |
139 choices_.push_back(*engineIter); | |
140 } | |
141 } | |
142 | |
143 // Randomize the order of the logos if the option has been set. | |
144 if (randomize_) { | |
145 int seed = static_cast<int>(base::Time::Now().ToInternalValue()); | |
146 srand(seed); | |
147 std::random_shuffle(choices_.begin(), choices_.end()); | |
148 } | |
149 | |
150 [self buildSearchEngineView]; | |
151 | |
152 // Display the dialog. | |
153 NSInteger choice = [NSApp runModalForWindow:win]; | |
154 searchEnginesModel_->SetDefaultSearchProvider(choices_.at(choice)); | |
155 } | |
156 | |
157 - (void)buildSearchEngineView { | |
158 scoped_nsobject<NSMutableArray> searchEngineViews | |
159 ([[NSMutableArray alloc] init]); | |
160 | |
161 for (size_t i = 0; i < choices_.size(); ++i) | |
162 [searchEngineViews addObject:[self viewForSearchEngine:choices_.at(i) | |
163 atIndex:i]]; | |
164 | |
165 NSSize newOverallSize = NSZeroSize; | |
166 for (NSView* view in searchEngineViews.get()) { | |
167 NSRect engineFrame = [view frame]; | |
168 engineFrame.origin = NSMakePoint(newOverallSize.width, 0); | |
169 [searchEngineView_ addSubview:view]; | |
170 [view setFrame:engineFrame]; | |
171 newOverallSize = NSMakeSize( | |
172 newOverallSize.width + NSWidth(engineFrame) + kSearchEngineSpacing, | |
173 std::max(newOverallSize.height, NSHeight(engineFrame))); | |
174 } | |
175 newOverallSize.width -= kSearchEngineSpacing; | |
176 | |
177 // Resize the window to fit (and because it's bound on all sides it will | |
178 // resize the search engine view). | |
179 NSSize currentOverallSize = [searchEngineView_ bounds].size; | |
180 NSSize deltaSize = NSMakeSize( | |
181 newOverallSize.width - currentOverallSize.width, | |
182 newOverallSize.height - currentOverallSize.height); | |
183 NSSize windowDeltaSize = [searchEngineView_ convertSize:deltaSize toView:nil]; | |
184 NSRect windowFrame = [[self window] frame]; | |
185 windowFrame.size.width += windowDeltaSize.width; | |
186 windowFrame.size.height += windowDeltaSize.height; | |
187 [[self window] setFrame:windowFrame display:NO]; | |
188 } | |
189 | |
190 - (NSView*)viewForSearchEngine:(const TemplateURL*)engine | |
191 atIndex:(size_t)index { | |
192 bool useImages = false; | |
193 #if defined(GOOGLE_CHROME_BUILD) | |
194 useImages = true; | |
195 #endif | |
196 | |
197 // Make the engine identifier. | |
198 NSView* engineIdentifier = nil; // either the logo or the text label | |
199 | |
200 int logoId = engine->logo_id(); | |
201 NSString* shortName = SysUTF16ToNSString(engine->short_name()); | |
202 if (useImages && logoId > 0) { | |
203 NSImage* logoImage = | |
204 ResourceBundle::GetSharedInstance().GetNativeImageNamed(logoId); | |
205 NSRect logoBounds = NSZeroRect; | |
206 logoBounds.size = [logoImage size]; | |
207 NSImageView* logoView = | |
208 [[[NSImageView alloc] initWithFrame:logoBounds] autorelease]; | |
209 [logoView setImage:logoImage]; | |
210 [logoView setEditable:NO]; | |
211 | |
212 // Tooltip text provides accessibility. | |
213 [logoView setToolTip:shortName]; | |
214 [logoView accessibilitySetOverrideValue:shortName | |
215 forAttribute:NSAccessibilityDescriptionAttribute]; | |
216 engineIdentifier = logoView; | |
217 } else { | |
218 // No logo -- we must show a text label. | |
219 NSRect labelBounds = NSMakeRect(0, 0, kLogoLabelWidth, kLogoLabelHeight); | |
220 NSTextField* labelField = | |
221 [[[NSTextField alloc] initWithFrame:labelBounds] autorelease]; | |
222 [labelField setBezeled:NO]; | |
223 [labelField setEditable:NO]; | |
224 [labelField setSelectable:NO]; | |
225 | |
226 scoped_nsobject<NSMutableParagraphStyle> paragraphStyle( | |
227 [[NSMutableParagraphStyle alloc] init]); | |
228 [paragraphStyle setAlignment:NSCenterTextAlignment]; | |
229 NSDictionary* attrs = [NSDictionary dictionaryWithObjectsAndKeys: | |
230 [NSFont boldSystemFontOfSize:13], NSFontAttributeName, | |
231 paragraphStyle.get(), NSParagraphStyleAttributeName, | |
232 nil]; | |
233 | |
234 scoped_nsobject<NSAttributedString> attrValue( | |
235 [[NSAttributedString alloc] initWithString:shortName | |
236 attributes:attrs]); | |
237 | |
238 [labelField setAttributedStringValue:attrValue.get()]; | |
239 | |
240 engineIdentifier = labelField; | |
241 } | |
242 | |
243 // Make the "Choose" button. | |
244 scoped_nsobject<NSButton> chooseButton( | |
245 [[NSButton alloc] initWithFrame:NSMakeRect(0, 0, 100, 34)]); | |
246 [chooseButton setBezelStyle:NSRoundedBezelStyle]; | |
247 [[chooseButton cell] setFont:[NSFont systemFontOfSize: | |
248 [NSFont systemFontSizeForControlSize:NSRegularControlSize]]]; | |
249 [chooseButton setTitle:l10n_util::GetNSStringWithFixup(IDS_FR_SEARCH_CHOOSE)]; | |
250 [GTMUILocalizerAndLayoutTweaker sizeToFitView:chooseButton.get()]; | |
251 [chooseButton setTag:index]; | |
252 [chooseButton setTarget:self]; | |
253 [chooseButton setAction:@selector(searchEngineSelected:)]; | |
254 | |
255 // Provide a more descriptive accessibility description. | |
256 id accElement = NSAccessibilityUnignoredDescendant(engineIdentifier); | |
257 [[chooseButton cell] accessibilitySetOverrideValue:accElement | |
258 forAttribute:NSAccessibilityTitleUIElementAttribute]; | |
259 | |
260 // Put 'em together. | |
261 NSRect engineIdentifierFrame = [engineIdentifier frame]; | |
262 NSRect chooseButtonFrame = [chooseButton frame]; | |
263 | |
264 NSRect containingViewFrame = NSZeroRect; | |
265 containingViewFrame.size.width += engineIdentifierFrame.size.width; | |
266 containingViewFrame.size.height += engineIdentifierFrame.size.height; | |
267 containingViewFrame.size.height += kLogoButtonSpacing; | |
268 containingViewFrame.size.height += chooseButtonFrame.size.height; | |
269 | |
270 NSView* containingView = | |
271 [[[NSView alloc] initWithFrame:containingViewFrame] autorelease]; | |
272 | |
273 [containingView addSubview:engineIdentifier]; | |
274 engineIdentifierFrame.origin.y = | |
275 chooseButtonFrame.size.height + kLogoButtonSpacing; | |
276 [engineIdentifier setFrame:engineIdentifierFrame]; | |
277 | |
278 [containingView addSubview:chooseButton]; | |
279 chooseButtonFrame.origin.x = | |
280 int((containingViewFrame.size.width - chooseButtonFrame.size.width) / 2); | |
281 [chooseButton setFrame:chooseButtonFrame]; | |
282 | |
283 return containingView; | |
284 } | |
285 | |
286 - (NSFont*)mainLabelFont { | |
287 return [NSFont boldSystemFontOfSize:13]; | |
288 } | |
289 | |
290 - (IBAction)searchEngineSelected:(id)sender { | |
291 [[self window] close]; | |
292 [NSApp stopModalWithCode:[sender tag]]; | |
293 } | |
294 | |
295 @end | |
OLD | NEW |