Home | History | Annotate | Download | only in extensions 
      1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef CHROME_BROWSER_EXTENSIONS_EXTENSION_ACTION_MANAGER_H_
      6 #define CHROME_BROWSER_EXTENSIONS_EXTENSION_ACTION_MANAGER_H_
      7 
      8 #include <map>
      9 #include <string>
     10 
     11 #include "base/memory/linked_ptr.h"
     12 #include "components/browser_context_keyed_service/browser_context_keyed_service.h"
     13 #include "content/public/browser/notification_observer.h"
     14 #include "content/public/browser/notification_registrar.h"
     15 
     16 class ExtensionAction;
     17 class Profile;
     18 
     19 namespace extensions {
     20 
     21 class Extension;
     22 
     23 // Owns the ExtensionActions associated with each extension.  These actions live
     24 // while an extension is loaded and are destroyed on unload.
     25 class ExtensionActionManager : public BrowserContextKeyedService,
     26                                public content::NotificationObserver {
     27  public:
     28   explicit ExtensionActionManager(Profile* profile);
     29   virtual ~ExtensionActionManager();
     30 
     31   // Returns this profile's ExtensionActionManager.  One instance is
     32   // shared between a profile and its incognito version.
     33   static ExtensionActionManager* Get(Profile* profile);
     34 
     35   // Retrieves the page action, browser action, or script badge for |extension|.
     36   // If the result is not NULL, it remains valid until the extension is
     37   // unloaded.
     38   ExtensionAction* GetPageAction(const extensions::Extension& extension) const;
     39   ExtensionAction* GetBrowserAction(
     40       const extensions::Extension& extension) const;
     41   ExtensionAction* GetScriptBadge(const extensions::Extension& extension) const;
     42   ExtensionAction* GetSystemIndicator(
     43       const extensions::Extension& extension) const;
     44 
     45  private:
     46   // Implement content::NotificationObserver.
     47   virtual void Observe(int type,
     48                        const content::NotificationSource& source,
     49                        const content::NotificationDetails& details) OVERRIDE;
     50 
     51   content::NotificationRegistrar registrar_;
     52   Profile* profile_;
     53 
     54   // Keyed by Extension ID.  These maps are populated lazily when their
     55   // ExtensionAction is first requested, and the entries are removed when the
     56   // extension is unloaded.  Not every extension has a page action or browser
     57   // action, but all have a script badge.
     58   typedef std::map<std::string, linked_ptr<ExtensionAction> > ExtIdToActionMap;
     59   mutable ExtIdToActionMap page_actions_;
     60   mutable ExtIdToActionMap browser_actions_;
     61   mutable ExtIdToActionMap script_badges_;
     62   mutable ExtIdToActionMap system_indicators_;
     63 };
     64 
     65 }  // namespace extensions
     66 
     67 #endif  // CHROME_BROWSER_EXTENSIONS_EXTENSION_ACTION_MANAGER_H_
     68