From eb18c073c60cbedfe3f1d2e63fc24c12b826fffa Mon Sep 17 00:00:00 2001
From: Alex Hart <alex@signal.org>
Date: Fri, 15 Jul 2022 15:29:59 -0300
Subject: [PATCH] Set contentIsReady flag if story attachment failed to
 download.

---
 .../stories/viewer/page/StoryViewerPageFragment.kt          | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/stories/viewer/page/StoryViewerPageFragment.kt b/app/src/main/java/org/thoughtcrime/securesms/stories/viewer/page/StoryViewerPageFragment.kt
index 95868c4509..561d309382 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/stories/viewer/page/StoryViewerPageFragment.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/stories/viewer/page/StoryViewerPageFragment.kt
@@ -640,8 +640,8 @@ class StoryViewerPageFragment :
   }
 
   private fun markViewedIfAble() {
-    val post = if (viewModel.hasPost()) viewModel.getPost() else null
-    if (post?.content?.transferState == AttachmentDatabase.TRANSFER_PROGRESS_DONE) {
+    val post = viewModel.getPost() ?: return
+    if (post.content.transferState == AttachmentDatabase.TRANSFER_PROGRESS_DONE) {
       if (isResumed) {
         viewModel.markViewed(post)
       }
@@ -707,7 +707,9 @@ class StoryViewerPageFragment :
       }
       AttachmentDatabase.TRANSFER_PROGRESS_FAILED -> {
         storySlate.moveToState(StorySlateView.State.NOT_FOUND, post.id)
+        sharedViewModel.setContentIsReady()
         viewModel.setIsDisplayingSlate(true)
+        markViewedIfAble()
       }
     }
   }