From a20c2ec63f77ae308d86da4d4af949c5d20815c6 Mon Sep 17 00:00:00 2001
From: Alex Hart <alex@signal.org>
Date: Fri, 8 Jul 2022 14:08:48 -0300
Subject: [PATCH] Fix stories check to account for registration.

---
 .../org/thoughtcrime/securesms/jobs/RefreshOwnProfileJob.java   | 2 ++
 app/src/main/java/org/thoughtcrime/securesms/stories/Stories.kt | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/org/thoughtcrime/securesms/jobs/RefreshOwnProfileJob.java b/app/src/main/java/org/thoughtcrime/securesms/jobs/RefreshOwnProfileJob.java
index 3fdc0bb8ad..bb74dcd373 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/jobs/RefreshOwnProfileJob.java
+++ b/app/src/main/java/org/thoughtcrime/securesms/jobs/RefreshOwnProfileJob.java
@@ -134,6 +134,8 @@ public class RefreshOwnProfileJob extends BaseJob {
     if (profileKeyCredential.isPresent()) {
       setProfileKeyCredential(self, ProfileKeyUtil.getSelfProfileKey(), profileKeyCredential.get());
     }
+
+    StoryOnboardingDownloadJob.Companion.enqueueIfNeeded();
   }
 
   private void setProfileKeyCredential(@NonNull Recipient recipient,
diff --git a/app/src/main/java/org/thoughtcrime/securesms/stories/Stories.kt b/app/src/main/java/org/thoughtcrime/securesms/stories/Stories.kt
index 65391d63a7..b7af67990e 100644
--- a/app/src/main/java/org/thoughtcrime/securesms/stories/Stories.kt
+++ b/app/src/main/java/org/thoughtcrime/securesms/stories/Stories.kt
@@ -54,7 +54,7 @@ object Stories {
 
   @JvmStatic
   fun isFeatureAvailable(): Boolean {
-    return FeatureFlags.stories() && Recipient.self().storiesCapability == Recipient.Capability.SUPPORTED
+    return SignalStore.account().isRegistered && FeatureFlags.stories() && Recipient.self().storiesCapability == Recipient.Capability.SUPPORTED
   }
 
   @JvmStatic