Add SSL icons on Mac OS X

chrome/browser/cocoa/location_bar_view_mac.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 "chrome/browser/cocoa/location_bar_view_mac.h"
 
 #include "app/l10n_util.h"
 #include "app/resource_bundle.h"
 #include "base/string_util.h"
 #include "base/sys_string_conversions.h"
@@ skipping 38 matching lines @@
                                        AutocompletePopupPositioner* positioner,
                                        CommandUpdater* command_updater,
                                        ToolbarModel* toolbar_model,
                                        Profile* profile)
     : edit_view_(new AutocompleteEditViewMac(this, positioner, toolbar_model,
                                              profile, command_updater, field)),
       command_updater_(command_updater),
       field_(field),
       disposition_(CURRENT_TAB),
       profile_(profile),
+      toolbar_model_(toolbar_model),
       transition_(PageTransition::TYPED) {
 }
 
 LocationBarViewMac::~LocationBarViewMac() {
   // TODO(shess): Placeholder for omnibox changes.
 }
 
 std::wstring LocationBarViewMac::GetInputString() const {
   return location_input_;
 }
@@ skipping 31 matching lines @@
   // TODO(pkasting): Focus the edit a la Linux/Win
 }
 
 void LocationBarViewMac::SaveStateToContents(TabContents* contents) {
   // TODO(shess): Why SaveStateToContents vs SaveStateToTab?
   edit_view_->SaveStateToTab(contents);
 }
 
 void LocationBarViewMac::Update(const TabContents* contents,
                                 bool should_restore_state) {
+  SetSecurityIcon(toolbar_model_->GetIcon());
   // AutocompleteEditView restores state if the tab is non-NULL.
   edit_view_->Update(should_restore_state ? contents : NULL);
 }
 
 void LocationBarViewMac::OnAutocompleteAccept(const GURL& url,
                                               WindowOpenDisposition disposition,
                                               PageTransition::Type transition,
                                               const GURL& alternate_nav_url) {
   if (!url.is_valid())
     return;
@@ skipping 94 matching lines @@
   // makes me sad.  I should fix that.
   OnChangedImpl(field_,
                 keyword,
                 short_name,
                 edit_view_->model()->is_keyword_hint(),
                 edit_view_->model()->show_search_hint(),
                 GetTabButtonImage());
 }
 
 void LocationBarViewMac::OnInputInProgress(bool in_progress) {
-  NOTIMPLEMENTED();
+  toolbar_model_->set_input_in_progress(in_progress);
+  Update(NULL, false);
 }
 
 void LocationBarViewMac::OnSetFocus() {
 }
 
 SkBitmap LocationBarViewMac::GetFavIcon() const {
   NOTIMPLEMENTED();
   return SkBitmap();
 }
 
@@ skipping 14 matching lines @@
 NSImage* LocationBarViewMac::GetTabButtonImage() {
   if (!tab_button_image_) {
     SkBitmap* skiaBitmap = ResourceBundle::GetSharedInstance().
         GetBitmapNamed(IDR_LOCATION_BAR_KEYWORD_HINT_TAB);
     if (skiaBitmap) {
       tab_button_image_.reset([gfx::SkBitmapToNSImage(*skiaBitmap) retain]);
     }
   }
   return tab_button_image_;
 }
+
+void LocationBarViewMac::SetSecurityIcon(ToolbarModel::Icon security_icon) {
+  ResourceBundle& rb = ResourceBundle::GetSharedInstance();
+  AutocompleteTextFieldCell* cell = [field_ autocompleteTextFieldCell];
+  switch (security_icon) {
+    case ToolbarModel::LOCK_ICON:
+      [cell setHintIcon:rb.GetNSImageNamed(IDR_LOCK)];
+      break;
+    case ToolbarModel::WARNING_ICON:
+      [cell setHintIcon:rb.GetNSImageNamed(IDR_WARNING)];
+      break;
+    case ToolbarModel::NO_ICON:
+      [cell setHintIcon:nil];
+      break;
+    default:
+      NOTREACHED();
+      break;
+  }
+  [field_ resetFieldEditorFrameIfNeeded];
+}