From cf7ab9a8c6c7c593e2479b579c4b02094eda55d2 Mon Sep 17 00:00:00 2001
From: Ashutosh Sharma <ashutosh.sharma068@gmail.com>
Date: Thu, 16 Feb 2023 23:34:46 +0530
Subject: [PATCH] Re-focus the composer on dialogue quit (#10007)

Co-authored-by: Michael Telatynski <7t3chguy@gmail.com>
---
 src/components/views/rooms/EditMessageComposer.tsx | 10 ++++++++--
 src/components/views/rooms/SendMessageComposer.tsx | 10 ++++++++--
 2 files changed, 16 insertions(+), 4 deletions(-)

diff --git a/src/components/views/rooms/EditMessageComposer.tsx b/src/components/views/rooms/EditMessageComposer.tsx
index a41487261b..beb076f96a 100644
--- a/src/components/views/rooms/EditMessageComposer.tsx
+++ b/src/components/views/rooms/EditMessageComposer.tsx
@@ -339,9 +339,15 @@ class EditMessageComposer extends React.Component<IEditMessageComposerProps, ISt
                     } else {
                         shouldSend = false;
                     }
-                } else if (!(await shouldSendAnyway(commandText))) {
+                } else {
+                    const sendAnyway = await shouldSendAnyway(commandText);
+                    // re-focus the composer after QuestionDialog is closed
+                    dis.dispatch({
+                        action: Action.FocusAComposer,
+                        context: this.context.timelineRenderingType,
+                    });
                     // if !sendAnyway bail to let the user edit the composer and try again
-                    return;
+                    if (!sendAnyway) return;
                 }
             }
             if (shouldSend) {
diff --git a/src/components/views/rooms/SendMessageComposer.tsx b/src/components/views/rooms/SendMessageComposer.tsx
index 68dda98e27..61cb72f90e 100644
--- a/src/components/views/rooms/SendMessageComposer.tsx
+++ b/src/components/views/rooms/SendMessageComposer.tsx
@@ -385,9 +385,15 @@ export class SendMessageComposer extends React.Component<ISendMessageComposerPro
                 } else {
                     shouldSend = false;
                 }
-            } else if (!(await shouldSendAnyway(commandText))) {
+            } else {
+                const sendAnyway = await shouldSendAnyway(commandText);
+                // re-focus the composer after QuestionDialog is closed
+                dis.dispatch({
+                    action: Action.FocusAComposer,
+                    context: this.context.timelineRenderingType,
+                });
                 // if !sendAnyway bail to let the user edit the composer and try again
-                return;
+                if (!sendAnyway) return;
             }
         }