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

chrome/browser/extensions/api/omnibox/omnibox_api.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_OMNIBOX_OMNIBOX_API_H_
 #define CHROME_BROWSER_EXTENSIONS_API_OMNIBOX_OMNIBOX_API_H_
 
+#include <set>
 #include <string>
 #include <vector>
 
 #include "base/memory/scoped_ptr.h"
 #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;
 }
 
 namespace content {
 class WebContents;
 }
 
+namespace gfx {
+class Image;
+}
+
 namespace extensions {
 
 // Event router class for events related to the omnibox API.
 class ExtensionOmniboxEventRouter {
  public:
   // The user has just typed the omnibox keyword. This is sent exactly once in
   // a given input session, before any OnInputChanged events.
   static void OnInputStarted(
       Profile* profile, const std::string& extension_id);
 
@@ skipping 24 matching lines @@
  public:
   DECLARE_EXTENSION_FUNCTION_NAME("omnibox.sendSuggestions");
 
  protected:
   virtual ~OmniboxSendSuggestionsFunction() {}
 
   // ExtensionFunction:
   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::set<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");
 
  protected:
   virtual ~OmniboxSetDefaultSuggestionFunction() {}
 
   // ExtensionFunction:
   virtual bool RunImpl() OVERRIDE;
 };
@@ skipping 40 matching lines @@
 // omnibox.setDefaultSuggestion, apply that to |match|. Otherwise, do nothing.
 void ApplyDefaultSuggestionForExtensionKeyword(
     Profile* profile,
     const TemplateURL* keyword,
     const string16& remaining_input,
     AutocompleteMatch* match);
 
 }  // namespace extensions
 
 #endif  // CHROME_BROWSER_EXTENSIONS_API_OMNIBOX_OMNIBOX_API_H_