Auto focus on text field in chat in various situations

Author: intitni

Core/Sources/ChatGPTChatTab/Chat.swift

@@ -32,7 +32,7 @@ struct Chat: ReducerProtocol {
         var history: [ChatMessage] = []
         @BindingState var isReceivingMessage = false
         var chatMenu = ChatMenu.State()
-        @BindingState var focusedField: Field? = .textField
+        @BindingState var focusedField: Field?
 
         enum Field: String, Hashable {
             case textField
@@ -95,6 +95,7 @@ struct Chat: ReducerProtocol {
                     await send(.isReceivingMessageChanged)
                     await send(.systemPromptChanged)
                     await send(.extraSystemPromptChanged)
+                    await send(.focusOnTextField)
                 }
 
             case .sendButtonTapped:

Tool/Sources/FocusedCodeFinder/FocusedCodeFinder.swift

@@ -67,14 +67,17 @@ public struct UnknownLanguageFocusedCodeFinder: FocusedCodeFinder {
         guard !activeDocumentContext.lines.isEmpty else { return .empty }
 
         // when user is not selecting any code.
-        if containingRange.start == containingRange.end {
+        if containingRange.start == containingRange.end{
             // search up and down for up to `proposedSearchRange * 2 + 1` lines.
             let lines = activeDocumentContext.lines
             let proposedLineCount = proposedSearchRange * 2 + 1
             let startLineIndex = max(containingRange.start.line - proposedSearchRange, 0)
-            let endLineIndex = max(
-                startLineIndex,
-                min(startLineIndex + proposedLineCount - 1, lines.count - 1)
+            let endLineIndex = min(
+                max(
+                    startLineIndex,
+                    min(startLineIndex + proposedLineCount - 1, lines.count - 1)
+                ),
+                lines.count - 1
             )
 
             let focusedLines = lines[startLineIndex...endLineIndex]