Adds navigator.registerProtocolHandler.

chrome/browser/tab_contents/tab_contents.cc

 // Copyright (c) 2011 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/tab_contents/tab_contents.h"
 
 #include <cmath>
 
 #include "base/auto_reset.h"
 #include "base/metrics/histogram.h"
 #include "base/metrics/stats_counters.h"
 #include "base/string16.h"
 #include "base/string_util.h"
 #include "base/time.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/autocomplete_history_manager.h"
 #include "chrome/browser/autofill/autofill_manager.h"
 #include "chrome/browser/blocked_content_container.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_shutdown.h"
 #include "chrome/browser/character_encoding.h"
 #include "chrome/browser/child_process_security_policy.h"
 #include "chrome/browser/content_settings/content_settings_details.h"
 #include "chrome/browser/content_settings/host_content_settings_map.h"
+#include "chrome/browser/custom_handlers/protocol_handler_registry.h"
+#include "chrome/browser/custom_handlers/register_protocol_handler_infobar_delegate.h"
 #include "chrome/browser/debugger/devtools_manager.h"
 #include "chrome/browser/defaults.h"
 #include "chrome/browser/desktop_notification_handler.h"
 #include "chrome/browser/dom_operation_notification_details.h"
 #include "chrome/browser/dom_ui/dom_ui.h"
 #include "chrome/browser/download/download_item_model.h"
 #include "chrome/browser/download/download_manager.h"
 #include "chrome/browser/download/download_request_limiter.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/external_protocol_handler.h"
@@ skipping 2578 matching lines @@
   render_manager_.dom_ui()->ProcessDOMUIMessage(params);
 }
 
 void TabContents::ProcessExternalHostMessage(const std::string& message,
                                              const std::string& origin,
                                              const std::string& target) {
   if (delegate())
     delegate()->ForwardMessageToExternalHost(message, origin, target);
 }
 
+void TabContents::RegisterProtocolHandler(const std::string& protocol,
+                                          const GURL& url,
+                                          const string16& title) {
+  ProtocolHandler* handler = ProtocolHandler::CreateProtocolHandler(protocol, url, title);
+
+  if (handler == NULL) {
+    return;
+  }
+
+  if (profile()->GetProtocolHandlerRegistry()->IsAlreadyRegistered(handler)) {
+      return;

[tony, 2011/02/14 23:40:07]

Nit: 2 space indent.  I would probably add a TODO(koz) about replacing existing
handlers.

[koz (OOO until 15th September), 2011/02/15 03:37:27]

Actually I've implemented the logic for showing a different message when we are
replacing a protocol handler - it's in
register_protocol_handler_infobar_delegate.cc. This line is so that if the site
tries to register an identical protocol handler to one that's already registered
then we don't bother the user with it and drop the request.

Actually, we probably want to at least notify the user of that. FireFox does
that, and if a site presents a link or button to let the user register a
protocol handler, then having no feedback for it in some cases would be
confusing.

I'll add that in this patch (it should only be one line here).

+  }
+
+  AddInfoBar(new RegisterProtocolHandlerInfoBarDelegate(
+      this, profile()->GetProtocolHandlerRegistry(), handler));
+}
+
 void TabContents::RunJavaScriptMessage(
     const std::wstring& message,
     const std::wstring& default_prompt,
     const GURL& frame_url,
     const int flags,
     IPC::Message* reply_msg,
     bool* did_suppress_message) {
   // Suppress javascript messages when requested and when inside a constrained
   // popup window (because that activates them and breaks them out of the
   // constrained window jail).
@@ skipping 495 matching lines @@
 }
 
 bool TabContents::MaybeUsePreloadedPage(const GURL& url) {
   PrerenderManager* pm = profile()->GetPrerenderManager();
   if (pm != NULL) {
     if (pm->MaybeUsePreloadedPage(this, url))
       return true;
   }
   return false;
 }