Connect ContentSuggestionsMediator to the ContentService

ios/chrome/browser/content_suggestions/content_suggestions_mediator.mm

 // Copyright 2017 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/content_suggestions/content_suggestions_mediator.h"
 
+#include "base/mac/bind_objc_block.h"
 #include "base/memory/ptr_util.h"
+#include "base/optional.h"
+#include "base/strings/sys_string_conversions.h"
 #include "components/ntp_snippets/category.h"
 #include "components/ntp_snippets/content_suggestion.h"
+#import "ios/chrome/browser/content_suggestions/content_suggestions_category_wrapper.h"
 #import "ios/chrome/browser/content_suggestions/content_suggestions_service_bridge_observer.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_image_updater.h"
+#import "ios/chrome/browser/ui/content_suggestions/content_suggestions_item.h"
+#import "ios/chrome/browser/ui/content_suggestions/content_suggestions_section_information.h"
+#include "ui/gfx/image/image.h"
 
 #if !defined(__has_feature) || !__has_feature(objc_arc)
 #error "This file requires ARC support."
 #endif
 
+namespace {
+
+// Returns the ItemType for this |category|.
+ContentSuggestionsItemType TypeForCategory(ntp_snippets::Category category) {
+  if (category.IsKnownCategory(ntp_snippets::KnownCategories::ARTICLES))
+    return ContentSuggestionsItemTypeArticle;
+  return ContentSuggestionsItemTypeText;
+}
+
+// Returns the section ID for this |category|.
+ContentSuggestionsSectionID SectionIDForCategory(
+    ntp_snippets::Category category) {
+  if (category.IsKnownCategory(ntp_snippets::KnownCategories::BOOKMARKS))
+    return ContentSuggestionsSectionBookmarks;
+  if (category.IsKnownCategory(ntp_snippets::KnownCategories::ARTICLES))
+    return ContentSuggestionsSectionArticles;
+
+  return ContentSuggestionsSectionCount;

[lpromero, 2017/02/14 10:52:27]

If this is used as a section identifier, it probably shouldn't be called count,
but maybe default or other?

[gambard, 2017/02/14 14:03:42]

It is not a section identifier in regard of the CollectionViewModel meaning.
It is a sectionID as in ContentSuggestionsSectionInformation.
If you have a better name, please tell :)

[lpromero, 2017/02/14 15:43:32]

Does it act as a "Unknown" category then?

[gambard, 2017/02/15 10:17:56]

Kind of. The name was "NotImplemented" before.

+}
+
+// Returns the section layout corresponding to the category |layout|.
+ContentSuggestionsSectionLayout SectionLayoutForLayout(
+    ntp_snippets::ContentSuggestionsCardLayout layout) {
+  // For now, only cards are relevant.
+  return ContentSuggestionsSectionLayoutCard;
+}
+
+}  // namespace
+
 @interface ContentSuggestionsMediator ()<ContentSuggestionsServiceObserver> {
   // Bridge for this class to become an observer of a ContentSuggestionsService.
   std::unique_ptr<ContentSuggestionsServiceBridge> _suggestionBridge;
 }
 
 @property(nonatomic, assign)
     ntp_snippets::ContentSuggestionsService* contentService;
+@property(nonatomic, strong) NSMutableDictionary* sectionInformationByCategory;
 
 // Converts the data in |category| to ContentSuggestion and adds them to the
 // |contentArray|.
 - (void)addContentInCategory:(ntp_snippets::Category&)category
                      toArray:(NSMutableArray<ContentSuggestion*>*)contentArray;
 
 // Converts the |contentsuggestion| to a ContentSuggestion.
 - (ContentSuggestion*)convertContentSuggestion:
-    (const ntp_snippets::ContentSuggestion&)contentsuggestion;
+    (const ntp_snippets::ContentSuggestion&)contentSuggestion;
+
+// Adds the section information for |category| in
+// self.sectionInformationByCategory.
+- (void)addSectionInformationForCategory:(ntp_snippets::Category)category;
 
 @end
 
 @implementation ContentSuggestionsMediator
 
 @synthesize contentService = _contentService;
 @synthesize dataSink = _dataSink;
