app_list ExtensionAppModelBuilder: Observe and handle changes to prefs::kHideWebStoreIcon

chrome/browser/ui/app_list/extension_app_model_builder.cc

 // Copyright (c) 2012 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/app_list/extension_app_model_builder.h"
 
 #include <algorithm>
 
 #include "base/auto_reset.h"
 #include "base/bind.h"
 #include "base/callback.h"
 #include "base/command_line.h"
 #include "chrome/browser/extensions/extension_ui_util.h"
 #include "chrome/browser/extensions/install_tracker.h"
 #include "chrome/browser/extensions/install_tracker_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/app_list/app_list_controller_delegate.h"
 #include "chrome/browser/ui/app_list/app_list_syncable_service.h"
 #include "chrome/browser/ui/app_list/app_list_syncable_service_factory.h"
 #include "chrome/browser/ui/app_list/extension_app_item.h"
 #include "chrome/common/chrome_switches.h"
+#include "chrome/common/pref_names.h"
 #include "extensions/browser/extension_prefs.h"
 #include "extensions/browser/extension_registry.h"
 #include "extensions/browser/extension_system.h"
 #include "extensions/browser/extensions_browser_client.h"
 #include "extensions/browser/pref_names.h"
 #include "extensions/common/constants.h"
 #include "extensions/common/extension.h"
 #include "extensions/common/extension_set.h"
 #include "ui/gfx/image/image_skia.h"
 #include "ui/gfx/image/image_skia_operations.h"
@@ skipping 17 matching lines @@
   if (!service_)
     model_->top_level_item_list()->RemoveObserver(this);
 }
 
 void ExtensionAppModelBuilder::InitializeWithService(
     app_list::AppListSyncableService* service) {
   DCHECK(!service_ && !profile_);
   model_ = service->model();
   service_ = service;
   profile_ = service->profile();
-  InitializePrefChangeRegistrar();
+  InitializePrefChangeRegistrars();
 
   BuildModel();
 }
 
 void ExtensionAppModelBuilder::InitializeWithProfile(
     Profile* profile,
     app_list::AppListModel* model) {
   DCHECK(!service_ && !profile_);
   model_ = model;
   model_->top_level_item_list()->AddObserver(this);
   profile_ = profile;
-  InitializePrefChangeRegistrar();
+  InitializePrefChangeRegistrars();
 
   BuildModel();
 }
 
-void ExtensionAppModelBuilder::InitializePrefChangeRegistrar() {
+void ExtensionAppModelBuilder::InitializePrefChangeRegistrars() {
+  profile_pref_change_registrar_.Init(profile_->GetPrefs());
+  profile_pref_change_registrar_.Add(
+      prefs::kHideWebStoreIcon,
+      base::Bind(&ExtensionAppModelBuilder::OnProfilePreferenceChanged,
+                 base::Unretained(this)));
+
   if (!CommandLine::ForCurrentProcess()->HasSwitch(
           switches::kEnableStreamlinedHostedApps))
     return;
 
   // TODO(calamity): analyze the performance impact of doing this every
   // extension pref change.
   extensions::ExtensionsBrowserClient* client =
       extensions::ExtensionsBrowserClient::Get();
   extension_pref_change_registrar_.Init(
       client->GetPrefServiceForContext(profile_));
   extension_pref_change_registrar_.Add(
     extensions::pref_names::kExtensions,
     base::Bind(&ExtensionAppModelBuilder::OnExtensionPreferenceChanged,
                base::Unretained(this)));
 }
 
+void ExtensionAppModelBuilder::OnProfilePreferenceChanged() {
+  extensions::ExtensionSet extensions;
+  controller_->GetApps(profile_, &extensions);
+
+  for (extensions::ExtensionSet::const_iterator app = extensions.begin();
+       app != extensions.end(); ++app) {
+    bool should_display =
+        extensions::ui_util::ShouldDisplayInAppLauncher(*app, profile_);
+    bool does_display = GetExtensionAppItem((*app)->id()) != NULL;
+
+    if (should_display == does_display)
+      continue;
+
+    if (should_display) {
+      InsertApp(CreateAppItem((*app)->id(),
+                              "",
+                              gfx::ImageSkia(),
+                              (*app)->is_platform_app()));
+    } else {
+      if (service_)
+        service_->RemoveItem((*app)->id());
+      else
+        model_->DeleteItem((*app)->id());
+    }
+  }
+}
+
 void ExtensionAppModelBuilder::OnExtensionPreferenceChanged() {
   model_->NotifyExtensionPreferenceChanged();
 }
 
 void ExtensionAppModelBuilder::OnBeginExtensionInstall(
     const ExtensionInstallParams& params) {
   if (!params.is_app || params.is_ephemeral)
     return;
 
   DVLOG(2) << service_ << ": OnBeginExtensionInstall: "
@@ skipping 224 matching lines @@
     app_list::AppListItem* item = item_list->item_at(idx + 1);
     if (item->GetItemType() == ExtensionAppItem::kItemType) {
       next = static_cast<ExtensionAppItem*>(item);
       break;
     }
   }
   // item->Move will call set_position, overriding the item's position.
   if (prev || next)
     static_cast<ExtensionAppItem*>(item)->Move(prev, next);
 }