Refactor the code for loading icons into the PageInfoModel.

chrome/browser/page_info_model.cc

 // Copyright (c) 2010 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.
 
 #include "chrome/browser/page_info_model.h"
 
 #include <string>
 
 #include "app/l10n_util.h"
+#include "app/resource_bundle.h"
 #include "base/command_line.h"
 #include "base/i18n/time_formatting.h"
 #include "base/string_number_conversions.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/cert_store.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profile.h"
 #include "chrome/browser/ssl/ssl_manager.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "grit/generated_resources.h"
+#include "grit/theme_resources.h"
 #include "net/base/cert_status_flags.h"
 #include "net/base/ssl_connection_status_flags.h"
 #include "net/base/ssl_cipher_suite_names.h"
 #include "net/base/x509_certificate.h"
 
+#if defined(OS_MACOSX)
+#include "base/mac_util.h"
+#endif
+
 PageInfoModel::PageInfoModel(Profile* profile,
                              const GURL& url,
                              const NavigationEntry::SSLStatus& ssl,
                              bool show_history,
                              PageInfoModelObserver* observer)
     : observer_(observer) {
-  SectionInfoState state = SECTION_STATE_OK;
+  Init();
+
+  SectionStateIcon icon_id = ICON_STATE_OK;
   string16 headline;
   string16 description;
   scoped_refptr<net::X509Certificate> cert;
 
   // Identity section.
   string16 subject_name(UTF8ToUTF16(url.host()));
   bool empty_subject_name = false;
   if (subject_name.empty()) {
     subject_name.assign(
         l10n_util::GetStringUTF16(IDS_PAGE_INFO_SECURITY_TAB_UNKNOWN_PARTY));
@@ skipping 24 matching lines @@
       description += ASCIIToUTF16("\n\n");
       if (cert_status & net::CERT_STATUS_UNABLE_TO_CHECK_REVOCATION) {
         description += l10n_util::GetStringUTF16(
             IDS_PAGE_INFO_SECURITY_TAB_UNABLE_TO_CHECK_REVOCATION);
       } else if (cert_status & net::CERT_STATUS_NO_REVOCATION_MECHANISM) {
         description += l10n_util::GetStringUTF16(
             IDS_PAGE_INFO_SECURITY_TAB_NO_REVOCATION_MECHANISM);
       } else {
         NOTREACHED() << "Need to specify string for this warning";
       }
-      state = SECTION_STATE_WARNING_MINOR;
+      icon_id = ICON_STATE_WARNING_MINOR;
     } else if ((ssl.cert_status() & net::CERT_STATUS_IS_EV) != 0) {
       // EV HTTPS page.
       DCHECK(!cert->subject().organization_names.empty());
       headline =
           l10n_util::GetStringFUTF16(IDS_PAGE_INFO_EV_IDENTITY_TITLE,
               UTF8ToUTF16(cert->subject().organization_names[0]),
               UTF8ToUTF16(url.host()));
       // An EV Cert is required to have a city (localityName) and country but
       // state is "if any".
       DCHECK(!cert->subject().locality_name.empty());
@@ skipping 36 matching lines @@
         issuer_name.assign(l10n_util::GetStringUTF16(
             IDS_PAGE_INFO_SECURITY_TAB_UNKNOWN_PARTY));
       }
       description.assign(l10n_util::GetStringFUTF16(
           IDS_PAGE_INFO_SECURITY_TAB_SECURE_IDENTITY, issuer_name));
     }
   } else {
     // HTTP or HTTPS with errors (not warnings).
     description.assign(l10n_util::GetStringUTF16(
         IDS_PAGE_INFO_SECURITY_TAB_INSECURE_IDENTITY));
-    state = ssl.security_style() == SECURITY_STYLE_UNAUTHENTICATED ?
-        SECTION_STATE_WARNING_MAJOR : SECTION_STATE_ERROR;
+    icon_id = ssl.security_style() == SECURITY_STYLE_UNAUTHENTICATED ?
+        ICON_STATE_WARNING_MAJOR : ICON_STATE_ERROR;
   }
   sections_.push_back(SectionInfo(
-      state,
+      icon_id,
       l10n_util::GetStringUTF16(IDS_PAGE_INFO_SECURITY_TAB_IDENTITY_TITLE),
       headline,
       description,
       SECTION_INFO_IDENTITY));
 
   // Connection section.
   // We consider anything less than 80 bits encryption to be weak encryption.
   // TODO(wtc): Bug 1198735: report mixed/unsafe content for unencrypted and
   // weakly encrypted connections.
