1 files changed, 70 insertions, 0 deletions
diff --git a/chrome/common/extensions/extension_set.h b/chrome/common/extensions/extension_set.h
new file mode 100644
index 0000000..9d00d1d
--- /dev/null
+++ b/chrome/common/extensions/extension_set.h
@@ -0,0 +1,70 @@
+// Copyright (c) 2011 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_COMMON_EXTENSIONS_EXTENSION_SET_H_
+#define CHROME_COMMON_EXTENSIONS_EXTENSION_SET_H_
+#pragma once
+
+#include <string>
+#include <vector>
+
+#include "base/gtest_prod_util.h"
+#include "base/ref_counted.h"
+#include "chrome/common/extensions/extension.h"
+#include "googleurl/src/gurl.h"
+
+// The one true extension container. Extensions are identified by their id.
+// Only one extension can be in the set with a given ID.
+class ExtensionSet {
+ public:
+  ExtensionSet();
+  ~ExtensionSet();
+
+  // Gets the number of extensions contained.
+  size_t size() const;
+
+  // Returns true if the set contains the specified extension.
+  bool Contains(const std::string& id);
+
+  // Adds the specified extension to the set. The set becomes an owner. Any
+  // previous extension with the same ID is removed.
+  void Insert(const scoped_refptr<const Extension>& extension);
+
+  // Removes the specified extension.
+  void Remove(const std::string& id);
+
+  // Returns the extension ID that the given URL is a part of, or empty if
+  // none. This includes web URLs that are part of an extension's web extent.
+  std::string GetIdByURL(const GURL& url) const;
+
+  // Returns the Extension that the given URL is a part of, or NULL if none.
+  // This includes web URLs that are part of an extension's web extent.
+  // NOTE: This can return NULL if called before UpdateExtensions receives
+  // bulk extension data (e.g. if called from
+  // EventBindings::HandleContextCreated)
+  const Extension* GetByURL(const GURL& url) const;
+
+  // Returns true if |new_url| is in the extent of the same extension as
+  // |old_url|.  Also returns true if neither URL is in an app.
+  bool InSameExtent(const GURL& old_url, const GURL& new_url) const;
+
+  // Look up an Extension object by id.
+  const Extension* GetByID(const std::string& id) const;
+
+  // Returns true if |url| should get extension api bindings and be permitted
+  // to make api calls. Note that this is independent of what extension
+  // permissions the given extension has been granted.
+  bool ExtensionBindingsAllowed(const GURL& url) const;
+
+ private:
+  FRIEND_TEST_ALL_PREFIXES(ExtensionSetTest, ExtensionSet);
+
+  // static
+  typedef std::map<std::string, scoped_refptr<const Extension> > ExtensionMap;
+  ExtensionMap extensions_;
+
+  DISALLOW_COPY_AND_ASSIGN(ExtensionSet);
+};
+
+#endif  // CHROME_COMMON_EXTENSIONS_EXTENSION_SET_H_