Implement scriptBadge.requestToAct.

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_
 
 #include <map>
 #include <set>
 #include <string>
@@ skipping 36 matching lines @@
       public ScriptExecutor::Observer,
       public content::WebContentsObserver,
       public content::NotificationObserver {
  public:
   explicit ScriptBadgeController(TabContents* tab_contents,
                                  ScriptExecutor* script_executor);
   virtual ~ScriptBadgeController();
 
   // LocationBarController implementation.
   virtual std::vector<ExtensionAction*> GetCurrentActions() const OVERRIDE;
+  virtual void GetAttentionFor(const std::string& extension_id) OVERRIDE;
   virtual Action OnClicked(const std::string& extension_id,
                            int mouse_button) OVERRIDE;
   virtual void NotifyChange() OVERRIDE;
 
   // ScriptExecutor::Observer implementation.
   virtual void OnExecuteScriptFinished(const std::string& extension_id,
                                        bool success,
                                        int32 page_id,
                                        const std::string& error) OVERRIDE;
 
@@ skipping 12 matching lines @@
 
   // 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);
+  // Adds the extension's icon to the list of script badges.  Returns
+  // the script badge ExtensionAction that was added, or NULL if
+  // extension_id isn't valid.
+  ExtensionAction* AddExtensionToCurrentActions(
+      const std::string& extension_id);
+
+  // Called when an extension is running script on the current tab,
+  // and tries to insert an extension into the relevant collections.
+  // Returns true if any change was made.
+  bool MarkExtensionExecuting(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);
 
   // Our parent TabContents.
   TabContents* tab_contents_;
 
-  // The current extension actions in the order they appeared.
+  // The current extension actions in the order they appeared.  These come from
+  // calls to ExecuteScript or getAttention on the current frame.
   std::vector<ExtensionAction*> current_actions_;
 
-  // The extensions that have called ExecuteScript on the current frame.
-  std::set<std::string> extensions_executing_scripts_;
+  // The extensions that have actions in current_actions_.
+  std::set<std::string> extensions_in_current_actions_;
 
   // Listen to extension unloaded notifications.
   content::NotificationRegistrar registrar_;
 
   DISALLOW_COPY_AND_ASSIGN(ScriptBadgeController);
 };
 
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_SCRIPT_BADGE_CONTROLLER_H_