-  state = SECTION_STATE_OK;
+  icon_id = ICON_STATE_OK;
   headline.clear();
   description.clear();
   if (ssl.security_bits() < 0) {
     // Security strength is unknown.  Say nothing.
-    state = SECTION_STATE_ERROR;
+    icon_id = ICON_STATE_ERROR;
   } else if (ssl.security_bits() == 0) {
-    state = ssl.security_style() == SECURITY_STYLE_UNAUTHENTICATED ?
-        SECTION_STATE_WARNING_MAJOR : SECTION_STATE_ERROR;
+    icon_id = ssl.security_style() == SECURITY_STYLE_UNAUTHENTICATED ?
+        ICON_STATE_WARNING_MAJOR : ICON_STATE_ERROR;
     description.assign(l10n_util::GetStringFUTF16(
         IDS_PAGE_INFO_SECURITY_TAB_NOT_ENCRYPTED_CONNECTION_TEXT,
         subject_name));
   } else if (ssl.security_bits() < 80) {
-    state = SECTION_STATE_ERROR;
+    icon_id = ICON_STATE_ERROR;
     description.assign(l10n_util::GetStringFUTF16(
         IDS_PAGE_INFO_SECURITY_TAB_WEAK_ENCRYPTION_CONNECTION_TEXT,
         subject_name));
   } else {
     description.assign(l10n_util::GetStringFUTF16(
         IDS_PAGE_INFO_SECURITY_TAB_ENCRYPTED_CONNECTION_TEXT,
         subject_name,
         base::IntToString16(ssl.security_bits())));
     if (ssl.displayed_insecure_content() || ssl.ran_insecure_content()) {
       // The old SSL dialog only had good and bad state, so for the old
       // implementation we raise an error on finding mixed content. The new
       // SSL info bubble has a warning state for displaying insecure content,
       // so we check. The command line check will go away once we eliminate
       // the old dialogs.
       const CommandLine* command_line(CommandLine::ForCurrentProcess());
       if (command_line->HasSwitch(switches::kEnableNewPageInfoBubble) &&
           !ssl.ran_insecure_content()) {
-        state = SECTION_STATE_WARNING_MINOR;
+        icon_id = ICON_STATE_WARNING_MINOR;
       } else {
-        state = SECTION_STATE_ERROR;
+        icon_id = ICON_STATE_ERROR;
       }
       description.assign(l10n_util::GetStringFUTF16(
           IDS_PAGE_INFO_SECURITY_TAB_ENCRYPTED_SENTENCE_LINK,
           description,
           l10n_util::GetStringUTF16(ssl.ran_insecure_content() ?
               IDS_PAGE_INFO_SECURITY_TAB_ENCRYPTED_INSECURE_CONTENT_ERROR :
               IDS_PAGE_INFO_SECURITY_TAB_ENCRYPTED_INSECURE_CONTENT_WARNING)));
     }
   }
 
@@ skipping 22 matching lines @@
       description += l10n_util::GetStringFUTF16(
           IDS_PAGE_INFO_SECURITY_TAB_COMPRESSION_DETAILS,
           ASCIIToUTF16(compression));
     } else {
       description += l10n_util::GetStringUTF16(
           IDS_PAGE_INFO_SECURITY_TAB_NO_COMPRESSION);
     }
 
     if (did_fallback) {
       // For now, only SSLv3 fallback will trigger a warning icon.
-      state = SECTION_STATE_ERROR;
+      icon_id = ICON_STATE_ERROR;
       description += ASCIIToUTF16("\n\n");
       description += l10n_util::GetStringUTF16(
           IDS_PAGE_INFO_SECURITY_TAB_FALLBACK_MESSAGE);
     }
     if (no_renegotiation) {
       description += ASCIIToUTF16("\n\n");
       description += l10n_util::GetStringUTF16(
           IDS_PAGE_INFO_SECURITY_TAB_RENEGOTIATION_MESSAGE);
     }
   }
 
   if (!description.empty()) {
     sections_.push_back(SectionInfo(
-        state,
+        icon_id,
         l10n_util::GetStringUTF16(IDS_PAGE_INFO_SECURITY_TAB_CONNECTION_TITLE),
         headline,
         description,
         SECTION_INFO_CONNECTION));
   }
 
   // Request the number of visits.
   HistoryService* history = profile->GetHistoryService(
       Profile::EXPLICIT_ACCESS);
   if (show_history && history) {
     history->GetVisitCountToHost(
         url,
         &request_consumer_,
         NewCallback(this, &PageInfoModel::OnGotVisitCountToHost));
   }
 }
 
 PageInfoModel::~PageInfoModel() {
+#if defined(OS_MACOSX)
+  // Release the NSImages.
+  for (std::vector<ImageType>::iterator it = icons_.begin();
+       it != icons_.end(); ++it) {
+    mac_util::NSObjectRelease(*it);
+  }
+#endif
 }
 
 int PageInfoModel::GetSectionCount() {
   return sections_.size();
 }
 
 PageInfoModel::SectionInfo PageInfoModel::GetSectionInfo(int index) {
   DCHECK(index < static_cast<int>(sections_.size()));
   return sections_[index];
 }
 
