profile_shortcut_manager_win.cc revision 7dbb3d5cf0c15f500944d211057644d6a2f37371
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 52a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "chrome/browser/profiles/profile_shortcut_manager_win.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 72a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include <shlobj.h> // For SHChangeNotify(). 82a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 92a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include <string> 102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include <vector> 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/command_line.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/file_util.h" 15868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/files/file_enumerator.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/path_service.h" 17868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string16.h" 18868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h" 19868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h" 20868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/win/shortcut.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/app_icon_win.h" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/browser_process.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile_info_cache_observer.h" 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile_info_util.h" 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile_manager.h" 2790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/browser/shell_integration.h" 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/chrome_switches.h" 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/browser_distribution.h" 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/product.h" 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/shell_util.h" 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/browser_thread.h" 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "grit/chrome_unscaled_resources.h" 342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "grit/chromium_strings.h" 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "skia/ext/image_operations.h" 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "skia/ext/platform_canvas.h" 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/l10n/l10n_util.h" 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/resource/resource_bundle.h" 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/icon_util.h" 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/image/image.h" 41c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "ui/gfx/image/image_family.h" 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "ui/gfx/rect.h" 432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "ui/gfx/skia_util.h" 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::BrowserThread; 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Characters that are not allowed in Windows filenames. Taken from 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// http://msdn.microsoft.com/en-us/library/aa365247.aspx 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char16 kReservedCharacters[] = L"<>:\"/\\|?*\x01\x02\x03\x04\x05\x06\x07" 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) L"\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19" 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) L"\x1A\x1B\x1C\x1D\x1E\x1F"; 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// The maximum number of characters allowed in profile shortcuts' file names. 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Warning: migration code will be needed if this is changed later, since 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// existing shortcuts might no longer be found if the name is generated 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// differently than it was when a shortcut was originally created. 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kMaxProfileShortcutFileNameLength = 64; 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kProfileAvatarBadgeSize = 28; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kShortcutIconSize = 48; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// 2x sized profile avatar icons. Mirrors |kDefaultAvatarIconResources| in 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// profile_info_cache.cc. 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kProfileAvatarIconResources2x[] = { 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_0, 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_1, 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_2, 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_3, 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_4, 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_5, 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_6, 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_7, 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_8, 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_9, 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_10, 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_11, 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_12, 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_13, 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_14, 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_15, 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_16, 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_17, 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_18, 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_19, 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_20, 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_21, 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_22, 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_23, 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_24, 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_25, 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Badges |app_icon_bitmap| with |avatar_bitmap| at the bottom right corner and 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// returns the resulting SkBitmap. 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SkBitmap BadgeIcon(const SkBitmap& app_icon_bitmap, 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap, 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int scale_factor) { 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rlp): Share this chunk of code with 1015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // avatar_menu_button::DrawTaskBarDecoration. 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap source_bitmap = avatar_bitmap; 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if ((avatar_bitmap.width() == scale_factor * profiles::kAvatarIconWidth) && 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (avatar_bitmap.height() == scale_factor * profiles::kAvatarIconHeight)) { 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Shave a couple of columns so the bitmap is more square. So when 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // resized to a square aspect ratio it looks pretty. 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) gfx::Rect frame(scale_factor * profiles::kAvatarIconWidth, 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scale_factor * profiles::kAvatarIconHeight); 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) frame.Inset(scale_factor * 2, 0, scale_factor * 2, 0); 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap.extractSubset(&source_bitmap, gfx::RectToSkIRect(frame)); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int avatar_badge_size = kProfileAvatarBadgeSize; 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (app_icon_bitmap.width() != kShortcutIconSize) { 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_badge_size = 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.width() * kProfileAvatarBadgeSize / kShortcutIconSize; 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkBitmap sk_icon = skia::ImageOperations::Resize( 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source_bitmap, skia::ImageOperations::RESIZE_LANCZOS3, avatar_badge_size, 1212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source_bitmap.height() * avatar_badge_size / source_bitmap.width()); 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Overlay the avatar on the icon, anchoring it to the bottom-right of the 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // icon. 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SkCanvas> offscreen_canvas( 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) skia::CreateBitmapCanvas(app_icon_bitmap.width(), 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.height(), 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) false)); 129b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DCHECK(offscreen_canvas); 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) offscreen_canvas->drawBitmap(app_icon_bitmap, 0, 0); 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) offscreen_canvas->drawBitmap(sk_icon, 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.width() - sk_icon.width(), 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.height() - sk_icon.height()); 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& badged_bitmap = 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) offscreen_canvas->getDevice()->accessBitmap(false); 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap badged_bitmap_copy; 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) badged_bitmap.deepCopyTo(&badged_bitmap_copy, badged_bitmap.getConfig()); 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return badged_bitmap_copy; 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Creates a desktop shortcut icon file (.ico) on the disk for a given profile, 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// badging the browser distribution icon with the profile avatar. 1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns a path to the shortcut icon file on disk, which is empty if this 1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// fails. Use index 0 when assigning the resulting file as the icon. 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath CreateChromeDesktopShortcutIconForProfile( 1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap_1x, 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap_2x) { 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<SkBitmap> app_icon_bitmap(GetAppIconForSize(kShortcutIconSize)); 151b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) if (!app_icon_bitmap) 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(); 1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 154c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) gfx::ImageFamily badged_bitmaps; 155c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) badged_bitmaps.Add(gfx::Image::CreateFrom1xBitmap( 156c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) BadgeIcon(*app_icon_bitmap, avatar_bitmap_1x, 1))); 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap = GetAppIconForSize(IconUtil::kLargeIconSize); 159b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) if (app_icon_bitmap) { 160c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) badged_bitmaps.Add(gfx::Image::CreateFrom1xBitmap( 161c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) BadgeIcon(*app_icon_bitmap, avatar_bitmap_2x, 2))); 162c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Finally, write the .ico file containing this new bitmap. 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath icon_path = 1662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.AppendASCII(profiles::internal::kProfileIconFileName); 167c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!IconUtil::CreateIconFileFromImageFamily(badged_bitmaps, icon_path)) 1682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(); 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return icon_path; 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Gets the user and system directories for desktop shortcuts. Parameters may 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// be NULL if a directory type is not needed. Returns true on success. 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool GetDesktopShortcutsDirectories( 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath* user_shortcuts_directory, 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath* system_shortcuts_directory) { 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (user_shortcuts_directory && 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, ShellUtil::CURRENT_USER, 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory)) { 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (system_shortcuts_directory && 1872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 1882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, ShellUtil::SYSTEM_LEVEL, 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory)) { 1902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns the long form of |path|, which will expand any shortened components 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// like "foo~2" to their full names. 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath ConvertToLongPath(const base::FilePath& path) { 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const size_t length = GetLongPathName(path.value().c_str(), NULL, 0); 2002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (length != 0 && length != path.value().length()) { 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<wchar_t> long_path(length); 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (GetLongPathName(path.value().c_str(), &long_path[0], length) != 0) 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(&long_path[0]); 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return path; 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if the file at |path| is a Chrome shortcut and returns its 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// command line in output parameter |command_line|. 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool IsChromeShortcut(const base::FilePath& path, 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& chrome_exe, 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16* command_line) { 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (path.Extension() != installer::kLnkExt) 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath target_path; 2192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!base::win::ResolveShortcut(path, &target_path, command_line)) 2202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // One of the paths may be in short (elided) form. Compare long paths to 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // ensure these are still properly matched. 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return ConvertToLongPath(target_path) == ConvertToLongPath(chrome_exe); 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Populates |paths| with the file paths of Chrome desktop shortcuts that have 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// the specified |command_line|. If |include_empty_command_lines| is true, 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Chrome desktop shortcuts with empty command lines will also be included. 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ListDesktopShortcutsWithCommandLine(const base::FilePath& chrome_exe, 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& command_line, 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool include_empty_command_lines, 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath>* paths) { 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, NULL)) 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 237868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) base::FileEnumerator enumerator(user_shortcuts_directory, false, 238868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) base::FileEnumerator::FILES); 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (base::FilePath path = enumerator.Next(); !path.empty(); 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) path = enumerator.Next()) { 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 shortcut_command_line; 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!IsChromeShortcut(path, chrome_exe, &shortcut_command_line)) 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue; 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // TODO(asvitkine): Change this to build a CommandLine object and ensure all 2462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // args from |command_line| are present in the shortcut's CommandLine. This 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // will be more robust when |command_line| contains multiple args. 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if ((shortcut_command_line.empty() && include_empty_command_lines) || 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (shortcut_command_line.find(command_line) != string16::npos)) { 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) paths->push_back(path); 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Renames the given desktop shortcut and informs the shell of this change. 256c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool RenameDesktopShortcut(const base::FilePath& old_shortcut_path, 257c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::FilePath& new_shortcut_path) { 258eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch if (!base::Move(old_shortcut_path, new_shortcut_path)) 259c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return false; 260c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 261c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Notify the shell of the rename, which allows the icon to keep its position 262c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // on the desktop when renamed. Note: This only works if either SHCNF_FLUSH or 263c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // SHCNF_FLUSHNOWAIT is specified as a flag. 264c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SHChangeNotify(SHCNE_RENAMEITEM, SHCNF_PATH | SHCNF_FLUSHNOWAIT, 265c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) old_shortcut_path.value().c_str(), 266c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) new_shortcut_path.value().c_str()); 267c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return true; 268c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 269c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Renames an existing Chrome desktop profile shortcut. Must be called on the 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// FILE thread. 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RenameChromeDesktopShortcutForProfile( 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& old_shortcut_filename, 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& new_shortcut_filename) { 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 2762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath system_shortcuts_directory; 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &system_shortcuts_directory)) { 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath old_shortcut_path = 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory.Append(old_shortcut_filename); 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath new_shortcut_path = 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory.Append(new_shortcut_filename); 2882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2897dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch if (base::PathExists(old_shortcut_path)) { 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Rename the old shortcut unless a system-level shortcut exists at the 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // destination, in which case the old shortcut is simply deleted. 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath possible_new_system_shortcut = 2932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory.Append(new_shortcut_filename); 2947dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch if (base::PathExists(possible_new_system_shortcut)) 2957dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::DeleteFile(old_shortcut_path, false); 296c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) else if (!RenameDesktopShortcut(old_shortcut_path, new_shortcut_path)) 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DLOG(ERROR) << "Could not rename Windows profile desktop shortcut."; 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the shortcut does not exist, it may have been renamed by the user. In 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // that case, its name should not be changed. 3012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // It's also possible that a system-level shortcut exists instead - this 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // should only be the case for the original Chrome shortcut from an 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // installation. If that's the case, copy that one over - it will get its 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // properties updated by |CreateOrUpdateDesktopShortcutsForProfile()|. 3052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath possible_old_system_shortcut = 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory.Append(old_shortcut_filename); 3077dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch if (base::PathExists(possible_old_system_shortcut)) 3087dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::CopyFile(possible_old_system_shortcut, new_shortcut_path); 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Updates all desktop shortcuts for the given profile to have the specified 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// parameters. If |create_mode| is CREATE_WHEN_NONE_FOUND, a new shortcut is 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// created if no existing ones were found. Whether non-profile shortcuts should 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// be updated is specified by |action|. Must be called on the FILE thread. 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void CreateOrUpdateDesktopShortcutsForProfile( 3172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 3182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& old_profile_name, 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& profile_name, 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_image_1x, 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_image_2x, 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileShortcutManagerWin::CreateOrUpdateMode create_mode, 3232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileShortcutManagerWin::NonProfileShortcutAction action) { 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 3255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Ensure that the distribution supports creating shortcuts. If it doesn't, 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // the following code may result in NOTREACHED() being hit. 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(distribution->CanCreateDesktopShortcuts()); 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (old_profile_name != profile_name) { 3382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 old_shortcut_filename = 3392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(old_profile_name, 3402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 new_shortcut_filename = 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(profile_name, 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) RenameChromeDesktopShortcutForProfile(old_shortcut_filename, 3452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_filename); 3462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ShellUtil::ShortcutProperties properties(ShellUtil::CURRENT_USER); 3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) installer::Product product(distribution); 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) product.AddDefaultShortcutProperties(chrome_exe, &properties); 3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 3532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Only set the profile-specific properties when |profile_name| is non empty. 3562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If it is empty, it means the shortcut being created should be a regular, 3572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // non-profile Chrome shortcut. 3582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!profile_name.empty()) { 3592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath shortcut_icon = 3602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateChromeDesktopShortcutIconForProfile(profile_path, 3612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_image_1x, 3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_image_2x); 3632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!shortcut_icon.empty()) 3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_icon(shortcut_icon, 0); 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_arguments(command_line); 3662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 3672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Set the arguments explicitly to the empty string to ensure that 3682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // |ShellUtil::CreateOrUpdateShortcut| updates that part of the shortcut. 3692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_arguments(string16()); 3702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 37290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) properties.set_app_id( 37390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ShellIntegration::GetChromiumModelIdForProfile(profile_path)); 37490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ShellUtil::ShortcutOperation operation = 3762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHELL_SHORTCUT_REPLACE_EXISTING; 3772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, 3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action == ProfileShortcutManagerWin::UPDATE_NON_PROFILE_SHORTCUTS, 3812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 3822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (create_mode == ProfileShortcutManagerWin::CREATE_WHEN_NONE_FOUND && 3832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts.empty()) { 3842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 shortcut_name = 3852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(profile_name, 3862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts.push_back(base::FilePath(shortcut_name)); 3882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) operation = ShellUtil::SHELL_SHORTCUT_CREATE_IF_NO_SYSTEM_LEVEL; 3892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < shortcuts.size(); ++i) { 3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath shortcut_name = 3932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts[i].BaseName().RemoveExtension(); 3942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_shortcut_name(shortcut_name.value()); 3952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::CreateOrUpdateShortcut(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 3962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, properties, operation); 3972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if any desktop shortcuts exist with target |chrome_exe|, 4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// regardless of their command line arguments. 4022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool ChromeDesktopShortcutsExist(const base::FilePath& chrome_exe) { 4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 4042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, NULL)) 4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 407868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) base::FileEnumerator enumerator(user_shortcuts_directory, false, 408868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) base::FileEnumerator::FILES); 4092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (base::FilePath path = enumerator.Next(); !path.empty(); 4102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) path = enumerator.Next()) { 4112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (IsChromeShortcut(path, chrome_exe, NULL)) 4122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 4132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 4162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Deletes all desktop shortcuts for the specified profile and also removes the 4192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// corresponding icon file. If |ensure_shortcuts_remain| is true, then a regular 4202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// non-profile shortcut will be created if this function would otherwise delete 4212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// the last Chrome desktop shortcut(s). Must be called on the FILE thread. 4222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void DeleteDesktopShortcutsAndIconFile(const base::FilePath& profile_path, 4232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool ensure_shortcuts_remain) { 4242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 4252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 4272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 4282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 4302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 4332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 4342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 4352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, false, 4362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 4372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 4392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < shortcuts.size(); ++i) { 4407dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch // Use base::DeleteFile() instead of ShellUtil::RemoveShortcut(), as the 4412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // latter causes non-profile taskbar shortcuts to be unpinned. 4427dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::DeleteFile(shortcuts[i], false); 4432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Notify the shell that the shortcut was deleted to ensure desktop refresh. 4442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SHChangeNotify(SHCNE_DELETE, SHCNF_PATH, shortcuts[i].value().c_str(), 4452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NULL); 4462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath icon_path = 4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.AppendASCII(profiles::internal::kProfileIconFileName); 4507dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::DeleteFile(icon_path, false); 4512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If |ensure_shortcuts_remain| is true and deleting this profile caused the 4532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // last shortcuts to be removed, re-create a regular non-profile shortcut. 4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const bool had_shortcuts = !shortcuts.empty(); 4552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (ensure_shortcuts_remain && had_shortcuts && 4562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ChromeDesktopShortcutsExist(chrome_exe)) { 4572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 4582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Ensure that the distribution supports creating shortcuts. If it doesn't, 4592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // the following code may result in NOTREACHED() being hit. 4602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(distribution->CanCreateDesktopShortcuts()); 4612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) installer::Product product(distribution); 4622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::ShortcutProperties properties(ShellUtil::CURRENT_USER); 4642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) product.AddDefaultShortcutProperties(chrome_exe, &properties); 4652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_shortcut_name( 4662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(string16(), 4672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution)); 4682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::CreateOrUpdateShortcut( 4692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHORTCUT_LOCATION_DESKTOP, distribution, properties, 4702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHELL_SHORTCUT_CREATE_IF_NO_SYSTEM_LEVEL); 4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if profile at |profile_path| has any shortcuts. Does not 4752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// consider non-profile shortcuts. Must be called on the FILE thread. 4762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool HasAnyProfileShortcuts(const base::FilePath& profile_path) { 4772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 4782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 4802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 4812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 4822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 4832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 4862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 4872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 4882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, false, 4892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 4902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return !shortcuts.empty(); 4912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Replaces any reserved characters with spaces, and trims the resulting string 4942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// to prevent any leading and trailing spaces. Also makes sure that the 4952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// resulting filename doesn't exceed |kMaxProfileShortcutFileNameLength|. 4962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// TODO(macourteau): find a way to limit the total path's length to MAX_PATH 4972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// instead of limiting the profile's name to |kMaxProfileShortcutFileNameLength| 4982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// characters. 4992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 SanitizeShortcutProfileNameString(const string16& profile_name) { 5002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 sanitized = profile_name; 5012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t pos = sanitized.find_first_of(kReservedCharacters); 5022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (pos != string16::npos) { 5032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sanitized[pos] = L' '; 5042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pos = sanitized.find_first_of(kReservedCharacters, pos + 1); 5052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TrimWhitespace(sanitized, TRIM_LEADING, &sanitized); 5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (sanitized.size() > kMaxProfileShortcutFileNameLength) 5092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sanitized.erase(kMaxProfileShortcutFileNameLength); 5102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TrimWhitespace(sanitized, TRIM_TRAILING, &sanitized); 5112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return sanitized; 5132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns a copied SkBitmap for the given resource id that can be safely passed 5162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// to another thread. 5172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SkBitmap GetImageResourceSkBitmapCopy(int resource_id) { 5182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const gfx::Image image = 5192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ResourceBundle::GetSharedInstance().GetNativeImageNamed(resource_id); 5202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(!image.IsEmpty()); 5212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap* image_bitmap = image.ToSkBitmap(); 5232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap bitmap_copy; 5242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) image_bitmap->deepCopyTo(&bitmap_copy, image_bitmap->getConfig()); 5252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return bitmap_copy; 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace profiles { 5312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace internal { 5322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kProfileIconFileName[] = "Google Profile.ico"; 5342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 GetShortcutFilenameForProfile(const string16& profile_name, 5362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution) { 5372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 shortcut_name; 5382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!profile_name.empty()) { 5392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(SanitizeShortcutProfileNameString(profile_name)); 5402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(L" - "); 5412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(l10n_util::GetStringUTF16(IDS_SHORT_PRODUCT_NAME)); 5422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 5432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(distribution->GetAppShortCutName()); 5442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 5452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return shortcut_name + installer::kLnkExt; 5462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 CreateProfileShortcutFlags(const base::FilePath& profile_path) { 5492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::StringPrintf(L"--%ls=\"%ls\"", 5502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASCIIToUTF16(switches::kProfileDirectory).c_str(), 5512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.BaseName().value().c_str()); 5522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace internal 5552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace profiles 5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ProfileShortcutManager::IsFeatureEnabled() { 5592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return BrowserDistribution::GetDistribution()->CanCreateDesktopShortcuts() && 560eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch !CommandLine::ForCurrentProcess()->HasSwitch(switches::kUserDataDir); 5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProfileShortcutManager* ProfileShortcutManager::Create( 5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProfileManager* manager) { 5665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return new ProfileShortcutManagerWin(manager); 5675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProfileShortcutManagerWin::ProfileShortcutManagerWin(ProfileManager* manager) 5705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : profile_manager_(manager) { 5712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ( 5722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) arraysize(kProfileAvatarIconResources2x), 5732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_manager_->GetProfileInfoCache().GetDefaultAvatarIconCount()); 5742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_manager_->GetProfileInfoCache().AddObserver(this); 5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProfileShortcutManagerWin::~ProfileShortcutManagerWin() { 5792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_manager_->GetProfileInfoCache().RemoveObserver(this); 5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::CreateProfileShortcut( 5832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 5842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, CREATE_WHEN_NONE_FOUND, 5852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 5862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::RemoveProfileShortcuts( 5892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 5902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::PostTask( 5912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::FILE, FROM_HERE, 5922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&DeleteDesktopShortcutsAndIconFile, profile_path, false)); 5932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::HasProfileShortcuts( 5962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 5972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::Callback<void(bool)>& callback) { 5982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::PostTaskAndReplyWithResult( 5992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::FILE, FROM_HERE, 6002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&HasAnyProfileShortcuts, profile_path), callback); 6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::OnProfileAdded( 6042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 60590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (profile_manager_->GetProfileInfoCache().GetNumberOfProfiles() == 2) { 60690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // When the second profile is added, make existing non-profile shortcuts 60790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // point to the first profile and be badged/named appropriately. 6082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(GetOtherProfilePath(profile_path), 6092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPDATE_EXISTING_ONLY, 6102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPDATE_NON_PROFILE_SHORTCUTS); 6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileWasRemoved( 6152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& profile_name) { 6172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache(); 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If there is only one profile remaining, remove the badging information 6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // from an existing shortcut. 6202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const bool deleting_down_to_last_profile = (cache.GetNumberOfProfiles() == 1); 6212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (deleting_down_to_last_profile) { 6222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(cache.GetPathOfProfileAtIndex(0), 6232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPDATE_EXISTING_ONLY, 6242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, 6282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&DeleteDesktopShortcutsAndIconFile, 6292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path, 6302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) deleting_down_to_last_profile)); 6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileNameChanged( 6342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& old_profile_name) { 6362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, UPDATE_EXISTING_ONLY, 6372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileAvatarChanged( 6412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 6422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, UPDATE_EXISTING_ONLY, 6432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath ProfileShortcutManagerWin::GetOtherProfilePath( 6472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 6482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache(); 6492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(2U, cache.GetNumberOfProfiles()); 6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get the index of the current profile, in order to find the index of the 6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // other profile. 6522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t current_profile_index = cache.GetIndexOfProfileWithPath(profile_path); 6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t other_profile_index = (current_profile_index == 0) ? 1 : 0; 6542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return cache.GetPathOfProfileAtIndex(other_profile_index); 6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::CreateOrUpdateShortcutsForProfileAtPath( 6582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateMode create_mode, 6602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NonProfileShortcutAction action) { 6612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileInfoCache* cache = &profile_manager_->GetProfileInfoCache(); 6622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t profile_index = cache->GetIndexOfProfileWithPath(profile_path); 6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (profile_index == std::string::npos) 6642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 6652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool remove_badging = cache->GetNumberOfProfiles() == 1; 6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string16 old_shortcut_appended_name = 6682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetShortcutNameOfProfileAtIndex(profile_index); 6692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 6702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Exit early if the mode is to update existing profile shortcuts only and 6712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // none were ever created for this profile, per the shortcut name not being 6722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // set in the profile info cache. 6732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (old_shortcut_appended_name.empty() && 6742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) create_mode == UPDATE_EXISTING_ONLY && 6752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action == IGNORE_NON_PROFILE_SHORTCUTS) { 6762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 new_shortcut_appended_name; 6802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!remove_badging) 6812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_appended_name = cache->GetNameOfProfileAtIndex(profile_index); 6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap avatar_bitmap_copy_1x; 6842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap avatar_bitmap_copy_2x; 6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!remove_badging) { 6862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const size_t icon_index = 6872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetAvatarIconIndexOfProfileAtIndex(profile_index); 6882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int resource_id_1x = 6892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetDefaultAvatarIconResourceIDAtIndex(icon_index); 6902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int resource_id_2x = kProfileAvatarIconResources2x[icon_index]; 6912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Make a copy of the SkBitmaps to ensure that we can safely use the image 6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // data on the FILE thread. 6932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_1x = GetImageResourceSkBitmapCopy(resource_id_1x); 6942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_2x = GetImageResourceSkBitmapCopy(resource_id_2x); 6955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::PostTask( 6975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::FILE, FROM_HERE, 6982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&CreateOrUpdateDesktopShortcutsForProfile, profile_path, 6992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) old_shortcut_appended_name, new_shortcut_appended_name, 7002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_1x, avatar_bitmap_copy_2x, create_mode, 7012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action)); 7025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->SetShortcutNameOfProfileAtIndex(profile_index, 7042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_appended_name); 7055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 706