profile_shortcut_manager_win.cc revision 2a99a7e74a7f215066514fe81d2bfa6639d9eddd
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" 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/common/chrome_switches.h" 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/browser_distribution.h" 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/product.h" 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/installer/util/shell_util.h" 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/browser_thread.h" 312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "grit/chrome_unscaled_resources.h" 322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "grit/chromium_strings.h" 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "skia/ext/image_operations.h" 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "skia/ext/platform_canvas.h" 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/l10n/l10n_util.h" 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/resource/resource_bundle.h" 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/icon_util.h" 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/gfx/image/image.h" 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "ui/gfx/rect.h" 402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "ui/gfx/skia_util.h" 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)using content::BrowserThread; 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Characters that are not allowed in Windows filenames. Taken from 472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// http://msdn.microsoft.com/en-us/library/aa365247.aspx 482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char16 kReservedCharacters[] = L"<>:\"/\\|?*\x01\x02\x03\x04\x05\x06\x07" 492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) L"\x08\x09\x0A\x0B\x0C\x0D\x0E\x0F\x10\x11\x12\x13\x14\x15\x16\x17\x18\x19" 502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) L"\x1A\x1B\x1C\x1D\x1E\x1F"; 512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// The maximum number of characters allowed in profile shortcuts' file names. 532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Warning: migration code will be needed if this is changed later, since 542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// existing shortcuts might no longer be found if the name is generated 552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// differently than it was when a shortcut was originally created. 562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kMaxProfileShortcutFileNameLength = 64; 572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kProfileAvatarBadgeSize = 28; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const int kShortcutIconSize = 48; 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// 2x sized profile avatar icons. Mirrors |kDefaultAvatarIconResources| in 622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// profile_info_cache.cc. 632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const int kProfileAvatarIconResources2x[] = { 642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_0, 652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_1, 662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_2, 672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_3, 682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_4, 692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_5, 702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_6, 712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_7, 722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_8, 732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_9, 742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_10, 752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_11, 762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_12, 772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_13, 782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_14, 792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_15, 802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_16, 812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_17, 822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_18, 832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_19, 842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_20, 852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_21, 862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_22, 872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_23, 882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_24, 892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IDR_PROFILE_AVATAR_2X_25, 902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)}; 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Badges |app_icon_bitmap| with |avatar_bitmap| at the bottom right corner and 932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// returns the resulting SkBitmap. 942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SkBitmap BadgeIcon(const SkBitmap& app_icon_bitmap, 952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap, 962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int scale_factor) { 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // TODO(rlp): Share this chunk of code with 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // avatar_menu_button::DrawTaskBarDecoration. 992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap source_bitmap = avatar_bitmap; 1002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if ((avatar_bitmap.width() == scale_factor * profiles::kAvatarIconWidth) && 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (avatar_bitmap.height() == scale_factor * profiles::kAvatarIconHeight)) { 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Shave a couple of columns so the bitmap is more square. So when 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // resized to a square aspect ratio it looks pretty. 1042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) gfx::Rect frame(scale_factor * profiles::kAvatarIconWidth, 1052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scale_factor * profiles::kAvatarIconHeight); 1062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) frame.Inset(scale_factor * 2, 0, scale_factor * 2, 0); 1072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap.extractSubset(&source_bitmap, gfx::RectToSkIRect(frame)); 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int avatar_badge_size = kProfileAvatarBadgeSize; 1122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (app_icon_bitmap.width() != kShortcutIconSize) { 1132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_badge_size = 1142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.width() * kProfileAvatarBadgeSize / kShortcutIconSize; 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) SkBitmap sk_icon = skia::ImageOperations::Resize( 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source_bitmap, skia::ImageOperations::RESIZE_LANCZOS3, avatar_badge_size, 1182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source_bitmap.height() * avatar_badge_size / source_bitmap.width()); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Overlay the avatar on the icon, anchoring it to the bottom-right of the 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // icon. 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_ptr<SkCanvas> offscreen_canvas( 1232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) skia::CreateBitmapCanvas(app_icon_bitmap.width(), 1242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.height(), 1255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) false)); 1265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(offscreen_canvas.get()); 1272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) offscreen_canvas->drawBitmap(app_icon_bitmap, 0, 0); 1282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) offscreen_canvas->drawBitmap(sk_icon, 1292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.width() - sk_icon.width(), 1302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap.height() - sk_icon.height()); 1312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& badged_bitmap = 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) offscreen_canvas->getDevice()->accessBitmap(false); 1332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap badged_bitmap_copy; 1342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) badged_bitmap.deepCopyTo(&badged_bitmap_copy, badged_bitmap.getConfig()); 1352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return badged_bitmap_copy; 1362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 1372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Creates a desktop shortcut icon file (.ico) on the disk for a given profile, 1392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// badging the browser distribution icon with the profile avatar. 1402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns a path to the shortcut icon file on disk, which is empty if this 1412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// fails. Use index 0 when assigning the resulting file as the icon. 1422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath CreateChromeDesktopShortcutIconForProfile( 1432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 1442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap_1x, 1452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_bitmap_2x) { 1462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 1472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) scoped_ptr<SkBitmap> app_icon_bitmap(GetAppIconForSize(kShortcutIconSize)); 1482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!app_icon_bitmap.get()) 1492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(); 1502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap badged_bitmap = BadgeIcon(*app_icon_bitmap, 1522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_1x, 1); 1532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 1542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap large_badged_bitmap; 1552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) app_icon_bitmap = GetAppIconForSize(IconUtil::kLargeIconSize); 1562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (app_icon_bitmap.get()) 1572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) large_badged_bitmap = BadgeIcon(*app_icon_bitmap, avatar_bitmap_2x, 2); 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Finally, write the .ico file containing this new bitmap. 1602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath icon_path = 1612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.AppendASCII(profiles::internal::kProfileIconFileName); 1622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!IconUtil::CreateIconFileFromSkBitmap(badged_bitmap, large_badged_bitmap, 1632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) icon_path)) 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(); 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return icon_path; 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Gets the user and system directories for desktop shortcuts. Parameters may 1702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// be NULL if a directory type is not needed. Returns true on success. 1712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool GetDesktopShortcutsDirectories( 1722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath* user_shortcuts_directory, 1732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath* system_shortcuts_directory) { 1742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 1752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (user_shortcuts_directory && 1762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 1772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, ShellUtil::CURRENT_USER, 1782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory)) { 1792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 1812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (system_shortcuts_directory && 1832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ShellUtil::GetShortcutPath(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 1842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, ShellUtil::SYSTEM_LEVEL, 1852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory)) { 1862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 1872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 1882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 1892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns the long form of |path|, which will expand any shortened components 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// like "foo~2" to their full names. 1942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath ConvertToLongPath(const base::FilePath& path) { 1952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const size_t length = GetLongPathName(path.value().c_str(), NULL, 0); 1962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (length != 0 && length != path.value().length()) { 1972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<wchar_t> long_path(length); 1982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (GetLongPathName(path.value().c_str(), &long_path[0], length) != 0) 1992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::FilePath(&long_path[0]); 2002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return path; 2022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if the file at |path| is a Chrome shortcut and returns its 2052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// command line in output parameter |command_line|. 2062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool IsChromeShortcut(const base::FilePath& path, 2072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& chrome_exe, 2082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16* command_line) { 2092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 2102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (path.Extension() != installer::kLnkExt) 2122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 2132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath target_path; 2152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!base::win::ResolveShortcut(path, &target_path, command_line)) 2162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 2172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // One of the paths may be in short (elided) form. Compare long paths to 2182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // ensure these are still properly matched. 2192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return ConvertToLongPath(target_path) == ConvertToLongPath(chrome_exe); 2202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 2212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Populates |paths| with the file paths of Chrome desktop shortcuts that have 2232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// the specified |command_line|. If |include_empty_command_lines| is true, 2242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Chrome desktop shortcuts with empty command lines will also be included. 2252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ListDesktopShortcutsWithCommandLine(const base::FilePath& chrome_exe, 2262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& command_line, 2272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool include_empty_command_lines, 2282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath>* paths) { 2292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 2302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, NULL)) 2312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::FileEnumerator enumerator(user_shortcuts_directory, false, 2342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::FileEnumerator::FILES); 2352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (base::FilePath path = enumerator.Next(); !path.empty(); 2362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) path = enumerator.Next()) { 2372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 shortcut_command_line; 2382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!IsChromeShortcut(path, chrome_exe, &shortcut_command_line)) 2392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) continue; 2402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // TODO(asvitkine): Change this to build a CommandLine object and ensure all 2422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // args from |command_line| are present in the shortcut's CommandLine. This 2432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // will be more robust when |command_line| contains multiple args. 2442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if ((shortcut_command_line.empty() && include_empty_command_lines) || 2452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) (shortcut_command_line.find(command_line) != string16::npos)) { 2462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) paths->push_back(path); 2472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Renames an existing Chrome desktop profile shortcut. Must be called on the 2525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// FILE thread. 2535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void RenameChromeDesktopShortcutForProfile( 2542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& old_shortcut_filename, 2552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& new_shortcut_filename) { 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 2572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 2592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath system_shortcuts_directory; 2602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, 2612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &system_shortcuts_directory)) { 2622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 2632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 2642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath old_shortcut_path = 2662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory.Append(old_shortcut_filename); 2672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath new_shortcut_path = 2682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) user_shortcuts_directory.Append(new_shortcut_filename); 2692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 2702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (file_util::PathExists(old_shortcut_path)) { 2712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Rename the old shortcut unless a system-level shortcut exists at the 2722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // destination, in which case the old shortcut is simply deleted. 2732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath possible_new_system_shortcut = 2742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory.Append(new_shortcut_filename); 2752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (file_util::PathExists(possible_new_system_shortcut)) 2762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::Delete(old_shortcut_path, false); 2772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) else if (!file_util::Move(old_shortcut_path, new_shortcut_path)) 2782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DLOG(ERROR) << "Could not rename Windows profile desktop shortcut."; 2792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 2802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If the shortcut does not exist, it may have been renamed by the user. In 2812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // that case, its name should not be changed. 2822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // It's also possible that a system-level shortcut exists instead - this 2832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // should only be the case for the original Chrome shortcut from an 2842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // installation. If that's the case, copy that one over - it will get its 2852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // properties updated by |CreateOrUpdateDesktopShortcutsForProfile()|. 2862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath possible_old_system_shortcut = 2872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) system_shortcuts_directory.Append(old_shortcut_filename); 2882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (file_util::PathExists(possible_old_system_shortcut)) 2892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::CopyFile(possible_old_system_shortcut, new_shortcut_path); 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Updates all desktop shortcuts for the given profile to have the specified 2942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// parameters. If |create_mode| is CREATE_WHEN_NONE_FOUND, a new shortcut is 2952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// created if no existing ones were found. Whether non-profile shortcuts should 2962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// be updated is specified by |action|. Must be called on the FILE thread. 2972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void CreateOrUpdateDesktopShortcutsForProfile( 2982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 2992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16& old_profile_name, 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& profile_name, 3012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_image_1x, 3022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap& avatar_image_2x, 3032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileShortcutManagerWin::CreateOrUpdateMode create_mode, 3042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileShortcutManagerWin::NonProfileShortcutAction action) { 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 3085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 3095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NOTREACHED(); 3105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 3115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 3122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 3142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Ensure that the distribution supports creating shortcuts. If it doesn't, 3152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // the following code may result in NOTREACHED() being hit. 3162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(distribution->CanCreateDesktopShortcuts()); 3172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (old_profile_name != profile_name) { 3192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 old_shortcut_filename = 3202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(old_profile_name, 3212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 new_shortcut_filename = 3232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(profile_name, 3242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) RenameChromeDesktopShortcutForProfile(old_shortcut_filename, 3262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_filename); 3272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ShellUtil::ShortcutProperties properties(ShellUtil::CURRENT_USER); 3302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) installer::Product product(distribution); 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) product.AddDefaultShortcutProperties(chrome_exe, &properties); 3322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 3342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 3352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Only set the profile-specific properties when |profile_name| is non empty. 3372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If it is empty, it means the shortcut being created should be a regular, 3382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // non-profile Chrome shortcut. 3392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!profile_name.empty()) { 3402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath shortcut_icon = 3412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateChromeDesktopShortcutIconForProfile(profile_path, 3422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_image_1x, 3432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_image_2x); 3442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!shortcut_icon.empty()) 3452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_icon(shortcut_icon, 0); 3462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_arguments(command_line); 3472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 3482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Set the arguments explicitly to the empty string to ensure that 3492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // |ShellUtil::CreateOrUpdateShortcut| updates that part of the shortcut. 3502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_arguments(string16()); 3512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ShellUtil::ShortcutOperation operation = 3542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHELL_SHORTCUT_REPLACE_EXISTING; 3552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 3572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, 3582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action == ProfileShortcutManagerWin::UPDATE_NON_PROFILE_SHORTCUTS, 3592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 3602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (create_mode == ProfileShortcutManagerWin::CREATE_WHEN_NONE_FOUND && 3612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts.empty()) { 3622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 shortcut_name = 3632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(profile_name, 3642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution); 3652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts.push_back(base::FilePath(shortcut_name)); 3662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) operation = ShellUtil::SHELL_SHORTCUT_CREATE_IF_NO_SYSTEM_LEVEL; 3672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < shortcuts.size(); ++i) { 3702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath shortcut_name = 3712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcuts[i].BaseName().RemoveExtension(); 3722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_shortcut_name(shortcut_name.value()); 3732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::CreateOrUpdateShortcut(ShellUtil::SHORTCUT_LOCATION_DESKTOP, 3742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution, properties, operation); 3752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if any desktop shortcuts exist with target |chrome_exe|, 3792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// regardless of their command line arguments. 3802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool ChromeDesktopShortcutsExist(const base::FilePath& chrome_exe) { 3812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath user_shortcuts_directory; 3822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!GetDesktopShortcutsDirectories(&user_shortcuts_directory, NULL)) 3832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 3842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::FileEnumerator enumerator(user_shortcuts_directory, false, 3862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::FileEnumerator::FILES); 3872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (base::FilePath path = enumerator.Next(); !path.empty(); 3882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) path = enumerator.Next()) { 3892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (IsChromeShortcut(path, chrome_exe, NULL)) 3902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return true; 3912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 3922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 3942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 3952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 3962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Deletes all desktop shortcuts for the specified profile and also removes the 3972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// corresponding icon file. If |ensure_shortcuts_remain| is true, then a regular 3982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// non-profile shortcut will be created if this function would otherwise delete 3992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// the last Chrome desktop shortcut(s). Must be called on the FILE thread. 4002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void DeleteDesktopShortcutsAndIconFile(const base::FilePath& profile_path, 4012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool ensure_shortcuts_remain) { 4022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 4032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 4052a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 4062a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 4072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 4082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 4112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 4122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 4132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, false, 4142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 4152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 4172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) for (size_t i = 0; i < shortcuts.size(); ++i) { 4182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Use file_util::Delete() instead of ShellUtil::RemoveShortcut(), as the 4192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // latter causes non-profile taskbar shortcuts to be unpinned. 4202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::Delete(shortcuts[i], false); 4212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Notify the shell that the shortcut was deleted to ensure desktop refresh. 4222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SHChangeNotify(SHCNE_DELETE, SHCNF_PATH, shortcuts[i].value().c_str(), 4232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NULL); 4242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath icon_path = 4272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.AppendASCII(profiles::internal::kProfileIconFileName); 4282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) file_util::Delete(icon_path, false); 4292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // If |ensure_shortcuts_remain| is true and deleting this profile caused the 4312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // last shortcuts to be removed, re-create a regular non-profile shortcut. 4322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const bool had_shortcuts = !shortcuts.empty(); 4332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (ensure_shortcuts_remain && had_shortcuts && 4342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !ChromeDesktopShortcutsExist(chrome_exe)) { 4352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution = BrowserDistribution::GetDistribution(); 4362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Ensure that the distribution supports creating shortcuts. If it doesn't, 4372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // the following code may result in NOTREACHED() being hit. 4382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(distribution->CanCreateDesktopShortcuts()); 4392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) installer::Product product(distribution); 4402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::ShortcutProperties properties(ShellUtil::CURRENT_USER); 4422a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) product.AddDefaultShortcutProperties(chrome_exe, &properties); 4432a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) properties.set_shortcut_name( 4442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::GetShortcutFilenameForProfile(string16(), 4452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) distribution)); 4462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::CreateOrUpdateShortcut( 4472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHORTCUT_LOCATION_DESKTOP, distribution, properties, 4482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ShellUtil::SHELL_SHORTCUT_CREATE_IF_NO_SYSTEM_LEVEL); 4492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns true if profile at |profile_path| has any shortcuts. Does not 4532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// consider non-profile shortcuts. Must be called on the FILE thread. 4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool HasAnyProfileShortcuts(const base::FilePath& profile_path) { 4552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(BrowserThread::CurrentlyOn(BrowserThread::FILE)); 4562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath chrome_exe; 4582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!PathService::Get(base::FILE_EXE, &chrome_exe)) { 4592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NOTREACHED(); 4602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return false; 4612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const string16 command_line = 4642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profiles::internal::CreateProfileShortcutFlags(profile_path); 4652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) std::vector<base::FilePath> shortcuts; 4662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ListDesktopShortcutsWithCommandLine(chrome_exe, command_line, false, 4672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) &shortcuts); 4682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return !shortcuts.empty(); 4692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Replaces any reserved characters with spaces, and trims the resulting string 4722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// to prevent any leading and trailing spaces. Also makes sure that the 4732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// resulting filename doesn't exceed |kMaxProfileShortcutFileNameLength|. 4742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// TODO(macourteau): find a way to limit the total path's length to MAX_PATH 4752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// instead of limiting the profile's name to |kMaxProfileShortcutFileNameLength| 4762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// characters. 4772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 SanitizeShortcutProfileNameString(const string16& profile_name) { 4782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 sanitized = profile_name; 4792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t pos = sanitized.find_first_of(kReservedCharacters); 4802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) while (pos != string16::npos) { 4812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sanitized[pos] = L' '; 4822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) pos = sanitized.find_first_of(kReservedCharacters, pos + 1); 4832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 4842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TrimWhitespace(sanitized, TRIM_LEADING, &sanitized); 4862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (sanitized.size() > kMaxProfileShortcutFileNameLength) 4872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) sanitized.erase(kMaxProfileShortcutFileNameLength); 4882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) TrimWhitespace(sanitized, TRIM_TRAILING, &sanitized); 4892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return sanitized; 4912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 4922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 4932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// Returns a copied SkBitmap for the given resource id that can be safely passed 4942a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)// to another thread. 4952a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)SkBitmap GetImageResourceSkBitmapCopy(int resource_id) { 4962a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const gfx::Image image = 4972a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ResourceBundle::GetSharedInstance().GetNativeImageNamed(resource_id); 4982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK(!image.IsEmpty()); 4992a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5002a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const SkBitmap* image_bitmap = image.ToSkBitmap(); 5012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap bitmap_copy; 5022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) image_bitmap->deepCopyTo(&bitmap_copy, image_bitmap->getConfig()); 5032a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return bitmap_copy; 5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace profiles { 5092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)namespace internal { 5102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)const char kProfileIconFileName[] = "Google Profile.ico"; 5122a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5132a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 GetShortcutFilenameForProfile(const string16& profile_name, 5142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserDistribution* distribution) { 5152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 shortcut_name; 5162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!profile_name.empty()) { 5172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(SanitizeShortcutProfileNameString(profile_name)); 5182a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(L" - "); 5192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(l10n_util::GetStringUTF16(IDS_SHORT_PRODUCT_NAME)); 5202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else { 5212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) shortcut_name.append(distribution->GetAppShortCutName()); 5222a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } 5232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return shortcut_name + installer::kLnkExt; 5242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)string16 CreateProfileShortcutFlags(const base::FilePath& profile_path) { 5272a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return base::StringPrintf(L"--%ls=\"%ls\"", 5282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ASCIIToUTF16(switches::kProfileDirectory).c_str(), 5292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path.BaseName().value().c_str()); 5302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5312a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5322a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace internal 5332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} // namespace profiles 5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)bool ProfileShortcutManager::IsFeatureEnabled() { 5372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return BrowserDistribution::GetDistribution()->CanCreateDesktopShortcuts() && 5382a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !CommandLine::ForCurrentProcess()->HasSwitch(switches::kUserDataDir) && 5392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) !CommandLine::ForCurrentProcess()->HasSwitch(switches::kShowAppList); 5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static 5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProfileShortcutManager* ProfileShortcutManager::Create( 5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ProfileManager* manager) { 5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return new ProfileShortcutManagerWin(manager); 5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProfileShortcutManagerWin::ProfileShortcutManagerWin(ProfileManager* manager) 5495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : profile_manager_(manager) { 5502a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ( 5512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) arraysize(kProfileAvatarIconResources2x), 5522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_manager_->GetProfileInfoCache().GetDefaultAvatarIconCount()); 5532a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_manager_->GetProfileInfoCache().AddObserver(this); 5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)ProfileShortcutManagerWin::~ProfileShortcutManagerWin() { 5582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_manager_->GetProfileInfoCache().RemoveObserver(this); 5595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::CreateProfileShortcut( 5622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 5632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, CREATE_WHEN_NONE_FOUND, 5642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 5652a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::RemoveProfileShortcuts( 5682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 5692a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::PostTask( 5702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::FILE, FROM_HERE, 5712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&DeleteDesktopShortcutsAndIconFile, profile_path, false)); 5722a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)} 5732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 5742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::HasProfileShortcuts( 5752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 5762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::Callback<void(bool)>& callback) { 5772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::PostTaskAndReplyWithResult( 5782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::FILE, FROM_HERE, 5792a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&HasAnyProfileShortcuts, profile_path), callback); 5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::OnProfileAdded( 5832a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 5842a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const size_t profile_count = 5852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_manager_->GetProfileInfoCache().GetNumberOfProfiles(); 5862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (profile_count == 1) { 5872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, 5882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CREATE_WHEN_NONE_FOUND, 5892a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPDATE_NON_PROFILE_SHORTCUTS); 5902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) } else if (profile_count == 2) { 5912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(GetOtherProfilePath(profile_path), 5922a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPDATE_EXISTING_ONLY, 5932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPDATE_NON_PROFILE_SHORTCUTS); 5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileWillBeRemoved( 5982a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileWasRemoved( 6022a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& profile_name) { 6042a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache(); 6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // If there is only one profile remaining, remove the badging information 6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // from an existing shortcut. 6072a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const bool deleting_down_to_last_profile = (cache.GetNumberOfProfiles() == 1); 6082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (deleting_down_to_last_profile) { 6092a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(cache.GetPathOfProfileAtIndex(0), 6102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) UPDATE_EXISTING_ONLY, 6112a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) BrowserThread::PostTask(BrowserThread::FILE, FROM_HERE, 6152a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&DeleteDesktopShortcutsAndIconFile, 6162a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) profile_path, 6172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) deleting_down_to_last_profile)); 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileNameChanged( 6212a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const string16& old_profile_name) { 6232a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, UPDATE_EXISTING_ONLY, 6242a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void ProfileShortcutManagerWin::OnProfileAvatarChanged( 6282a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 6292a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateShortcutsForProfileAtPath(profile_path, UPDATE_EXISTING_ONLY, 6302a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) IGNORE_NON_PROFILE_SHORTCUTS); 6315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6332a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)base::FilePath ProfileShortcutManagerWin::GetOtherProfilePath( 6342a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path) { 6352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const ProfileInfoCache& cache = profile_manager_->GetProfileInfoCache(); 6362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) DCHECK_EQ(2U, cache.GetNumberOfProfiles()); 6375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get the index of the current profile, in order to find the index of the 6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // other profile. 6392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t current_profile_index = cache.GetIndexOfProfileWithPath(profile_path); 6405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) size_t other_profile_index = (current_profile_index == 0) ? 1 : 0; 6412a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return cache.GetPathOfProfileAtIndex(other_profile_index); 6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void ProfileShortcutManagerWin::CreateOrUpdateShortcutsForProfileAtPath( 6452a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const base::FilePath& profile_path, 6462a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) CreateOrUpdateMode create_mode, 6472a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) NonProfileShortcutAction action) { 6482a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) ProfileInfoCache* cache = &profile_manager_->GetProfileInfoCache(); 6492a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) size_t profile_index = cache->GetIndexOfProfileWithPath(profile_path); 6505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (profile_index == std::string::npos) 6512a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 6522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) bool remove_badging = cache->GetNumberOfProfiles() == 1; 6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) string16 old_shortcut_appended_name = 6552a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetShortcutNameOfProfileAtIndex(profile_index); 6562a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) 6572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Exit early if the mode is to update existing profile shortcuts only and 6582a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // none were ever created for this profile, per the shortcut name not being 6592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // set in the profile info cache. 6602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (old_shortcut_appended_name.empty() && 6612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) create_mode == UPDATE_EXISTING_ONLY && 6622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action == IGNORE_NON_PROFILE_SHORTCUTS) { 6632a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) return; 6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6662a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) string16 new_shortcut_appended_name; 6672a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) if (!remove_badging) 6682a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_appended_name = cache->GetNameOfProfileAtIndex(profile_index); 6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6702a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap avatar_bitmap_copy_1x; 6712a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) SkBitmap avatar_bitmap_copy_2x; 6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!remove_badging) { 6732a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const size_t icon_index = 6742a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetAvatarIconIndexOfProfileAtIndex(profile_index); 6752a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int resource_id_1x = 6762a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->GetDefaultAvatarIconResourceIDAtIndex(icon_index); 6772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) const int resource_id_2x = kProfileAvatarIconResources2x[icon_index]; 6782a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) // Make a copy of the SkBitmaps to ensure that we can safely use the image 6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // data on the FILE thread. 6802a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_1x = GetImageResourceSkBitmapCopy(resource_id_1x); 6812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_2x = GetImageResourceSkBitmapCopy(resource_id_2x); 6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::PostTask( 6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) BrowserThread::FILE, FROM_HERE, 6852a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::Bind(&CreateOrUpdateDesktopShortcutsForProfile, profile_path, 6862a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) old_shortcut_appended_name, new_shortcut_appended_name, 6872a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) avatar_bitmap_copy_1x, avatar_bitmap_copy_2x, create_mode, 6882a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) action)); 6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6902a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) cache->SetShortcutNameOfProfileAtIndex(profile_index, 6912a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) new_shortcut_appended_name); 6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 693