[Win] Add a fast profile switcher to the Windows taskbar item.

chrome/browser/jumplist_win.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/jumplist_win.h"
 
 #include "base/bind.h"
 #include "base/bind_helpers.h"
 #include "base/command_line.h"
 #include "base/files/file_util.h"
 #include "base/path_service.h"
 #include "base/prefs/pref_change_registrar.h"
 #include "base/strings/string_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/threading/thread.h"
+#include "chrome/browser/browser_process.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/favicon/favicon_service.h"
 #include "chrome/browser/favicon/favicon_service_factory.h"
 #include "chrome/browser/history/history_service.h"
 #include "chrome/browser/history/top_sites.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/profiles/profile_avatar_icon_util.h"
+#include "chrome/browser/profiles/profile_info_cache.h"
+#include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/browser/sessions/session_types.h"
 #include "chrome/browser/sessions/tab_restore_service.h"
 #include "chrome/browser/sessions/tab_restore_service_factory.h"
 #include "chrome/browser/shell_integration.h"
 #include "chrome/common/chrome_constants.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/pref_names.h"
 #include "chrome/common/url_constants.h"
 #include "chrome/grit/generated_resources.h"
 #include "components/favicon_base/favicon_types.h"
@@ skipping 84 matching lines @@
         l10n_util::GetStringUTF16(IDS_NEW_INCOGNITO_WINDOW);
     ReplaceSubstringsAfterOffset(&incognito_title, 0, L"&", L"");
     incognito->set_title(incognito_title);
     incognito->set_icon(chrome_path.value(), 0);
     items.push_back(incognito);
   }
 
   return jumplist_updater->AddTasks(items);
 }
 
+bool UpdateProfilesCategory(JumpListUpdater* jumplist_updater,
+                            AvatarMenu* avatar_menu,
+                            const std::vector<gfx::Image>& profile_avatars,
+                            const base::FilePath& icon_dir) {
+  base::FilePath chrome_path;
+  if (!PathService::Get(base::FILE_EXE, &chrome_path))
+    return false;
+
+  ShellLinkItemList profile_items;
+
+  for (size_t i = 0; i < avatar_menu->GetNumberOfItems(); ++i) {
+    scoped_refptr<ShellLinkItem> link = CreateShellLink();
+    const AvatarMenu::Item& item = avatar_menu->GetItemAt(i);
+
+    link->set_title(base::UTF16ToWide(item.name));

[tapted, 2014/11/03 23:57:06]

is the UTF16ToWide needed? (pretty sure std::wstring is utf16 on windows, so
it's just a typedef, and there's a lingering TODO to convert wstring to
string16)

[noms (inactive), 2014/11/04 20:03:08]

Done.

+    link->GetCommandLine()->AppendSwitchPath(
+        switches::kProfileDirectory, item.profile_path.BaseName());
+    link->GetCommandLine()->AppendSwitch(
+        switches::kReuseExistingProfileBrowser);
+
+    // Icons need to be saved on disk to be used by a ShellLinkItem.
+    base::FilePath icon_path;
+    if (CreateIconFile(profile_avatars[i].AsBitmap(), icon_dir, &icon_path))
+      link->set_icon(icon_path.value(), 0);
+
+    profile_items.push_back(link);
+  }
+
+  return jumplist_updater->AddCustomCategory(
+      base::UTF16ToWide(

[tapted, 2014/11/03 23:57:05]

remove UTF16ToWide?

[noms (inactive), 2014/11/04 20:03:08]

Done.

+          l10n_util::GetStringUTF16(IDS_PROFILES_OPTIONS_GROUP_NAME)),
+      profile_items, profile_items.size());
+}
+
 // Updates the application JumpList.
 bool UpdateJumpList(const wchar_t* app_id,
                     const ShellLinkItemList& most_visited_pages,
                     const ShellLinkItemList& recently_closed_pages,
-                    IncognitoModePrefs::Availability incognito_availability) {
+                    IncognitoModePrefs::Availability incognito_availability,
+                    AvatarMenu* avatar_menu,
+                    const std::vector<gfx::Image>& profile_avatars,
+                    const base::FilePath& icon_dir) {
   // JumpList is implemented only on Windows 7 or later.
   // So, we should return now when this function is called on earlier versions
   // of Windows.
   if (!JumpListUpdater::IsEnabled())
     return true;
 
   JumpListUpdater jumplist_updater(app_id);
   if (!jumplist_updater.BeginUpdate())
     return false;
 
@@ skipping 24 matching lines @@
   }
 
   // Update the "Recently Closed" category of the JumpList.
   if (!jumplist_updater.AddCustomCategory(
           base::UTF16ToWide(
               l10n_util::GetStringUTF16(IDS_NEW_TAB_RECENTLY_CLOSED)),
           recently_closed_pages, recently_closed_items)) {
     return false;
   }
 
+  // Update the "Profiles" category of the JumpList.
+  if (!UpdateProfilesCategory(&jumplist_updater, avatar_menu,
+                              profile_avatars, icon_dir)) {
+    return false;
+  }
+
   // Update the "Tasks" category of the JumpList.
   if (!UpdateTaskCategory(&jumplist_updater, incognito_availability))
     return false;
 
   // Commit this transaction and send the updated JumpList to Windows.
   if (!jumplist_updater.CommitUpdate())
     return false;
 
   return true;
 }
