SupportsUserData and manifest handlers for Extension; use them for the Omnibox API.

chrome/browser/extensions/api/omnibox/omnibox_api.h

Index: chrome/browser/extensions/api/omnibox/omnibox_api.h
diff --git a/chrome/browser/extensions/api/omnibox/omnibox_api.h b/chrome/browser/extensions/api/omnibox/omnibox_api.h
index ee195d435eb71ece41cb7531be2f75a0f5bfe3e7..7e27b4fd2c62daf23e0ee9c95420f26b194d8a99 100644
--- a/chrome/browser/extensions/api/omnibox/omnibox_api.h
+++ b/chrome/browser/extensions/api/omnibox/omnibox_api.h
@@ -12,8 +12,14 @@
 #include "base/string16.h"
 #include "chrome/browser/autocomplete/autocomplete_match.h"
 #include "chrome/browser/extensions/extension_function.h"
+#include "chrome/browser/extensions/extension_icon_manager.h"
+#include "chrome/browser/profiles/profile_keyed_service.h"
+#include "content/public/browser/notification_observer.h"
+#include "content/public/browser/notification_registrar.h"
 
+class Profile;
 class TemplateURL;
+class TemplateURLService;
 
 namespace base {
 class ListValue;
@@ -23,6 +29,10 @@ namespace content {
 class WebContents;
 }
 
+namespace gfx {
+class Image;
+}
+
 namespace extensions {
 
 // Event router class for events related to the omnibox API.
@@ -67,6 +77,46 @@ class OmniboxSendSuggestionsFunction : public SyncExtensionFunction {
   virtual bool RunImpl() OVERRIDE;
 };
 
+class OmniboxAPI : public ProfileKeyedService,
+                   public content::NotificationObserver {
+ public:
+  explicit OmniboxAPI(Profile* profile);
+  virtual ~OmniboxAPI();
+
+  // ProfileKeyedService implementation.
+  virtual void Shutdown() OVERRIDE;
+
+  // Convenience method to get the OmniboxAPI for a profile.
+  static OmniboxAPI* Get(Profile* profile);
+
+  // content::NotificationObserver implementation.
+  virtual void Observe(int type,
+                       const content::NotificationSource& source,
+                       const content::NotificationDetails& details) OVERRIDE;
+
+  // Returns the icon to display in the omnibox for the given extension.
+  gfx::Image GetOmniboxIcon(const std::string& extension_id);
+
+  // Returns the icon to display in the omnibox popup window for the given
+  // extension.
+  gfx::Image GetOmniboxPopupIcon(const std::string& extension_id);
+
+ private:
+  typedef std::list<const Extension*> PendingExtensions;
+
+  TemplateURLService* url_service_;
+
+  // List of extensions waiting for the TemplateURLService to Load to
+  // have keywords registered.
+  PendingExtensions pending_extensions_;
+
+  content::NotificationRegistrar registrar_;
+
+  // Keeps track of favicon-sized omnibox icons for extensions.
+  ExtensionIconManager omnibox_icon_manager_;
+  ExtensionIconManager omnibox_popup_icon_manager_;
+};
+
 class OmniboxSetDefaultSuggestionFunction : public SyncExtensionFunction {
  public:
   DECLARE_EXTENSION_FUNCTION_NAME("omnibox.setDefaultSuggestion");