Show content suggestions from all categories on the NTP

components/ntp_snippets/offline_pages/offline_page_suggestions_provider.cc

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/ntp_snippets/offline_pages/offline_page_suggestions_provider.h"
 
 #include "base/bind.h"
 #include "base/strings/string_number_conversions.h"
 
 using offline_pages::MultipleOfflinePageItemResult;
 using offline_pages::OfflinePageModel;
 using offline_pages::OfflinePageItem;
 
 namespace ntp_snippets {
 
+namespace {
+
+const int kMaxSuggestionsCount = 5;
+
+}  // namespace
+
 OfflinePageSuggestionsProvider::OfflinePageSuggestionsProvider(
     OfflinePageModel* offline_page_model)
     : ContentSuggestionsProvider({ContentSuggestionsCategory::OFFLINE_PAGES}),
       category_status_(ContentSuggestionsCategoryStatus::AVAILABLE_LOADING),
       observer_(nullptr),
       offline_page_model_(offline_page_model) {
   offline_page_model_->AddObserver(this);
 }
 
 OfflinePageSuggestionsProvider::~OfflinePageSuggestionsProvider() {}
@@ skipping 64 matching lines @@
 }
 
 void OfflinePageSuggestionsProvider::OnOfflinePagesLoaded(
     const MultipleOfflinePageItemResult& result) {
   NotifyStatusChanged(ContentSuggestionsCategoryStatus::AVAILABLE);
   if (!observer_)
     return;
 
   std::vector<ContentSuggestion> suggestions;
   for (const OfflinePageItem& item : result) {
+    // TODO(pke): Make sure the URL is actually opened as an offline URL.
+    // Currently, the browser opens the offline URL and then immediately
+    // redirects to the online URL if the device is online.
     ContentSuggestion suggestion(
         MakeUniqueID(ContentSuggestionsCategory::OFFLINE_PAGES,
                      base::IntToString(item.offline_id)),
         item.GetOfflineURL());
 
     // TODO(pke): Sort my most recently visited and only keep the top one of
     // multiple entries for the same URL.
     // TODO(pke): Get more reasonable data from the OfflinePageModel here.
     suggestion.set_title(item.url.spec());
     suggestion.set_snippet_text(std::string());
     suggestion.set_publish_date(item.creation_time);
     suggestion.set_publisher_name(item.url.host());
     suggestions.emplace_back(std::move(suggestion));
+    if (suggestions.size() == kMaxSuggestionsCount)
+      break;
   }
 
   observer_->OnNewSuggestions(ContentSuggestionsCategory::OFFLINE_PAGES,
                               std::move(suggestions));
 }
 
 void OfflinePageSuggestionsProvider::NotifyStatusChanged(
     ContentSuggestionsCategoryStatus new_status) {
   if (category_status_ == new_status)
     return;
   category_status_ = new_status;
 
   if (!observer_)
     return;
   observer_->OnCategoryStatusChanged(ContentSuggestionsCategory::OFFLINE_PAGES,
                                      new_status);
 }
 
 }  // namespace ntp_snippets