Reorder browser list on workspace switch

chrome/browser/ui/browser_list.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/ui/browser_list.h"
 
 #include <algorithm>
 
 #include "base/auto_reset.h"
 #include "base/logging.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/browser_shutdown.h"
 #include "chrome/browser/chrome_notification_types.h"
 #include "chrome/browser/lifetime/application_lifetime.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/browser_finder.h"
 #include "chrome/browser/ui/browser_list.h"
 #include "chrome/browser/ui/browser_list_observer.h"
 #include "chrome/browser/ui/browser_window.h"
 #include "content/public/browser/notification_service.h"
 #include "content/public/browser/user_metrics.h"
+#include "ui/display/desktop.h"
 
 using base::UserMetricsAction;
 using content::WebContents;
 
 // static
 base::LazyInstance<base::ObserverList<chrome::BrowserListObserver>>::Leaky
     BrowserList::observers_ = LAZY_INSTANCE_INITIALIZER;
 
 // static
 BrowserList* BrowserList::instance_ = NULL;
@@ skipping 177 matching lines @@
   for (auto* browser : *BrowserList::GetInstance()) {
     if (browser->profile()->IsSameProfile(profile) &&
         browser->profile()->IsOffTheRecord()) {
       return true;
     }
   }
   return false;
 }
 
 ////////////////////////////////////////////////////////////////////////////////
+// BrowserList, display::DesktopObserver implementation:
+
+void BrowserList::OnWorkspaceChanged(const std::string& new_workspace) {
+  DCHECK(!new_workspace.empty());
+
+  Browser* old_last_active = GetLastActive();
+
+  // Reorder the browsers in the list so that the browsers in the new workspace
+  // appear after the browsers in the other workspaces.
+  //
+  // For example, if we have a list of browser-workspace pairs
+  // [{b1, 0}, {b2, 1}, {b3, 0}, {b4, 1}]
+  // and we switch to workspace 1, we want the resulting browser list to look
+  // like [{b1, 0}, {b3, 0}, {b2, 1}, {b4, 1}].
+  BrowserVector in_this_workspace;
+  BrowserVector in_different_workspace;
+  for (Browser* browser : last_active_browsers_) {
+    if (browser->window()->GetWorkspace() == new_workspace)
+      in_this_workspace.push_back(browser);
+    else
+      in_different_workspace.push_back(browser);
+  }
+  last_active_browsers_.clear();
+  last_active_browsers_.insert(last_active_browsers_.end(),
+                               in_different_workspace.begin(),
+                               in_different_workspace.end());
+  last_active_browsers_.insert(last_active_browsers_.end(),
+                               in_this_workspace.begin(),
+                               in_this_workspace.end());
+
+  Browser* new_last_active = GetLastActive();
+  if (old_last_active != new_last_active) {
+    FOR_EACH_OBSERVER(chrome::BrowserListObserver, observers_.Get(),
+                      OnBrowserSetLastActive(new_last_active));
+  }
+}
+
+////////////////////////////////////////////////////////////////////////////////
 // BrowserList, private:
 
 BrowserList::BrowserList() {
+  display::Desktop::AddObserver(this);
 }
 
 BrowserList::~BrowserList() {
+  display::Desktop::RemoveObserver(this);
 }
 
 // static
 void BrowserList::RemoveBrowserFrom(Browser* browser,
                                     BrowserVector* browser_list) {
   BrowserVector::iterator remove_browser =
       std::find(browser_list->begin(), browser_list->end(), browser);
   if (remove_browser != browser_list->end())
     browser_list->erase(remove_browser);
 }