Enhance chrome.app.window API for shelf integration with pinning support

chrome/browser/ui/ash/launcher/extension_app_window_launcher_controller.cc

 // Copyright 2016 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/ash/launcher/extension_app_window_launcher_controller.h"
 
 #include "ash/common/shelf/shelf_delegate.h"
 #include "ash/common/wm_shell.h"
 #include "ash/shelf/shelf_util.h"
 #include "ash/wm/window_util.h"
@@ skipping 134 matching lines @@
     LauncherItemController::Type type =
         app_window->window_type_is_panel()
             ? LauncherItemController::TYPE_APP_PANEL
             : LauncherItemController::TYPE_APP;
     ExtensionAppWindowLauncherItemController* controller =
         new ExtensionAppWindowLauncherItemController(type, app_shelf_id, app_id,
                                                      owner());
     controller->AddAppWindow(app_window);
     // If there is already a shelf id mapped to this app_shelf_id (e.g. pinned),
     // use that shelf item.
-    AppShelfIdToShelfIdMap::iterator app_shelf_id_iter =
-        app_shelf_id_to_shelf_id_map_.find(app_shelf_id);
-    if (app_shelf_id_iter != app_shelf_id_to_shelf_id_map_.end()) {
-      if (owner()->IsPinned(app_shelf_id_iter->second)) {
-        shelf_id = app_shelf_id_iter->second;
-      } else {
-        app_shelf_id_to_shelf_id_map_.erase(app_shelf_id);
-      }
-    } else if (app_shelf_id == app_id) {
-      // show_in_shelf in false and not a panel
-      shelf_id =
-          ash::WmShell::Get()->shelf_delegate()->GetShelfIDForAppID(app_id);
-      // Check if the shelf_id corresponds to an already opened
-      // showInShelf=true window that has the same app_id. The current
-      // showInShelf=false window should not fold under this shelf item,
-      // so the shelf_id is set to 0 to get a new shelf_id.
-      auto&& id_map = app_shelf_id_to_shelf_id_map_;
-      if (std::find_if(
-              id_map.begin(), id_map.end(),
-              [shelf_id](const AppShelfIdToShelfIdMap::value_type& pair) {
-                return pair.second == shelf_id;
-              }) != id_map.end()) {
-        shelf_id = 0;
-      }
-    }
+    shelf_id = ash::WmShell::Get()->shelf_delegate()->GetShelfIDForAppID(
+        app_id, app_shelf_id);
 
     if (shelf_id == 0) {
       shelf_id = owner()->CreateAppLauncherItem(controller, app_id, status);
       // Restore any existing app icon and flag as set.
       const gfx::Image& app_icon = app_window->app_icon();
       if (!app_icon.IsEmpty()) {
         owner()->SetLauncherItemImage(shelf_id, app_icon.AsImageSkia());
         controller->set_image_set_by_controller(true);
       }
     } else {
       owner()->SetItemController(shelf_id, controller);
     }
 
     // We need to change the controller associated with app_shelf_id.
     app_controller_map_[app_shelf_id] = controller;
-    app_shelf_id_to_shelf_id_map_[app_shelf_id] = shelf_id;
   }
   owner()->SetItemStatus(shelf_id, status);
   ash::SetShelfIDForWindow(shelf_id, window);
 }
 
 void ExtensionAppWindowLauncherController::UnregisterApp(aura::Window* window) {
   WindowToAppShelfIdMap::iterator window_iter =
       window_to_app_shelf_id_map_.find(window);
   DCHECK(window_iter != window_to_app_shelf_id_map_.end());
   std::string app_shelf_id = window_iter->second;
   window_to_app_shelf_id_map_.erase(window_iter);
   window->RemoveObserver(this);
 
   AppControllerMap::iterator app_controller_iter =
       app_controller_map_.find(app_shelf_id);
   DCHECK(app_controller_iter != app_controller_map_.end());
   ExtensionAppWindowLauncherItemController* controller;
   controller = app_controller_iter->second;
 
   controller->RemoveWindow(controller->GetAppWindow(window));
   if (controller->window_count() == 0) {
     // If this is the last window associated with the app window shelf id,
     // close the shelf item.
     ash::ShelfID shelf_id = controller->shelf_id();
-    if (!owner()->IsPinned(shelf_id)) {
-      app_shelf_id_to_shelf_id_map_.erase(app_shelf_id);
-    }
     owner()->CloseLauncherItem(shelf_id);
     app_controller_map_.erase(app_controller_iter);
   }
 }
 
 bool ExtensionAppWindowLauncherController::IsRegisteredApp(
     aura::Window* window) {
   return window_to_app_shelf_id_map_.find(window) !=
          window_to_app_shelf_id_map_.end();
 }
 
 // Private Methods
 
 AppWindowLauncherItemController*
 ExtensionAppWindowLauncherController::ControllerForWindow(
     aura::Window* window) {
   WindowToAppShelfIdMap::iterator window_iter =
       window_to_app_shelf_id_map_.find(window);
   if (window_iter == window_to_app_shelf_id_map_.end())
     return nullptr;
   AppControllerMap::iterator app_controller_iter =
       app_controller_map_.find(window_iter->second);
   if (app_controller_iter == app_controller_map_.end())
     return nullptr;
   return app_controller_iter->second;
 }