Make application shortcuts point to app_host.exe, install App Host during app installation.

chrome/browser/extensions/unpacked_installer.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/extensions/unpacked_installer.h"
 
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/file_util.h"
 #include "base/string_util.h"
 #include "base/threading/thread_restrictions.h"
+#include "chrome/browser/extensions/app_host_installer.h"
 #include "chrome/browser/extensions/extension_install_prompt.h"
 #include "chrome/browser/extensions/extension_install_ui.h"
 #include "chrome/browser/extensions/extension_prefs.h"
 #include "chrome/browser/extensions/extension_service.h"
 #include "chrome/browser/extensions/permissions_updater.h"
 #include "chrome/browser/extensions/requirements_checker.h"
 #include "chrome/common/extensions/extension.h"
 #include "chrome/common/extensions/extension_file_util.h"
 #include "sync/api/string_ordinal.h"
 
@@ skipping 12 matching lines @@
                             base::WeakPtr<ExtensionService> extension_service,
                             const Extension* extension);
   ~SimpleExtensionLoadPrompt();
 
   void ShowPrompt();
 
   // ExtensionInstallUI::Delegate
   virtual void InstallUIProceed() OVERRIDE;
   virtual void InstallUIAbort(bool user_initiated) OVERRIDE;
 
+  void OnAppHostInstallation(bool success);
+  void CompleteInstall();
+
  private:
   base::WeakPtr<ExtensionService> service_weak_;
   scoped_ptr<ExtensionInstallPrompt> install_ui_;
   scoped_refptr<const Extension> extension_;
+  extensions::AppHostInstaller app_host_installer_;
 };
 
 SimpleExtensionLoadPrompt::SimpleExtensionLoadPrompt(
     Profile* profile,
     base::WeakPtr<ExtensionService> extension_service,
     const Extension* extension)
     : service_weak_(extension_service),
       extension_(extension) {
   install_ui_.reset(
       ExtensionInstallUI::CreateInstallPromptWithProfile(profile));
 }
 
 SimpleExtensionLoadPrompt::~SimpleExtensionLoadPrompt() {
 }
 
 void SimpleExtensionLoadPrompt::ShowPrompt() {
   install_ui_->ConfirmInstall(this, extension_);
 }
 
 void SimpleExtensionLoadPrompt::InstallUIProceed() {
   if (service_weak_.get()) {
-    extensions::PermissionsUpdater perms_updater(service_weak_->profile());
-    perms_updater.GrantActivePermissions(extension_, false);
-    service_weak_->OnExtensionInstalled(
-        extension_,
-        false,  // Not from web store.
-        syncer::StringOrdinal(),
-        false /* no requirement errors */);
+    if (extensions::AppHostInstaller::IsAppHostInstallRequired(*extension_)) {
+      // TODO(huangs): Should this be Unretained, or not?
+      app_host_installer_.InstallAppHost(
+          base::Bind(&SimpleExtensionLoadPrompt::OnAppHostInstallation,
+                     base::Unretained(this)));
+    } else {
+      CompleteInstall();
+    }
+  } else {
+    delete this;
   }
+}
+
+void SimpleExtensionLoadPrompt::OnAppHostInstallation(bool success) {
+  if (success) {
+    CompleteInstall();
+  } else {
+    // TODO(huangs): Error message.
+    delete this;
+  }
+}
+
+void SimpleExtensionLoadPrompt::CompleteInstall() {
+  extensions::PermissionsUpdater perms_updater(service_weak_->profile());
+  perms_updater.GrantActivePermissions(extension_, false);
+  service_weak_->OnExtensionInstalled(
+      extension_,
+      false,  // Not from web store.
+      syncer::StringOrdinal(),
+      false /* no requirement errors */);
   delete this;
 }
 
 void SimpleExtensionLoadPrompt::InstallUIAbort(bool user_initiated) {
   delete this;
 }
 
 }  // namespace
 
 namespace extensions {
@@ skipping 165 matching lines @@
       !extension_->plugins().empty() &&
       !disabled_extensions->Contains(extension_->id())) {
     SimpleExtensionLoadPrompt* prompt = new SimpleExtensionLoadPrompt(
         service_weak_->profile(),
         service_weak_,
         extension_);
     prompt->ShowPrompt();
     return;  // continues in SimpleExtensionLoadPrompt::InstallPrompt*
   }
 
+  if (AppHostInstaller::IsAppHostInstallRequired(*extension_)) {
+    // Calls CompleteInstall() after.
+    app_host_installer_.InstallAppHost(
+        base::Bind(&UnpackedInstaller::OnAppHostInstallation, this));
+  } else {
+    CompleteInstall();
+  }
+}
+
+void UnpackedInstaller::OnAppHostInstallation(bool success) {
+  if (success) {
+    CompleteInstall();
+  } else {
+    // TODO(huangs): Error message.
+    std::string error = "Some random error message";
+    UnpackedInstaller::ReportExtensionLoadError(error);
+  }
+}
+
+void UnpackedInstaller::CompleteInstall() {
+  CHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
   PermissionsUpdater perms_updater(service_weak_->profile());
   perms_updater.GrantActivePermissions(extension_, false);
   service_weak_->OnExtensionInstalled(extension_,
                                       false,  // Not from web store.
                                       syncer::StringOrdinal(),
                                       false /* no requirement errors */);
 }
 
 }  // namespace extensions