Add ContentSuggestion for ReadingList

ios/chrome/browser/ui/content_suggestions/content_suggestions_collection_updater.mm

 // 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.
 
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_collection_updater.h"
 
 #include "base/logging.h"
 #include "base/mac/foundation_util.h"
+#include "base/strings/sys_string_conversions.h"
 #include "base/time/time.h"
 #include "components/strings/grit/components_strings.h"
 #import "ios/chrome/browser/ui/collection_view/cells/collection_view_text_item.h"
 #import "ios/chrome/browser/ui/collection_view/collection_view_controller.h"
 #import "ios/chrome/browser/ui/collection_view/collection_view_model.h"
 #import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_article_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_button_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_expandable_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_favicon_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_footer_item.h"
+#import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_reading_list_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_stack_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_text_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestion.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_data_sink.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_data_source.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_image_fetcher.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_view_controller.h"
 #import "ios/chrome/browser/ui/content_suggestions/identifier/content_suggestion_identifier.h"
 #import "ios/chrome/browser/ui/content_suggestions/identifier/content_suggestions_section_information.h"
 #include "ui/base/l10n/l10n_util.h"
@@ skipping 13 matching lines @@
 // Enum defining the ItemType of this ContentSuggestionsCollectionUpdater.
 typedef NS_ENUM(NSInteger, ItemType) {
   ItemTypeText = kItemTypeEnumZero,
   ItemTypeArticle,
   ItemTypeExpand,
   ItemTypeStack,
   ItemTypeFavicon,
   ItemTypeFooter,
   ItemTypeHeader,
   ItemTypeEmpty,
+  ItemTypeReadingList,
 };
 
 typedef NS_ENUM(NSInteger, SectionIdentifier) {
   SectionIdentifierBookmarks = kSectionIdentifierEnumZero,
   SectionIdentifierArticles,
+  SectionIdentifierReadingList,
   SectionIdentifierDefault,
 };
 
+// Update ContentSuggestionTypeForItemType if you update this function.
 ItemType ItemTypeForContentSuggestionType(ContentSuggestionType type) {
   switch (type) {
     case ContentSuggestionTypeArticle:
       return ItemTypeArticle;
     case ContentSuggestionTypeEmpty:
       return ItemTypeEmpty;
+    case ContentSuggestionTypeReadingList:
+      return ItemTypeReadingList;
   }
 }
 
 ContentSuggestionType ContentSuggestionTypeForItemType(NSInteger type) {
   if (type == ItemTypeArticle)
     return ContentSuggestionTypeArticle;
   if (type == ItemTypeEmpty)
     return ContentSuggestionTypeEmpty;
+  if (type == ItemTypeReadingList)
+    return ContentSuggestionTypeReadingList;
   // Add new type here
 
   // Default type.
   return ContentSuggestionTypeEmpty;
 }
 
 // Returns the section identifier corresponding to the section |info|.
 SectionIdentifier SectionIdentifierForInfo(
     ContentSuggestionsSectionInformation* info) {
   switch (info.sectionID) {
     case ContentSuggestionsSectionBookmarks:
       return SectionIdentifierBookmarks;
 
     case ContentSuggestionsSectionArticles:
       return SectionIdentifierArticles;
 
+    case ContentSuggestionsSectionReadingList:
+      return SectionIdentifierReadingList;
+
     case ContentSuggestionsSectionUnknown:
       return SectionIdentifierDefault;
   }
 }
 
 }  // namespace
 
 @interface ContentSuggestionsCollectionUpdater ()<
     ContentSuggestionsArticleItemDelegate,
     ContentSuggestionsDataSink>
@@ skipping 150 matching lines @@
         articleItem.publisher = suggestion.publisher;
         articleItem.publishDate = suggestion.publishDate;
 
         articleItem.suggestionIdentifier = suggestion.suggestionIdentifier;
 
         NSIndexPath* addedIndexPath = [self addItem:articleItem
                             toSectionWithIdentifier:sectionIdentifier];
         [indexPaths addObject:addedIndexPath];
         break;
       }
+      case ContentSuggestionTypeReadingList: {
+        ContentSuggestionsReadingListItem* readingListItem =
+            [[ContentSuggestionsReadingListItem alloc]
+                     initWithType:ItemTypeReadingList
+                              url:suggestion.url
+                distillationState:ReadingListEntry::PROCESSING];
+        readingListItem.title = suggestion.title;
+        readingListItem.subtitle = suggestion.publisher;
+
+        readingListItem.suggestionIdentifier = suggestion.suggestionIdentifier;
+
+        NSIndexPath* addedIndexPath = [self addItem:readingListItem
+                            toSectionWithIdentifier:sectionIdentifier];
+        [indexPaths addObject:addedIndexPath];
+        break;
+      }
     }
   }
 
   return indexPaths;
 }
 
 - (NSIndexSet*)addSectionsForSuggestionsToModel:
     (NSArray<ContentSuggestion*>*)suggestions {
   NSMutableIndexSet* indexSet = [NSMutableIndexSet indexSet];
 
@@ skipping 158 matching lines @@
   CSCollectionViewModel* model =
       self.collectionViewController.collectionViewModel;
   NSInteger section = [model sectionForSectionIdentifier:sectionIdentifier];
   NSInteger itemNumber = [model numberOfItemsInSection:section];
   [model addItem:item toSectionWithIdentifier:sectionIdentifier];
 
   return [NSIndexPath indexPathForItem:itemNumber inSection:section];
 }
 
 @end