Recover post if it failed to send.

Nerimity · Dec 23, 2024 · aae3100 · aae3100
1 parent d1bddf3
commit aae3100
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 12 deletions.
diff --git a/src/chat-api/store/usePosts.ts b/src/chat-api/store/usePosts.ts
@@ -207,6 +207,7 @@ export function usePosts() {
               ...this.commentIds!,
             ]);
           });
+          return true;
         },
         cachedComments() {
           return this.commentIds?.map((postId) => state.posts[postId] as Post);
@@ -247,9 +248,12 @@ export function usePosts() {
       alert(err.message);
     });
     if (!post) return;
-    pushPost(post, account.user()?.id!);
-    setState("feedPostIds", [post.id, ...state.feedPostIds]);
-    setState("discoverPostIds", [post.id, ...state.discoverPostIds]);
+    batch(() => {
+      pushPost(post, account.user()?.id!);
+      setState("feedPostIds", [post.id, ...state.feedPostIds]);
+      setState("discoverPostIds", [post.id, ...state.discoverPostIds]);
+    });
+    return true;
   };
 
   const fetchUserPosts = async (userId: string, withReplies?: boolean) => {

diff --git a/src/components/PostsArea.tsx b/src/components/PostsArea.tsx
@@ -149,18 +149,32 @@ function NewPostArea(props: {
   };
 
   const onCreateClick = () => {
+    const tempContent = content();
     const formattedContent = formatMessage(content().trim());
     if (props.postId) {
-      posts.cachedPost(props.postId)?.submitReply({
-        content: formattedContent,
-        attachment: attachedFile(),
-      });
+      posts
+        .cachedPost(props.postId)
+        ?.submitReply({
+          content: formattedContent,
+          attachment: attachedFile(),
+        })
+        .then((res) => {
+          if (!res) {
+            setContent(tempContent);
+          }
+        });
     } else {
-      posts.submitPost({
-        content: formattedContent,
-        file: attachedFile(),
-        poll: showPollOptions() ? { choices: pollOptions } : undefined,
-      });
+      posts
+        .submitPost({
+          content: formattedContent,
+          file: attachedFile(),
+          poll: showPollOptions() ? { choices: pollOptions } : undefined,
+        })
+        .then((res) => {
+          if (!res) {
+            setContent(tempContent);
+          }
+        });
     }
     setContent("");
     setPollOptions(reconcile([""]));