Badge Windows profile shortcuts with multi-user avatar.

chrome/installer/setup/install.cc

 // Copyright (c) 2011 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 <vector>
 
@@ skipping 113 matching lines @@
       installer_state.target_path().Append(installer::kChromeExe));
 
   if ((install_status == installer::FIRST_INSTALL_SUCCESS) ||
       (install_status == installer::INSTALL_REPAIRED)) {
     if (!file_util::PathExists(shortcut_path))
       file_util::CreateDirectoryW(shortcut_path);
 
     VLOG(1) << "Creating shortcut to " << chrome_exe.value() << " at "
             << chrome_link.value();
     ret = ShellUtil::UpdateChromeShortcut(browser_dist, chrome_exe.value(),
-        chrome_link.value(), L"", product_desc, true);
+        chrome_link.value(), L"", product_desc, chrome_exe.value(),
+        browser_dist->GetIconIndex(), true);
   } else if (file_util::PathExists(chrome_link)) {
     VLOG(1) << "Updating shortcut at " << chrome_link.value()
             << " to point to " << chrome_exe.value();
     ret = ShellUtil::UpdateChromeShortcut(browser_dist, chrome_exe.value(),
-        chrome_link.value(), L"", product_desc, false);
+        chrome_link.value(), L"", product_desc, chrome_exe.value(),
+        browser_dist->GetIconIndex(), false);
   } else {
     VLOG(1)
         << "not first or repaired install, link file doesn't exist. status: "
         << install_status;
   }
 
   // Create/update uninstall link if we are not an MSI install. MSI
   // installations are, for the time being, managed only through the
   // Add/Remove Programs dialog.
   // TODO(robertshield): We could add a shortcut to msiexec /X {GUID} here.
@@ skipping 22 matching lines @@
           setup_exe.value().c_str(),
           0,
           NULL);
     }
   }
 
   // Update Desktop and Quick Launch shortcuts. If --create-new-shortcuts
   // is specified we want to create them, otherwise we update them only if
   // they exist.
   if (ret) {
+    ShellUtil::ShellChange desktop_level = ShellUtil::CURRENT_USER;
+    int quick_launch_levels = ShellUtil::CURRENT_USER;
     if (installer_state.system_install()) {
-      ret = ShellUtil::CreateChromeDesktopShortcut(product.distribution(),
-          chrome_exe.value(), product_desc, L"", L"", ShellUtil::SYSTEM_LEVEL,
-          alt_shortcut, create_all_shortcut);
-      if (ret) {
-        ret = ShellUtil::CreateChromeQuickLaunchShortcut(
-            product.distribution(), chrome_exe.value(),
-            ShellUtil::CURRENT_USER | ShellUtil::SYSTEM_LEVEL,
-            create_all_shortcut);
-      }
-    } else {
-      ret = ShellUtil::CreateChromeDesktopShortcut(product.distribution(),
-          chrome_exe.value(), product_desc, L"", L"", ShellUtil::CURRENT_USER,
-          alt_shortcut, create_all_shortcut);
-      if (ret) {
-        ret = ShellUtil::CreateChromeQuickLaunchShortcut(
-            product.distribution(), chrome_exe.value(), ShellUtil::CURRENT_USER,
-            create_all_shortcut);
-      }
+      desktop_level = ShellUtil::SYSTEM_LEVEL;
+      quick_launch_levels |= ShellUtil::SYSTEM_LEVEL;
+    }
+    ret = ShellUtil::CreateChromeDesktopShortcut(browser_dist,
+        chrome_exe.value(), product_desc, L"", L"", chrome_exe.value(),
+        browser_dist->GetIconIndex(), desktop_level, alt_shortcut,
+        create_all_shortcut);
+    if (ret) {
+      ret = ShellUtil::CreateChromeQuickLaunchShortcut(
+          browser_dist, chrome_exe.value(), quick_launch_levels,
+          create_all_shortcut);
     }
   }
 
   return ret;
 }
 
 void RegisterChromeOnMachine(const InstallerState& installer_state,
                              const Product& product,
                              bool make_chrome_default) {
   DCHECK(product.is_chrome());
@@ skipping 210 matching lines @@
     installer_state.RemoveOldVersionDirectories(
         new_version,
         existing_version.get(),
         install_temp_path);
   }
 
   return result;
 }
 
 }  // namespace installer