Bring up a content settings icon for ungestured registerProtocolHandler call.

chrome/browser/ui/content_settings/content_setting_bubble_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_bubble_model.h"
 
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/content_settings/content_settings_utils.h"
 #include "chrome/browser/content_settings/cookie_settings.h"
 #include "chrome/browser/content_settings/tab_specific_content_settings.h"
+#include "chrome/browser/custom_handlers/protocol_handler_registry.h"
 #include "chrome/browser/favicon/favicon_tab_helper.h"
 #include "chrome/browser/infobars/infobar_tab_helper.h"
 #include "chrome/browser/prefs/pref_service.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/blocked_content/blocked_content_tab_helper.h"
 #include "chrome/browser/ui/blocked_content/blocked_content_tab_helper_delegate.h"
 #include "chrome/browser/ui/collected_cookies_infobar_delegate.h"
 #include "chrome/browser/ui/content_settings/content_setting_bubble_model_delegate.h"
 #include "chrome/browser/ui/tab_contents/tab_contents.h"
 #include "chrome/common/chrome_notification_types.h"
@@ skipping 100 matching lines @@
 
   void SetManageLink() {
     static const ContentSettingsTypeIdEntry kLinkIDs[] = {
       {CONTENT_SETTINGS_TYPE_COOKIES, IDS_BLOCKED_COOKIES_LINK},
       {CONTENT_SETTINGS_TYPE_IMAGES, IDS_BLOCKED_IMAGES_LINK},
       {CONTENT_SETTINGS_TYPE_JAVASCRIPT, IDS_BLOCKED_JAVASCRIPT_LINK},
       {CONTENT_SETTINGS_TYPE_PLUGINS, IDS_BLOCKED_PLUGINS_LINK},
       {CONTENT_SETTINGS_TYPE_POPUPS, IDS_BLOCKED_POPUPS_LINK},
       {CONTENT_SETTINGS_TYPE_GEOLOCATION, IDS_GEOLOCATION_BUBBLE_MANAGE_LINK},
       {CONTENT_SETTINGS_TYPE_MIXEDSCRIPT, IDS_LEARN_MORE},
+      {CONTENT_SETTINGS_TYPE_REGISTER_PROTOCOL_HANDLER,
+        IDS_HANDLERS_BUBBLE_MANAGE_LINK}

[Bernhard Bauer, 2012/06/25 16:52:20]

Nit: Indent three more spaces?

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

I'd do two more.

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

Done.

     };
     set_manage_link(l10n_util::GetStringUTF8(
         GetIdForContentType(kLinkIDs, arraysize(kLinkIDs), content_type())));
   }
 
   virtual void OnManageLinkClicked() {
     if (delegate_)
       delegate_->ShowContentSettingsPage(content_type());
   }
 
@@ skipping 392 matching lines @@
   virtual void OnCustomLinkClicked() OVERRIDE {
     content::RecordAction(UserMetricsAction("MixedScript_LoadAnyway_Bubble"));
     DCHECK(tab_contents());
     content::RenderViewHost* host =
         tab_contents()->web_contents()->GetRenderViewHost();
     host->Send(new ChromeViewMsg_SetAllowRunningInsecureContent(
         host->GetRoutingID(), true));
   }
 };
 
