Add ARC app icons to the Task Manager UI

chrome/browser/task_management/providers/arc/arc_process_task.cc

 // Copyright 2015 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/task_management/providers/arc/arc_process_task.h"
 
+#include "base/bind.h"
 #include "base/i18n/rtl.h"
 #include "base/strings/utf_string_conversions.h"
 #include "chrome/grit/generated_resources.h"
 #include "components/arc/arc_bridge_service.h"
+#include "components/arc/arc_service_manager.h"
 #include "components/arc/common/process.mojom.h"
+#include "content/public/browser/browser_thread.h"
 #include "content/public/common/child_process_host.h"
 #include "ui/base/l10n/l10n_util.h"
 
 namespace task_management {
 
 namespace {
 
 base::string16 MakeTitle(const std::string& process_name) {
   base::string16 title =
       l10n_util::GetStringFUTF16(
           IDS_TASK_MANAGER_ARC_PREFIX, base::UTF8ToUTF16(process_name));
   base::i18n::AdjustStringForLocaleDirection(&title);
   return title;
 }
 
+scoped_refptr<arc::ActivityIconLoader> GetIconLoader() {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  arc::ArcServiceManager* arc_service_manager = arc::ArcServiceManager::Get();
+  if (!arc_service_manager)
+    return nullptr;
+  return arc_service_manager->icon_loader();
+}
+
 }  // namespace
 
-ArcProcessTask::ArcProcessTask(
-    base::ProcessId pid,
-    base::ProcessId nspid,
-    const std::string& process_name,
-    arc::mojom::ProcessState process_state)
-    : Task(MakeTitle(process_name), process_name, nullptr, pid),
+ArcProcessTask::ArcProcessTask(base::ProcessId pid,
+                               base::ProcessId nspid,
+                               const std::string& process_name,
+                               arc::mojom::ProcessState process_state,
+                               const std::vector<std::string>& packages)
+    : Task(MakeTitle(process_name), process_name, nullptr /* icon */, pid),
       nspid_(nspid),
       process_name_(process_name),
-      process_state_(process_state) {
+      process_state_(process_state),
+      weak_factory_(this) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  if (packages.empty())
+    return;
+
+  std::vector<arc::ActivityIconLoader::ActivityName> activity;
+  activity.emplace_back(packages.at(0), "" /* don't care */);

[Luis Héctor Chávez, 2016/05/31 22:10:53]

nit: kEmptyActivityName?

[Luis Héctor Chávez, 2016/05/31 22:10:53]

Maybe add a comment explaining why you only choose the first package entry.

[Yusuke Sato, 2016/05/31 23:11:52]

Done.

[Yusuke Sato, 2016/05/31 23:11:52]

Good idea. Done.

+
+  scoped_refptr<arc::ActivityIconLoader> icon_loader = GetIconLoader();
+  if (!icon_loader)
+    return;
+
+  // TODO(yusukes): ArcProcessTask might be created before intent_helper
+  // instance becomes ready. Because of the race, OnIconLoaded() might
+  // be called synchronously with no icon. Fix the race to ensure that
+  // ArcProcessTask can always fetch icons.
+  icon_loader->GetActivityIcons(
+      activity,
+      base::Bind(&ArcProcessTask::OnIconLoaded, weak_factory_.GetWeakPtr()));
 }
 
 ArcProcessTask::~ArcProcessTask() {
 }
 
 Task::Type ArcProcessTask::GetType() const {
   return Task::ARC;
 }
 
 int ArcProcessTask::GetChildProcessUniqueID() const {
@@ skipping 18 matching lines @@
     return;
   }
   arc_process_instance->KillProcess(
       nspid_, "Killed manually from Task Manager");
 }
 
 void ArcProcessTask::SetProcessState(arc::mojom::ProcessState process_state) {
   process_state_ = process_state;
 }
 
+void ArcProcessTask::OnIconLoaded(
+    std::unique_ptr<arc::ActivityIconLoader::ActivityToIconsMap> icons) {
+  DCHECK_CURRENTLY_ON(content::BrowserThread::UI);
+  for (const auto& kv : *icons) {
+    const gfx::Image& icon = kv.second.icon16;
+    if (icon.IsEmpty())
+      continue;
+    icon_ = icon;
+    set_icon(*icon_.ToImageSkia());  // pass a weak pointer of the |icon_|.

[Luis Héctor Chávez, 2016/05/31 22:10:53]

Given that you only use one package name (and expect one icon back), break after
this? (Adding a comment about it).

[Yusuke Sato, 2016/05/31 23:11:52]

Done.

+  }
+}
+
 }  // namespace task_management