WIP

chrome/browser/ui/ash/launcher/arc_app_window_launcher_item_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/arc_app_window_launcher_item_controller.h"
 
 #include <utility>
 
 #include "base/memory/ptr_util.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/ui/app_icon_loader.h"
 #include "chrome/browser/ui/app_list/arc/arc_app_list_prefs.h"
 #include "chrome/browser/ui/app_list/arc/arc_app_utils.h"
+#include "chrome/browser/ui/ash/launcher/arc_app_window.h"
 #include "chrome/browser/ui/ash/launcher/arc_app_window_launcher_controller.h"
 #include "chrome/browser/ui/ash/launcher/chrome_launcher_controller.h"
 #include "chrome/browser/ui/ash/launcher/launcher_controller_helper.h"
 #include "ui/aura/window.h"
 #include "ui/base/base_window.h"
 
 ArcAppWindowLauncherItemController::ArcAppWindowLauncherItemController(
-    const std::string& arc_app_id)
-    : AppWindowLauncherItemController(ash::ShelfID(arc_app_id)) {}
+    const std::string& arc_app_id,
+    ChromeLauncherController* owner)
+    : AppWindowLauncherItemController(ash::ShelfID(arc_app_id)),
+      owner_(owner) {}
 
 ArcAppWindowLauncherItemController::~ArcAppWindowLauncherItemController() {}
 
 void ArcAppWindowLauncherItemController::AddTaskId(int task_id) {
   task_ids_.insert(task_id);
 }
 
 void ArcAppWindowLauncherItemController::RemoveTaskId(int task_id) {
   task_ids_.erase(task_id);
 }
@@ skipping 23 matching lines @@
 }
 
 void ArcAppWindowLauncherItemController::ExecuteCommand(uint32_t command_id,
                                                         int32_t event_flags) {
   ActivateIndexedApp(command_id);
 }
 
 ash::MenuItemList ArcAppWindowLauncherItemController::GetAppMenuItems(
     int event_flags) {
   ash::MenuItemList items;
-  base::string16 app_title = LauncherControllerHelper::GetAppTitle(
-      ChromeLauncherController::instance()->profile(), app_id());
+  base::string16 app_title =
+      LauncherControllerHelper::GetAppTitle(owner_->profile(), app_id());
   for (auto it = windows().begin(); it != windows().end(); ++it) {
     // TODO(khmel): resolve correct icon here.
     size_t i = std::distance(windows().begin(), it);
     aura::Window* window = (*it)->GetNativeWindow();
     ash::mojom::MenuItemPtr item = ash::mojom::MenuItem::New();
     item->command_id = base::checked_cast<uint32_t>(i);
     item->label = (window && !window->GetTitle().empty()) ? window->GetTitle()
                                                           : app_title;
     items.push_back(std::move(item));
   }
+
   return items;
 }
+
+void ArcAppWindowLauncherItemController::OnWindowChanged(
+    ArcAppWindow* arc_app_window) {
+  DCHECK(arc_app_window && arc_app_window->controller() == this);
+  if (GetLastActiveWindow() == arc_app_window)
+    UpdateIcon(arc_app_window);
+}
+
+void ArcAppWindowLauncherItemController::UpdateIcon(
+    ArcAppWindow* arc_app_window) {
+  if (!arc_app_window || arc_app_window->image_skia().isNull()) {
+    if (image_set_by_controller()) {
+      set_image_set_by_controller(false);
+      owner_->SetLauncherItemImage(shelf_id(), gfx::ImageSkia());
+      AppIconLoader* icon_loader = owner_->GetAppIconLoaderForApp(app_id());
+      if (icon_loader)
+        icon_loader->UpdateImage(app_id());
+    }
+  } else {
+    owner_->SetLauncherItemImage(shelf_id(), arc_app_window->image_skia());
+    set_image_set_by_controller(true);
+  }
+}
+
+void ArcAppWindowLauncherItemController::OnActiveWindowChanged(
+    ui::BaseWindow* active_window) {
+  UpdateIcon(static_cast<ArcAppWindow*>(active_window));
+}