From b9ff6bd20ffbad9fc75b9ce9ec42e30e9c07a77e Mon Sep 17 00:00:00 2001
From: sawka <mike.sawka@gmail.com>
Date: Fri, 15 Sep 2023 17:24:38 -0700
Subject: [PATCH] codeedit: on close, if we're focused, give focus back to the
 input box

---
 src/view/code.tsx | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/src/view/code.tsx b/src/view/code.tsx
index a00d381f..d1c4823b 100644
--- a/src/view/code.tsx
+++ b/src/view/code.tsx
@@ -203,6 +203,9 @@ class SourceCodeRenderer extends React.Component<
                     this.setState({ message: null });
                 }, 3000);
             });
+        if (this.props.shouldFocus) {
+            GlobalCommandRunner.screenSetFocus("input");
+        }
     };
 
     handleEditorChange = (code) => {