Home | History | Annotate | Download | only in extensions 
      1 // Copyright (c) 2014 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 #include "chrome/browser/extensions/settings_api_helpers.h"
      6 
      7 #include "chrome/browser/extensions/api/preference/preference_api.h"
      8 #include "chrome/common/pref_names.h"
      9 #include "components/search_engines/search_engines_pref_names.h"
     10 #include "extensions/browser/extension_pref_value_map.h"
     11 #include "extensions/browser/extension_pref_value_map_factory.h"
     12 #include "extensions/browser/extension_registry.h"
     13 #include "extensions/common/extension_set.h"
     14 
     15 namespace extensions {
     16 
     17 namespace {
     18 
     19 // Returns which |extension| (if any) is overriding a particular |type| of
     20 // setting.
     21 const Extension* FindOverridingExtension(
     22     content::BrowserContext* browser_context,
     23     SettingsApiOverrideType type) {
     24   const ExtensionSet& extensions =
     25       ExtensionRegistry::Get(browser_context)->enabled_extensions();
     26 
     27   for (ExtensionSet::const_iterator it = extensions.begin();
     28        it != extensions.end();
     29        ++it) {
     30     const SettingsOverrides* settings = SettingsOverrides::Get(it->get());
     31     if (settings) {
     32       if (type == BUBBLE_TYPE_HOME_PAGE && !settings->homepage)
     33         continue;
     34       if (type == BUBBLE_TYPE_STARTUP_PAGES && settings->startup_pages.empty())
     35         continue;
     36       if (type == BUBBLE_TYPE_SEARCH_ENGINE && !settings->search_engine)
     37         continue;
     38 
     39       std::string key;
     40       switch (type) {
     41         case BUBBLE_TYPE_HOME_PAGE:
     42           key = prefs::kHomePage;
     43           break;
     44         case BUBBLE_TYPE_STARTUP_PAGES:
     45           key = prefs::kRestoreOnStartup;
     46           break;
     47         case BUBBLE_TYPE_SEARCH_ENGINE:
     48           key = prefs::kDefaultSearchProviderEnabled;
     49           break;
     50       }
     51 
     52       // Found an extension overriding the current type, check if primary.
     53       PreferenceAPI* preference_api = PreferenceAPI::Get(browser_context);
     54       if (preference_api &&  // Expected to be NULL in unit tests.
     55           !preference_api->DoesExtensionControlPref((*it)->id(), key, NULL))
     56         continue;  // Not primary.
     57 
     58       // Found the primary extension.
     59       return it->get();
     60     }
     61   }
     62 
     63   return NULL;
     64 }
     65 
     66 }  // namespace
     67 
     68 const Extension* GetExtensionOverridingHomepage(
     69     content::BrowserContext* browser_context) {
     70   return FindOverridingExtension(browser_context, BUBBLE_TYPE_HOME_PAGE);
     71 }
     72 
     73 const Extension* GetExtensionOverridingStartupPages(
     74     content::BrowserContext* browser_context) {
     75   return FindOverridingExtension(browser_context, BUBBLE_TYPE_STARTUP_PAGES);
     76 }
     77 
     78 const Extension* GetExtensionOverridingSearchEngine(
     79     content::BrowserContext* browser_context) {
     80   return FindOverridingExtension(browser_context, BUBBLE_TYPE_SEARCH_ENGINE);
     81 }
     82 
     83 const Extension* GetExtensionOverridingProxy(
     84     content::BrowserContext* browser_context) {
     85   ExtensionPrefValueMap* extension_prefs_value_map =
     86       ExtensionPrefValueMapFactory::GetForBrowserContext(browser_context);
     87   if (!extension_prefs_value_map)
     88     return NULL;  // Can be null during testing.
     89   std::string extension_id =
     90       extension_prefs_value_map->GetExtensionControllingPref(prefs::kProxy);
     91   if (extension_id.empty())
     92     return NULL;
     93   return ExtensionRegistry::Get(browser_context)->GetExtensionById(
     94       extension_id, ExtensionRegistry::ENABLED);
     95 }
     96 
     97 }  // namespace extensions
     98