Expose notification info in ContentSuggestion

components/ntp_snippets/remote/ntp_snippet.h

 // Copyright 2015 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_REMOTE_NTP_SNIPPET_H_
 #define COMPONENTS_NTP_SNIPPETS_REMOTE_NTP_SNIPPET_H_
 
 #include <map>
 #include <memory>
 #include <string>
 #include <vector>
 
 #include "base/macros.h"
 #include "base/time/time.h"
+#include "components/ntp_snippets/content_suggestion.h"
 #include "url/gurl.h"
 
 namespace base {
 class DictionaryValue;
 }  // namespace base
 
 namespace ntp_snippets {
 
 // Exposed for tests.
 extern const int kArticlesRemoteId;
@@ skipping 29 matching lines @@
   static std::unique_ptr<NTPSnippet> CreateForTesting(
       const std::string& id,
       int remote_category_id,
       const GURL& url,
       const std::string& publisher_name,
       const GURL& amp_url);
 
   // Creates a protocol buffer corresponding to this snippet, for persisting.
   SnippetProto ToProto() const;
 
+  // Coverts to general content suggestion form
+  ContentSuggestion ToContentSuggestion(Category category) const;
+
   // Returns all ids of the snippet.
   const std::vector<std::string>& GetAllIDs() const { return ids_; }
 
   // The unique, primary ID for identifying the snippet.
   const std::string& id() const { return ids_.front(); }
 
   // Title of the snippet.
   const std::string& title() const { return title_; }
 
   // The main URL pointing to the content web page.
@@ skipping 24 matching lines @@
 
   // If this snippet has all the data we need to show a full card to the user
   bool is_complete() const {
     return !id().empty() && !title().empty() && !snippet().empty() &&
            salient_image_url().is_valid() && !publish_date().is_null() &&
            !expiry_date().is_null() && !publisher_name().empty();
   }
 
   float score() const { return score_; }
 
+  bool should_notify() const { return should_notify_; }
+  base::Time notification_deadline() const { return notification_deadline_; }
+
   bool is_dismissed() const { return is_dismissed_; }
   void set_dismissed(bool dismissed) { is_dismissed_ = dismissed; }
 
   // The ID of the remote category this snippet belongs to, for use with
   // CategoryFactory::FromRemoteCategory.
   int remote_category_id() const { return remote_category_id_; }
 
   // Public for testing.
   static base::Time TimeFromJsonString(const std::string& timestamp_str);
   static std::string TimeToJsonString(const base::Time& time);
@@ skipping 15 matching lines @@
   GURL amp_url_;
 
   GURL salient_image_url_;
   std::string snippet_;
   base::Time publish_date_;
   base::Time expiry_date_;
   float score_;
   bool is_dismissed_;
   int remote_category_id_;
 
+  bool should_notify_;
+  base::Time notification_deadline_;
+
   DISALLOW_COPY_AND_ASSIGN(NTPSnippet);
 };
 
 }  // namespace ntp_snippets
 
 #endif  // COMPONENTS_NTP_SNIPPETS_REMOTE_NTP_SNIPPET_H_