WIP

chrome/browser/ui/ash/launcher/arc_app_window_launcher_item_controller.h

 // 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.
 
 #ifndef CHROME_BROWSER_UI_ASH_LAUNCHER_ARC_APP_WINDOW_LAUNCHER_ITEM_CONTROLLER_H_
 #define CHROME_BROWSER_UI_ASH_LAUNCHER_ARC_APP_WINDOW_LAUNCHER_ITEM_CONTROLLER_H_
 
 #include <string>
 #include <unordered_set>
 
 #include "base/macros.h"
 #include "chrome/browser/ui/ash/launcher/app_window_launcher_item_controller.h"
 
+class ArcAppWindow;
+class ChromeLauncherController;
+
 // Shelf item delegate for ARC app windows.
 class ArcAppWindowLauncherItemController
     : public AppWindowLauncherItemController {
  public:
-  explicit ArcAppWindowLauncherItemController(const std::string& arc_app_id);
+  ArcAppWindowLauncherItemController(const std::string& arc_app_id,
+                                     ChromeLauncherController* owner);
 
   ~ArcAppWindowLauncherItemController() override;
 
   // AppWindowLauncherItemController overrides:
   void ItemSelected(std::unique_ptr<ui::Event> event,
                     int64_t display_id,
                     ash::ShelfLaunchSource source,
                     ItemSelectedCallback callback) override;
   ash::MenuItemList GetAppMenuItems(int event_flags) override;
   void ExecuteCommand(uint32_t command_id, int32_t event_flags) override;
 
   void AddTaskId(int task_id);
   void RemoveTaskId(int task_id);
   bool HasAnyTasks() const;
 
+  void OnWindowChanged(ArcAppWindow* arc_app_window);
+
  private:
+  // Updates controller icon from the active window. If custom icon of the
+  // window exists and decoded then set this custom icon for the controller.
+  // If not and controller has custom icon set then update the icon from default
+  // AppIconLoader.
+  void UpdateIcon(ArcAppWindow* arc_app_window);
+
+  // AppWindowLauncherItemController:
+  void OnActiveWindowChanged(ui::BaseWindow* active_window) override;
+
   std::unordered_set<int> task_ids_;
 
+  // Unowned property.
+  ChromeLauncherController* const owner_;
+
   DISALLOW_COPY_AND_ASSIGN(ArcAppWindowLauncherItemController);
 };
 
 #endif  // CHROME_BROWSER_UI_ASH_LAUNCHER_ARC_APP_WINDOW_LAUNCHER_ITEM_CONTROLLER_H_