From dca848389c2c5910e6d4a4cd54cbae0b0d46fe18 Mon Sep 17 00:00:00 2001
From: Josh Perez <60019601+josh-signal@users.noreply.github.com>
Date: Thu, 4 Aug 2022 21:13:26 -0400
Subject: [PATCH] Sync stories sent to a group

---
 ts/components/MyStories.tsx           |  6 ++--
 ts/state/selectors/stories.ts         | 46 +++++++++++++++++----------
 ts/test-both/helpers/getFakeStory.tsx |  5 ++-
 ts/textsecure/MessageReceiver.ts      | 18 +++++++++++
 ts/types/Stories.ts                   |  4 +--
 5 files changed, 54 insertions(+), 25 deletions(-)

diff --git a/ts/components/MyStories.tsx b/ts/components/MyStories.tsx
index b9675c186d..4931c5adba 100644
--- a/ts/components/MyStories.tsx
+++ b/ts/components/MyStories.tsx
@@ -68,12 +68,12 @@ export const MyStories = ({
       </div>
       <div className="Stories__pane__list">
         {myStories.map(list => (
-          <div className="MyStories__distribution" key={list.distributionId}>
+          <div className="MyStories__distribution" key={list.id}>
             <div className="MyStories__distribution__title">
               <StoryDistributionListName
                 i18n={i18n}
-                id={list.distributionId}
-                name={list.distributionName}
+                id={list.id}
+                name={list.name}
               />
             </div>
             {list.stories.map(story => (
diff --git a/ts/state/selectors/stories.ts b/ts/state/selectors/stories.ts
index ed7a01c5ee..48192b4acd 100644
--- a/ts/state/selectors/stories.ts
+++ b/ts/state/selectors/stories.ts
@@ -283,36 +283,48 @@ export const getStories = createSelector(
         return;
       }
 
-      if (story.sendStateByConversationId && story.storyDistributionListId) {
-        const list =
-          story.storyDistributionListId === MY_STORIES_ID
-            ? { id: MY_STORIES_ID, name: MY_STORIES_ID }
-            : distributionListSelector(
-                story.storyDistributionListId.toLowerCase()
-              );
+      const conversationStory = getConversationStory(
+        conversationSelector,
+        story
+      );
 
-        if (!list) {
+      if (story.sendStateByConversationId) {
+        let sentId = story.conversationId;
+        let sentName = conversationStory.group?.title;
+
+        if (story.storyDistributionListId) {
+          const list =
+            story.storyDistributionListId === MY_STORIES_ID
+              ? { id: MY_STORIES_ID, name: MY_STORIES_ID }
+              : distributionListSelector(
+                  story.storyDistributionListId.toLowerCase()
+                );
+
+          if (!list) {
+            return;
+          }
+
+          sentId = list.id;
+          sentName = list.name;
+        }
+
+        if (!sentName) {
           return;
         }
 
         const storyView = getStoryView(conversationSelector, story);
 
-        const existingMyStory = myStoriesById.get(list.id) || { stories: [] };
+        const existingMyStory = myStoriesById.get(sentId) || { stories: [] };
 
-        myStoriesById.set(list.id, {
-          distributionId: list.id,
-          distributionName: list.name,
+        myStoriesById.set(sentId, {
+          id: sentId,
+          name: sentName,
           stories: [...existingMyStory.stories, storyView],
         });
 
         return;
       }
 
-      const conversationStory = getConversationStory(
-        conversationSelector,
-        story
-      );
-
       let storiesMap: Map<string, ConversationStoryType>;
 
       if (conversationStory.isHidden) {
diff --git a/ts/test-both/helpers/getFakeStory.tsx b/ts/test-both/helpers/getFakeStory.tsx
index f43da9e72b..b30b8f25d5 100644
--- a/ts/test-both/helpers/getFakeStory.tsx
+++ b/ts/test-both/helpers/getFakeStory.tsx
@@ -27,9 +27,8 @@ export function getFakeMyStory(id?: string, name?: string): MyStoryType {
   const storyCount = casual.integer(2, 6);
 
   return {
-    distributionId: id || UUID.generate().toString(),
-    distributionName:
-      name || id === MY_STORIES_ID ? 'My Stories' : casual.catch_phrase,
+    id: id || UUID.generate().toString(),
+    name: name || id === MY_STORIES_ID ? 'My Stories' : casual.catch_phrase,
     stories: Array.from(Array(storyCount), () => ({
       ...getFakeStoryView(),
       sendState: [],
diff --git a/ts/textsecure/MessageReceiver.ts b/ts/textsecure/MessageReceiver.ts
index 4fccb6ecf6..1c2daad12c 100644
--- a/ts/textsecure/MessageReceiver.ts
+++ b/ts/textsecure/MessageReceiver.ts
@@ -1932,6 +1932,24 @@ export default class MessageReceiver
       timestamp: envelope.timestamp,
     };
 
+    if (sentMessage && message.groupV2) {
+      const ev = new SentEvent(
+        {
+          destinationUuid: envelope.destinationUuid.toString(),
+          isRecipientUpdate: Boolean(sentMessage.isRecipientUpdate),
+          message,
+          receivedAtCounter: envelope.receivedAtCounter,
+          receivedAtDate: envelope.receivedAtDate,
+          serverTimestamp: envelope.serverTimestamp,
+          timestamp: envelope.timestamp,
+          unidentifiedStatus: sentMessage.unidentifiedStatus,
+        },
+        this.removeFromCache.bind(this, envelope)
+      );
+      this.dispatchAndWait(ev);
+      return;
+    }
+
     if (sentMessage) {
       const { storyMessageRecipients } = sentMessage;
       const recipients = storyMessageRecipients ?? [];
diff --git a/ts/types/Stories.ts b/ts/types/Stories.ts
index 2b4efbc7eb..5ec3e7e37e 100644
--- a/ts/types/Stories.ts
+++ b/ts/types/Stories.ts
@@ -85,8 +85,8 @@ export type StoryViewType = {
 };
 
 export type MyStoryType = {
-  distributionId: string;
-  distributionName: string;
+  id: string;
+  name: string;
   stories: Array<StoryViewType>;
 };