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_
 #pragma once
 
 #include <map>
 #include <set>
@@ skipping 39 matching lines @@
     : public base::RefCountedThreadSafe<ScriptBadgeController>,
       public LocationBarController,
       public ScriptExecutor,
       public content::WebContentsObserver,
       public content::NotificationObserver {
  public:
   explicit ScriptBadgeController(TabContents* tab_contents);
 
   // LocationBarController implementation.
   virtual std::vector<ExtensionAction*> GetCurrentActions() const OVERRIDE;
+  virtual void RequestScriptPermission(
+      const std::string& extension_id) OVERRIDE;
   virtual Action OnClicked(const std::string& extension_id,
                            int mouse_button) OVERRIDE;
   virtual void NotifyChange() OVERRIDE;
 
   // ScriptExecutor implementation.
   virtual void ExecuteScript(const std::string& extension_id,
                              ScriptType script_type,
                              const std::string& code,
                              FrameScope frame_scope,
                              UserScript::RunLocation run_at,
@@ skipping 26 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);
 
   // Delegate ScriptExecutorImpl for running ExecuteScript.
   ScriptExecutorImpl script_executor_;
 
   // 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 requestToAct 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_