Home | History | Annotate | Download | only in extensions 
      1 // Copyright 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/extension_management_test_util.h"
      6 
      7 #include "components/crx_file/id_util.h"
      8 
      9 namespace extensions {
     10 
     11 namespace schema = schema_constants;
     12 
     13 namespace {
     14 
     15 std::string make_path(std::string a, std::string b) {
     16   return a + "." + b;
     17 }
     18 
     19 const char kInstallSourcesPath[] = "*.install_sources";
     20 const char kAllowedTypesPath[] = "*.allowed_types";
     21 
     22 }  // namespace
     23 
     24 ExtensionManagementPrefUpdaterBase::ExtensionManagementPrefUpdaterBase() {
     25 }
     26 
     27 ExtensionManagementPrefUpdaterBase::~ExtensionManagementPrefUpdaterBase() {
     28 }
     29 
     30 void ExtensionManagementPrefUpdaterBase::UnsetPerExtensionSettings(
     31     const ExtensionId& id) {
     32   DCHECK(crx_file::id_util::IdIsValid(id));
     33   pref_->RemoveWithoutPathExpansion(id, NULL);
     34 }
     35 
     36 void ExtensionManagementPrefUpdaterBase::ClearPerExtensionSettings(
     37     const ExtensionId& id) {
     38   DCHECK(crx_file::id_util::IdIsValid(id));
     39   pref_->SetWithoutPathExpansion(id, new base::DictionaryValue());
     40 }
     41 
     42 void ExtensionManagementPrefUpdaterBase::SetBlacklistedByDefault(bool value) {
     43   pref_->SetString(make_path(schema::kWildcard, schema::kInstallationMode),
     44                    value ? schema::kBlocked : schema::kAllowed);
     45 }
     46 
     47 void ExtensionManagementPrefUpdaterBase::
     48     ClearInstallationModesForIndividualExtensions() {
     49   for (base::DictionaryValue::Iterator it(*pref_.get()); !it.IsAtEnd();
     50        it.Advance()) {
     51     DCHECK(it.value().IsType(base::Value::TYPE_DICTIONARY));
     52     if (it.key() != schema::kWildcard) {
     53       DCHECK(crx_file::id_util::IdIsValid(it.key()));
     54       pref_->Remove(make_path(it.key(), schema::kInstallationMode), NULL);
     55       pref_->Remove(make_path(it.key(), schema::kUpdateUrl), NULL);
     56     }
     57   }
     58 }
     59 
     60 void
     61 ExtensionManagementPrefUpdaterBase::SetIndividualExtensionInstallationAllowed(
     62     const ExtensionId& id,
     63     bool allowed) {
     64   DCHECK(crx_file::id_util::IdIsValid(id));
     65   pref_->SetString(make_path(id, schema::kInstallationMode),
     66                    allowed ? schema::kAllowed : schema::kBlocked);
     67   pref_->Remove(make_path(id, schema::kUpdateUrl), NULL);
     68 }
     69 
     70 void ExtensionManagementPrefUpdaterBase::SetIndividualExtensionAutoInstalled(
     71     const ExtensionId& id,
     72     const std::string& update_url,
     73     bool forced) {
     74   DCHECK(crx_file::id_util::IdIsValid(id));
     75   pref_->SetString(make_path(id, schema::kInstallationMode),
     76                    forced ? schema::kForceInstalled : schema::kNormalInstalled);
     77   pref_->SetString(make_path(id, schema::kUpdateUrl), update_url);
     78 }
     79 
     80 void ExtensionManagementPrefUpdaterBase::UnsetInstallSources() {
     81   pref_->Remove(kInstallSourcesPath, NULL);
     82 }
     83 
     84 void ExtensionManagementPrefUpdaterBase::ClearInstallSources() {
     85   ClearList(kInstallSourcesPath);
     86 }
     87 
     88 void ExtensionManagementPrefUpdaterBase::AddInstallSource(
     89     const std::string& install_source) {
     90   AddStringToList(kInstallSourcesPath, install_source);
     91 }
     92 
     93 void ExtensionManagementPrefUpdaterBase::RemoveInstallSource(
     94     const std::string& install_source) {
     95   RemoveStringFromList(kInstallSourcesPath, install_source);
     96 }
     97 
     98 void ExtensionManagementPrefUpdaterBase::UnsetAllowedTypes() {
     99   pref_->Remove(kAllowedTypesPath, NULL);
    100 }
    101 
    102 void ExtensionManagementPrefUpdaterBase::ClearAllowedTypes() {
    103   ClearList(kAllowedTypesPath);
    104 }
    105 
    106 void ExtensionManagementPrefUpdaterBase::AddAllowedType(
    107     const std::string& allowed_type) {
    108   AddStringToList(kAllowedTypesPath, allowed_type);
    109 }
    110 
    111 const base::DictionaryValue* ExtensionManagementPrefUpdaterBase::GetPref() {
    112   return pref_.get();
    113 }
    114 
    115 void ExtensionManagementPrefUpdaterBase::SetPref(base::DictionaryValue* pref) {
    116   pref_.reset(pref);
    117 }
    118 
    119 scoped_ptr<base::DictionaryValue>
    120 ExtensionManagementPrefUpdaterBase::TakePref() {
    121   return pref_.Pass();
    122 }
    123 
    124 void ExtensionManagementPrefUpdaterBase::RemoveAllowedType(
    125     const std::string& allowd_type) {
    126   RemoveStringFromList(kAllowedTypesPath, allowd_type);
    127 }
    128 
    129 void ExtensionManagementPrefUpdaterBase::ClearList(const std::string& path) {
    130   pref_->Set(path, new base::ListValue());
    131 }
    132 
    133 void ExtensionManagementPrefUpdaterBase::AddStringToList(
    134     const std::string& path,
    135     const std::string& str) {
    136   base::ListValue* list_value = NULL;
    137   if (!pref_->GetList(path, &list_value)) {
    138     list_value = new base::ListValue();
    139     pref_->Set(path, list_value);
    140   }
    141   CHECK(list_value->AppendIfNotPresent(new base::StringValue(str)));
    142 }
    143 
    144 void ExtensionManagementPrefUpdaterBase::RemoveStringFromList(
    145     const std::string& path,
    146     const std::string& str) {
    147   base::ListValue* list_value = NULL;
    148   if (pref_->GetList(path, &list_value))
    149     CHECK(list_value->Remove(base::StringValue(str), NULL));
    150 }
    151 
    152 }  // namespace extensions
    153