Allow chrome to become the os default handler for arbitrary protocols on mac/win.

chrome/browser/shell_integration.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.

[Mark Mentovai, 2011/05/23 00:23:52]

I am not yet examining this file heavily as it will likely change substantially
in light of the comments I made in its header.

 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/shell_integration.h"
 
 #include "base/command_line.h"
 #include "base/file_util.h"
 #include "base/path_service.h"
 #include "base/string_util.h"
 #include "base/utf_string_conversions.h"
@@ skipping 42 matching lines @@
   } else {
     // Use '--app=url' instead of just 'url' to launch the browser with minimal
     // chrome.
     // Note: Do not change this flag!  Old Gears shortcuts will break if you do!
     new_cmd_line.AppendSwitchASCII(switches::kApp, url.spec());
   }
   return new_cmd_line;
 }
 
 ///////////////////////////////////////////////////////////////////////////////
-// ShellIntegration::DefaultBrowserWorker
+// ShellIntegration::DefaultClientAppWorker
 //
 
-ShellIntegration::DefaultBrowserWorker::DefaultBrowserWorker(
-    DefaultBrowserObserver* observer)
-    : observer_(observer) {
+ShellIntegration::DefaultClientAppWorker::DefaultClientAppWorker(
+    DefaultClientAppObserver* observer)
+    : observer_(observer), protocol_() {
 }
 
-void ShellIntegration::DefaultBrowserWorker::StartCheckDefaultBrowser() {
-  observer_->SetDefaultBrowserUIState(STATE_PROCESSING);
+void ShellIntegration::DefaultClientAppWorker::StartCheckDefaultBrowser() {
+  if (observer_) {
+    observer_->SetDefaultClientAppUIState(STATE_PROCESSING);
+  }
   BrowserThread::PostTask(
       BrowserThread::FILE, FROM_HERE,
       NewRunnableMethod(
-          this, &DefaultBrowserWorker::ExecuteCheckDefaultBrowser));
+          this, &DefaultClientAppWorker::ExecuteCheckDefaultBrowser));
 }
 
-void ShellIntegration::DefaultBrowserWorker::StartSetAsDefaultBrowser() {
-  observer_->SetDefaultBrowserUIState(STATE_PROCESSING);
+void ShellIntegration::DefaultClientAppWorker::StartCheckDefaultProtocolClient(
+    const std::string& protocol) {
+  if (observer_) {
+    observer_->SetDefaultClientAppUIState(STATE_PROCESSING);
+  }
+  protocol_.assign(protocol);
   BrowserThread::PostTask(
       BrowserThread::FILE, FROM_HERE,
       NewRunnableMethod(
-          this, &DefaultBrowserWorker::ExecuteSetAsDefaultBrowser));
+          this, &DefaultClientAppWorker::ExecuteCheckDefaultProtocolClient));
 }
 
-void ShellIntegration::DefaultBrowserWorker::ObserverDestroyed() {
+void ShellIntegration::DefaultClientAppWorker::StartSetAsDefaultBrowser() {
+  if (observer_) {
+    observer_->SetDefaultClientAppUIState(STATE_PROCESSING);
+  }
+  BrowserThread::PostTask(
+      BrowserThread::FILE, FROM_HERE,
+      NewRunnableMethod(
+          this, &DefaultClientAppWorker::ExecuteSetAsDefaultBrowser));
+}
+
+void ShellIntegration::DefaultClientAppWorker::StartSetAsDefaultProtocolClient(
+    const std::string& protocol) {
+  if (observer_) {
+    observer_->SetDefaultClientAppUIState(STATE_PROCESSING);
+  }
+  protocol_.assign(protocol);
+  BrowserThread::PostTask(
+      BrowserThread::FILE, FROM_HERE,
+      NewRunnableMethod(
+          this, &DefaultClientAppWorker::ExecuteSetAsDefaultProtocolClient));
+}
+
+void ShellIntegration::DefaultClientAppWorker::ObserverDestroyed() {
   // Our associated view has gone away, so we shouldn't call back to it if
   // our worker thread returns after the view is dead.
   observer_ = NULL;
 }
 
 ///////////////////////////////////////////////////////////////////////////////
-// DefaultBrowserWorker, private:
+// DefaultClientAppWorker, private:
 
-void ShellIntegration::DefaultBrowserWorker::ExecuteCheckDefaultBrowser() {
+void ShellIntegration::DefaultClientAppWorker::ExecuteCheckDefaultBrowser() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
-  DefaultBrowserState state = ShellIntegration::IsDefaultBrowser();
+  DefaultClientAppState state = ShellIntegration::IsDefaultBrowser();
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       NewRunnableMethod(
-          this, &DefaultBrowserWorker::CompleteCheckDefaultBrowser, state));
+          this, &DefaultClientAppWorker::CompleteCheckDefaultApplication,
+          state));
 }
 
