Fix mynah UI progress indicator (#374)

* Fix progress indicator not disappearing after complete result has streamed * Fix Amazon Q progress spinner bug (#376)
aws · Feb 18, 2025 · 7bdc736 · 7bdc736
1 parent 680a01a
commit 7bdc736
Showing 1 changed file with 23 additions and 2 deletions.
diff --git a/plugin/src/software/aws/toolkits/eclipse/amazonq/views/AmazonQChatWebview.java b/plugin/src/software/aws/toolkits/eclipse/amazonq/views/AmazonQChatWebview.java
@@ -250,6 +250,9 @@ private String generateCss() {
                     textarea:placeholder-shown {
                         line-height: 1.5rem;
                     }
+                    .mynah-ui-spinner-container > span.mynah-ui-spinner-logo-part > .mynah-ui-spinner-logo-mask.text {
+                        opacity: 1 !important;
+                    }
                 </style>
                 """;
     }
@@ -268,7 +271,8 @@ private String generateJS(final String jsEntrypoint) {
                                     postMessage: (message) => {
                                         ideCommand(JSON.stringify(message));
                                     }
-                                }, {
+                                },
+                                {
                                     quickActionCommands: %s,
                                     disclaimerAcknowledged: %b
                                 });
@@ -284,10 +288,12 @@ private String generateJS(final String jsEntrypoint) {
 
                     %s
 
+                    %s
+
                 </script>
                 """, jsEntrypoint, getWaitFunction(), chatQuickActionConfig,
                 "true".equals(disclaimerAcknowledged), getArrowKeyBlockingFunction(),
-                getSelectAllAndCopySupportFunctions(), getPreventEmptyPopupFunction());
+                getSelectAllAndCopySupportFunctions(), getPreventEmptyPopupFunction(), getFocusOnChatPromptFunction());
     }
 
     /*
@@ -423,6 +429,21 @@ function attachEventListeners(element) {
                 """.formatted(selector);
     }
 
+    private String getFocusOnChatPromptFunction() {
+        return """
+                window.addEventListener('load', () => {
+                    const chatContainer = document.querySelector('.mynah-chat-prompt');
+                    if (chatContainer) {
+                        chatContainer.addEventListener('click', (event) => {
+                            if (!event.target.closest('.mynah-chat-prompt-input')) {
+                                keepFocusOnPrompt();
+                            }
+                        });
+                    }
+                });
+                """;
+    }
+
     @Override
     public final void onSendToChatUi(final String message) {
         String script = "window.postMessage(" + message + ");";