Fetch images for ContentSuggestions

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/time/time.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/content_suggestion.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestion_identifier.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_article_item.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_expandable_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_favicon_item.h"
+#import "ios/chrome/browser/ui/content_suggestions/content_suggestions_image_fetcher.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_section_information.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_stack_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_text_item.h"
 #import "ios/chrome/browser/ui/content_suggestions/content_suggestions_view_controller.h"
+#include "ui/gfx/image/image.h"
 #include "url/gurl.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
 namespace {
 
 // Enum defining the ItemType of this ContentSuggestionsCollectionUpdater.
 typedef NS_ENUM(NSInteger, ItemType) {
@@ skipping 36 matching lines @@
     case ContentSuggestionsSectionArticles:
       return SectionIdentifierArticles;
 
     case ContentSuggestionsSectionUnknown:
       return SectionIdentifierDefault;
   }
 }
 
 }  // namespace
 
-@interface ContentSuggestionsCollectionUpdater ()<ContentSuggestionsDataSink>
+@interface ContentSuggestionsCollectionUpdater ()<
+    ContentSuggestionsArticleItemDelegate,
+    ContentSuggestionsDataSink>
 
 @property(nonatomic, weak) id<ContentSuggestionsDataSource> dataSource;
 @property(nonatomic, strong)
     NSMutableDictionary<NSNumber*, ContentSuggestionsSectionInformation*>*
         sectionInfoBySectionIdentifier;
 
 // Reloads all the data from the data source, deleting all the current items.
 - (void)reloadData;
 // Adds a new section if needed and returns the section identifier.
 - (NSInteger)addSectionIfNeeded:
@@ skipping 43 matching lines @@
   ContentSuggestionsSectionInformation* sectionInformation =
       self.sectionInfoBySectionIdentifier[identifier];
   return sectionInformation.layout == ContentSuggestionsSectionLayoutCustom;
 }
 
 - (ContentSuggestionType)contentSuggestionTypeForItem:
     (CollectionViewItem*)item {
   return ContentSuggestionTypeForItemType(item.type);
 }
 
+#pragma mark - ContentSuggestionsArticleItemDelegate
+
+- (void)loadImageForArticleItem:(ContentSuggestionsArticleItem*)articleItem {
+  NSInteger sectionIdentifier =
+      SectionIdentifierForInfo(articleItem.suggestionIdentifier.sectionInfo);
+
+  __weak ContentSuggestionsCollectionUpdater* weakSelf = self;
+  __weak ContentSuggestionsArticleItem* weakArticle = articleItem;
+  void (^imageFetchedCallback)(const gfx::Image&) = ^(const gfx::Image& image) {
+    ContentSuggestionsCollectionUpdater* strongSelf = weakSelf;
+    ContentSuggestionsArticleItem* strongArticle = weakArticle;
+    if (!strongSelf || !strongArticle) {
+      return;
+    }
+
+    strongArticle.imageBeingFetched = NO;
+    strongArticle.image = image.CopyUIImage();
+    [strongSelf.collectionViewController
+        reconfigureCellsForItems:@[ strongArticle ]
+         inSectionWithIdentifier:sectionIdentifier];
+  };
+
+  articleItem.imageBeingFetched = YES;
+
+  [self.dataSource.imageFetcher
+      fetchImageForSuggestion:articleItem.suggestionIdentifier
+                     callback:imageFetchedCallback];
+}
+
 #pragma mark - Private methods
 
 - (void)reloadData {
   [self resetModels];
   CollectionViewModel* model =
       self.collectionViewController.collectionViewModel;
 
   NSArray<ContentSuggestion*>* suggestions = [self.dataSource allSuggestions];
 
   for (ContentSuggestion* suggestion in suggestions) {
     NSInteger sectionIdentifier =
         [self addSectionIfNeeded:suggestion.suggestionIdentifier.sectionInfo];
     ContentSuggestionsArticleItem* articleItem =
         [[ContentSuggestionsArticleItem alloc]
             initWithType:ItemTypeForContentSuggestionType(suggestion.type)
                    title:suggestion.title
                 subtitle:suggestion.text
-                   image:suggestion.image
+                delegate:self
                      url:suggestion.url];
 
     articleItem.publisher = suggestion.publisher;
     articleItem.publishDate = suggestion.publishDate;
 
     articleItem.suggestionIdentifier = suggestion.suggestionIdentifier;
 
     [model addItem:articleItem toSectionWithIdentifier:sectionIdentifier];
   }
 
@@ skipping 17 matching lines @@
   }
   return sectionIdentifier;
 }
 
 - (void)resetModels {
   [self.collectionViewController loadModel];
   self.sectionInfoBySectionIdentifier = [[NSMutableDictionary alloc] init];
 }
 
 @end