Refactor extension app icon.

chrome/browser/extensions/extension_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/extension_app_icon_service.h"
+
+#include "base/bind.h"
+#include "base/memory/ptr_util.h"
+#include "chrome/browser/extensions/extension_app_icon.h"
+#include "chrome/browser/extensions/extension_app_icon_service_factory.h"
+#include "extensions/browser/extension_registry.h"
+
+#if defined(OS_CHROMEOS)
+#include "chrome/browser/ui/ash/launcher/launcher_extension_app_updater.h"
+#endif
+
+namespace extensions {
+
+#if defined(OS_CHROMEOS)
+
+class ExtensionAppIconService::Updater : public LauncherAppUpdater::Delegate {
+ public:
+  Updater(ExtensionAppIconService* host, content::BrowserContext* context)
+      : host_(host), extension_app_updater_(this, context) {}
+
+  ~Updater() override = default;
+
+ private:
+  // LauncherAppUpdater::Delegate:
+  void OnAppUpdated(content::BrowserContext* browser_context,
+                    const std::string& app_id) override {
+    host_->OnAppUpdated(app_id);
+  }
+
+  // Unowned pointer.
+  ExtensionAppIconService* const host_;
+
+  LauncherExtensionAppUpdater extension_app_updater_;
+
+  DISALLOW_COPY_AND_ASSIGN(Updater);
+};
+
+#endif
+
+// static
+ExtensionAppIconService* ExtensionAppIconService::Create(
+    content::BrowserContext* context) {
+  return new ExtensionAppIconService(context);
+}
+
+// static
+ExtensionAppIconService* ExtensionAppIconService::Get(
+    content::BrowserContext* context) {
+  return ExtensionAppIconServiceFactory::GetInstance()->GetForBrowserContext(
+      context);
+}
+
+ExtensionAppIconService::ExtensionAppIconService(
+    content::BrowserContext* context)
+    : context_(context), weak_ptr_factory_(this) {
+#if defined(OS_CHROMEOS)
+  app_updater_ = base::MakeUnique<Updater>(this, context);
+#endif
+  ExtensionRegistry::Get(context_)->AddObserver(this);

[Devlin, 2017/04/26 22:41:57]

Prefer a ScopedObserver

[khmel, 2017/04/27 00:18:12]

Done.

+}
+
+ExtensionAppIconService::~ExtensionAppIconService() = default;
+
+void ExtensionAppIconService::Shutdown() {
+#if defined(OS_CHROMEOS)
+  app_updater_.reset();
+#endif
+  ExtensionRegistry::Get(context_)->RemoveObserver(this);
+}
+
+std::unique_ptr<ExtensionAppIcon> ExtensionAppIconService::CreateIcon(
+    ExtensionAppIconDelegate* delegate,
+    const std::string& app_id,
+    int resource_size_in_dip) {
+  std::unique_ptr<ExtensionAppIcon> icon = base::MakeUnique<ExtensionAppIcon>(
+      delegate, context_,
+      base::Bind(&ExtensionAppIconService::OnIconDestroyed,
+                 weak_ptr_factory_.GetWeakPtr()),
+      app_id, resource_size_in_dip);
+
+  icon_map_.insert(std::make_pair(icon->app_id(), icon.get()));
+  return icon;
+}
+
+void ExtensionAppIconService::OnExtensionLoaded(
+    content::BrowserContext* browser_context,
+    const Extension* extension) {
+  OnAppUpdated(extension->id());
+}
+
+void ExtensionAppIconService::OnExtensionUnloaded(
+    content::BrowserContext* browser_context,
+    const Extension* extension,
+    UnloadedExtensionInfo::Reason reason) {
+  OnAppUpdated(extension->id());
+}
+
+void ExtensionAppIconService::OnAppUpdated(const std::string& app_id) {
+  auto icon_range = icon_map_.equal_range(app_id);
+  for (auto icon = icon_range.first; icon != icon_range.second; ++icon)
+    icon->second->UpdateIcon();
+}
+
+void ExtensionAppIconService::OnIconDestroyed(ExtensionAppIcon* icon) {
+  DCHECK(icon);
+  auto icon_range = icon_map_.equal_range(icon->app_id());
+  for (auto scan_icon = icon_range.first; scan_icon != icon_range.second;
+       ++scan_icon) {
+    if (scan_icon->second == icon) {
+      icon_map_.erase(scan_icon);
+      return;
+    }
+  }
+  NOTREACHED();
+}
+
+}  // namespace extensions