Extensions: create a simple model for the browser action buttons toolstrip an...

chrome/browser/gtk/browser_actions_toolbar_gtk.cc

 // Copyright (c) 2009 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/gtk/browser_actions_toolbar_gtk.h"
 
 #include <gtk/gtk.h>
 #include <vector>
 
 #include "app/gfx/canvas_paint.h"
@@ skipping 185 matching lines @@
   GdkPixbuf* default_icon_;
 
   NotificationRegistrar registrar_;
 
   friend class BrowserActionsToolbarGtk;
 };
 
 BrowserActionsToolbarGtk::BrowserActionsToolbarGtk(Browser* browser)
     : browser_(browser),
       profile_(browser->profile()),
+      model_(NULL),
       hbox_(gtk_hbox_new(FALSE, kBrowserActionButtonPadding)) {
-  registrar_.Add(this, NotificationType::EXTENSION_LOADED,
-                 Source<Profile>(profile_));
-  registrar_.Add(this, NotificationType::EXTENSION_UNLOADED,
-                 Source<Profile>(profile_));
-  registrar_.Add(this, NotificationType::EXTENSION_UNLOADED_DISABLED,
-                 Source<Profile>(profile_));
-
-  CreateAllButtons();
+  ExtensionsService* extension_service = profile_->GetExtensionsService();
+  // The |extension_service| can be NULL in Incognito.
+  if (extension_service) {
+    model_ = extension_service->toolbar_model();
+    model_->AddObserver(this);
+    CreateAllButtons();
+  }
 }
 
 BrowserActionsToolbarGtk::~BrowserActionsToolbarGtk() {
+  if (model_)
+    model_->RemoveObserver(this);
   hbox_.Destroy();
 }
 
 int BrowserActionsToolbarGtk::GetCurrentTabId() {
   TabContents* selected_tab = browser_->GetSelectedTabContents();
   if (!selected_tab)
     return -1;
 
   return selected_tab->controller().session_id().id();
 }
 
 void BrowserActionsToolbarGtk::Update() {
   for (ExtensionButtonMap::iterator iter = extension_button_map_.begin();
        iter != extension_button_map_.end(); ++iter) {
     iter->second->UpdateState();
   }
 }
 
-void BrowserActionsToolbarGtk::Observe(NotificationType type,
-                                       const NotificationSource& source,
-                                       const NotificationDetails& details) {
-  Extension* extension = Details<Extension>(details).ptr();
-
-  if (type == NotificationType::EXTENSION_LOADED) {
-    CreateButtonForExtension(extension);
-  } else if (type == NotificationType::EXTENSION_UNLOADED ||
-             type == NotificationType::EXTENSION_UNLOADED_DISABLED) {
-    RemoveButtonForExtension(extension);
-  } else {
-    NOTREACHED() << "Received unexpected notification";
-  }
-}
-
 void BrowserActionsToolbarGtk::CreateAllButtons() {
-  ExtensionsService* extension_service = profile_->GetExtensionsService();
-  if (!extension_service)  // The |extension_service| can be NULL in Incognito.
-    return;
-
-  for (size_t i = 0; i < extension_service->extensions()->size(); ++i) {
-    Extension* extension = extension_service->GetExtensionById(
-        extension_service->extensions()->at(i)->id(), false);
-    CreateButtonForExtension(extension);
+  for (ExtensionList::iterator iter = model_->begin();
+       iter != model_->end(); ++iter) {
+    CreateButtonForExtension(*iter);
   }
 }
 
 void BrowserActionsToolbarGtk::CreateButtonForExtension(Extension* extension) {
-  // Only show extensions with browser actions.
-  if (!extension->browser_action())
-    return;
-
   RemoveButtonForExtension(extension);
   linked_ptr<BrowserActionButton> button(
       new BrowserActionButton(this, extension));
   gtk_box_pack_end(GTK_BOX(hbox_.get()), button->widget(), FALSE, FALSE, 0);
   gtk_widget_show(button->widget());
   extension_button_map_[extension->id()] = button;
 
   UpdateVisibility();
 }
 
 void BrowserActionsToolbarGtk::RemoveButtonForExtension(Extension* extension) {
   if (extension_button_map_.erase(extension->id()))
     UpdateVisibility();
 }
 
 void BrowserActionsToolbarGtk::UpdateVisibility() {
   if (button_count() == 0)
     gtk_widget_hide(widget());
   else
     gtk_widget_show(widget());
 }
+
+void BrowserActionsToolbarGtk::ExtensionAdded(Extension* extension,
+                                              int index) {
+  // TODO(estade): respect |index|.
+  CreateButtonForExtension(extension);
+}
+
+void BrowserActionsToolbarGtk::ExtensionRemoved(Extension* extension) {
+  RemoveButtonForExtension(extension);
+}