Add bookmark provider for content suggestions

components/ntp_snippets/bookmarks/bookmark_suggestions_provider.h

+// 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.
+
+#ifndef COMPONENTS_NTP_SNIPPETS_BOOKMARKS_BOOKMARK_SUGGESTIONS_PROVIDER_H_
+#define COMPONENTS_NTP_SNIPPETS_BOOKMARKS_BOOKMARK_SUGGESTIONS_PROVIDER_H_
+
+#include <set>
+#include <string>
+
+#include "components/bookmarks/browser/bookmark_model_observer.h"
+#include "components/keyed_service/core/keyed_service.h"
+#include "components/ntp_snippets/content_suggestions_category.h"
+#include "components/ntp_snippets/content_suggestions_category_status.h"
+#include "components/ntp_snippets/content_suggestions_provider.h"
+
+namespace gfx {
+class Image;
+}
+
+namespace ntp_snippets {
+
+// Provides content suggestions from the bookmarks model.
+class BookmarkSuggestionsProvider : public KeyedService,
+                                    public ContentSuggestionsProvider,
+                                    public bookmarks::BookmarkModelObserver {
+ public:
+  explicit BookmarkSuggestionsProvider(
+      bookmarks::BookmarkModel* bookmark_model);
+  ~BookmarkSuggestionsProvider() override;
+
+  // Inherited from KeyedService.
+  void Shutdown() override;
+
+ private:
+  // ContentSuggestionsProvider implementation.
+  void SetObserver(ContentSuggestionsProvider::Observer* observer) override;
+  ContentSuggestionsCategoryStatus GetCategoryStatus(
+      ContentSuggestionsCategory category) override;
+  void DismissSuggestion(const std::string& suggestion_id) override;
+  void FetchSuggestionImage(const std::string& suggestion_id,
+                            const ImageFetchedCallback& callback) override;
+  void ClearCachedSuggestionsForDebugging() override;
+  void ClearDismissedSuggestionsForDebugging() override;
+
+  // bookmarks::BookmarkModelObserver implementation.
+  void BookmarkModelLoaded(bookmarks::BookmarkModel* model,
+                           bool ids_reassigned) override;
+  void OnWillChangeBookmarkMetaInfo(
+      bookmarks::BookmarkModel* model,
+      const bookmarks::BookmarkNode* node) override;
+  void BookmarkMetaInfoChanged(bookmarks::BookmarkModel* model,
+                               const bookmarks::BookmarkNode* node) override;
+
+  void BookmarkNodeMoved(bookmarks::BookmarkModel* model,
+                         const bookmarks::BookmarkNode* old_parent,
+                         int old_index,
+                         const bookmarks::BookmarkNode* new_parent,
+                         int new_index) override {}
+  void BookmarkNodeAdded(bookmarks::BookmarkModel* model,
+                         const bookmarks::BookmarkNode* parent,
+                         int index) override {}
+  void BookmarkNodeRemoved(
+      bookmarks::BookmarkModel* model,
+      const bookmarks::BookmarkNode* parent,
+      int old_index,
+      const bookmarks::BookmarkNode* node,
+      const std::set<GURL>& no_longer_bookmarked) override {}
+  void BookmarkNodeChanged(bookmarks::BookmarkModel* model,
+                           const bookmarks::BookmarkNode* node) override {}
+  void BookmarkNodeFaviconChanged(
+      bookmarks::BookmarkModel* model,
+      const bookmarks::BookmarkNode* node) override {}
+  void BookmarkNodeChildrenReordered(
+      bookmarks::BookmarkModel* model,
+      const bookmarks::BookmarkNode* node) override {}
+  void BookmarkAllUserNodesRemoved(
+      bookmarks::BookmarkModel* model,
+      const std::set<GURL>& removed_urls) override {}
+
+  // The actual method to fetch bookmarks - follows each call to FetchBookmarks
+  // but not sooner than the BookmarkModel gets loaded.
+  void FetchBookmarksInternal();
+
+  // Queries the BookmarkModel for recently visited bookmarks and pushes the
+  // results to the ContentSuggestionService. The actual fetching does not
+  // happen before the Bookmark model gets loaded.
+  void FetchBookmarks();
+
+  // Updates the |category_status_| and notifies the |observer_|, if necessary.
+  void NotifyStatusChanged(ContentSuggestionsCategoryStatus new_status);
+
+  ContentSuggestionsCategoryStatus category_status_;
+  ContentSuggestionsProvider::Observer* observer_;
+  bookmarks::BookmarkModel* bookmark_model_;
+  bool fetch_requested_;
+
+  base::Time node_to_change_last_visit_date_;
+  base::Time end_of_list_last_visit_date_;
+
+  DISALLOW_COPY_AND_ASSIGN(BookmarkSuggestionsProvider);
+};
+
+}  // namespace ntp_snippets
+
+#endif  // COMPONENTS_NTP_SNIPPETS_BOOKMARKS_BOOKMARK_SUGGESTIONS_PROVIDER_H_