+class ContentSettingRPHBubbleModel : public ContentSettingTitleAndLinkModel {
+ public:
+  ContentSettingRPHBubbleModel(Delegate* delegate,

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

Nit: For consistency, people should de-inline class member definitions even when
the declaration is in a .cc file.  The whole rest of this file is also wrong,
and I won't make you fix that (unless you want to!), but if you could separate
this class into a declaration and definition that'd be great.

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

Fixed up me and one neighbor. :-)

On 2012/06/25 23:55:30, Peter Kasting wrote:

+                               TabContents* tab_contents,
+                               Profile* profile,
+                               ContentSettingsType content_type)
+      : ContentSettingTitleAndLinkModel(
+            delegate, tab_contents, profile, content_type),
+        selected_item_(0),
+        pending_handler_(ProtocolHandler::EmptyProtocolHandler()),
+        previous_handler_(ProtocolHandler::EmptyProtocolHandler()) {
+    DCHECK_EQ(CONTENT_SETTINGS_TYPE_REGISTER_PROTOCOL_HANDLER, content_type);
+
+    TabSpecificContentSettings* content_settings =
+        tab_contents->content_settings();
+    pending_handler_ = content_settings->GetPendingProtocolHandler();
+    previous_handler_ = content_settings->GetPreviousProtocolHandler();
+
+    string16 protocol = UTF8ToUTF16(pending_handler_.protocol());
+    if (pending_handler_.protocol() == "mailto") {
+      protocol = l10n_util::GetStringUTF16(
+          IDS_REGISTER_PROTOCOL_HANDLER_MAILTO_NAME);
+    }
+    if (pending_handler_.protocol() == "webcal") {

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

Nit: For clarity I'd make this "else if" and put the UTF8ToUTF16(...) call from
above in a final "else" block.

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

Done.

+      protocol = l10n_util::GetStringUTF16(
+          IDS_REGISTER_PROTOCOL_HANDLER_WEBCAL_NAME);
+    }
+    if (previous_handler_.IsEmpty()) {
+      set_title(l10n_util::GetStringFUTF8(
+          IDS_REGISTER_PROTOCOL_HANDLER_CONFIRM,
+          pending_handler_.title(), UTF8ToUTF16(pending_handler_.url().host()),
+          protocol));
+    } else {
+      set_title(l10n_util::GetStringFUTF8(
+          IDS_REGISTER_PROTOCOL_HANDLER_CONFIRM_REPLACE,
+          pending_handler_.title(), UTF8ToUTF16(pending_handler_.url().host()),
+          protocol, previous_handler_.title()));
+    }
+
+    std::string radio_allow_label =
+        l10n_util::GetStringFUTF8(IDS_REGISTER_PROTOCOL_HANDLER_ACCEPT,
+                                  pending_handler_.title());
+    std::string radio_deny_label =
+        l10n_util::GetStringUTF8(IDS_REGISTER_PROTOCOL_HANDLER_DENY);
+    std::string radio_ignore_label =
+        l10n_util::GetStringUTF8(IDS_REGISTER_PROTOCOL_HANDLER_IGNORE);
+
+    GURL url = tab_contents->web_contents()->GetURL();
+    RadioGroup radio_group;
+    radio_group.url = url;
+
+    radio_group.radio_items.push_back(radio_allow_label);
+    radio_group.radio_items.push_back(radio_deny_label);
+    radio_group.radio_items.push_back(radio_ignore_label);
+    ContentSetting setting =
+        content_settings->GetPendingProtocolHandlerSetting();
+    if (setting == CONTENT_SETTING_ALLOW)
+      radio_group.default_item = RPH_ALLOW;
+    if (setting == CONTENT_SETTING_BLOCK)

[Bernhard Bauer, 2012/06/25 16:52:20]

Nit: else if?

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

Not just a nit -- all "allow"s will get set to RPH_IGNORE without this :)

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

Done.

