Index: components/ntp_snippets/content_suggestions_service.cc |
diff --git a/components/ntp_snippets/content_suggestions_service.cc b/components/ntp_snippets/content_suggestions_service.cc |
index f52a9809d09dcae42dbad4cd9a62ff9598870423..b689990b54198c6ae380ac04e6fbb0b1f707fa39 100644 |
--- a/components/ntp_snippets/content_suggestions_service.cc |
+++ b/components/ntp_snippets/content_suggestions_service.cc |
@@ -150,11 +150,19 @@ void ContentSuggestionsService::DismissCategory(Category category) { |
if (providers_it == providers_by_category_.end()) |
return; |
+ dismissed_providers_by_category_[providers_it->first] = providers_it->second; |
providers_by_category_.erase(providers_it); |
categories_.erase( |
std::find(categories_.begin(), categories_.end(), category)); |
} |
+void ContentSuggestionsService::RestoreDismissedCategories() { |
+ for (const auto& category_provider_pair : dismissed_providers_by_category_) |
Marc Treib
2016/10/11 07:35:58
Braces if the body doesn't fit on one line.
Michael van Ouwerkerk
2016/10/11 12:04:16
Done.
|
+ RegisterCategoryIfRequired(category_provider_pair.second, |
+ category_provider_pair.first); |
+ dismissed_providers_by_category_.clear(); |
Marc Treib
2016/10/11 07:35:58
Should RegisterCategoryIfRequired remove the entry
Michael van Ouwerkerk
2016/10/11 12:04:16
Done.
|
+} |
+ |
void ContentSuggestionsService::AddObserver(Observer* observer) { |
observers_.AddObserver(observer); |
} |