Refactor extension app icon.

chrome/browser/extensions/chrome_app_icon_service.cc

+// Copyright 2017 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/extensions/chrome_app_icon_service.h"
+
+#include "base/bind.h"
+#include "base/memory/ptr_util.h"
+#include "chrome/browser/extensions/chrome_app_icon.h"
+#include "chrome/browser/extensions/chrome_app_icon_service_factory.h"
+#include "extensions/browser/extension_registry.h"
+
+namespace extensions {
+
+// static
+ChromeAppIconService* ChromeAppIconService::Get(
+    content::BrowserContext* context) {
+  return ChromeAppIconServiceFactory::GetInstance()->GetForBrowserContext(
+      context);
+}
+
+ChromeAppIconService::ChromeAppIconService(content::BrowserContext* context)
+    : context_(context), observer_(this), weak_ptr_factory_(this) {
+#if defined(OS_CHROMEOS)
+  app_updater_ = base::MakeUnique<LauncherExtensionAppUpdater>(this, context);
+#endif
+
+  observer_.Add(ExtensionRegistry::Get(context_));
+}
+
+ChromeAppIconService::~ChromeAppIconService() = default;
+
+void ChromeAppIconService::Shutdown() {
+#if defined(OS_CHROMEOS)
+  app_updater_.reset();
+#endif
+}
+
+std::unique_ptr<ChromeAppIcon> ChromeAppIconService::CreateIcon(
+    ChromeAppIconDelegate* delegate,
+    const std::string& app_id,
+    int resource_size_in_dip) {
+  std::unique_ptr<ChromeAppIcon> icon = base::MakeUnique<ChromeAppIcon>(
+      delegate, context_,
+      base::Bind(&ChromeAppIconService::OnIconDestroyed,
+                 weak_ptr_factory_.GetWeakPtr()),
+      app_id, resource_size_in_dip);
+
+  icon_map_[icon->app_id()].insert(icon.get());
+  return icon;
+}
+
+void ChromeAppIconService::OnExtensionLoaded(
+    content::BrowserContext* browser_context,
+    const Extension* extension) {
+  OnAppUpdated(extension->id());
+}
+
+void ChromeAppIconService::OnExtensionUnloaded(
+    content::BrowserContext* browser_context,
+    const Extension* extension,
+    UnloadedExtensionReason reason) {
+  OnAppUpdated(extension->id());
+}
+
+#if defined(OS_CHROMEOS)
+void ChromeAppIconService::OnAppUpdated(
+    content::BrowserContext* browser_context,
+    const std::string& app_id) {
+  OnAppUpdated(app_id);
+}
+#endif
+
+void ChromeAppIconService::OnAppUpdated(const std::string& app_id) {
+  IconMap::const_iterator it = icon_map_.find(app_id);
+  if (it == icon_map_.end())
+    return;
+  for (auto* icon : it->second)
+    icon->UpdateIcon();
+}
+
+void ChromeAppIconService::OnIconDestroyed(ChromeAppIcon* icon) {
+  DCHECK(icon);
+  IconMap::iterator it = icon_map_.find(icon->app_id());
+  DCHECK(it != icon_map_.end());
+  it->second.erase(icon);
+  if (it->second.empty())
+    icon_map_.erase(it);
+}
+
+}  // namespace extensions