+      radio_group.default_item = RPH_BLOCK;
+    else
+      radio_group.default_item = RPH_IGNORE;
+    selected_item_ = radio_group.default_item;
+    set_radio_group_enabled(true);
+    set_radio_group(radio_group);
+  }
+
+  virtual void OnCustomLinkClicked() {
+    RegisterProtocolHandler();
+  }
+
+  virtual void OnRadioClicked(int radio_index) {
+    if (selected_item_ != radio_index) {
+      selected_item_ = radio_index;
+
+      if (radio_index == RPH_ALLOW) {

[Bernhard Bauer, 2012/06/25 16:52:20]

You can either leave out the braces or leave them in, but you should be
consistent with above.

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

Done.

+        RegisterProtocolHandler();
+      } else if (radio_index == RPH_BLOCK) {
+        UnregisterProtocolHandler();
+      } else if (radio_index == RPH_IGNORE) {
+        IgnoreProtocolHandler();
+      }
+    }
+  }
+
+  void RegisterProtocolHandler() {
+    // A no-op if the handler hasn't been ignored, but needed in case the user
+    // selects sequences like register/ignore/register.
+    profile()->GetProtocolHandlerRegistry()->RemoveIgnoredHandler(
+        pending_handler_);
+
+    profile()->GetProtocolHandlerRegistry()->OnAcceptRegisterProtocolHandler(
+        pending_handler_);
+    tab_contents()->content_settings()->SetPendingProtocolHandlerSetting(
+        CONTENT_SETTING_ALLOW);
+  }
+
+  void UnregisterProtocolHandler() {
+    profile()->GetProtocolHandlerRegistry()->OnDenyRegisterProtocolHandler(
+        pending_handler_);
+    tab_contents()->content_settings()->SetPendingProtocolHandlerSetting(
+        CONTENT_SETTING_BLOCK);
+    ClearOrSetPreviousHandler();
+  }
+
+  void IgnoreProtocolHandler() {
+    profile()->GetProtocolHandlerRegistry()->OnIgnoreRegisterProtocolHandler(
+        pending_handler_);
+    tab_contents()->content_settings()->SetPendingProtocolHandlerSetting(
+        CONTENT_SETTING_DEFAULT);
+    ClearOrSetPreviousHandler();
+  }
+
+  void ClearOrSetPreviousHandler() {
+    if (previous_handler_.IsEmpty()) {
+      profile()->GetProtocolHandlerRegistry()->ClearDefault(
+          pending_handler_.protocol());
+    } else {
+      profile()->GetProtocolHandlerRegistry()->OnAcceptRegisterProtocolHandler(
+          previous_handler_);
+    }
+  }
+
+ private:
+  enum RPHState {
+    RPH_ALLOW = 0,
+    RPH_BLOCK,
+    RPH_IGNORE,
+  };
+
+  int selected_item_;
+  ProtocolHandler pending_handler_;
+  ProtocolHandler previous_handler_;
+};
+
 // static
 ContentSettingBubbleModel*
     ContentSettingBubbleModel::CreateContentSettingBubbleModel(
         Delegate* delegate,
         TabContents* tab_contents,
         Profile* profile,
         ContentSettingsType content_type) {
   if (content_type == CONTENT_SETTINGS_TYPE_COOKIES) {
     return new ContentSettingCookiesBubbleModel(delegate, tab_contents, profile,
                                                 content_type);
   }
   if (content_type == CONTENT_SETTINGS_TYPE_POPUPS) {
     return new ContentSettingPopupBubbleModel(delegate, tab_contents, profile,
                                               content_type);
   }
   if (content_type == CONTENT_SETTINGS_TYPE_GEOLOCATION) {
     return new ContentSettingDomainListBubbleModel(delegate, tab_contents,
                                                    profile, content_type);
   }
   if (content_type == CONTENT_SETTINGS_TYPE_PLUGINS) {
     return new ContentSettingPluginBubbleModel(delegate, tab_contents, profile,
                                                content_type);
   }
   if (content_type == CONTENT_SETTINGS_TYPE_MIXEDSCRIPT) {
     return new ContentSettingMixedScriptBubbleModel(delegate, tab_contents,
                                                     profile, content_type);
   }
+  if (content_type == CONTENT_SETTINGS_TYPE_REGISTER_PROTOCOL_HANDLER) {
+    return new ContentSettingRPHBubbleModel(delegate, tab_contents, profile,
+                                            content_type);
+  }
   return new ContentSettingSingleRadioGroup(delegate, tab_contents, profile,
                                             content_type);
 }
 
 ContentSettingBubbleModel::ContentSettingBubbleModel(
     TabContents* tab_contents,
     Profile* profile,
     ContentSettingsType content_type)
     : tab_contents_(tab_contents),
       profile_(profile),
@@ skipping 34 matching lines @@
     const content::NotificationDetails& details) {
   if (type == chrome::NOTIFICATION_TAB_CONTENTS_DESTROYED) {
     DCHECK_EQ(tab_contents_, content::Source<TabContents>(source).ptr());
     tab_contents_ = NULL;
   } else {
     DCHECK_EQ(chrome::NOTIFICATION_PROFILE_DESTROYED, type);
     DCHECK_EQ(profile_, content::Source<Profile>(source).ptr());
     profile_ = NULL;
   }
 }