Implementing --app-launcher install/uninstall flow.

chrome/installer/setup/install.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/installer/setup/install.h"
 
 #include <shlobj.h>
 #include <time.h>
 #include <winuser.h>
 
 #include <string>
 
 #include "base/command_line.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/memory/scoped_ptr.h"
 #include "base/path_service.h"
 #include "base/string_util.h"
 #include "base/stringprintf.h"
 #include "base/utf_string_conversions.h"
 #include "base/win/shortcut.h"
 #include "base/win/windows_version.h"
 #include "chrome/common/chrome_constants.h"
+#include "chrome/common/chrome_switches.h"
 #include "chrome/installer/setup/setup_constants.h"
 #include "chrome/installer/setup/install_worker.h"
 #include "chrome/installer/util/auto_launch_util.h"
 #include "chrome/installer/util/browser_distribution.h"
 #include "chrome/installer/util/create_reg_key_work_item.h"
 #include "chrome/installer/util/delete_after_reboot_helper.h"
 #include "chrome/installer/util/google_update_constants.h"
 #include "chrome/installer/util/helper.h"
 #include "chrome/installer/util/install_util.h"
+#include "chrome/installer/util/installation_state.h"
 #include "chrome/installer/util/master_preferences.h"
 #include "chrome/installer/util/master_preferences_constants.h"
 #include "chrome/installer/util/set_reg_value_work_item.h"
 #include "chrome/installer/util/shell_util.h"
 #include "chrome/installer/util/util_constants.h"
 #include "chrome/installer/util/work_item_list.h"
 
 // Build-time generated include file.
 #include "registered_dlls.h"  // NOLINT
 
@@ skipping 219 matching lines @@
   ShellUtil::ShellChange shortcut_level = installer_state.system_install() ?
       ShellUtil::SYSTEM_LEVEL : ShellUtil::CURRENT_USER;
   FilePath uninstall_shortcut_path;
   ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_START_MENU, dist,
                              shortcut_level, &uninstall_shortcut_path);
   uninstall_shortcut_path = uninstall_shortcut_path.Append(
       dist->GetUninstallLinkName() + installer::kLnkExt);
   file_util::Delete(uninstall_shortcut_path, false);
 }
 
