Bring up a content settings icon for ungestured registerProtocolHandler call.

chrome/browser/ui/content_settings/content_setting_image_model.cc

 // 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.
 
 #include "chrome/browser/ui/content_settings/content_setting_image_model.h"
 
 #include "chrome/browser/content_settings/host_content_settings_map.h"
 #include "chrome/browser/content_settings/tab_specific_content_settings.h"
 #include "chrome/browser/prerender/prerender_manager.h"
 #include "chrome/browser/profiles/profile.h"
@@ skipping 14 matching lines @@
   virtual void UpdateFromWebContents(WebContents* web_contents) OVERRIDE;
 };
 
 class ContentSettingGeolocationImageModel : public ContentSettingImageModel {
  public:
   ContentSettingGeolocationImageModel();
 
   virtual void UpdateFromWebContents(WebContents* web_contents) OVERRIDE;
 };
 
+class ContentSettingRPHImageModel : public ContentSettingImageModel {
+ public:
+  ContentSettingRPHImageModel();
+
+  virtual void UpdateFromWebContents(WebContents* web_contents) OVERRIDE;
+};
+
 class ContentSettingNotificationsImageModel : public ContentSettingImageModel {
  public:
   ContentSettingNotificationsImageModel();
 
   virtual void UpdateFromWebContents(WebContents* web_contents) OVERRIDE;
 };
 
 namespace {
 
 struct ContentSettingsTypeIdEntry {
@@ skipping 104 matching lines @@
   unsigned int tab_state_flags = 0;
   settings_state.GetDetailedInfo(NULL, &tab_state_flags);
   bool allowed =
       !!(tab_state_flags & GeolocationSettingsState::TABSTATE_HAS_ANY_ALLOWED);
   set_icon(allowed ? IDR_GEOLOCATION_ALLOWED_LOCATIONBAR_ICON :
       IDR_GEOLOCATION_DENIED_LOCATIONBAR_ICON);
   set_tooltip(l10n_util::GetStringUTF8(allowed ?
       IDS_GEOLOCATION_ALLOWED_TOOLTIP : IDS_GEOLOCATION_BLOCKED_TOOLTIP));
 }
 
+ContentSettingRPHImageModel::ContentSettingRPHImageModel()
+    : ContentSettingImageModel(
+        CONTENT_SETTINGS_TYPE_REGISTER_PROTOCOL_HANDLER) {
+}
+
+void ContentSettingRPHImageModel::UpdateFromWebContents(
+    WebContents* web_contents) {
+  set_visible(false);
+  if (!web_contents)
+    return;
+
+  TabSpecificContentSettings* content_settings =
+      TabContents::FromWebContents(web_contents)->content_settings();
+  if (content_settings->GetPendingProtocolHandler().IsEmpty())
+    return;
+
+  set_visible(true);
+  set_icon(IDR_REGISTER_PROTOCOL_HANDLER_LOCATIONBAR_ICON);

[Peter Kasting, 2012/06/25 23:55:30]

Nit: Can you set the icon and tooltip in the constructor?

[Greg Billock, 2012/06/26 19:17:04]

Done.

+
+  set_tooltip(l10n_util::GetStringUTF8(IDS_REGISTER_PROTOCOL_HANDLER_TOOLTIP));
+}
+
 ContentSettingNotificationsImageModel::ContentSettingNotificationsImageModel()
     : ContentSettingImageModel(CONTENT_SETTINGS_TYPE_NOTIFICATIONS) {
 }
 
 void ContentSettingNotificationsImageModel::UpdateFromWebContents(
     WebContents* web_contents) {
   // Notifications do not have a bubble.
   set_visible(false);
 }
 
 ContentSettingImageModel::ContentSettingImageModel(
     ContentSettingsType content_settings_type)
     : content_settings_type_(content_settings_type),
       is_visible_(false),
       icon_(0),
       explanatory_string_id_(0) {
 }
 
 // static
 ContentSettingImageModel*
     ContentSettingImageModel::CreateContentSettingImageModel(
     ContentSettingsType content_settings_type) {
   switch (content_settings_type) {
     case CONTENT_SETTINGS_TYPE_GEOLOCATION:
       return new ContentSettingGeolocationImageModel();
     case CONTENT_SETTINGS_TYPE_NOTIFICATIONS:
       return new ContentSettingNotificationsImageModel();
+    case CONTENT_SETTINGS_TYPE_REGISTER_PROTOCOL_HANDLER:
+      return new ContentSettingRPHImageModel();
     default:
       return new ContentSettingBlockedImageModel(content_settings_type);
   }
 }