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"
 
 using content::BrowserThread;
 using extensions::Extension;
 
 namespace {
 
 const char kUnpackedExtensionsBlacklistedError[] =
     "Loading of unpacked extensions is disabled by the administrator.";
 
 // Manages an ExtensionInstallPrompt for a particular extension.
 class SimpleExtensionLoadPrompt : public ExtensionInstallPrompt::Delegate {
  public:
   SimpleExtensionLoadPrompt(Profile* profile,
                             base::WeakPtr<ExtensionService> extension_service,
-                            const Extension* extension);
+                            const Extension* extension,
+                            const base::Closure& callback);
   ~SimpleExtensionLoadPrompt();
 
   void ShowPrompt();
 
   // ExtensionInstallUI::Delegate
   virtual void InstallUIProceed() OVERRIDE;
   virtual void InstallUIAbort(bool user_initiated) OVERRIDE;
 
+  void BeginInstall();
+  void OnAppHostInstallation(bool success);
+  void CompleteInstall();
+
  private:
   base::WeakPtr<ExtensionService> service_weak_;
   scoped_ptr<ExtensionInstallPrompt> install_ui_;
   scoped_refptr<const Extension> extension_;
+  const base::Closure& callback_;
 };
 
 SimpleExtensionLoadPrompt::SimpleExtensionLoadPrompt(
     Profile* profile,
     base::WeakPtr<ExtensionService> extension_service,
-    const Extension* extension)
+    const Extension* extension,
+    const base::Closure& callback)
     : service_weak_(extension_service),
-      extension_(extension) {
+      extension_(extension),
+      callback_(callback) {
   install_ui_.reset(
       ExtensionInstallUI::CreateInstallPromptWithProfile(profile));
 }
 
 SimpleExtensionLoadPrompt::~SimpleExtensionLoadPrompt() {
 }
 
 void SimpleExtensionLoadPrompt::ShowPrompt() {
   install_ui_->ConfirmInstall(this, extension_);
 }
 
 void SimpleExtensionLoadPrompt::InstallUIProceed() {
   if (service_weak_.get()) {

[erikwright (departed), 2012/10/04 01:28:16]

Remove this check, remove the extension_service member and constructor
parameter.

See my comment below to add the check.

[huangs, 2012/10/04 22:56:42]

Keeping this the way it was, now that we changed interface to AppHostInstaller.

-    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 */);
+    callback_.Run();
   }
   delete this;
 }
 
 void SimpleExtensionLoadPrompt::InstallUIAbort(bool user_initiated) {
   delete this;
 }
 
 }  // namespace
 
@@ skipping 161 matching lines @@
     return;
   const ExtensionSet* disabled_extensions =
       service_weak_->disabled_extensions();
   if (service_weak_->show_extensions_prompts() &&
       prompt_for_plugins_ &&
       !extension_->plugins().empty() &&
       !disabled_extensions->Contains(extension_->id())) {
     SimpleExtensionLoadPrompt* prompt = new SimpleExtensionLoadPrompt(
         service_weak_->profile(),
         service_weak_,
-        extension_);
+        extension_,
+        base::Bind(&UnpackedInstaller::OnPromptAccepted, this));
     prompt->ShowPrompt();
     return;  // continues in SimpleExtensionLoadPrompt::InstallPrompt*
   }
+  OnPromptAccepted();
+}
 
+void UnpackedInstaller::OnPromptAccepted() {
+  if (!BrowserThread::PostTask(
+          BrowserThread::FILE, FROM_HERE,
+          base::Bind(&UnpackedInstaller::BeginInstall, this)))
+    NOTREACHED();
+}
+
+void UnpackedInstaller::BeginInstall() {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
+  if (AppHostInstaller::IsAppHostInstallRequired(*extension_)) {
+    // Calls CompleteInstall() after.
+    app_host_installer_.InstallAppHost(
+        base::Bind(&UnpackedInstaller::OnAppHostInstallation, this));

[benwells, 2012/10/04 07:53:01]

early return and remove the else.

[huangs, 2012/10/04 22:56:42]

Done.

+  } else {
+    CompleteInstall();
+  }
+}
+
+void UnpackedInstaller::OnAppHostInstallation(bool success) {
+  if (success) {
+    CompleteInstall();

[benwells, 2012/10/04 07:53:01]

early return and remove the else.

[huangs, 2012/10/04 22:56:42]

Done.

+  } else {
+    // TODO(huangs): Error message.
+    std::string error = "Some random error message";
+    UnpackedInstaller::ReportExtensionLoadError(error);
+  }
+}
+
+void UnpackedInstaller::CompleteInstall() {
+  CHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE));
   PermissionsUpdater perms_updater(service_weak_->profile());

[erikwright (departed), 2012/10/04 01:28:16]

Do a check on service_weak_ around lines 298-303.

[erikwright (departed), 2012/10/04 01:28:16]

This used to all happen on the UI thread. Have you checked that it is safe to do
on the file thread too?

[huangs, 2012/10/04 22:56:42]

No check, in light of changed interface to AppHostInstaller.

[huangs, 2012/10/04 22:56:42]

Changing interface to AppHostInstaller, so still in UI thread here.

   perms_updater.GrantActivePermissions(extension_, false);
   service_weak_->OnExtensionInstalled(extension_,
                                       false,  // Not from web store.
                                       syncer::StringOrdinal(),
                                       false /* no requirement errors */);
 }
 
 }  // namespace extensions