feat: add autofocus parameter (#134)

Author: miquelbeltran

lib/src/views/llm_chat_view/llm_chat_view.dart

@@ -89,6 +89,7 @@ class LlmChatView extends StatefulWidget {
     this.errorMessage = 'ERROR',
     this.enableAttachments = true,
     this.enableVoiceNotes = true,
+    this.autofocus,
     super.key,
   }) : viewModel = ChatViewModel(
          provider: provider,
@@ -141,6 +142,13 @@ class LlmChatView extends StatefulWidget {
   /// Defaults to 'ERROR'.
   final String errorMessage;
 
+  /// Whether to autofocus the chat input field when the view is displayed.
+  ///
+  /// Defaults to `null`, which means it will be determined based on the
+  /// presence of suggestions. If there are no suggestions, the input field
+  /// will be focused automatically.
+  final bool? autofocus;
+
   @override
   State<LlmChatView> createState() => _LlmChatViewState();
 }
@@ -204,7 +212,9 @@ class _LlmChatViewState extends State<LlmChatView>
                     ),
                     ChatInput(
                       initialMessage: _initialMessage,
-                      autofocus: widget.viewModel.suggestions.isEmpty,
+                      autofocus:
+                          widget.autofocus ??
+                          widget.viewModel.suggestions.isEmpty,
                       onCancelEdit:
                           _associatedResponse != null ? _onCancelEdit : null,
                       onSendMessage: _onSendMessage,

test/suggestions_test.dart

@@ -6,7 +6,7 @@ import 'package:flutter_test/flutter_test.dart';
 
 void main() {
   group('Suggestions - Welcome message overlap tests', () {
-    Widget materialApp(int suggestionsCount) => MaterialApp(
+    Widget materialApp(int suggestionsCount, {bool? autofocus}) => MaterialApp(
       home: Scaffold(
         appBar: AppBar(title: const Text('Title')),
         body: LlmChatView(
@@ -17,6 +17,7 @@ void main() {
             (index) => 'Suggestion sample ${index + 1}',
           ),
           provider: EchoProvider(),
+          autofocus: autofocus,
         ),
       ),
     );
@@ -69,6 +70,32 @@ void main() {
       // TextField must be focused now
       expect((tester.widget<TextField>(textField)).focusNode?.hasFocus, true);
     });
+    testWidgets('force autofocus false even if no suggestions provided', (
+      tester,
+    ) async {
+      // No suggestions provided, but autofocus is set to false
+      await tester.pumpWidget(materialApp(0, autofocus: false));
+
+      // ChatTextField must be autofocus false and TextField must not be focused
+      // because parameter is set to false
+      final chatTextField = find.byType(ChatTextField);
+      final textField = find.byType(TextField);
+      expect((tester.widget<ChatTextField>(chatTextField)).autofocus, false);
+      expect((tester.widget<TextField>(textField)).focusNode?.hasFocus, false);
+    });
+    testWidgets('force autofocus true even if suggestions provided', (
+      tester,
+    ) async {
+      // Suggestions provided, but autofocus is set to true
+      await tester.pumpWidget(materialApp(40, autofocus: true));
+
+      // ChatTextField must be autofocus true and TextField must be focused
+      // because parameter is set to true
+      final chatTextField = find.byType(ChatTextField);
+      final textField = find.byType(TextField);
+      expect((tester.widget<ChatTextField>(chatTextField)).autofocus, true);
+      expect((tester.widget<TextField>(textField)).focusNode?.hasFocus, true);
+    });
     testWidgets('Welcome message with a lot of suggestions allowing scroll', (
       tester,
     ) async {