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 490 matching lines @@
                         OnUpdateContentRestrictions)
     IPC_MESSAGE_HANDLER(ViewHostMsg_PDFHasUnsupportedFeature,
                         OnPDFHasUnsupportedFeature)
     IPC_MESSAGE_HANDLER(ViewHostMsg_GoToEntryAtOffset, OnGoToEntryAtOffset)
     IPC_MESSAGE_HANDLER(ViewHostMsg_DidGetApplicationInfo,
                         OnDidGetApplicationInfo)
     IPC_MESSAGE_HANDLER(ViewHostMsg_InstallApplication,
                         OnInstallApplication)
     IPC_MESSAGE_HANDLER(ViewHostMsg_PageContents, OnPageContents)
     IPC_MESSAGE_HANDLER(ViewHostMsg_PageTranslated, OnPageTranslated)
+    IPC_MESSAGE_HANDLER(ViewHostMsg_RegisterProtocolHandler,
+                        OnRegisterProtocolHandler)
     IPC_MESSAGE_HANDLER(ViewHostMsg_SetSuggestions, OnSetSuggestions)
     IPC_MESSAGE_HANDLER(ViewHostMsg_InstantSupportDetermined,
                         OnInstantSupportDetermined)
     IPC_MESSAGE_HANDLER(ViewHostMsg_RunFileChooser, OnRunFileChooser)
     IPC_MESSAGE_UNHANDLED(handled = false)
   IPC_END_MESSAGE_MAP_EX()
 
   if (!message_is_ok) {
     UserMetrics::RecordAction(UserMetricsAction("BadMessageTerminate_RVD"));
     GetRenderProcessHost()->ReceivedBadMessage();
@@ skipping 1618 matching lines @@
                                    TranslateErrors::Type error_type) {
   language_state_.set_current_language(translated_lang);
   language_state_.set_translation_pending(false);
   PageTranslatedDetails details(original_lang, translated_lang, error_type);
   NotificationService::current()->Notify(
       NotificationType::PAGE_TRANSLATED,
       Source<TabContents>(this),
       Details<PageTranslatedDetails>(&details));
 }
 
+void TabContents::OnRegisterProtocolHandler(const std::string& protocol,
+                                               const GURL& url,
+                                               const string16& title) {
+  ProtocolHandlerRegistry* registry = profile()->GetProtocolHandlerRegistry();
+  if (!registry->CanSchemeBeOverridden(protocol)) {
+    return;
+  }
+
+  ProtocolHandler* handler = ProtocolHandler::CreateProtocolHandler(protocol,
+                                                                    url, title);
+  if (handler == NULL) {
+    return;
+  }
+
+  if (registry->IsAlreadyRegistered(handler)) {
+      AddInfoBar(new SimpleAlertInfoBarDelegate(this, NULL,
+          l10n_util::GetStringFUTF16(
+              IDS_REGISTER_PROTOCOL_HANDLER_ALREADY_REGISTERED,
+              handler->title(), UTF8ToUTF16(handler->protocol())), true));
+    return;
+  }
+
+  AddInfoBar(new RegisterProtocolHandlerInfoBarDelegate(this, registry,
+        handler));
+}
+
 void TabContents::OnSetSuggestions(
     int32 page_id,
     const std::vector<std::string>& suggestions) {
   if (delegate())
     delegate()->OnSetSuggestions(page_id, suggestions);
 }
 
 void TabContents::OnInstantSupportDetermined(int32 page_id, bool result) {
   if (delegate())
     delegate()->OnInstantSupportDetermined(page_id, result);
@@ skipping 945 matching lines @@
 }
 
 bool TabContents::MaybeUsePreloadedPage(const GURL& url) {
   PrerenderManager* pm = profile()->GetPrerenderManager();
   if (pm != NULL) {
     if (pm->MaybeUsePreloadedPage(this, url))
       return true;
   }
   return false;
 }