Merge pull request #687 from Mobile-Artificial-Intelligence/chat_listener

Chat listener

Author: danemadsen

lib/controllers/chat_controller.dart

@@ -161,6 +161,8 @@ class ChatController extends ChangeNotifier {
       _rootKey = chat.id;
     }
 
+    _rootKey = _mapping.keys.firstWhere((key) => _mapping[key]!.parent == null);
+
     notifyListeners();
   }
 

lib/main.dart

@@ -63,6 +63,7 @@ part 'widgets/dropdowns/theme_mode_dropdown.dart';
 part 'widgets/layout/main_drawer.dart';
 
 part 'widgets/listeners/artificial_intelligence_listener.dart';
+part 'widgets/listeners/chat_listener.dart';
 
 part 'widgets/message/message_view.dart';
 part 'widgets/message/message.dart';

lib/widgets/listeners/chat_listener.dart

@@ -0,0 +1,23 @@
+part of 'package:maid/main.dart';
+
+class ChatListener extends StatelessWidget{
+  final Widget Function(BuildContext context, Widget? child) builder;
+
+  const ChatListener({
+    super.key,
+    required this.builder,
+  });
+
+  @override
+  Widget build(BuildContext context) => ListenableBuilder(
+    listenable: ChatController.instance,
+    builder: builder,
+  );
+
+  Widget buildMessageListener(int index) => ListenableBuilder(
+    listenable: ChatController.instance.root.chain[index],
+    builder: ChatController.instance.root.chain.length - 1 >= index 
+      ? (context, child) => builder(context, child)
+      : (context, child) => buildMessageListener(index + 1),
+  );
+}

lib/widgets/message/message.dart

@@ -3,23 +3,26 @@ part of 'package:maid/main.dart';
 class MessageWidget extends StatefulWidget {
   final ChatMessage node;
 
-  /// The chain position is used to determine the position of the message in the chain.
-  /// Where a position of 0 would indicate the bottom of the chain.
-  final int chainPosition;
-
   ChatMessage get message => node.currentChild!;
   int get siblingsIndex => node.currentChildIndex;
   int get siblingCount => node.children.length;
   bool get onNextEnabled => (siblingsIndex + 1) < siblingCount;
   bool get onPreviousEnabled => siblingsIndex > 0;
-  bool get buildChild => node.currentChild?.currentChild != null && chainPosition > 0;
 
   const MessageWidget({
     required super.key, 
-    required this.node,
-    required this.chainPosition,
+    required this.node
   });
 
+  factory MessageWidget.itemBuilder(int index) {
+    final message = ChatController.instance.root.chain[index];
+
+    return MessageWidget(
+      key: message.id,
+      node: message,
+    );
+  }
+
   @override
   State<MessageWidget> createState() => MessageWidgetState();
 }
@@ -103,25 +106,11 @@ class MessageWidgetState extends State<MessageWidget> {
   @override
   Widget build(BuildContext context) => ListenableBuilder(
     listenable: widget.node.currentChild ?? widget.node, 
-    builder: buildColumn
-  );
-
-  Widget buildColumn(BuildContext context, Widget? child) => Column(
-    children: [
-      // Build the current node
-      if (widget.node.currentChild != null) buildCurrentMessage(),
-
-      /// Builds the child node/s if it exists.
-      if (widget.buildChild) MessageWidget(
-        key: childKey,
-        node: widget.node.currentChild!,
-        chainPosition: widget.chainPosition - 1,
-      ),
-    ],
+    builder: buildMessage
   );
 
   // The buildCurrentMessage method will build the padding and the appropriate column based on the editing state.
-  Widget buildCurrentMessage() => Padding(
+  Widget buildMessage(BuildContext context, Widget? child) => Padding(
     padding: const EdgeInsets.symmetric(vertical: 8.0, horizontal: 16.0),
     child: editing && !AIController.instance.busy ? 
       buildMessageEditingColumn() : 

lib/widgets/message/message_view.dart

@@ -73,12 +73,7 @@ class MessageViewState extends State<MessageView> {
   void initState() {
     super.initState();
     controller.addListener(onScroll);
-
-    WidgetsBinding.instance.addPostFrameCallback((_) {
-      final root = ChatController.instance.root;
-      rootPosition = math.max(0, root.chain.length - maxMessages);
-      controller.jumpTo(controller.position.maxScrollExtent);
-    });
+    rootPosition = math.max(0, ChatController.instance.root.chain.length - maxMessages);
   }
 
   void onScroll() {
@@ -128,7 +123,7 @@ class MessageViewState extends State<MessageView> {
       if (rootPosition >= 0 && rootPosition < ChatController.instance.root.chain.length - maxMessages)
         buildClearButton(),
       Expanded(
-        child: buildScrollView()
+        child: buildSettingsListener()
       )
     ],
   );
@@ -142,18 +137,12 @@ class MessageViewState extends State<MessageView> {
     )
   );
 
-  Widget buildScrollView() => SingleChildScrollView(
-    controller: controller,
-    child: buildChatListener(),
-  );
-
-  Widget buildChatListener() => ListenableBuilder(
-    listenable: ChatController.instance,
-    builder: buildSettingsListener
+  Widget buildSettingsListener() => ListenableBuilder(
+    listenable: AppSettings.instance,
+    builder: buildChatListener
   );
 
-  Widget buildSettingsListener(BuildContext context, Widget? child) => ListenableBuilder(
-    listenable: AppSettings.instance,
+  Widget buildChatListener(BuildContext context, Widget? child) => ChatListener(
     builder: buildMessage
   );
 
@@ -178,10 +167,10 @@ class MessageViewState extends State<MessageView> {
       WidgetsBinding.instance.addPostFrameCallback(jumpToExtent);
     }
 
-    return MessageWidget(
-      key: rootKey,
-      node: localRoot,
-      chainPosition: maxMessages - 1,
+    return ListView.builder(
+      controller: controller,
+      itemBuilder: (context, index) => MessageWidget.itemBuilder(rootPosition + index),
+      itemCount: math.min(maxMessages, ChatController.instance.root.chain.length - 1),
     );
   }
 }