Implement Keybinding for script badges.

chrome/browser/extensions/api/commands/command_service.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_API_COMMANDS_COMMAND_SERVICE_H_
 #define CHROME_BROWSER_EXTENSIONS_API_COMMANDS_COMMAND_SERVICE_H_
 
 #include <string>
 
 #include "base/basictypes.h"
@@ skipping 31 matching lines @@
     ACTIVE_ONLY,
   };
 
   // Register prefs for keybinding.
   static void RegisterUserPrefs(PrefService* user_prefs);
 
   // Constructs a CommandService object for the given profile.
   explicit CommandService(Profile* profile);
   virtual ~CommandService();
 
-  // Gets the keybinding (if any) for the browser action of an extension given
+  // Gets the command (if any) for the browser action of an extension given
   // its |extension_id|. The function consults the master list to see if
-  // the keybinding is active. Returns false if the extension has no browser
-  // action. Returns false if the keybinding is not active and |type| requested
+  // the command is active. Returns false if the extension has no browser
+  // action. Returns false if the command is not active and |type| requested
   // is ACTIVE_ONLY. |command| contains the command found and |active| (if not
   // NULL) contains whether |command| is active.
   bool GetBrowserActionCommand(const std::string& extension_id,
                                QueryType type,
                                extensions::Command* command,
                                bool* active);
 
-  // Gets the keybinding (if any) for the page action of an extension given
+  // Gets the command (if any) for the page action of an extension given
   // its |extension_id|. The function consults the master list to see if
-  // the keybinding is active. Returns false if the extension has no page
-  // action. Returns false if the keybinding is not active and |type| requested
+  // the command is active. Returns false if the extension has no page
+  // action. Returns false if the command is not active and |type| requested
   // is ACTIVE_ONLY. |command| contains the command found and |active| (if not
   // NULL) contains whether |command| is active.
   bool GetPageActionCommand(const std::string& extension_id,
                             QueryType type,
                             extensions::Command* command,
                             bool* active);
 
-  // Gets the active keybinding (if any) for the named commands of an extension
+  // Gets the command (if any) for the script badge of an extension given
+  // its |extension_id|. The function consults the master list to see if
+  // the command is active. Returns false if the extension has no script
+  // badge. Returns false if the command is not active and |type| requested
+  // is ACTIVE_ONLY. |command| contains the command found and |active| (if not
+  // NULL) contains whether |command| is active.
+  bool GetScriptBadgeCommand(const std::string& extension_id,
+                             QueryType type,
+                             extensions::Command* command,
+                             bool* active);
+
+  // Gets the active command (if any) for the named commands of an extension
   // given its |extension_id|. The function consults the master list to see if
-  // the keybinding is active. Returns an empty map if the extension has no
+  // the command is active. Returns an empty map if the extension has no
   // named commands or no active named commands when |type| requested is
   // ACTIVE_ONLY.
   bool GetNamedCommands(const std::string& extension_id,
                         QueryType type,
                         extensions::CommandMap* command_map);
 
   // Records a keybinding |accelerator| as active for an extension with id
   // |extension_id| and command with the name |command_name|. If
   // |allow_overrides| is false, the keybinding must be free for the change to
   // be recorded (as determined by the master list in |user_prefs|). If
@@ skipping 18 matching lines @@
   // command is not found.
   ui::Accelerator FindShortcutForCommand(const std::string& extension_id,
                                          const std::string& command);
 
   // Overridden from content::NotificationObserver.
   virtual void Observe(int type,
                        const content::NotificationSource& source,
                        const content::NotificationDetails& details) OVERRIDE;
 
  private:
+   // An enum specifying the types of icons that can have a command.
+   enum ExtensionActionType {
+     BROWSER_ACTION,
+     PAGE_ACTION,
+     SCRIPT_BADGE,
+   };
+
   // Assigns initial keybinding for a given |extension|'s page action, browser
   // action and named commands. In each case, if the suggested keybinding is
   // free, it will be taken by this extension. If not, that keybinding request
   // is ignored. |user_pref| is the PrefService used to record the new
   // keybinding assignment.
   void AssignInitialKeybindings(const extensions::Extension* extension);
 
   // Removes all keybindings for a given extension by its |extension_id|.
   // |command_name| is optional and if specified, causes only the command with
   // the name |command_name| to be removed.
   void RemoveKeybindingPrefs(const std::string& extension_id,
                              const std::string& command_name);
 
   bool GetExtensionActionCommand(const std::string& extension_id,
-                                 QueryType type,
+                                 QueryType query_type,
                                  extensions::Command* command,
                                  bool* active,
-                                 bool browser_action);
+                                 ExtensionActionType action_type);
 
   // The content notification registrar for listening to extension events.
   content::NotificationRegistrar registrar_;
 
   // A weak pointer to the profile we are associated with. Not owned by us.
   Profile* profile_;
 
   DISALLOW_COPY_AND_ASSIGN(CommandService);
 };
 
 }  //  namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_API_COMMANDS_COMMAND_SERVICE_H_