Implementing unified Chrome / App Launcher flow, and migrating old stand-alone App Launcher.

chrome/installer/util/shell_util.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.
 //
 // This file defines functions that integrate Chrome in Windows shell. These
 // functions can be used by Chrome as well as Chrome installer. All of the
 // work is done by the local functions defined in anonymous namespace in
 // this class.
 
 #include "chrome/installer/util/shell_util.h"
@@ skipping 1189 matching lines @@
 
 bool ShortcutOpDelete(const base::FilePath& shortcut_path) {
   bool ret = file_util::Delete(shortcut_path, false);
   LOG_IF(ERROR, !ret) << "Failed to remove " << shortcut_path.value();
   return ret;
 }
 
 bool ShortcutOpUpdate(const base::win::ShortcutProperties& shortcut_properties,
                       const base::FilePath& shortcut_path) {
   bool ret = base::win::CreateOrUpdateShortcutLink(
-      shortcut_path, shortcut_properties, base::win::SHORTCUT_REPLACE_EXISTING);
+      shortcut_path, shortcut_properties, base::win::SHORTCUT_UPDATE_EXISTING);
   LOG_IF(ERROR, !ret) << "Failed to update " << shortcut_path.value();
   return ret;
 }
 
 // {|location|, |dist|, |level|} determine |shortcut_folder|.
-// Applies |shortcut_operation| to each shortcut in |shortcut_folder| that
-// targets |target_exe|.
+// Applies |shortcut_operation| to shortcuts in |shortcut_folder| that
+// targets |target_exe|, and have filenames matching |filename_filter|.
 // Returns true if all operations are successful. All intended operations are
 // attempted even if failures occur.