+@synthesize sectionInformationByCategory = _sectionInformationByCategory;
 
 - (instancetype)initWithContentService:
     (ntp_snippets::ContentSuggestionsService*)contentService {
   self = [super init];
   if (self) {
     _suggestionBridge =
         base::MakeUnique<ContentSuggestionsServiceBridge>(self, contentService);
     _contentService = contentService;
+    _sectionInformationByCategory = [[NSMutableDictionary alloc] init];
   }
   return self;
 }
 
 #pragma mark - ContentSuggestionsDataSource
 
 - (NSArray<ContentSuggestion*>*)allSuggestions {
   std::vector<ntp_snippets::Category> categories =
       self.contentService->GetCategories();
   NSMutableArray<ContentSuggestion*>* dataHolders = [NSMutableArray array];
   for (auto& category : categories) {
+    if (self.contentService->GetCategoryStatus(category) !=
+        ntp_snippets::CategoryStatus::AVAILABLE) {
+      continue;
+    }
+    if (![self.sectionInformationByCategory
+            objectForKey:[[ContentSuggestionsCategoryWrapper alloc]
+                             initWithCategory:category]]) {
+      [self addSectionInformationForCategory:category];
+    }
     [self addContentInCategory:category toArray:dataHolders];
   }
   return dataHolders;
 }
 
 #pragma mark - ContentSuggestionsServiceObserver
 
 - (void)contentSuggestionsService:
             (ntp_snippets::ContentSuggestionsService*)suggestionsService
          newSuggestionsInCategory:(ntp_snippets::Category)category {
@@ skipping 23 matching lines @@
     (ntp_snippets::ContentSuggestionsService*)suggestionsService {
   // Update dataSink.
 }
 
 #pragma mark - Private
 
 - (void)addContentInCategory:(ntp_snippets::Category&)category
                      toArray:(NSMutableArray<ContentSuggestion*>*)contentArray {
   const std::vector<ntp_snippets::ContentSuggestion>& suggestions =
       self.contentService->GetSuggestionsForCategory(category);
+  ContentSuggestionsCategoryWrapper* categoryWrapper =
+      [[ContentSuggestionsCategoryWrapper alloc] initWithCategory:category];
   for (auto& contentSuggestion : suggestions) {
-    [contentArray addObject:[self convertContentSuggestion:contentSuggestion]];
+    ContentSuggestion* suggestion =
+        [self convertContentSuggestion:contentSuggestion];
+    suggestion.type = TypeForCategory(category);
+    suggestion.section = self.sectionInformationByCategory[categoryWrapper];
+    [contentArray addObject:suggestion];
   }
 }
 
 - (ContentSuggestion*)convertContentSuggestion:
-    (const ntp_snippets::ContentSuggestion&)contentsuggestion {
-  return [[ContentSuggestion alloc] init];
+    (const ntp_snippets::ContentSuggestion&)contentSuggestion {
+  ContentSuggestionsImageUpdater* imageUpdater =
+      [[ContentSuggestionsImageUpdater alloc] init];
+
+  __weak ContentSuggestionsImageUpdater* weakImageUpdater = imageUpdater;
+  self.contentService->FetchSuggestionImage(
+      contentSuggestion.id(), base::BindBlockArc(^(const gfx::Image& image) {
+        [weakImageUpdater receivedImage:image.CopyUIImage()];
+      }));
+
+  ContentSuggestion* suggestion = [[ContentSuggestion alloc] init];
+  suggestion.title = base::SysUTF16ToNSString(contentSuggestion.title());
+  suggestion.text = base::SysUTF16ToNSString(contentSuggestion.snippet_text());
+  suggestion.url = contentSuggestion.url();
+  suggestion.imageUpdater = imageUpdater;
+
+  return suggestion;
+}
+
+- (void)addSectionInformationForCategory:(ntp_snippets::Category)category {
+  base::Optional<ntp_snippets::CategoryInfo> categoryInfo =
+      self.contentService->GetCategoryInfo(category);
+  ContentSuggestionsSectionInformation* sectionInfo =
+      [[ContentSuggestionsSectionInformation alloc]
+          initWithID:SectionIDForCategory(category)];
+  if (categoryInfo) {
+    sectionInfo.layout = SectionLayoutForLayout(categoryInfo->card_layout());
+    if (categoryInfo->show_if_empty()) {
+      NSString* title =
+          base::SysUTF16ToNSString(categoryInfo->no_suggestions_message());
+      sectionInfo.emptyCell = [[ContentSuggestionsItem alloc] initWithType:0

[lpromero, 2017/02/14 10:52:27]

item type 0 should trigger a breakpoint in the model. Can you check?

[gambard, 2017/02/14 14:03:42]

A breakpoint or a DCHECK?
I don't see when it should trigger one.

[lpromero, 2017/02/14 15:43:32]

Sorry, yes, a DCHECK. 
https://cs.chromium.org/chromium/src/ios/chrome/browser/ui/collection_view/co...

[gambard, 2017/02/15 10:17:56]

Yes. I am removing it and adding a TODO for now.

+                                                                     title:title
+                                                                  subtitle:nil];
+    }
+    sectionInfo.title = base::SysUTF16ToNSString(categoryInfo->title());
+  }
+
+  [self.sectionInformationByCategory
+      setObject:sectionInfo
+         forKey:[[ContentSuggestionsCategoryWrapper alloc]
+                    initWithCategory:category]];
 }
 
 @end