@@ skipping 32 matching lines @@
     // observers are detatched at that time.
     registrar_->Add(this, chrome::NOTIFICATION_PROFILE_DESTROYED,
                     content::Source<Profile>(profile_));
   }
   tab_restore_service->AddObserver(this);
   pref_change_registrar_.reset(new PrefChangeRegistrar);
   pref_change_registrar_->Init(profile_->GetPrefs());
   pref_change_registrar_->Add(
       prefs::kIncognitoModeAvailability,
       base::Bind(&JumpList::OnIncognitoAvailabilityChanged, this));
+
+  ProfileManager* profile_manager = g_browser_process->profile_manager();
+  DCHECK(profile_manager);
+  avatar_menu_.reset(new AvatarMenu(
+      &profile_manager->GetProfileInfoCache(), this, NULL));
+  avatar_menu_->RebuildMenu();
+  UpdateProfileAvatars();
 }
 
 JumpList::~JumpList() {
   Terminate();
 }
 
 // static
 bool JumpList::Enabled() {
   return JumpListUpdater::IsEnabled();
 }
@@ skipping 106 matching lines @@
     recently_closed_pages_ = temp_list;
   }
 
   // Send a query that retrieves the first favicon.
   StartLoadingFavicon();
 }
 
 void JumpList::TabRestoreServiceDestroyed(TabRestoreService* service) {
 }
 
+void JumpList::OnAvatarMenuChanged(AvatarMenu* avatar_menu) {
+  UpdateProfileAvatars();
+  PostRunUpdate();
+}
+
 bool JumpList::AddTab(const TabRestoreService::Tab* tab,
                       ShellLinkItemList* list,
                       size_t max_items) {
   // This code adds the URL and the title strings of the given tab to the
   // specified list.
   if (list->size() >= max_items)
     return false;
 
   scoped_refptr<ShellLinkItem> link = CreateShellLink();
   const sessions::SerializedNavigationEntry& current_navigation =
@@ skipping 45 matching lines @@
       url,
       base::Bind(&JumpList::OnFaviconDataAvailable, base::Unretained(this)),
       &cancelable_task_tracker_);
 }
 
 void JumpList::OnFaviconDataAvailable(
     const favicon_base::FaviconImageResult& image_result) {
   // If there is currently a favicon request in progress, it is now outdated,
   // as we have received another, so nullify the handle from the old request.
   task_id_ = base::CancelableTaskTracker::kBadTaskId;
-  // lock the list to set icon data and pop the url
+  // Lock the list to set icon data and pop the url.
   {
     base::AutoLock auto_lock(list_lock_);
     // Attach the received data to the ShellLinkItem object.
     // This data will be decoded by the RunUpdate method.
     if (!image_result.image.IsEmpty()) {
       if (!icon_urls_.empty() && icon_urls_.front().second)
         icon_urls_.front().second->set_icon_data(image_result.image.AsBitmap());
     }
 
     if (!icon_urls_.empty())
@@ skipping 56 matching lines @@
   CreateIconFiles(local_most_visited_pages);
 
   // Create temporary icon files for shortcuts in the "Recently Closed"
   // category.
   CreateIconFiles(local_recently_closed_pages);
 
   // We finished collecting all resources needed for updating an application
   // JumpList. So, create a new JumpList and replace the current JumpList
   // with it.
   UpdateJumpList(app_id_.c_str(), local_most_visited_pages,
-                 local_recently_closed_pages, incognito_availability);
+                 local_recently_closed_pages, incognito_availability,
+                 avatar_menu_.get(), profile_avatars_, icon_dir_);

[tapted, 2014/11/03 23:57:05]

RunUpdate runs on the FILE thread, so accessing |avatar_menu_| and
|profile_avatars_| here isn't threadsafe. I think you'll want to more closely
follow the pattern used for most_visited_pages_ and recently_closed_pages_
above.  E.g. a data member like `ShellLinkItemList profile_switcher_;`

I'd probably also start by renaming RunUpdate to RunUpdateOnFileThread so this
is clearer.

[noms (inactive), 2014/11/04 20:03:08]

Done.

 }
 
 void JumpList::CreateIconFiles(const ShellLinkItemList& item_list) {
   for (ShellLinkItemList::const_iterator item = item_list.begin();
       item != item_list.end(); ++item) {
     base::FilePath icon_path;
     if (CreateIconFile((*item)->icon_data(), icon_dir_, &icon_path))
       (*item)->set_icon(icon_path.value(), 0);
   }
 }
+
+void JumpList::UpdateProfileAvatars() {
+  profile_avatars_.clear();
+  for (size_t i = 0; i < avatar_menu_->GetNumberOfItems(); ++i) {
+    const AvatarMenu::Item& item = avatar_menu_->GetItemAt(i);
+    gfx::Image avatar;
+    bool is_rectangle;
+    profiles::GetTransparentBackgroundProfileAvatar(
+        item.profile_path, &avatar, &is_rectangle);
+    profile_avatars_.push_back(avatar);
+  }
+}