Move ExtensionService::GetExtensionById() to ExtensionRegistry

extensions/browser/extension_registry.h

 // Copyright 2013 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 EXTENSIONS_BROWSER_EXTENSION_REGISTRY_H_
 #define EXTENSIONS_BROWSER_EXTENSION_REGISTRY_H_
 
 #include <string>
 
 #include "base/compiler_specific.h"
 #include "base/memory/ref_counted.h"
 #include "components/browser_context_keyed_service/browser_context_keyed_service.h"
 #include "extensions/common/extension_set.h"
 
 namespace content {
 class BrowserContext;
 }
 
 namespace extensions {
 class Extension;
 
 // ExtensionRegistry holds sets of the installed extensions for a given
 // BrowserContext. An incognito browser context and its master browser context
 // share a single registry.
 class ExtensionRegistry : public BrowserContextKeyedService {
  public:
+  // Flags to pass to GetExtensionById() to select which sets to look in.
+  enum IncludeFlag {
+    NONE        = 0,
+    ENABLED     = 1 << 0,
+    DISABLED    = 1 << 1,
+    TERMINATED  = 1 << 2,
+    BLACKLISTED = 1 << 3,
+    EVERYTHING = (1 << 4) - 1,
+  };
+
   ExtensionRegistry();
   virtual ~ExtensionRegistry();
 
   // Returns the instance for the given |browser_context|.
   static ExtensionRegistry* Get(content::BrowserContext* browser_context);
 
   // NOTE: These sets are *eventually* mututally exclusive, but an extension can
   // appear in two sets for short periods of time.
   const ExtensionSet& enabled_extensions() const {
     return enabled_extensions_;
   }
   const ExtensionSet& disabled_extensions() const {
     return disabled_extensions_;
   }
   const ExtensionSet& terminated_extensions() const {
     return terminated_extensions_;
   }
   const ExtensionSet& blacklisted_extensions() const {
     return blacklisted_extensions_;
   }
 
+  // Find an extension by ID using |include_mask| to pick the sets to search:
+  //  * enabled_extensions()     --> ExtensionRegistry::ENABLED
+  //  * disabled_extensions()    --> ExtensionRegistry::DISABLED
+  //  * terminated_extensions()  --> ExtensionRegistry::TERMINATED
+  //  * blacklisted_extensions() --> ExtensionRegistry::BLACKLISTED
+  // Returns NULL if the extension is not found in the selected sets.
+  const Extension* GetExtensionById(const std::string& id,
+                                    int include_mask) const;
+
   // Adds the specified extension to the enabled set. The registry becomes an
   // owner. Any previous extension with the same ID is removed.
   // Returns true if there is no previous extension.
   // NOTE: You probably want to use ExtensionService instead of calling this
   // method directly.
   bool AddEnabled(const scoped_refptr<const Extension>& extension);
 
   // Removes the specified extension from the enabled set.
   // Returns true if the set contained the specified extension.
   // NOTE: You probably want to use ExtensionService instead of calling this
@@ skipping 39 matching lines @@
   // so that if extensions are blacklisted by mistake they can easily be
   // un-blacklisted.
   ExtensionSet blacklisted_extensions_;
 
   DISALLOW_COPY_AND_ASSIGN(ExtensionRegistry);
 };
 
 }  // namespace extensions
 
 #endif  // EXTENSIONS_BROWSER_EXTENSION_REGISTRY_H_