ContentSuggestionsDataSource returns CollectionViewItem

ios/chrome/browser/ui/content_suggestions/content_suggestions_view_controller.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_view_controller.h"
 
 #include "base/mac/foundation_util.h"
 #import "ios/chrome/browser/ui/collection_view/cells/MDCCollectionViewCell+Chrome.h"
 #import "ios/chrome/browser/ui/collection_view/cells/collection_view_item.h"
 #import "ios/chrome/browser/ui/collection_view/collection_view_model.h"
 #import "ios/chrome/browser/ui/content_suggestions/cells/content_suggestions_item.h"
-#import "ios/chrome/browser/ui/content_suggestions/content_suggestion.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_collection_updater.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_commands.h"
 #include "url/gurl.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
+namespace {
+using CSCollectionViewItem = CollectionViewItem<SuggestedContent>;
+}
+
 @interface ContentSuggestionsViewController ()
 
 @property(nonatomic, strong)
     ContentSuggestionsCollectionUpdater* collectionUpdater;
 
 @end
 
 @implementation ContentSuggestionsViewController
 
 @synthesize suggestionCommandHandler = _suggestionCommandHandler;
@@ skipping 43 matching lines @@
   [self.collectionView performBatchUpdates:^{
     [self.collectionViewModel removeSectionWithIdentifier:sectionIdentifier];
     [self.collectionView deleteSections:[NSIndexSet indexSetWithIndex:section]];
   }
       completion:^(BOOL) {
         // The context menu could be displayed for the deleted entries.
         [self.suggestionCommandHandler dismissContextMenu];
       }];
 }
 
-- (void)addSuggestions:(NSArray<ContentSuggestion*>*)suggestions {
+- (void)addSuggestions:(NSArray<CSCollectionViewItem*>*)suggestions
+         toSectionInfo:(ContentSuggestionsSectionInformation*)sectionInfo {
   if (suggestions.count == 0) {
     return;
   }
 
   [self.collectionView performBatchUpdates:^{
-    NSIndexSet* addedSections =
-        [self.collectionUpdater addSectionsForSuggestionsToModel:suggestions];
+    NSIndexSet* addedSections = [self.collectionUpdater
+        addSectionsForSectionInfoToModel:@[ sectionInfo ]];
     [self.collectionView insertSections:addedSections];
   }
                                 completion:nil];
 
   [self.collectionView performBatchUpdates:^{
     NSArray<NSIndexPath*>* addedItems =
-        [self.collectionUpdater addSuggestionsToModel:suggestions];
+        [self.collectionUpdater addSuggestionsToModel:suggestions
+                                      withSectionInfo:sectionInfo];
     [self.collectionView insertItemsAtIndexPaths:addedItems];
   }
                                 completion:nil];
 }
 
 #pragma mark - UIViewController
 
 - (void)viewDidLoad {
   [super viewDidLoad];
 
@@ skipping 118 matching lines @@
 - (void)addEmptySectionPlaceholderIfNeeded:(NSInteger)section {
   if ([self.collectionViewModel numberOfItemsInSection:section] > 0)
     return;
 
   NSIndexPath* emptyItem =
       [self.collectionUpdater addEmptyItemForSection:section];
   [self.collectionView insertItemsAtIndexPaths:@[ emptyItem ]];
 }
 
 @end