Refactor security-icon code to a more general form

chrome/browser/cocoa/location_bar_view_mac.h

 // 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.
 
 #ifndef CHROME_BROWSER_COCOA_LOCATION_BAR_VIEW_MAC_H_
 #define CHROME_BROWSER_COCOA_LOCATION_BAR_VIEW_MAC_H_
 
 #import <Cocoa/Cocoa.h>
 
 #include "base/scoped_nsobject.h"
@@ skipping 66 matching lines @@
   // Internals of OnChanged(), pulled out for purposes of unit
   // testing.  Sets up |field| based on the parameters, which are
   // pulled from edit_view->model().
   static void OnChangedImpl(AutocompleteTextField* field,
                             const std::wstring& keyword,
                             const std::wstring& short_name,
                             const bool is_keyword_hint,
                             const bool show_search_hint,
                             NSImage* image);
 
+  // Used to display a clickable icon in the location bar.
+  class LocationBarImageView {
+   public:
+    explicit LocationBarImageView() : image_(nil),
+                                      label_(nil),
+                                      visible_(false) {}
+    virtual ~LocationBarImageView() {}
+
+    // Sets the image.
+    void SetImage(NSImage* image);
+
+    // Sets the label text, font, and color. |text| may be nil; |color| and
+    // |font| are ignored if |text| is nil.
+    void SetLabel(NSString* text, NSFont* baseFont, NSColor* color);
+
+    // Sets the visibility. SetImage() should be called with a valid image
+    // before the visibility is set to |true|.
+    void SetVisible(bool visible);
+
+    const NSImage* GetImage() const { return image_; }
+    const NSAttributedString* GetLabel() const { return label_; }
+    bool IsVisible() const { return visible_; }
+
+    virtual bool OnMousePressed() = 0;
+
+   private:
+    scoped_nsobject<NSImage> image_;
+
+    // The label shown next to the icon, or nil if none.
+    scoped_nsobject<NSAttributedString> label_;
+
+    bool visible_;
+
+    DISALLOW_COPY_AND_ASSIGN(LocationBarImageView);
+  };
+
+  // SecurityImageView is used to display the lock or warning icon when the
+  // current URL's scheme is https.
+  class SecurityImageView : public LocationBarImageView {
+   public:
+    enum Image {
+      LOCK = 0,
+      WARNING
+    };
+
+    SecurityImageView(Profile* profile, ToolbarModel* model);
+    virtual ~SecurityImageView();
+
+    // Sets the image to the appropriate icon.
+    void SetImageShown(Image image);
+
+    // Shows the page info dialog.
+    virtual bool OnMousePressed();
+
+   private:
+    // The lock icon shown when using HTTPS. Loaded lazily, the first time it's
+    // needed.
+    scoped_nsobject<NSImage> lock_icon_;
+
+    // The warning icon shown when HTTPS is broken. Loaded lazily, the first
+    // time it's needed.
+    scoped_nsobject<NSImage> warning_icon_;
+
+    Profile* profile_;
+    ToolbarModel* model_;
+
+    DISALLOW_COPY_AND_ASSIGN(SecurityImageView);
+  };
+
  private:
-  // Set the SSL icon we should be showing.
+  // Sets the SSL icon we should be showing.
   void SetSecurityIcon(ToolbarModel::Icon icon);
 
+  // Sets the label for the SSL icon.
+  void SetSecurityIconLabel();
+
   scoped_ptr<AutocompleteEditViewMac> edit_view_;
 
   CommandUpdater* command_updater_;  // Weak, owned by Browser.
 
   AutocompleteTextField* field_;  // owned by tab controller
 
   // When we get an OnAutocompleteAccept notification from the autocomplete
   // edit, we save the input string so we can give it back to the browser on
   // the LocationBar interface via GetInputString().
   std::wstring location_input_;
 
   // The user's desired disposition for how their input should be opened.
   WindowOpenDisposition disposition_;
 
+  // The view that shows the lock/warning when in HTTPS mode.
+  SecurityImageView security_image_view_;
+
   Profile* profile_;
 
   ToolbarModel* toolbar_model_;  // Weak, owned by Browser.
 
   // Image used in drawing keyword hint.
   scoped_nsobject<NSImage> tab_button_image_;
 
   // The transition type to use for the navigation.
   PageTransition::Type transition_;
 
   DISALLOW_COPY_AND_ASSIGN(LocationBarViewMac);
 };
 
 #endif  // CHROME_BROWSER_COCOA_LOCATION_BAR_VIEW_MAC_H_