-bool BatchShortcutAction(const FileOperationCallback& shortcut_operation,
-                         ShellUtil::ShortcutLocation location,
-                         BrowserDistribution* dist,
-                         ShellUtil::ShellChange level,
-                         const base::FilePath& target_exe) {
+bool BatchShortcutActionFilteredByName(
+    const FileOperationCallback& shortcut_operation,
+    ShellUtil::ShortcutLocation location,
+    BrowserDistribution* dist,
+    ShellUtil::ShellChange level,
+    const base::FilePath& target_exe,
+    const string16& filename_filter) {
   DCHECK(!shortcut_operation.is_null());
   base::FilePath shortcut_folder;
   if (!ShellUtil::GetShortcutPath(location, dist, level, &shortcut_folder)) {
     LOG(WARNING) << "Cannot find path at location " << location;
     return false;
   }
 
   bool success = true;
   InstallUtil::ProgramCompare target_compare(target_exe);
   file_util::FileEnumerator enumerator(
       shortcut_folder, false, file_util::FileEnumerator::FILES,
-      string16(L"*") + installer::kLnkExt);
-  base::FilePath target_path;
-  for (base::FilePath shortcut_path = enumerator.Next();
-       !shortcut_path.empty();
-       shortcut_path = enumerator.Next()) {
-    if (base::win::ResolveShortcut(shortcut_path, &target_path, NULL)) {
-      if (target_compare.EvaluatePath(target_path) &&
-          !shortcut_operation.Run(shortcut_path)) {
+      filename_filter);
+  base::FilePath original_target_exe;
+  for (base::FilePath shortcut_file = enumerator.Next();
+       !shortcut_file.empty();
+       shortcut_file = enumerator.Next()) {
+    if (base::win::ResolveShortcut(shortcut_file, &original_target_exe, NULL)) {
+      if (target_compare.EvaluatePath(original_target_exe) &&
+          !shortcut_operation.Run(shortcut_file)) {
         success = false;
       }
     } else {
-      LOG(ERROR) << "Cannot resolve shortcut at " << shortcut_path.value();
+      LOG(ERROR) << "Cannot resolve shortcut at " << shortcut_file.value();
       success = false;
     }
   }
   return success;
 }
 
+// Adaptor to FilteredBatchShortcutAction(), to process all files.

[gab, 2013/05/15 22:42:20]

Remove comma.

[huangs, 2013/05/17 20:59:24]

Done.

+bool BatchShortcutAction(const FileOperationCallback& shortcut_operation,
+                         ShellUtil::ShortcutLocation location,
+                         BrowserDistribution* dist,
+                         ShellUtil::ShellChange level,
+                         const base::FilePath& target_exe) {
+  return BatchShortcutActionFilteredByName(
+      shortcut_operation, location, dist, level, target_exe,
+      string16(L"*") + installer::kLnkExt);
+}
+
 // Removes folder spsecified by {|location|, |dist|, |level|}.
 bool RemoveShortcutFolder(ShellUtil::ShortcutLocation location,
                           BrowserDistribution* dist,
                           ShellUtil::ShellChange level) {
 
   // Explicitly whitelist locations, since accidental calls can be very harmful.
   if (location != ShellUtil::SHORTCUT_LOCATION_START_MENU &&
       location != ShellUtil::SHORTCUT_LOCATION_APP_SHORTCUTS) {
     NOTREACHED();
     return false;
@@ skipping 77 matching lines @@
 
 bool ShellUtil::ShortcutLocationIsSupported(
     ShellUtil::ShortcutLocation location) {
   switch (location) {
     case SHORTCUT_LOCATION_DESKTOP:
       return true;
     case SHORTCUT_LOCATION_QUICK_LAUNCH:
       return true;
     case SHORTCUT_LOCATION_START_MENU:
       return true;
+    case SHORTCUT_LOCATION_START_MENU_ROOT:
+      return true;
     case SHORTCUT_LOCATION_TASKBAR_PINS:
       return base::win::GetVersion() >= base::win::VERSION_WIN7;
     case SHORTCUT_LOCATION_APP_SHORTCUTS:
       return base::win::GetVersion() >= base::win::VERSION_WIN8;
     default:
       NOTREACHED();
       return false;
   }
 }
 
@@ skipping 11 matching lines @@
       break;
     case SHORTCUT_LOCATION_QUICK_LAUNCH:
       dir_key = (level == CURRENT_USER) ? base::DIR_USER_QUICK_LAUNCH :
                                           base::DIR_DEFAULT_USER_QUICK_LAUNCH;
       break;
     case SHORTCUT_LOCATION_START_MENU:
       dir_key = (level == CURRENT_USER) ? base::DIR_START_MENU :
                                           base::DIR_COMMON_START_MENU;
       add_folder_for_dist = true;
       break;
+    case SHORTCUT_LOCATION_START_MENU_ROOT:

[gab, 2013/05/15 22:42:20]

Coordinate with https://codereview.chromium.org/13864015/

[huangs, 2013/05/17 20:59:24]

Deleted for now.

+      dir_key = (level == CURRENT_USER) ? base::DIR_START_MENU :
+                                          base::DIR_COMMON_START_MENU;
+      break;
     case SHORTCUT_LOCATION_TASKBAR_PINS:
       dir_key = base::DIR_TASKBAR_PINS;
       break;
     case SHORTCUT_LOCATION_APP_SHORTCUTS:
       // TODO(huangs): Move GetAppShortcutsFolder() logic into base_paths_win.
       return GetAppShortcutsFolder(dist, level, path);
 
     default:
       NOTREACHED();
       return false;
   }
 
   if (!PathService::Get(dir_key, path) || path->empty()) {
     NOTREACHED() << dir_key;
     return false;
   }
 
   if (add_folder_for_dist)
-    *path = path->Append(dist->GetAppShortCutName());
+    *path = path->Append(dist->GetAppShortCutFolderName());
 
   return true;
 }
 
 bool ShellUtil::CreateOrUpdateShortcut(
     ShellUtil::ShortcutLocation location,
     BrowserDistribution* dist,
     const ShellUtil::ShortcutProperties& properties,
     ShellUtil::ShortcutOperation operation) {
   // Explicitly whitelist locations to which this is applicable.
   if (location != SHORTCUT_LOCATION_DESKTOP &&
       location != SHORTCUT_LOCATION_QUICK_LAUNCH &&
-      location != SHORTCUT_LOCATION_START_MENU) {
+      location != SHORTCUT_LOCATION_START_MENU &&
+      location != SHORTCUT_LOCATION_START_MENU_ROOT) {
     NOTREACHED();
     return false;
   }
 
   DCHECK(dist);
   // |pin_to_taskbar| is only acknowledged when first creating the shortcut.
   DCHECK(!properties.pin_to_taskbar ||
          operation == SHELL_SHORTCUT_CREATE_ALWAYS ||
          operation == SHELL_SHORTCUT_CREATE_IF_NO_SYSTEM_LEVEL);
 
@@ skipping 567 matching lines @@
 
   switch (location) {
     case SHORTCUT_LOCATION_START_MENU:  // Falls through.
     case SHORTCUT_LOCATION_APP_SHORTCUTS:
       return RemoveShortcutFolder(location, dist, level);
 
     case SHORTCUT_LOCATION_TASKBAR_PINS:
       return BatchShortcutAction(base::Bind(&ShortcutOpUnpin), location, dist,
                                  level, target_exe);
 
+    case SHORTCUT_LOCATION_START_MENU_ROOT:  // For emphasis; falls through.

[gab, 2013/05/15 22:42:20]

See https://codereview.chromium.org/13864015/ (I think that's explicit enough
with adding this extra case).

[huangs, 2013/05/17 20:59:24]

Deleted.

     default:
       return BatchShortcutAction(base::Bind(&ShortcutOpDelete), location, dist,
                                  level, target_exe);
   }
 }
 
 // static
 bool ShellUtil::UpdateShortcuts(
     ShellUtil::ShortcutLocation location,
     BrowserDistribution* dist,
     ShellChange level,
     const base::FilePath& target_exe,
     const ShellUtil::ShortcutProperties& properties) {
   if (!ShellUtil::ShortcutLocationIsSupported(location))
     return true;  // Vacuous success.
 
   base::win::ShortcutProperties shortcut_properties(
       TranslateShortcutProperties(properties));
   return BatchShortcutAction(base::Bind(&ShortcutOpUpdate, shortcut_properties),
                              location, dist, level, target_exe);
 }
 
+// static
+bool ShellUtil::UpdateShortcutsFilteredByName(
+    ShellUtil::ShortcutLocation location,
+    BrowserDistribution* dist,
+    ShellChange level,
+    const string16& name_filter,
+    const base::FilePath& target_exe,
+    const ShellUtil::ShortcutProperties& properties) {
+  DCHECK(EndsWith(name_filter, installer::kLnkExt, false));
+
+  if (!ShellUtil::ShortcutLocationIsSupported(location))
+    return true;  // Vacuous success.
+
+  base::win::ShortcutProperties shortcut_properties(
+      TranslateShortcutProperties(properties));
+  return BatchShortcutActionFilteredByName(
+      base::Bind(&ShortcutOpUpdate, shortcut_properties),
+      location, dist, level, target_exe, name_filter);
+}
+
 bool ShellUtil::GetUserSpecificRegistrySuffix(string16* suffix) {
   // Use a thread-safe cache for the user's suffix.
   static base::LazyInstance<UserSpecificRegistrySuffix>::Leaky suffix_instance =
       LAZY_INSTANCE_INITIALIZER;
   return suffix_instance.Get().GetSuffix(suffix);
 }
 
 bool ShellUtil::GetOldUserSpecificRegistrySuffix(string16* suffix) {
   wchar_t user_name[256];
   DWORD size = arraysize(user_name);
@@ skipping 47 matching lines @@
     // are any left...).
     if (free_bits >= 8 && next_byte_index < size) {
       free_bits -= 8;
       bit_stream += bytes[next_byte_index++] << free_bits;
     }
   }
 
   DCHECK_EQ(ret.length(), encoded_length);
   return ret;
 }