profile_shortcut_manager_win.cc revision 90dce4d38c5ff5333bea97d859d4e484e27edf0c
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" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/path_service.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/string16.h" 172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/string_util.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/stringprintf.h" 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/utf_string_conversions.h" 202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/win/shortcut.h" 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/app_icon_win.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/browser_process.h" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile_info_cache_observer.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile_info_util.h" 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile_manager.h" 2690dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "chrome/browser/shell_integration.h" 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/chrome_switches.h" 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/browser_distribution.h" 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/product.h" 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/shell_util.h" 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/browser_thread.h" 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "grit/chrome_unscaled_resources.h" 332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "grit/chromium_strings.h" 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "skia/ext/image_operations.h" 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "skia/ext/platform_canvas.h" 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/l10n/l10n_util.h" 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/resource/resource_bundle.h" 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/icon_util.h" 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/image/image.h" 40c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)#include "ui/gfx/image/image_family.h" 412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "ui/gfx/rect.h" 422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "ui/gfx/skia_util.h" 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::BrowserThread; 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Characters that are not allowed in Windows filenames. Taken from 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// http://msdn.microsoft.com/en-us/library/aa365247.aspx 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char16 kReservedCharacters[] = L"<>:\"/\\|?*\x01\x02\x03\x04\x05\x06\x07" 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) L"\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19" 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) L"\x1A\x1B\x1C\x1D\x1E\x1F"; 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// The maximum number of characters allowed in profile shortcuts' file names. 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Warning: migration code will be needed if this is changed later, since 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// existing shortcuts might no longer be found if the name is generated 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// differently than it was when a shortcut was originally created. 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kMaxProfileShortcutFileNameLength = 64; 592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kProfileAvatarBadgeSize = 28; 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kShortcutIconSize = 48; 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// 2x sized profile avatar icons. Mirrors |kDefaultAvatarIconResources| in 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// profile_info_cache.cc. 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kProfileAvatarIconResources2x[] = { 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_0, 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_1, 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_2, 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_3, 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_4, 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_5, 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_6, 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_7, 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_8, 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_9, 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_10, 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_11, 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_12, 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_13, 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_14, 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_15, 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_16, 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_17, 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_18, 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_19, 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_20, 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_21, 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_22, 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_23, 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_24, 912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_25, 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Badges |app_icon_bitmap| with |avatar_bitmap| at the bottom right corner and 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// returns the resulting SkBitmap. 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SkBitmap BadgeIcon(const SkBitmap& app_icon_bitmap, 972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap, 982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int scale_factor) { 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rlp): Share this chunk of code with 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // avatar_menu_button::DrawTaskBarDecoration. 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap source_bitmap = avatar_bitmap; 1022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if ((avatar_bitmap.width() == scale_factor * profiles::kAvatarIconWidth) && 1032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (avatar_bitmap.height() == scale_factor * profiles::kAvatarIconHeight)) { 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Shave a couple of columns so the bitmap is more square. So when 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // resized to a square aspect ratio it looks pretty. 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) gfx::Rect frame(scale_factor * profiles::kAvatarIconWidth, 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scale_factor * profiles::kAvatarIconHeight); 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) frame.Inset(scale_factor * 2, 0, scale_factor * 2, 0); 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap.extractSubset(&source_bitmap, gfx::RectToSkIRect(frame)); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int avatar_badge_size = kProfileAvatarBadgeSize; 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (app_icon_bitmap.width() != kShortcutIconSize) { 1152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_badge_size = 1162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.width() * kProfileAvatarBadgeSize / kShortcutIconSize; 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkBitmap sk_icon = skia::ImageOperations::Resize( 1192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source_bitmap, skia::ImageOperations::RESIZE_LANCZOS3, avatar_badge_size, 1202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source_bitmap.height() * avatar_badge_size / source_bitmap.width()); 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Overlay the avatar on the icon, anchoring it to the bottom-right of the 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // icon. 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SkCanvas> offscreen_canvas( 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) skia::CreateBitmapCanvas(app_icon_bitmap.width(), 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.height(), 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) false)); 128b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) DCHECK(offscreen_canvas); 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) offscreen_canvas->drawBitmap(app_icon_bitmap, 0, 0); 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) offscreen_canvas->drawBitmap(sk_icon, 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.width() - sk_icon.width(), 1322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.height() - sk_icon.height()); 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& badged_bitmap = 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) offscreen_canvas->getDevice()->accessBitmap(false); 1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap badged_bitmap_copy; 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) badged_bitmap.deepCopyTo(&badged_bitmap_copy, badged_bitmap.getConfig()); 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return badged_bitmap_copy; 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Creates a desktop shortcut icon file (.ico) on the disk for a given profile, 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// badging the browser distribution icon with the profile avatar. 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns a path to the shortcut icon file on disk, which is empty if this 1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// fails. Use index 0 when assigning the resulting file as the icon. 1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath CreateChromeDesktopShortcutIconForProfile( 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap_1x, 1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap_2x) { 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<SkBitmap> app_icon_bitmap(GetAppIconForSize(kShortcutIconSize)); 150b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) if (!app_icon_bitmap) 1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(); 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 153c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) gfx::ImageFamily badged_bitmaps; 154c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) badged_bitmaps.Add(gfx::Image::CreateFrom1xBitmap( 155c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) BadgeIcon(*app_icon_bitmap, avatar_bitmap_1x, 1))); 1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap = GetAppIconForSize(IconUtil::kLargeIconSize); 158b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) if (app_icon_bitmap) { 159c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) badged_bitmaps.Add(gfx::Image::CreateFrom1xBitmap( 160c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) BadgeIcon(*app_icon_bitmap, avatar_bitmap_2x, 2))); 161c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Finally, write the .ico file containing this new bitmap. 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath icon_path = 1652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.AppendASCII(profiles::internal::kProfileIconFileName); 166c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!IconUtil::CreateIconFileFromImageFamily(badged_bitmaps, icon_path)) 1672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(); 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return icon_path; 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Gets the user and system directories for desktop shortcuts. Parameters may 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// be NULL if a directory type is not needed. Returns true on success. 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool GetDesktopShortcutsDirectories( 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath* user_shortcuts_directory, 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath* system_shortcuts_directory) { 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (user_shortcuts_directory && 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, ShellUtil::CURRENT_USER, 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory)) { 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (system_shortcuts_directory && 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 1872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, ShellUtil::SYSTEM_LEVEL, 1882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory)) { 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 1912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 1935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns the long form of |path|, which will expand any shortened components 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// like "foo~2" to their full names. 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath ConvertToLongPath(const base::FilePath& path) { 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const size_t length = GetLongPathName(path.value().c_str(), NULL, 0); 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (length != 0 && length != path.value().length()) { 2002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<wchar_t> long_path(length); 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (GetLongPathName(path.value().c_str(), &long_path[0], length) != 0) 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(&long_path[0]); 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return path; 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if the file at |path| is a Chrome shortcut and returns its 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// command line in output parameter |command_line|. 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool IsChromeShortcut(const base::FilePath& path, 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& chrome_exe, 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16* command_line) { 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (path.Extension() != installer::kLnkExt) 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath target_path; 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!base::win::ResolveShortcut(path, &target_path, command_line)) 2192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 2202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // One of the paths may be in short (elided) form. Compare long paths to 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // ensure these are still properly matched. 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return ConvertToLongPath(target_path) == ConvertToLongPath(chrome_exe); 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Populates |paths| with the file paths of Chrome desktop shortcuts that have 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// the specified |command_line|. If |include_empty_command_lines| is true, 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Chrome desktop shortcuts with empty command lines will also be included. 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ListDesktopShortcutsWithCommandLine(const base::FilePath& chrome_exe, 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& command_line, 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool include_empty_command_lines, 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath>* paths) { 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, NULL)) 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::FileEnumerator enumerator(user_shortcuts_directory, false, 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::FileEnumerator::FILES); 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (base::FilePath path = enumerator.Next(); !path.empty(); 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) path = enumerator.Next()) { 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 shortcut_command_line; 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!IsChromeShortcut(path, chrome_exe, &shortcut_command_line)) 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue; 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // TODO(asvitkine): Change this to build a CommandLine object and ensure all 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // args from |command_line| are present in the shortcut's CommandLine. This 2462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // will be more robust when |command_line| contains multiple args. 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if ((shortcut_command_line.empty() && include_empty_command_lines) || 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (shortcut_command_line.find(command_line) != string16::npos)) { 2492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) paths->push_back(path); 2502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 254c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)// Renames the given desktop shortcut and informs the shell of this change. 255c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)bool RenameDesktopShortcut(const base::FilePath& old_shortcut_path, 256c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::FilePath& new_shortcut_path) { 257c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) if (!file_util::Move(old_shortcut_path, new_shortcut_path)) 258c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return false; 259c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 260c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // Notify the shell of the rename, which allows the icon to keep its position 261c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // on the desktop when renamed. Note: This only works if either SHCNF_FLUSH or 262c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) // SHCNF_FLUSHNOWAIT is specified as a flag. 263c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) SHChangeNotify(SHCNE_RENAMEITEM, SHCNF_PATH | SHCNF_FLUSHNOWAIT, 264c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) old_shortcut_path.value().c_str(), 265c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) new_shortcut_path.value().c_str()); 266c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return true; 267c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles)} 268c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Renames an existing Chrome desktop profile shortcut. Must be called on the 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// FILE thread. 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RenameChromeDesktopShortcutForProfile( 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& old_shortcut_filename, 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& new_shortcut_filename) { 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath system_shortcuts_directory; 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &system_shortcuts_directory)) { 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath old_shortcut_path = 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory.Append(old_shortcut_filename); 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath new_shortcut_path = 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory.Append(new_shortcut_filename); 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (file_util::PathExists(old_shortcut_path)) { 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Rename the old shortcut unless a system-level shortcut exists at the 2902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // destination, in which case the old shortcut is simply deleted. 2912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath possible_new_system_shortcut = 2922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory.Append(new_shortcut_filename); 2932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (file_util::PathExists(possible_new_system_shortcut)) 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::Delete(old_shortcut_path, false); 295c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) else if (!RenameDesktopShortcut(old_shortcut_path, new_shortcut_path)) 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DLOG(ERROR) << "Could not rename Windows profile desktop shortcut."; 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the shortcut does not exist, it may have been renamed by the user. In 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // that case, its name should not be changed. 3002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // It's also possible that a system-level shortcut exists instead - this 3012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // should only be the case for the original Chrome shortcut from an 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // installation. If that's the case, copy that one over - it will get its 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // properties updated by |CreateOrUpdateDesktopShortcutsForProfile()|. 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath possible_old_system_shortcut = 3052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory.Append(old_shortcut_filename); 3062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (file_util::PathExists(possible_old_system_shortcut)) 3072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::CopyFile(possible_old_system_shortcut, new_shortcut_path); 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Updates all desktop shortcuts for the given profile to have the specified 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// parameters. If |create_mode| is CREATE_WHEN_NONE_FOUND, a new shortcut is 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// created if no existing ones were found. Whether non-profile shortcuts should 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// be updated is specified by |action|. Must be called on the FILE thread. 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void CreateOrUpdateDesktopShortcutsForProfile( 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 3172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& old_profile_name, 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& profile_name, 3192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_image_1x, 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_image_2x, 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileShortcutManagerWin::CreateOrUpdateMode create_mode, 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileShortcutManagerWin::NonProfileShortcutAction action) { 3235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 3245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 3265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 3275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Ensure that the distribution supports creating shortcuts. If it doesn't, 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // the following code may result in NOTREACHED() being hit. 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(distribution->CanCreateDesktopShortcuts()); 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (old_profile_name != profile_name) { 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 old_shortcut_filename = 3382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(old_profile_name, 3392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 new_shortcut_filename = 3412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(profile_name, 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) RenameChromeDesktopShortcutForProfile(old_shortcut_filename, 3442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_filename); 3452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ShellUtil::ShortcutProperties properties(ShellUtil::CURRENT_USER); 3482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) installer::Product product(distribution); 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) product.AddDefaultShortcutProperties(chrome_exe, &properties); 3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 3522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 3532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Only set the profile-specific properties when |profile_name| is non empty. 3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If it is empty, it means the shortcut being created should be a regular, 3562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // non-profile Chrome shortcut. 3572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!profile_name.empty()) { 3582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath shortcut_icon = 3592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateChromeDesktopShortcutIconForProfile(profile_path, 3602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_image_1x, 3612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_image_2x); 3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!shortcut_icon.empty()) 3632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_icon(shortcut_icon, 0); 3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_arguments(command_line); 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 3662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Set the arguments explicitly to the empty string to ensure that 3672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // |ShellUtil::CreateOrUpdateShortcut| updates that part of the shortcut. 3682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_arguments(string16()); 3692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 37190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) properties.set_app_id( 37290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ShellIntegration::GetChromiumModelIdForProfile(profile_path)); 37390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ShellUtil::ShortcutOperation operation = 3752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHELL_SHORTCUT_REPLACE_EXISTING; 3762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 3782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action == ProfileShortcutManagerWin::UPDATE_NON_PROFILE_SHORTCUTS, 3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 3812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (create_mode == ProfileShortcutManagerWin::CREATE_WHEN_NONE_FOUND && 3822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts.empty()) { 3832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 shortcut_name = 3842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(profile_name, 3852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts.push_back(base::FilePath(shortcut_name)); 3872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) operation = ShellUtil::SHELL_SHORTCUT_CREATE_IF_NO_SYSTEM_LEVEL; 3882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < shortcuts.size(); ++i) { 3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath shortcut_name = 3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts[i].BaseName().RemoveExtension(); 3932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_shortcut_name(shortcut_name.value()); 3942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::CreateOrUpdateShortcut(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 3952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, properties, operation); 3962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if any desktop shortcuts exist with target |chrome_exe|, 4002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// regardless of their command line arguments. 4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool ChromeDesktopShortcutsExist(const base::FilePath& chrome_exe) { 4022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, NULL)) 4042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::FileEnumerator enumerator(user_shortcuts_directory, false, 4072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::FileEnumerator::FILES); 4082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (base::FilePath path = enumerator.Next(); !path.empty(); 4092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) path = enumerator.Next()) { 4102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (IsChromeShortcut(path, chrome_exe, NULL)) 4112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 4122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 4152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Deletes all desktop shortcuts for the specified profile and also removes the 4182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// corresponding icon file. If |ensure_shortcuts_remain| is true, then a regular 4192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// non-profile shortcut will be created if this function would otherwise delete 4202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// the last Chrome desktop shortcut(s). Must be called on the FILE thread. 4212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void DeleteDesktopShortcutsAndIconFile(const base::FilePath& profile_path, 4222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool ensure_shortcuts_remain) { 4232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 4242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 4262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 4272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 4282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 4322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 4332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 4342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, false, 4352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 4362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 4382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < shortcuts.size(); ++i) { 4392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Use file_util::Delete() instead of ShellUtil::RemoveShortcut(), as the 4402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // latter causes non-profile taskbar shortcuts to be unpinned. 4412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::Delete(shortcuts[i], false); 4422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Notify the shell that the shortcut was deleted to ensure desktop refresh. 4432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SHChangeNotify(SHCNE_DELETE, SHCNF_PATH, shortcuts[i].value().c_str(), 4442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NULL); 4452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath icon_path = 4482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.AppendASCII(profiles::internal::kProfileIconFileName); 4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::Delete(icon_path, false); 4502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If |ensure_shortcuts_remain| is true and deleting this profile caused the 4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // last shortcuts to be removed, re-create a regular non-profile shortcut. 4532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const bool had_shortcuts = !shortcuts.empty(); 4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (ensure_shortcuts_remain && had_shortcuts && 4552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ChromeDesktopShortcutsExist(chrome_exe)) { 4562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 4572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Ensure that the distribution supports creating shortcuts. If it doesn't, 4582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // the following code may result in NOTREACHED() being hit. 4592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(distribution->CanCreateDesktopShortcuts()); 4602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) installer::Product product(distribution); 4612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::ShortcutProperties properties(ShellUtil::CURRENT_USER); 4632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) product.AddDefaultShortcutProperties(chrome_exe, &properties); 4642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_shortcut_name( 4652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(string16(), 4662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution)); 4672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::CreateOrUpdateShortcut( 4682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHORTCUT_LOCATION_DESKTOP, distribution, properties, 4692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHELL_SHORTCUT_CREATE_IF_NO_SYSTEM_LEVEL); 4702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if profile at |profile_path| has any shortcuts. Does not 4742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// consider non-profile shortcuts. Must be called on the FILE thread. 4752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool HasAnyProfileShortcuts(const base::FilePath& profile_path) { 4762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 4772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 4792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 4802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 4812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 4822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 4852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 4862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 4872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, false, 4882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 4892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return !shortcuts.empty(); 4902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Replaces any reserved characters with spaces, and trims the resulting string 4932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// to prevent any leading and trailing spaces. Also makes sure that the 4942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// resulting filename doesn't exceed |kMaxProfileShortcutFileNameLength|. 4952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// TODO(macourteau): find a way to limit the total path's length to MAX_PATH 4962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// instead of limiting the profile's name to |kMaxProfileShortcutFileNameLength| 4972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// characters. 4982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 SanitizeShortcutProfileNameString(const string16& profile_name) { 4992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 sanitized = profile_name; 5002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t pos = sanitized.find_first_of(kReservedCharacters); 5012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (pos != string16::npos) { 5022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sanitized[pos] = L' '; 5032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pos = sanitized.find_first_of(kReservedCharacters, pos + 1); 5042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 5052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TrimWhitespace(sanitized, TRIM_LEADING, &sanitized); 5072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (sanitized.size() > kMaxProfileShortcutFileNameLength) 5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sanitized.erase(kMaxProfileShortcutFileNameLength); 5092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TrimWhitespace(sanitized, TRIM_TRAILING, &sanitized); 5102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return sanitized; 5122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns a copied SkBitmap for the given resource id that can be safely passed 5152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// to another thread. 5162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SkBitmap GetImageResourceSkBitmapCopy(int resource_id) { 5172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const gfx::Image image = 5182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ResourceBundle::GetSharedInstance().GetNativeImageNamed(resource_id); 5192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(!image.IsEmpty()); 5202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap* image_bitmap = image.ToSkBitmap(); 5222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap bitmap_copy; 5232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) image_bitmap->deepCopyTo(&bitmap_copy, image_bitmap->getConfig()); 5242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return bitmap_copy; 5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace profiles { 5302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace internal { 5312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kProfileIconFileName[] = "Google Profile.ico"; 5332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 GetShortcutFilenameForProfile(const string16& profile_name, 5352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution) { 5362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 shortcut_name; 5372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!profile_name.empty()) { 5382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(SanitizeShortcutProfileNameString(profile_name)); 5392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(L" - "); 5402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(l10n_util::GetStringUTF16(IDS_SHORT_PRODUCT_NAME)); 5412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 5422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(distribution->GetAppShortCutName()); 5432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 5442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return shortcut_name + installer::kLnkExt; 5452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 CreateProfileShortcutFlags(const base::FilePath& profile_path) { 5482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::StringPrintf(L"--%ls=\"%ls\"", 5492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASCIIToUTF16(switches::kProfileDirectory).c_str(), 5502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.BaseName().value().c_str()); 5512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace internal 5542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace profiles 5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ProfileShortcutManager::IsFeatureEnabled() { 5582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return BrowserDistribution::GetDistribution()->CanCreateDesktopShortcuts() && 5592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !CommandLine::ForCurrentProcess()->HasSwitch(switches::kUserDataDir) && 5602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !CommandLine::ForCurrentProcess()->HasSwitch(switches::kShowAppList); 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::OnProfileWillBeRemoved( 6152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 6165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileWasRemoved( 6192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& profile_name) { 6212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache(); 6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If there is only one profile remaining, remove the badging information 6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // from an existing shortcut. 6242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const bool deleting_down_to_last_profile = (cache.GetNumberOfProfiles() == 1); 6252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (deleting_down_to_last_profile) { 6262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(cache.GetPathOfProfileAtIndex(0), 6272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPDATE_EXISTING_ONLY, 6282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, 6322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&DeleteDesktopShortcutsAndIconFile, 6332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path, 6342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) deleting_down_to_last_profile)); 6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileNameChanged( 6382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& old_profile_name) { 6402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, UPDATE_EXISTING_ONLY, 6412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileAvatarChanged( 6452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 6462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, UPDATE_EXISTING_ONLY, 6472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath ProfileShortcutManagerWin::GetOtherProfilePath( 6512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 6522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache(); 6532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(2U, cache.GetNumberOfProfiles()); 6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get the index of the current profile, in order to find the index of the 6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // other profile. 6562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t current_profile_index = cache.GetIndexOfProfileWithPath(profile_path); 6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t other_profile_index = (current_profile_index == 0) ? 1 : 0; 6582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return cache.GetPathOfProfileAtIndex(other_profile_index); 6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::CreateOrUpdateShortcutsForProfileAtPath( 6622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateMode create_mode, 6642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NonProfileShortcutAction action) { 6652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileInfoCache* cache = &profile_manager_->GetProfileInfoCache(); 6662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t profile_index = cache->GetIndexOfProfileWithPath(profile_path); 6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (profile_index == std::string::npos) 6682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 6692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool remove_badging = cache->GetNumberOfProfiles() == 1; 6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string16 old_shortcut_appended_name = 6722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetShortcutNameOfProfileAtIndex(profile_index); 6732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 6742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Exit early if the mode is to update existing profile shortcuts only and 6752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // none were ever created for this profile, per the shortcut name not being 6762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // set in the profile info cache. 6772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (old_shortcut_appended_name.empty() && 6782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) create_mode == UPDATE_EXISTING_ONLY && 6792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action == IGNORE_NON_PROFILE_SHORTCUTS) { 6802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 new_shortcut_appended_name; 6842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!remove_badging) 6852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_appended_name = cache->GetNameOfProfileAtIndex(profile_index); 6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap avatar_bitmap_copy_1x; 6882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap avatar_bitmap_copy_2x; 6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!remove_badging) { 6902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const size_t icon_index = 6912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetAvatarIconIndexOfProfileAtIndex(profile_index); 6922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int resource_id_1x = 6932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetDefaultAvatarIconResourceIDAtIndex(icon_index); 6942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int resource_id_2x = kProfileAvatarIconResources2x[icon_index]; 6952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Make a copy of the SkBitmaps to ensure that we can safely use the image 6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // data on the FILE thread. 6972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_1x = GetImageResourceSkBitmapCopy(resource_id_1x); 6982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_2x = GetImageResourceSkBitmapCopy(resource_id_2x); 6995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::PostTask( 7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::FILE, FROM_HERE, 7022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&CreateOrUpdateDesktopShortcutsForProfile, profile_path, 7032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) old_shortcut_appended_name, new_shortcut_appended_name, 7042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_1x, avatar_bitmap_copy_2x, create_mode, 7052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action)); 7065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->SetShortcutNameOfProfileAtIndex(profile_index, 7082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_appended_name); 7095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 710