extension_set.h revision ddb351dbec246cf1fab5ec20d2d5520909041de1
172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen// Copyright (c) 2011 The Chromium Authors. All rights reserved. 272a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen// Use of this source code is governed by a BSD-style license that can be 372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen// found in the LICENSE file. 472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#ifndef CHROME_COMMON_EXTENSIONS_EXTENSION_SET_H_ 672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#define CHROME_COMMON_EXTENSIONS_EXTENSION_SET_H_ 772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#pragma once 872a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include <string> 1072a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include <vector> 1172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 1272a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include "base/gtest_prod_util.h" 13ddb351dbec246cf1fab5ec20d2d5520909041de1Kristian Monsen#include "base/memory/ref_counted.h" 1472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include "chrome/common/extensions/extension.h" 1572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#include "googleurl/src/gurl.h" 1672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 1772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen// The one true extension container. Extensions are identified by their id. 1872a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen// Only one extension can be in the set with a given ID. 1972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsenclass ExtensionSet { 2072a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen public: 2172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen ExtensionSet(); 2272a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen ~ExtensionSet(); 2372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 2472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Gets the number of extensions contained. 2572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen size_t size() const; 2672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 2772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Returns true if the set contains the specified extension. 2872a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen bool Contains(const std::string& id); 2972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 3072a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Adds the specified extension to the set. The set becomes an owner. Any 3172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // previous extension with the same ID is removed. 3272a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen void Insert(const scoped_refptr<const Extension>& extension); 3372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 3472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Removes the specified extension. 3572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen void Remove(const std::string& id); 3672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 3772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Returns the extension ID that the given URL is a part of, or empty if 3872a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // none. This includes web URLs that are part of an extension's web extent. 3972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen std::string GetIdByURL(const GURL& url) const; 4072a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 4172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Returns the Extension that the given URL is a part of, or NULL if none. 4272a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // This includes web URLs that are part of an extension's web extent. 4372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // NOTE: This can return NULL if called before UpdateExtensions receives 4472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // bulk extension data (e.g. if called from 4572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // EventBindings::HandleContextCreated) 4672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen const Extension* GetByURL(const GURL& url) const; 4772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 4872a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Returns true if |new_url| is in the extent of the same extension as 4972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // |old_url|. Also returns true if neither URL is in an app. 5072a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen bool InSameExtent(const GURL& old_url, const GURL& new_url) const; 5172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 5272a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Look up an Extension object by id. 5372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen const Extension* GetByID(const std::string& id) const; 5472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 5572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // Returns true if |url| should get extension api bindings and be permitted 5672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // to make api calls. Note that this is independent of what extension 5772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // permissions the given extension has been granted. 5872a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen bool ExtensionBindingsAllowed(const GURL& url) const; 5972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 6072a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen private: 6172a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen FRIEND_TEST_ALL_PREFIXES(ExtensionSetTest, ExtensionSet); 6272a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 6372a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen // static 6472a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen typedef std::map<std::string, scoped_refptr<const Extension> > ExtensionMap; 6572a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen ExtensionMap extensions_; 6672a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 6772a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen DISALLOW_COPY_AND_ASSIGN(ExtensionSet); 6872a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen}; 6972a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen 7072a454cd3513ac24fbdd0e0cb9ad70b86a99b801Kristian Monsen#endif // CHROME_COMMON_EXTENSIONS_EXTENSION_SET_H_ 71