From c326356accb9de24f359b0b770fb92a057a88194 Mon Sep 17 00:00:00 2001
From: Austin Huang <im@austinhuang.me>
Date: Tue, 29 Jun 2021 11:07:38 -0400
Subject: [PATCH] some story touch-ups

---
 .../java/awais/instagrabber/repositories/StoriesService.kt   | 4 ++--
 .../java/awais/instagrabber/webservices/StoriesRepository.kt | 5 +++--
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/app/src/main/java/awais/instagrabber/repositories/StoriesService.kt b/app/src/main/java/awais/instagrabber/repositories/StoriesService.kt
index f9190548..4d217fef 100644
--- a/app/src/main/java/awais/instagrabber/repositories/StoriesService.kt
+++ b/app/src/main/java/awais/instagrabber/repositories/StoriesService.kt
@@ -11,10 +11,10 @@ interface StoriesService {
     suspend fun fetch(@Path("mediaId") mediaId: Long): String
 
     @GET("/api/v1/feed/reels_tray/")
-    suspend fun getFeedStories(): ReelsTrayResponse
+    suspend fun getFeedStories(): ReelsTrayResponse?
 
     @GET("/api/v1/highlights/{uid}/highlights_tray/")
-    suspend fun fetchHighlights(@Path("uid") uid: Long): ReelsTrayResponse
+    suspend fun fetchHighlights(@Path("uid") uid: Long): ReelsTrayResponse?
 
     @GET("/api/v1/archive/reel/day_shells/")
     suspend fun fetchArchive(@QueryMap queryParams: Map<String, String>): ArchiveResponse?
diff --git a/app/src/main/java/awais/instagrabber/webservices/StoriesRepository.kt b/app/src/main/java/awais/instagrabber/webservices/StoriesRepository.kt
index bc5b18ff..6d82d70f 100644
--- a/app/src/main/java/awais/instagrabber/webservices/StoriesRepository.kt
+++ b/app/src/main/java/awais/instagrabber/webservices/StoriesRepository.kt
@@ -26,8 +26,8 @@ open class StoriesRepository(private val service: StoriesService) {
 
     suspend fun getFeedStories(): List<Story> {
         val response = service.getFeedStories()
-        val result = response.tray?.toMutableList() ?: mutableListOf()
-        if (response.broadcasts != null) {
+        val result: MutableList<Story> = mutableListOf()
+        if (response?.broadcasts != null) {
             val length = response.broadcasts.size
             for (i in 0 until length) {
                 val broadcast = response.broadcasts.get(i)
@@ -49,6 +49,7 @@ open class StoriesRepository(private val service: StoriesService) {
                 )
             }
         }
+        if (response?.tray != null) result.addAll(response.tray)
         return sort(result.toList())
     }