Index: components/search_engines/template_url_service.cc |
diff --git a/components/search_engines/template_url_service.cc b/components/search_engines/template_url_service.cc |
index 123a6d05c88a733f1742792042fbe298b208b5d9..2c88e36da810e758229ec0e00f72146f1668bd3e 100644 |
--- a/components/search_engines/template_url_service.cc |
+++ b/components/search_engines/template_url_service.cc |
@@ -484,18 +484,23 @@ |
TemplateURL* TemplateURLService::AddExtensionControlledTURL( |
std::unique_ptr<TemplateURL> template_url, |
std::unique_ptr<TemplateURL::AssociatedExtensionInfo> info) { |
+ DCHECK(loaded_); |
DCHECK(template_url); |
DCHECK_EQ(kInvalidTemplateURLID, template_url->id()); |
DCHECK(info); |
DCHECK_NE(TemplateURL::NORMAL, template_url->type()); |
DCHECK( |
!FindTemplateURLForExtension(info->extension_id, template_url->type())); |
- |
template_url->extension_info_.swap(info); |
+ KeywordWebDataService::BatchModeScoper scoper(web_data_service_.get()); |
TemplateURL* template_url_ptr = AddNoNotify(std::move(template_url), true); |
- if (template_url_ptr) |
+ if (template_url_ptr) { |
+ if (template_url_ptr->extension_info_->wants_to_be_default_engine) { |
+ UpdateExtensionDefaultSearchEngine(); |
+ } |
NotifyObservers(); |
+ } |
return template_url_ptr; |
} |
@@ -513,9 +518,12 @@ |
if (!url) |
return; |
// NULL this out so that we can call RemoveNoNotify. |
+ // UpdateExtensionDefaultSearchEngine will cause it to be reset. |
if (default_search_provider_ == url) |
default_search_provider_ = nullptr; |
+ KeywordWebDataService::BatchModeScoper scoper(web_data_service_.get()); |
RemoveNoNotify(url); |
+ UpdateExtensionDefaultSearchEngine(); |
NotifyObservers(); |
} |
@@ -1971,7 +1979,6 @@ |
} else if (source == DefaultSearchManager::FROM_EXTENSION) { |
default_search_provider_ = FindMatchingExtensionTemplateURL( |
*data, TemplateURL::NORMAL_CONTROLLED_BY_EXTENSION); |
- DCHECK(default_search_provider_); |
} else if (source == DefaultSearchManager::FROM_FALLBACK) { |
default_search_provider_ = |
FindPrepopulatedTemplateURL(data->prepopulate_id); |
@@ -2492,3 +2499,25 @@ |
} |
return nullptr; |
} |
+ |
+void TemplateURLService::UpdateExtensionDefaultSearchEngine() { |
+ TemplateURL* most_recently_intalled_default = nullptr; |
+ for (const auto& turl : template_urls_) { |
+ if ((turl->type() == TemplateURL::NORMAL_CONTROLLED_BY_EXTENSION) && |
+ turl->extension_info_->wants_to_be_default_engine && |
+ turl->SupportsReplacement(search_terms_data()) && |
+ (!most_recently_intalled_default || |
+ (most_recently_intalled_default->extension_info_->install_time < |
+ turl->extension_info_->install_time))) |
+ most_recently_intalled_default = turl.get(); |
+ } |
+ |
+ if (most_recently_intalled_default) { |
+ base::AutoReset<DefaultSearchChangeOrigin> change_origin( |
+ &dsp_change_origin_, DSP_CHANGE_OVERRIDE_SETTINGS_EXTENSION); |
+ default_search_manager_.SetExtensionControlledDefaultSearchEngine( |
+ most_recently_intalled_default->data()); |
+ } else { |
+ default_search_manager_.ClearExtensionControlledDefaultSearchEngine(); |
+ } |
+} |