+PageInfoModel::ImageType PageInfoModel::GetIconImage(SectionStateIcon icon_id) {
+  if (icon_id == ICON_NONE)
+    return NULL;
+  // TODO(rsesek): Remove once the window is replaced with the bubble.
+  const CommandLine* command_line(CommandLine::ForCurrentProcess());
+  if (!command_line->HasSwitch(switches::kEnableNewPageInfoBubble) &&
+      icon_id != ICON_STATE_OK) {
+    return icons_[ICON_STATE_WARNING_MAJOR];
+  }
+  // The buble uses new, various icons.
+  return icons_[icon_id];
+}
+
 void PageInfoModel::OnGotVisitCountToHost(HistoryService::Handle handle,
                                           bool found_visits,
                                           int count,
                                           base::Time first_visit) {
   if (!found_visits) {
     // This indicates an error, such as the page wasn't http/https; do nothing.
     return;
   }
 
   bool visited_before_today = false;
   if (count) {
     base::Time today = base::Time::Now().LocalMidnight();
     base::Time first_visit_midnight = first_visit.LocalMidnight();
     visited_before_today = (first_visit_midnight < today);
   }
 
   // We only show the Site Information heading for the new dialogs.
   string16 title;
   const CommandLine* command_line(CommandLine::ForCurrentProcess());
-  if (command_line->HasSwitch(switches::kEnableNewPageInfoBubble))
+  bool as_bubble = command_line->HasSwitch(switches::kEnableNewPageInfoBubble);
+  if (as_bubble)
     title = l10n_util::GetStringUTF16(IDS_PAGE_INFO_SITE_INFO_TITLE);
 
   if (!visited_before_today) {
     sections_.push_back(SectionInfo(
-        SECTION_STATE_ERROR,
+        ICON_STATE_WARNING_MAJOR,
         l10n_util::GetStringUTF16(
             IDS_PAGE_INFO_SECURITY_TAB_PERSONAL_HISTORY_TITLE),
         title,
         l10n_util::GetStringUTF16(
             IDS_PAGE_INFO_SECURITY_TAB_FIRST_VISITED_TODAY),
         SECTION_INFO_FIRST_VISIT));
   } else {
     sections_.push_back(SectionInfo(
-        SECTION_STATE_OK,
+        // TODO(rsesek): Remove once the window is replaced with the bubble.
+        as_bubble ? ICON_STATE_INFO : ICON_STATE_OK,
         l10n_util::GetStringUTF16(
             IDS_PAGE_INFO_SECURITY_TAB_PERSONAL_HISTORY_TITLE),
         title,
         l10n_util::GetStringFUTF16(
             IDS_PAGE_INFO_SECURITY_TAB_VISITED_BEFORE_TODAY,
             WideToUTF16(base::TimeFormatShortDate(first_visit))),
         SECTION_INFO_FIRST_VISIT));
   }
   observer_->ModelChanged();
 }
 
 // static
 void PageInfoModel::RegisterPrefs(PrefService* prefs) {
   prefs->RegisterDictionaryPref(prefs::kPageInfoWindowPlacement);
 }
 
 PageInfoModel::PageInfoModel() {
+  Init();
 }
+
+void PageInfoModel::Init() {
+  // Loads the icons into the vector. The order must match the SectionStateIcon
+  // enum.
+  icons_.push_back(GetBitmapNamed(IDR_PAGEINFO_GOOD));
+  icons_.push_back(GetBitmapNamed(IDR_PAGEINFO_WARNING_MINOR));
+  icons_.push_back(GetBitmapNamed(IDR_PAGEINFO_WARNING_MAJOR));
+  icons_.push_back(GetBitmapNamed(IDR_PAGEINFO_BAD));
+  icons_.push_back(GetBitmapNamed(IDR_PAGEINFO_INFO));
+}
+
+PageInfoModel::ImageType PageInfoModel::GetBitmapNamed(int resource_id) {
+  ResourceBundle& rb = ResourceBundle::GetSharedInstance();
+#if defined(OS_MACOSX)
+  // Unlike other platforms, the Mac ResourceBundle does not keep a shared image
+  // cache. These are released in the dtor.
+  ImageType image = rb.GetNSImageNamed(resource_id);
+  mac_util::NSObjectRetain(image);
+  return image;
+#elif defined(USE_X11) && !defined(TOOLKIT_VIEWS)
+  return rb.GetPixbufNamed(resource_id);
+#else
+  return rb.GetBitmapNamed(resource_id);
+#endif
+}