Move/replace/rename URL-based extension getters from ExtensionService to/in ExtensionSet.

chrome/browser/extensions/extension_service.cc

 // 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.
 
 #include "chrome/browser/extensions/extension_service.h"
 
 #include <algorithm>
 #include <set>
 
 #include "base/basictypes.h"
@@ skipping 235 matching lines @@
 
 // If a download url matches one of these patterns and has a referrer of the
 // webstore, then we're willing to treat that as a gallery download.
 static const char* kAllowedDownloadURLPatterns[] = {
   "https://clients2.google.com/service/update2*",
   "https://clients2.googleusercontent.com/crx/*"
 };
 
 bool ExtensionService::IsDownloadFromGallery(const GURL& download_url,
                                              const GURL& referrer_url) {
-  const Extension* download_extension = GetExtensionByWebExtent(download_url);
-  const Extension* referrer_extension = GetExtensionByWebExtent(referrer_url);
+  const Extension* download_extension =
+      extensions_.GetByWebExtent(ExtensionURLInfo(download_url));
+  const Extension* referrer_extension =
+      extensions_.GetByWebExtent(ExtensionURLInfo(referrer_url));
   const Extension* webstore_app = GetWebStoreApp();
 
   bool referrer_valid = (referrer_extension == webstore_app);
   bool download_valid = (download_extension == webstore_app);
 
   // We also allow the download to be from a small set of trusted paths.
   if (!download_valid) {
     for (size_t i = 0; i < arraysize(kAllowedDownloadURLPatterns); i++) {
       URLPattern pattern(URLPattern::SCHEME_HTTPS,
                          kAllowedDownloadURLPatterns[i]);
@@ skipping 30 matching lines @@
 
       // Otherwise, the TLD must match the TLD of the command-line url.
       download_valid = (download_tld == store_tld);
     }
   }
 
   return (referrer_valid && download_valid);
 }
 
 const Extension* ExtensionService::GetInstalledApp(const GURL& url) {
-  // Check for hosted app.
-  const Extension* app = GetExtensionByWebExtent(url);
-  if (app)
-    return app;
-
-  // Check for packaged app.
-  app = GetExtensionByURL(url);
+  const Extension* app = extensions_.GetByURL(ExtensionURLInfo(url));

[Aaron Boodman, 2011/12/09 16:05:20]

This reads a little funny. Can you change the variable name to 'extension' as
long as you're in here?

[Yoyo Zhou, 2011/12/09 19:59:09]

Uh, sure, I can see why this seems off.

   if (app && app->is_app())
     return app;
 
   return NULL;
 }
 
 bool ExtensionService::IsInstalledApp(const GURL& url) {
   return !!GetInstalledApp(url);
 }
 
@@ skipping 1820 matching lines @@
 
 const Extension* ExtensionService::GetInstalledExtension(
     const std::string& id) const {
   return GetExtensionByIdInternal(id, true, true, true);
 }
 
 const Extension* ExtensionService::GetWebStoreApp() {
   return GetExtensionById(extension_misc::kWebStoreAppId, false);
 }
 
-const Extension* ExtensionService::GetExtensionByURL(const GURL& url) {
-  return url.scheme() != chrome::kExtensionScheme ? NULL :
-      GetExtensionById(url.host(), false);
-}
-
-const Extension* ExtensionService::GetExtensionByWebExtent(const GURL& url) {
-  // TODO(yoz): Should be ExtensionSet::GetByURL.
-  for (ExtensionSet::const_iterator iter = extensions_.begin();
-       iter != extensions_.end(); ++iter) {
-    if ((*iter)->web_extent().MatchesURL(url))
-      return *iter;
-  }
-  return NULL;
-}
-
-const Extension* ExtensionService::GetDisabledExtensionByWebExtent(
-    const GURL& url) {
-  // TODO(yoz): Should be ExtensionSet::GetByURL.
-  for (ExtensionSet::const_iterator iter = disabled_extensions_.begin();
-       iter != disabled_extensions_.end(); ++iter) {
-    if ((*iter)->web_extent().MatchesURL(url))
-      return *iter;
-  }
-  return NULL;
-}
-
 bool ExtensionService::ExtensionBindingsAllowed(const GURL& url) {
-  // Allow bindings for all packaged extensions.
-  // Note that GetExtensionByURL may return an Extension for hosted apps
-  // (excluding bookmark apps) if the URL came from GetEffectiveURL.
-  const Extension* extension = GetExtensionByURL(url);
-  if (extension && extension->GetType() != Extension::TYPE_HOSTED_APP)
-    return true;
-
-  // Allow bindings for all component, hosted apps.
-  if (!extension)
-    extension = GetExtensionByWebExtent(url);
-  return (extension && extension->location() == Extension::COMPONENT);
-}
-
-const Extension* ExtensionService::GetExtensionByOverlappingWebExtent(
-    const URLPatternSet& extent) {
-  // TODO(yoz): Should be in ExtensionSet.
-  for (ExtensionSet::const_iterator iter = extensions_.begin();
-       iter != extensions_.end(); ++iter) {
-    if ((*iter)->web_extent().OverlapsWith(extent))
-      return *iter;
-  }
-
-  return NULL;
+  // Allow bindings for all packaged extensions and component hosted apps.
+  const Extension* extension = extensions_.GetByURL(ExtensionURLInfo(url));
+  return extension && (!extension->is_hosted_app() ||
+                       extension->location() == Extension::COMPONENT);
 }
 
 const SkBitmap& ExtensionService::GetOmniboxIcon(
     const std::string& extension_id) {
   return omnibox_icon_manager_.GetIcon(extension_id);
 }
 
 const SkBitmap& ExtensionService::GetOmniboxPopupIcon(
     const std::string& extension_id) {
   return omnibox_popup_icon_manager_.GetIcon(extension_id);
@@ skipping 290 matching lines @@
 
 ExtensionService::NaClModuleInfoList::iterator
     ExtensionService::FindNaClModule(const GURL& url) {
   for (NaClModuleInfoList::iterator iter = nacl_module_list_.begin();
        iter != nacl_module_list_.end(); ++iter) {
     if (iter->url == url)
       return iter;
   }
   return nacl_module_list_.end();
 }