-void ShellIntegration::DefaultBrowserWorker::CompleteCheckDefaultBrowser(
-    DefaultBrowserState state) {
+void
+ShellIntegration::DefaultClientAppWorker::ExecuteCheckDefaultProtocolClient() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+  DefaultClientAppState state =
+      ShellIntegration::IsDefaultProtocolClient(protocol_);
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      NewRunnableMethod(
+          this, &DefaultClientAppWorker::CompleteCheckDefaultApplication,
+          state));
+}
+
+void ShellIntegration::DefaultClientAppWorker::CompleteCheckDefaultApplication(
+    DefaultClientAppState state) {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   UpdateUI(state);
 }
 
-void ShellIntegration::DefaultBrowserWorker::ExecuteSetAsDefaultBrowser() {
+void ShellIntegration::DefaultClientAppWorker::ExecuteSetAsDefaultBrowser() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
   ShellIntegration::SetAsDefaultBrowser();
   BrowserThread::PostTask(
       BrowserThread::UI, FROM_HERE,
       NewRunnableMethod(
-          this, &DefaultBrowserWorker::CompleteSetAsDefaultBrowser));
+          this, &DefaultClientAppWorker::CompleteSetAsDefaultBrowser));
 }
 
-void ShellIntegration::DefaultBrowserWorker::CompleteSetAsDefaultBrowser() {
+void ShellIntegration::DefaultClientAppWorker::CompleteSetAsDefaultBrowser() {
   DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   if (observer_) {
     // Set as default completed, check again to make sure it stuck...
     StartCheckDefaultBrowser();
   }
 }
 
-void ShellIntegration::DefaultBrowserWorker::UpdateUI(
-    DefaultBrowserState state) {
+void
+ShellIntegration::DefaultClientAppWorker::ExecuteSetAsDefaultProtocolClient() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+  ShellIntegration::SetAsDefaultProtocolClient(protocol_);
+  BrowserThread::PostTask(
+      BrowserThread::UI, FROM_HERE,
+      NewRunnableMethod(
+          this, &DefaultClientAppWorker::CompleteSetAsDefaultProtocolClient));
+}
+
+void
+ShellIntegration::DefaultClientAppWorker::CompleteSetAsDefaultProtocolClient() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  if (observer_) {
+    // Set as default completed, check again to make sure it stuck...
+    StartCheckDefaultProtocolClient(protocol_);
+  }
+}
+
+void ShellIntegration::DefaultClientAppWorker::UpdateUI(
+    DefaultClientAppState state) {
   if (observer_) {
     switch (state) {
-      case NOT_DEFAULT_BROWSER:
-        observer_->SetDefaultBrowserUIState(STATE_NOT_DEFAULT);
+      case NOT_DEFAULT_CLIENT_APP:
+        observer_->SetDefaultClientAppUIState(STATE_NOT_DEFAULT);
         break;
-      case IS_DEFAULT_BROWSER:
-        observer_->SetDefaultBrowserUIState(STATE_IS_DEFAULT);
+      case IS_DEFAULT_CLIENT_APP:
+        observer_->SetDefaultClientAppUIState(STATE_IS_DEFAULT);
         break;
-      case UNKNOWN_DEFAULT_BROWSER:
-        observer_->SetDefaultBrowserUIState(STATE_UNKNOWN);
+      case UNKNOWN_DEFAULT_CLIENT_APP:
+        observer_->SetDefaultClientAppUIState(STATE_UNKNOWN);
         break;
       default:
         break;
     }
   }
 }