From 99680baf83cd494a97b4db115ba2f1a2a4f7ce06 Mon Sep 17 00:00:00 2001
From: Adam <2363879+adamdotdevin@users.noreply.github.com>
Date: Tue, 16 Dec 2025 15:25:00 -0600
Subject: fix(desktop): focus prompt input after dialog close

---
 packages/ui/src/context/dialog.tsx | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'packages/ui/src/context')

diff --git a/packages/ui/src/context/dialog.tsx b/packages/ui/src/context/dialog.tsx
index 71fc63806..56be9ee47 100644
--- a/packages/ui/src/context/dialog.tsx
+++ b/packages/ui/src/context/dialog.tsx
@@ -33,6 +33,10 @@ function init() {
     },
     close() {
       active()?.onClose?.()
+      if (!active()?.onClose) {
+        const promptInput = document.querySelector("[data-component=prompt-input]") as HTMLElement
+        promptInput?.focus()
+      }
       setActive(undefined)
     },
     show(element: DialogElement, owner: Owner, onClose?: () => void) {
@@ -48,7 +52,6 @@ function init() {
                 open={true}
                 onOpenChange={(open) => {
                   if (!open) {
-                    console.log("closing")
                     result.close()
                   }
                 }}
-- 
cgit v1.2.3