Updated GraphQL parsing logic

InMobiGraphQLSource/InMobiGraphQLSource/InMobiGraphQLSource.swift

@@ -72,8 +72,8 @@ public class InMobiGraphQLSource: ShortSource {
             decoder.keyDecodingStrategy = .useDefaultKeys
             do {
                 let graphQLResponse = try decoder.decode(GrapQLResponse.self, from: data)
-                let items = graphQLResponse.weatherItems.map{ self.toAppModel(item: $0) }
-                completion(items, nil)
+                let shorts = self.toAppModel(items: graphQLResponse.weatherItems)
+                completion(shorts, nil)
             }
             catch {
                 self.log.error("Parsing error: \(error)")
@@ -83,44 +83,54 @@ public class InMobiGraphQLSource: ShortSource {
         dataTask.resume()
     }
     
-    private func toAppModel(item: WeatherCardItem) -> ShortsItem {
+    private func toAppModel(items: [WeatherCardItem]) -> [ShortsItem] {
         let lowImageSize = CGSize(width: 440, height: 783)
         let highImageSize = CGSize(width: 783, height: 1701)
         
-        var overlayImages = [ShortsItemImage]()
-        var baseImages = [ShortsItemImage]()
-        for mediaObjectCollectionItem in item.mediaArrayObjectCollection.items {
-            let image = mediaObjectCollectionItem.image
-            
-            for overlayImage in image.overlayImage {
-                overlayImages.append(ShortsItemImage(width: Int(lowImageSize.width),
-                                                     height: Int(lowImageSize.height),
-                                                     url: overlayImage.lowresolution))
-                
-                overlayImages.append(ShortsItemImage(width: Int(highImageSize.width),
-                                                     height: Int(highImageSize.height),
-                                                     url: overlayImage.highresolution))
+        var shortItems = [ShortsItem]()
+        for cardItem in items {
+            guard cardItem.mediaArrayObjectCollection.items.count == 1 else {
+                continue
             }
             
-            for baseImage in image.baseImage {
-                baseImages.append(ShortsItemImage(width: Int(lowImageSize.width),
-                                                  height: Int(lowImageSize.height),
-                                                  url: baseImage.lowresolution))
-                baseImages.append(ShortsItemImage(width: Int(highImageSize.width),
-                                                  height: Int(highImageSize.height),
-                                                  url: baseImage.highresolution))
+            for mediaObjectCollectionItem in cardItem.mediaArrayObjectCollection.items {
+                let image = mediaObjectCollectionItem.image
+                var overlayImages = [ShortsItemImage]()
+                var baseImages = [ShortsItemImage]()
+                
+                for overlayImage in image.overlayImage {
+                    overlayImages.append(ShortsItemImage(width: Int(lowImageSize.width),
+                                                         height: Int(lowImageSize.height),
+                                                         url: overlayImage.lowresolution))
+                    
+                    overlayImages.append(ShortsItemImage(width: Int(highImageSize.width),
+                                                         height: Int(highImageSize.height),
+                                                         url: overlayImage.highresolution))
+                }
+                
+                for baseImage in image.baseImage {
+                    baseImages.append(ShortsItemImage(width: Int(lowImageSize.width),
+                                                      height: Int(lowImageSize.height),
+                                                      url: baseImage.lowresolution))
+                    baseImages.append(ShortsItemImage(width: Int(highImageSize.width),
+                                                      height: Int(highImageSize.height),
+                                                      url: baseImage.highresolution))
+                }
+                
+                let shortItem = ShortsItem(id: cardItem.id,
+                                           images: baseImages,
+                                           overlayImages: overlayImages,
+                                           shareURL: cardItem.sourceUrl,
+                                           title: mediaObjectCollectionItem.title,
+                                           summaryText: mediaObjectCollectionItem.summary,
+                                           sourceName: "",
+                                           shortURL: cardItem.sourceUrl,
+                                           ctaText: cardItem.buttonText,
+                                           ctaURL: cardItem.sourceUrl)
+                shortItems.append(shortItem)
             }
         }
         
-        return ShortsItem(id: item.id,
-                          images: baseImages,
-                          overlayImages: overlayImages,
-                          shareURL: item.sourceUrl,
-                          title: item.mediaArrayObjectCollection.items.first?.title ?? "",
-                          summaryText: item.mediaArrayObjectCollection.items.first?.summary ?? "",
-                          sourceName: "",
-                          shortURL: item.sourceUrl,
-                          ctaText: item.buttonText,
-                          ctaURL: item.sourceUrl)
+        return shortItems
     }
 }