+// Returns the appropriate shortcut operations for App Launcher,
+// based on state of installation and master preference.
+installer::InstallShortcutOperation GetAppLauncherShortcutOperation(
+    const InstallationState& original_state,
+    const InstallerState& installer_state,
+    int version_compare) {
+  const installer::ProductState* original_app_host_state =
+      original_state.GetProductState(installer_state.system_install(),
+                                     BrowserDistribution::CHROME_APP_HOST);
+  bool already_had_app_launcher = original_app_host_state &&
+      original_app_host_state->channel().IsAppLauncher();

[grt (UTC plus 2), 2012/10/25 14:46:53]

using the channel id like this will lead to problems.  look in the
uninstall_command() for the switch you care about instead.

[huangs, 2012/10/29 21:15:16]

Done.

+
+  if (!already_had_app_launcher)
+    return installer::INSTALL_SHORTCUT_CREATE_ALL;
+
+  if (version_compare >= 0)
+    return installer::INSTALL_SHORTCUT_REPLACE_EXISTING;
+
+  LOG(ERROR) << "Not sure how we got here";
+  return installer::INSTALL_SHORTCUT_REPLACE_EXISTING;
+}
+
 }  // end namespace
 
 namespace installer {
 
 void EscapeXmlAttributeValueInSingleQuotes(string16* att_value) {
   ReplaceChars(*att_value, L"&", L"&amp;", att_value);
   ReplaceChars(*att_value, L"'", L"&apos;", att_value);
   ReplaceChars(*att_value, L"<", L"&lt;", att_value);
 }
 
@@ skipping 51 matching lines @@
               << " to " << src_path.value();
       return true;
     } else {
       PLOG(ERROR) << "Error writing " << installer::kVisualElementsManifest
                   << " to " << src_path.value();
       return false;
     }
   }
 }
 
 void CreateOrUpdateShortcuts(const InstallerState& installer_state,

[grt (UTC plus 2), 2012/10/25 14:46:53]

this function was written to very specifically handle Chrome's shortcuts and may
not be appropriate for your use.  please consult with gab@ about the right way
to have your shortcut(s) created.

[huangs, 2012/10/29 21:15:16]

Decided to do shortcut creation as a separate CL.

                              const FilePath& setup_exe,
                              const Product& product,
                              InstallShortcutOperation install_operation,
                              bool alternate_desktop_shortcut) {
   // TODO(tommi): Change this function to use WorkItemList.
-  DCHECK(product.is_chrome());
+  DCHECK(product.is_chrome() || product.is_chrome_app_host());
 
   BrowserDistribution* dist = product.distribution();
-  const FilePath chrome_exe(
-      installer_state.target_path().Append(installer::kChromeExe));
   ShellUtil::ShellChange install_level =
       installer_state.system_install() ? ShellUtil::SYSTEM_LEVEL :
                                          ShellUtil::CURRENT_USER;
+  // |base_properties|: The basic properties to set on every shortcut installed
+  // (to be refined on a per-shortcut basis).
+  ShellUtil::ChromeShortcutProperties base_properties(install_level);
+
+  base_properties.set_chrome_exe(
+      installer_state.target_path().Append(installer::kChromeExe));
+  if (product.is_chrome_app_host()) {

[erikwright (departed), 2012/10/25 02:23:03]

needs to be app_host.exe, not chrome.exe

[huangs, 2012/10/29 21:15:16]

Done.  (Deferring to different CL).

+    // Should consult AppListController::GetAppListCommandLine().
+    // TODO(huangs): Append switches::kUserDataDir, perhaps?

[benwells, 2012/10/25 04:50:51]

I don't think you can put kUserDataDir here as you're in the installer. Do you
have any concept of overriding the user data dir?

[huangs, 2012/10/29 21:15:16]

Nope.  I was looking at
AppListController::GetAppListCommandLine()
and decided to worry about this later.  I should have just asked.  Removing.   
(Deferring to different CL).

+    base_properties.set_arguments(
+        string16(L"--").append(ASCIIToUTF16(::switches::kShowAppList)));

[grt (UTC plus 2), 2012/10/25 14:46:53]

i don't like this hacky prepend.  please use CommandLine instead:
    CommandLine app_host_cmd(CommandLine::NO_PROGRAM);
    app_host_cmd.AppendSwitch(switches::kShowAppList);
    base_properties.set_arguments(app_host_cmd.GetCommandLineString());

[huangs, 2012/10/29 21:15:16]

Done.  However, note that this leads to 2 spaces, since
app_host_cmd.GetCommandLineString() would return
" --show-app-list", and another gets added later.    (Deferring to different
CL).

[grt (UTC plus 2), 2012/10/30 13:02:55]

If the spaces cause no harm, just do the simple thing and leave them.

+  }
 
   // The default operation on update is to overwrite shortcuts with the
   // currently desired properties, but do so only for shortcuts that still
   // exist.
   ShellUtil::ChromeShortcutOperation shortcut_operation =
       ShellUtil::SHORTCUT_REPLACE_EXISTING;
-
-  // |base_properties|: The basic properties to set on every shortcut installed
-  // (to be refined on a per-shortcut basis).
-  ShellUtil::ChromeShortcutProperties base_properties(install_level);
-  base_properties.set_chrome_exe(chrome_exe);
-
   // If |install_operation| is INSTALL_SHORTCUT_CREATE_ALL, create optional
   // shortcuts (Desktop and Quick Launch) immediately; otherwise delay their
   // creation until first run.
   if (install_operation == INSTALL_SHORTCUT_CREATE_ALL)
     shortcut_operation = ShellUtil::SHORTCUT_CREATE_ALWAYS;
 
   ShellUtil::ChromeShortcutProperties desktop_properties(base_properties);
   if (alternate_desktop_shortcut)
     desktop_properties.set_shortcut_name(dist->GetAlternateApplicationName());
   ExecuteAndLogShortcutOperation(
@@ skipping 14 matching lines @@
         ShellUtil::SHORTCUT_QUICK_LAUNCH, dist, user_ql_properties,
         shortcut_operation);
   }
 
   // Turn on shortcut creation for mandatory shortcuts if requested.
   if (install_operation == INSTALL_SHORTCUT_CREATE_MANDATORY)
     shortcut_operation = ShellUtil::SHORTCUT_CREATE_ALWAYS;
 
   ShellUtil::ChromeShortcutProperties start_menu_properties(base_properties);
   // IMPORTANT: Only the default (no arguments and default browserappid) browser
