arc: Set custom icon in shelf for ARC apps.

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::UpdateLauncherItem() {
+  const ArcAppWindow* arc_app_window =
+      static_cast<const ArcAppWindow*>(GetLastActiveWindow());
+  if (!arc_app_window || arc_app_window->icon().isNull()) {
+    if (!image_set_by_controller())
+      return;
+    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());
+    return;
+  }
+
+  owner_->SetLauncherItemImage(shelf_id(), arc_app_window->icon());
+  set_image_set_by_controller(true);
+}