| OLD | NEW |
| 1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/extensions/component_migration_helper.h" | 5 #include "chrome/browser/extensions/component_migration_helper.h" |
| 6 | 6 |
| 7 #include "base/stl_util.h" | 7 #include "base/stl_util.h" |
| 8 #include "base/values.h" | 8 #include "base/values.h" |
| 9 #include "chrome/browser/extensions/extension_service.h" | 9 #include "chrome/browser/extensions/extension_service.h" |
| 10 #include "chrome/browser/extensions/extension_system_impl.h" | 10 #include "chrome/browser/extensions/extension_system_impl.h" |
| (...skipping 61 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 72 if (IsExtensionInstalledAndEnabled(id)) { | 72 if (IsExtensionInstalledAndEnabled(id)) { |
| 73 extension_was_installed = true; | 73 extension_was_installed = true; |
| 74 UnloadExtension(id); | 74 UnloadExtension(id); |
| 75 } | 75 } |
| 76 } | 76 } |
| 77 | 77 |
| 78 // Read the pref to determine component action status. If not set, set to | 78 // Read the pref to determine component action status. If not set, set to |
| 79 // true if we unloaded an extension. | 79 // true if we unloaded an extension. |
| 80 const base::DictionaryValue* migration_pref = pref_service_->GetDictionary( | 80 const base::DictionaryValue* migration_pref = pref_service_->GetDictionary( |
| 81 ::prefs::kToolbarMigratedComponentActionStatus); | 81 ::prefs::kToolbarMigratedComponentActionStatus); |
| 82 bool has_component_action_pref = migration_pref->HasKey(component_action_id); | |
| 83 bool component_action_pref = false; | 82 bool component_action_pref = false; |
| 84 if (has_component_action_pref) { | 83 if (migration_pref->HasKey(component_action_id)) { |
| 85 bool success = | 84 component_action_pref = GetComponentActionPref(component_action_id); |
| 86 migration_pref->GetBoolean(component_action_id, &component_action_pref); | 85 } else if (extension_was_installed) { |
| 87 DCHECK(success); // Shouldn't fail, but can in case of pref corruption. | |
| 88 } | |
| 89 | |
| 90 if (!has_component_action_pref && extension_was_installed) { | |
| 91 SetComponentActionPref(component_action_id, true); | 86 SetComponentActionPref(component_action_id, true); |
| 92 component_action_pref = true; | 87 component_action_pref = true; |
| 93 } | 88 } |
| 94 | 89 |
| 95 if (component_action_pref && | 90 if (component_action_pref && |
| 96 !delegate_->HasComponentAction(component_action_id)) { | 91 !delegate_->HasComponentAction(component_action_id)) { |
| 97 delegate_->AddComponentAction(component_action_id); | 92 delegate_->AddComponentAction(component_action_id); |
| 98 } | 93 } |
| 99 } | 94 } |
| 100 | 95 |
| (...skipping 31 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 132 return; | 127 return; |
| 133 if (base::ContainsKey(enabled_actions_, component_action_id)) { | 128 if (base::ContainsKey(enabled_actions_, component_action_id)) { |
| 134 UnloadExtension(extension_id); | 129 UnloadExtension(extension_id); |
| 135 SetComponentActionPref(component_action_id, true); | 130 SetComponentActionPref(component_action_id, true); |
| 136 | 131 |
| 137 if (!delegate_->HasComponentAction(component_action_id)) | 132 if (!delegate_->HasComponentAction(component_action_id)) |
| 138 delegate_->AddComponentAction(component_action_id); | 133 delegate_->AddComponentAction(component_action_id); |
| 139 } | 134 } |
| 140 } | 135 } |
| 141 | 136 |
| 137 bool ComponentMigrationHelper::GetComponentActionPref( |
| 138 const std::string& component_action_id) const { |
| 139 const base::DictionaryValue* migration_pref = pref_service_->GetDictionary( |
| 140 ::prefs::kToolbarMigratedComponentActionStatus); |
| 141 bool component_action_pref = false; |
| 142 |
| 143 // If the entry for |component_action_id| doesn't exist, GetBoolean() does not |
| 144 // modify |component_action_pref|. |
| 145 migration_pref->GetBoolean(component_action_id, &component_action_pref); |
| 146 return component_action_pref; |
| 147 } |
| 148 |
| 142 void ComponentMigrationHelper::SetComponentActionPref( | 149 void ComponentMigrationHelper::SetComponentActionPref( |
| 143 const std::string& component_action_id, | 150 const std::string& component_action_id, |
| 144 bool enabled) { | 151 bool enabled) { |
| 145 DictionaryPrefUpdate update(pref_service_, | 152 DictionaryPrefUpdate update(pref_service_, |
| 146 ::prefs::kToolbarMigratedComponentActionStatus); | 153 ::prefs::kToolbarMigratedComponentActionStatus); |
| 147 update->SetBoolean(component_action_id, enabled); | 154 update->SetBoolean(component_action_id, enabled); |
| 148 } | 155 } |
| 149 | 156 |
| 150 bool ComponentMigrationHelper::IsExtensionInstalledAndEnabled( | 157 bool ComponentMigrationHelper::IsExtensionInstalledAndEnabled( |
| 151 const ExtensionId& extension_id) const { | 158 const ExtensionId& extension_id) const { |
| (...skipping 27 matching lines...) Expand all Loading... |
| 179 std::string ComponentMigrationHelper::GetActionIdForExtensionId( | 186 std::string ComponentMigrationHelper::GetActionIdForExtensionId( |
| 180 const ExtensionId& extension_id) const { | 187 const ExtensionId& extension_id) const { |
| 181 for (const auto& i : migrated_actions_) { | 188 for (const auto& i : migrated_actions_) { |
| 182 if (i.second == extension_id) | 189 if (i.second == extension_id) |
| 183 return i.first; | 190 return i.first; |
| 184 } | 191 } |
| 185 return ""; | 192 return ""; |
| 186 } | 193 } |
| 187 | 194 |
| 188 } // namespace extensions | 195 } // namespace extensions |
| OLD | NEW |