[TEMP - DO NOT SUBMIT] OfflinePagesProvider

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/stringprintf.h"
+
+using offline_pages::MultipleOfflinePageItemResult;
+using offline_pages::OfflinePageModel;
+using offline_pages::OfflinePageItem;
+
+namespace ntp_snippets {
+
+OfflinePageSuggestionsProvider::OfflinePageSuggestionsProvider(
+    OfflinePageModel* offline_page_model)
+    : ContentSuggestionsProvider(
+          {ContentSuggestionsCategory::OFFLINE_PAGES_CACHED}),
+      category_status_(ContentSuggestionsCategoryStatus::AVAILABLE_LOADING),
+      offline_page_model_(offline_page_model) {
+  offline_page_model_->AddObserver(this);
+  FetchOfflinePages();
+}
+
+OfflinePageSuggestionsProvider::~OfflinePageSuggestionsProvider() {}
+
+// Inherited from KeyedService.
+void OfflinePageSuggestionsProvider::Shutdown() {
+  offline_page_model_->RemoveObserver(this);
+  category_status_ = ContentSuggestionsCategoryStatus::NOT_PROVIDED;
+}
+
+////////////////////////////////////////////////////////////////////////////////
+// Private methods
+
+void OfflinePageSuggestionsProvider::SetObserver(
+    ContentSuggestionsProvider::Observer* observer) {
+  observer_ = observer;
+}
+
+ContentSuggestionsCategoryStatus
+OfflinePageSuggestionsProvider::GetCategoryStatus(
+    ContentSuggestionsCategory category) {
+  // TODO(pke) Is this ok?
+  return ContentSuggestionsCategoryStatus::AVAILABLE;
+}
+
+void OfflinePageSuggestionsProvider::DiscardSuggestion(
+    const std::string& suggestion_id) {
+  // TODO(pke) Implement some "dont show on NTP anymore" behaviour,
+  // then also implement ClearDiscardedSuggestionsForDebugging.
+}
+
+void OfflinePageSuggestionsProvider::FetchSuggestionImage(
+    const std::string& suggestion_id,
+    const ImageFetchedCallback& callback) {
+  // TODO(pke) Implement
+}
+
+void OfflinePageSuggestionsProvider::ClearCachedSuggestionsForDebugging() {
+  // Ignore
+}
+
+void OfflinePageSuggestionsProvider::ClearDiscardedSuggestionsForDebugging() {
+  // Ignore
+}
+
+void OfflinePageSuggestionsProvider::OfflinePageModelLoaded(
+    OfflinePageModel* model) {
+  // Ignore (when offline pages have been loaded, OnOfflinePagesLoaded will be
+  // called as a delayed callback, see ctor).
+}
+
+void OfflinePageSuggestionsProvider::OfflinePageModelChanged(
+    OfflinePageModel* model) {
+  LOG(WARNING) << "Fetching ...";
+  FetchOfflinePages();
+}
+
+void OfflinePageSuggestionsProvider::OfflinePageDeleted(
+    int64_t offline_id,
+    const offline_pages::ClientId& client_id) {
+  // TODO(pke) Implement (does the ofp have to be cleared from UI immediately?)
+}
+
+void OfflinePageSuggestionsProvider::FetchOfflinePages() {
+  offline_page_model_->GetAllPages(
+      base::Bind(&OfflinePageSuggestionsProvider::OnOfflinePagesLoaded,
+                 base::Unretained(this)));
+}
+
+void OfflinePageSuggestionsProvider::OnOfflinePagesLoaded(
+    const MultipleOfflinePageItemResult& result) {
+  if (!observer_) {
+    category_status_ = ContentSuggestionsCategoryStatus::AVAILABLE;
+    return;
+  }
+
+  if (category_status_ != ContentSuggestionsCategoryStatus::AVAILABLE) {
+    category_status_ = ContentSuggestionsCategoryStatus::AVAILABLE;
+    observer_->OnCategoryStatusChanged(
+        ContentSuggestionsCategory::OFFLINE_PAGES_CACHED, category_status_);
+  }
+
+  std::vector<ContentSuggestion> suggestions;
+  for (const OfflinePageItem& item : result) {
+    ContentSuggestion suggestion = ContentSuggestion(
+        MakeUniqueID(ContentSuggestionsCategory::OFFLINE_PAGES_CACHED,
+                     std::to_string(item.offline_id)),
+        item.GetOfflineURL());
+
+    // TODO(pke): Possibly filter duplicates (only the newest per unique URL),
+    // or use better OfflinePageModel::Get method that chooses the "best"
+    // version automatically.
+    // TODO(pke): Get 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));
+  }
+  observer_->OnNewSuggestions(ContentSuggestionsCategory::OFFLINE_PAGES_CACHED,
+                              std::move(suggestions));
+}
+
+}  // namespace ntp_snippets