| OLD | NEW |
| 1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 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 "components/offline_pages/core/prefetch/suggested_articles_observer.h" | 5 #include "components/offline_pages/core/prefetch/suggested_articles_observer.h" |
| 6 | 6 |
| 7 #include <unordered_set> | 7 #include <unordered_set> |
| 8 | 8 |
| 9 #include "base/memory/ptr_util.h" | 9 #include "base/memory/ptr_util.h" |
| 10 #include "components/ntp_snippets/category.h" | 10 #include "components/ntp_snippets/category.h" |
| 11 #include "components/ntp_snippets/category_status.h" | 11 #include "components/ntp_snippets/category_status.h" |
| 12 #include "components/offline_pages/core/client_namespace_constants.h" | 12 #include "components/offline_pages/core/client_namespace_constants.h" |
| 13 #include "components/offline_pages/core/offline_page_feature.h" | 13 #include "components/offline_pages/core/offline_page_feature.h" |
| 14 #include "components/offline_pages/core/offline_page_item.h" | 14 #include "components/offline_pages/core/offline_page_item.h" |
| 15 #include "components/offline_pages/core/prefetch/prefetch_dispatcher.h" | 15 #include "components/offline_pages/core/prefetch/prefetch_dispatcher.h" |
| 16 #include "components/offline_pages/core/prefetch/prefetch_service_impl.h" | 16 #include "components/offline_pages/core/prefetch/prefetch_service_impl.h" |
| 17 #include "components/offline_pages/core/prefetch/prefetch_types.h" |
| 17 | 18 |
| 18 using ntp_snippets::Category; | 19 using ntp_snippets::Category; |
| 19 using ntp_snippets::ContentSuggestion; | 20 using ntp_snippets::ContentSuggestion; |
| 20 | 21 |
| 21 namespace offline_pages { | 22 namespace offline_pages { |
| 22 | 23 |
| 23 namespace { | 24 namespace { |
| 24 | 25 |
| 25 const ntp_snippets::Category& ArticlesCategory() { | 26 const ntp_snippets::Category& ArticlesCategory() { |
| 26 static ntp_snippets::Category articles = | 27 static ntp_snippets::Category articles = |
| (...skipping 27 matching lines...) Expand all Loading... |
| 54 return; | 55 return; |
| 55 } | 56 } |
| 56 | 57 |
| 57 const std::vector<ContentSuggestion>& suggestions = | 58 const std::vector<ContentSuggestion>& suggestions = |
| 58 test_articles_ ? *test_articles_ | 59 test_articles_ ? *test_articles_ |
| 59 : content_suggestions_service_->GetSuggestionsForCategory( | 60 : content_suggestions_service_->GetSuggestionsForCategory( |
| 60 ArticlesCategory()); | 61 ArticlesCategory()); |
| 61 if (suggestions.empty()) | 62 if (suggestions.empty()) |
| 62 return; | 63 return; |
| 63 | 64 |
| 64 std::vector<PrefetchDispatcher::PrefetchURL> prefetch_urls; | 65 std::vector<PrefetchURL> prefetch_urls; |
| 65 for (const ContentSuggestion& suggestion : suggestions) { | 66 for (const ContentSuggestion& suggestion : suggestions) { |
| 66 prefetch_urls.push_back( | 67 prefetch_urls.push_back( |
| 67 {CreateClientIDFromSuggestionId(suggestion.id()), suggestion.url()}); | 68 {CreateClientIDFromSuggestionId(suggestion.id()), suggestion.url()}); |
| 68 } | 69 } |
| 69 | 70 |
| 70 prefetch_service_->GetDispatcher()->AddCandidatePrefetchURLs(prefetch_urls); | 71 prefetch_service_->GetPrefetchDispatcher()->AddCandidatePrefetchURLs( |
| 72 prefetch_urls); |
| 71 } | 73 } |
| 72 | 74 |
| 73 void SuggestedArticlesObserver::OnCategoryStatusChanged( | 75 void SuggestedArticlesObserver::OnCategoryStatusChanged( |
| 74 Category category, | 76 Category category, |
| 75 ntp_snippets::CategoryStatus new_status) { | 77 ntp_snippets::CategoryStatus new_status) { |
| 76 if (category != ArticlesCategory() || category_status_ == new_status) | 78 if (category != ArticlesCategory() || category_status_ == new_status) |
| 77 return; | 79 return; |
| 78 | 80 |
| 79 category_status_ = new_status; | 81 category_status_ = new_status; |
| 80 | 82 |
| 81 if (category_status_ == | 83 if (category_status_ == |
| 82 ntp_snippets::CategoryStatus::CATEGORY_EXPLICITLY_DISABLED || | 84 ntp_snippets::CategoryStatus::CATEGORY_EXPLICITLY_DISABLED || |
| 83 category_status_ == | 85 category_status_ == |
| 84 ntp_snippets::CategoryStatus::ALL_SUGGESTIONS_EXPLICITLY_DISABLED) { | 86 ntp_snippets::CategoryStatus::ALL_SUGGESTIONS_EXPLICITLY_DISABLED) { |
| 85 prefetch_service_->GetDispatcher()->RemoveAllUnprocessedPrefetchURLs( | 87 prefetch_service_->GetPrefetchDispatcher() |
| 86 kSuggestedArticlesNamespace); | 88 ->RemoveAllUnprocessedPrefetchURLs(kSuggestedArticlesNamespace); |
| 87 } | 89 } |
| 88 } | 90 } |
| 89 | 91 |
| 90 void SuggestedArticlesObserver::OnSuggestionInvalidated( | 92 void SuggestedArticlesObserver::OnSuggestionInvalidated( |
| 91 const ContentSuggestion::ID& suggestion_id) { | 93 const ContentSuggestion::ID& suggestion_id) { |
| 92 prefetch_service_->GetDispatcher()->RemovePrefetchURLsByClientId( | 94 prefetch_service_->GetPrefetchDispatcher()->RemovePrefetchURLsByClientId( |
| 93 CreateClientIDFromSuggestionId(suggestion_id)); | 95 CreateClientIDFromSuggestionId(suggestion_id)); |
| 94 } | 96 } |
| 95 | 97 |
| 96 void SuggestedArticlesObserver::OnFullRefreshRequired() { | 98 void SuggestedArticlesObserver::OnFullRefreshRequired() { |
| 97 prefetch_service_->GetDispatcher()->RemoveAllUnprocessedPrefetchURLs( | 99 prefetch_service_->GetPrefetchDispatcher()->RemoveAllUnprocessedPrefetchURLs( |
| 98 kSuggestedArticlesNamespace); | 100 kSuggestedArticlesNamespace); |
| 99 OnNewSuggestions(ArticlesCategory()); | 101 OnNewSuggestions(ArticlesCategory()); |
| 100 } | 102 } |
| 101 | 103 |
| 102 void SuggestedArticlesObserver::ContentSuggestionsServiceShutdown() { | 104 void SuggestedArticlesObserver::ContentSuggestionsServiceShutdown() { |
| 103 // No need to do anything here, we will just stop getting events. | 105 // No need to do anything here, we will just stop getting events. |
| 104 } | 106 } |
| 105 | 107 |
| 106 std::vector<ntp_snippets::ContentSuggestion>* | 108 std::vector<ntp_snippets::ContentSuggestion>* |
| 107 SuggestedArticlesObserver::GetTestingArticles() { | 109 SuggestedArticlesObserver::GetTestingArticles() { |
| 108 if (!test_articles_) { | 110 if (!test_articles_) { |
| 109 test_articles_ = | 111 test_articles_ = |
| 110 base::MakeUnique<std::vector<ntp_snippets::ContentSuggestion>>(); | 112 base::MakeUnique<std::vector<ntp_snippets::ContentSuggestion>>(); |
| 111 } | 113 } |
| 112 return test_articles_.get(); | 114 return test_articles_.get(); |
| 113 } | 115 } |
| 114 | 116 |
| 115 } // namespace offline_pages | 117 } // namespace offline_pages |
| OLD | NEW |