Index: components/ntp_snippets/bookmarks/bookmark_suggestions_provider.h |
diff --git a/components/ntp_snippets/bookmarks/bookmark_suggestions_provider.h b/components/ntp_snippets/bookmarks/bookmark_suggestions_provider.h |
new file mode 100644 |
index 0000000000000000000000000000000000000000..667dad671fa69bf25ba784504d8949438326e72b |
--- /dev/null |
+++ b/components/ntp_snippets/bookmarks/bookmark_suggestions_provider.h |
@@ -0,0 +1,102 @@ |
+// 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/category.h" |
+#include "components/ntp_snippets/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 ContentSuggestionsProvider, |
+ public bookmarks::BookmarkModelObserver { |
+ public: |
+ BookmarkSuggestionsProvider(ContentSuggestionsProvider::Observer* observer, |
+ CategoryFactory* category_factory, |
+ bookmarks::BookmarkModel* bookmark_model); |
+ ~BookmarkSuggestionsProvider() override; |
+ |
+ private: |
+ // ContentSuggestionsProvider implementation. |
+ std::vector<Category> GetProvidedCategories() override; |
+ CategoryStatus GetCategoryStatus(Category 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(CategoryStatus new_status); |
+ |
+ CategoryStatus category_status_; |
+ const Category provided_category_; |
+ 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_ |