Order the script badges in the location bar consistently

chrome/browser/extensions/script_badge_controller.h

 // 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.
 
 #ifndef CHROME_BROWSER_EXTENSIONS_SCRIPT_BADGE_CONTROLLER_H_
 #define CHROME_BROWSER_EXTENSIONS_SCRIPT_BADGE_CONTROLLER_H_
 #pragma once
 
 #include <map>
 #include <set>
 #include <string>
 
 #include "base/compiler_specific.h"
 #include "base/memory/linked_ptr.h"
 #include "base/memory/ref_counted.h"
 #include "chrome/browser/extensions/location_bar_controller.h"
 #include "chrome/browser/extensions/script_executor.h"
 #include "chrome/browser/extensions/script_executor_impl.h"
+#include "content/public/browser/notification_observer.h"
+#include "content/public/browser/notification_registrar.h"
 #include "content/public/browser/web_contents_observer.h"
 
 class ExtensionAction;
 class ExtensionService;
 class TabContents;
 
 namespace IPC {
 class Message;
 }
 
@@ skipping 12 matching lines @@
 //
 // When extension IDs are recorded a NOTIFICATION_EXTENSION_LOCATION_BAR_UPDATED
 // is sent, and those extensions will be returned from GetCurrentActions until
 // the next page navigation.
 //
 // Ref-counted so that it can be safely bound in a base::Bind.
 class ScriptBadgeController
     : public base::RefCountedThreadSafe<ScriptBadgeController>,
       public LocationBarController,
       public ScriptExecutor,
-      public content::WebContentsObserver {
+      public content::WebContentsObserver,
+      public content::NotificationObserver {
  public:
   explicit ScriptBadgeController(TabContents* tab_contents);
 
   // LocationBarController implementation.
   virtual scoped_ptr<std::vector<ExtensionAction*> > GetCurrentActions()
       OVERRIDE;
   virtual Action OnClicked(const std::string& extension_id,
                            int mouse_button) OVERRIDE;
 
   // ScriptExecutor implementation.
@@ skipping 25 matching lines @@
 
   // Sends a LOCATION_BAR_UPDATED notification.
   void Notify();
 
   // content::WebContentsObserver implementation.
   virtual void DidNavigateMainFrame(
       const content::LoadCommittedDetails& details,
       const content::FrameNavigateParams& params) OVERRIDE;
   virtual bool OnMessageReceived(const IPC::Message& message) OVERRIDE;
 
+  // content::NotificationObserver implementation.
+  virtual void Observe(int type,
+                       const content::NotificationSource& source,
+                       const content::NotificationDetails& details) OVERRIDE;
+
   // IPC::Message handlers.
   void OnContentScriptsExecuting(const std::set<std::string>& extension_ids,
                                  int32 page_id);
 
+  // Tries to insert an extension into the relevant collections, and returns
+  // whether any change was made.
+  bool InsertExtension(const std::string& extension_id);
+
+  // Tries to erase an extension from the relevant collections, and returns
+  // whether any change was made.
+  bool EraseExtension(const Extension* extension);
+
   // Delegate ScriptExecutorImpl for running ExecuteScript.
   ScriptExecutorImpl script_executor_;
 
   // Our parent TabContents.
   TabContents* tab_contents_;
 
+  // The current extension actions in the order they appeared.
+  std::vector<ExtensionAction*> current_actions_;
+
   // The extensions that have called ExecuteScript on the current frame.
   std::set<std::string> extensions_executing_scripts_;
 
+  // Listen to extension unloaded notifications.
+  content::NotificationRegistrar registrar_;
+
   DISALLOW_COPY_AND_ASSIGN(ScriptBadgeController);
 };
 
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_SCRIPT_BADGE_CONTROLLER_H_