-  // shortcut in the Start menu (Start screen on Win8+) should be  made dual
+  // shortcut in the Start menu (Start screen on Win8+) should be made dual
   // mode.
   start_menu_properties.set_dual_mode(true);
   if (shortcut_operation == ShellUtil::SHORTCUT_CREATE_ALWAYS)
     start_menu_properties.set_pin_to_taskbar(true);
   ExecuteAndLogShortcutOperation(
       ShellUtil::SHORTCUT_START_MENU, dist, start_menu_properties,
       shortcut_operation);
 }
 
 void RegisterChromeOnMachine(const InstallerState& installer_state,
@@ skipping 75 matching lines @@
 
     // Update the modifiers on the channel values for the product(s) being
     // installed and for the binaries in case of multi-install.
     installer_state.UpdateChannels();
 
     installer_state.UpdateStage(installer::COPYING_PREFERENCES_FILE);
 
     if (result == FIRST_INSTALL_SUCCESS && !prefs_path.empty())
       CopyPreferenceFileForFirstRun(installer_state, prefs_path);
 
-    // Currently this only creates shortcuts for Chrome, but for other products
-    // we might want to create shortcuts.
+    FilePath setup_exe(installer_state.GetInstallerDirectory(new_version)
+        .Append(setup_path.BaseName()));
+
+    // Creates shortcuts for App Launcher.
+    const Product* app_host_install =
+        installer_state.FindProduct(BrowserDistribution::CHROME_APP_HOST);
+    if (app_host_install && app_host_install->HasOption(kOptionAppLauncher)) {
+      installer_state.UpdateStage(installer::CREATING_SHORTCUTS);
+
+      InstallShortcutOperation app_launcher_shortcut_operation =
+          GetAppLauncherShortcutOperation(
+              original_state, installer_state,
+              new_version.CompareTo(*existing_version));
+
+      bool alt_shortcut = false;
+      prefs.GetBool(master_preferences::kAltShortcutText, &alt_shortcut);
+
+      CreateOrUpdateShortcuts(installer_state, setup_exe, *app_host_install,
+                              app_launcher_shortcut_operation, alt_shortcut);
+    }
+
+    // Creates shortcuts for Chrome.
     const Product* chrome_install =
         installer_state.FindProduct(BrowserDistribution::CHROME_BROWSER);
     if (chrome_install) {
       installer_state.UpdateStage(installer::CREATING_SHORTCUTS);
 
       BrowserDistribution* dist = chrome_install->distribution();
       CleanupLegacyShortcuts(installer_state, dist);
 
       bool create_all_shortcuts = false;
       prefs.GetBool(master_preferences::kCreateAllShortcuts,
@@ skipping 74 matching lines @@
   if (base::win::GetVersion() >= base::win::VERSION_WIN8) {
     VLOG(1) << "Updating and registering shortcuts.";
     CreateOrUpdateShortcuts(
         installer_state, setup_exe, chrome, INSTALL_SHORTCUT_REPLACE_EXISTING,
         false);
     RegisterChromeOnMachine(installer_state, chrome, false);
   }
 }
 
 }  // namespace installer