[New CL] Add a tab helper to record the last visit date for each bookmark

components/ntp_snippets/bookmarks/bookmark_last_visit_utils.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/bookmarks/bookmark_last_visit_utils.h"
+
+#include <algorithm>
+#include <string>
+#include <utility>
+
+#include "base/strings/string_number_conversions.h"
+
+using bookmarks::BookmarkModel;
+using bookmarks::BookmarkNode;
+
+namespace ntp_snippets {
+
+namespace {
+
+const char kBookmarkLastVisitDateKey[] = "last_visited";
+
+base::Time ParseLastVisitDate(const std::string& date_string) {
+  int64_t date = 0;
+  if (!base::StringToInt64(date_string, &date))
+    return base::Time::UnixEpoch();
+  return base::Time::FromInternalValue(date);
+}
+
+std::string FormatLastVisitDate(const base::Time& date) {
+  return base::Int64ToString(date.ToInternalValue());
+}
+
+bool CompareBookmarksByLastVisitDate(const BookmarkNode* a,
+                                     const BookmarkNode* b) {
+  return GetLastVisitDateForBookmark(a) > GetLastVisitDateForBookmark(b);
+}
+
+}  // namespace
+
+void UpdateBookmarkOnURLVisitedInMainFrame(BookmarkModel* bookmark_model,
+                                           const GURL& url) {
+  std::vector<const BookmarkNode*> bookmarks_for_url;
+  bookmark_model->GetNodesByURL(url, &bookmarks_for_url);
+  if (bookmarks_for_url.empty())
+    return;
+
+  // If there are bookmark for |url|, set their last visit date to now.

[Marc Treib, 2016/08/02 13:37:11]

nit: bookmarks, plural

[Philipp Keck, 2016/08/02 14:06:51]

Done.

+  base::Time now = base::Time::Now();
+  for (auto* node : bookmarks_for_url) {
+    bookmark_model->SetNodeMetaInfo(node, kBookmarkLastVisitDateKey,
+                                    FormatLastVisitDate(now));

[Marc Treib, 2016/08/02 13:37:11]

nit: You could pull this out of the loop

[Philipp Keck, 2016/08/02 14:06:51]

Done.

+  }
+}
+
+base::Time GetLastVisitDateForBookmark(const BookmarkNode* node) {
+  if (!node)
+    return base::Time::UnixEpoch();
+
+  std::string last_visit_date_string;
+  node->GetMetaInfo(kBookmarkLastVisitDateKey, &last_visit_date_string);
+
+  if (last_visit_date_string.empty()) {
+    // Use creation date if no last visit info present.
+    return node->date_added();
+  }
+
+  return ParseLastVisitDate(last_visit_date_string);
+}
+
+std::vector<const BookmarkNode*> GetRecentlyVisitedBookmarks(
+    BookmarkModel* bookmark_model,
+    int max_count,
+    const base::Time& min_visit_time) {
+  // Get all the bookmarks.
+  std::vector<BookmarkModel::URLAndTitle> bookmarks;
+  bookmark_model->GetBookmarks(&bookmarks);
+  // Remove bookmarks that have not been visited after |min_time|.

[Marc Treib, 2016/08/02 13:37:11]

nit: The empty line should be before this

[Philipp Keck, 2016/08/02 14:06:51]

Done.

+
+  bookmarks.erase(
+      std::remove_if(
+          bookmarks.begin(), bookmarks.end(),
+          [&bookmark_model,
+           &min_visit_time](const BookmarkModel::URLAndTitle& bookmark) {
+            // Get all bookmarks for the given URL.
+            std::vector<const BookmarkNode*> bookmarks_for_url;
+            bookmark_model->GetNodesByURL(bookmark.url, &bookmarks_for_url);
+            // Find the most recent visit time in the set.
+            base::Time max_visit_time = min_visit_time;
+            for (const auto* node : bookmarks_for_url) {
+              max_visit_time =
+                  std::max(max_visit_time, GetLastVisitDateForBookmark(node));

[Marc Treib, 2016/08/02 13:37:11]

You could just do
if (GetLastVisitDateForBookmark(node) > min_visit_time) return false;
and "return true" after the loop.

[Philipp Keck, 2016/08/02 14:06:51]

Done.

+            }
+            return max_visit_time <= min_visit_time;
+          }),
+      bookmarks.end());
+  // Sort the remaining bookmarks by date.
+  std::sort(bookmarks.begin(), bookmarks.end(),
+            [&bookmark_model](const BookmarkModel::URLAndTitle& a,
+                              const BookmarkModel::URLAndTitle& b) {
+              return CompareBookmarksByLastVisitDate(
+                  bookmark_model->GetMostRecentlyAddedUserNodeForURL(a.url),
+                  bookmark_model->GetMostRecentlyAddedUserNodeForURL(b.url));
+            });
+
+  std::vector<const BookmarkNode*> result;
+  // Insert the first |max_count| items from |bookmarks| into |result|.
+  for (const BookmarkModel::URLAndTitle& bookmark : bookmarks) {
+    result.push_back(
+        bookmark_model->GetMostRecentlyAddedUserNodeForURL(bookmark.url));
+    if (result.size() >= static_cast<size_t>(max_count))
+      break;
+  }
+  return result;
+}
+
+}  // namespace ntp_snippets