improve on reaction added

Author: renefloor

packages/stream_feeds/lib/src/client/feeds_client_impl.dart

@@ -383,6 +383,7 @@ class StreamFeedsClientImpl implements StreamFeedsClient {
       query: query,
       commentsRepository: _commentsRepository,
       eventsEmitter: events,
+      currentUserId: user.id,
     );
   }
 

packages/stream_feeds/lib/src/models/comment_data.dart

@@ -1,6 +1,7 @@
 // ignore_for_file: avoid_redundant_argument_values
 
 import 'package:freezed_annotation/freezed_annotation.dart';
+import 'package:stream_core/stream_core.dart';
 
 import '../generated/api/models.dart';
 import '../state/query/comments_query.dart';
@@ -197,3 +198,99 @@ extension CommentResponseMapper on CommentResponse {
     );
   }
 }
+
+extension CommentDataMutations on CommentData {
+  /// Adds a reaction to the comment, updating the latest reactions, reaction groups, reaction count,
+  /// and own reactions if applicable.
+  ///
+  /// @param reaction The reaction to add.
+  /// @param currentUserId The ID of the current user, used to update own reactions.
+  /// @return A new [CommentData] instance with the updated reaction data.
+  CommentData addReaction(
+    FeedsReactionData reaction,
+    String currentUserId,
+  ) {
+    final updatedOwnReactions = switch (reaction.user.id == currentUserId) {
+      true => ownReactions.upsert(reaction, key: (it) => it.id),
+      false => ownReactions,
+    };
+
+    final updatedLatestReactions = latestReactions.upsert(
+      reaction,
+      key: (reaction) => reaction.id,
+    );
+
+    final reactionGroup = switch (reactionGroups[reaction.type]) {
+      final existingGroup? => existingGroup,
+      _ => ReactionGroupData(
+          count: 1,
+          firstReactionAt: reaction.createdAt,
+          lastReactionAt: reaction.createdAt,
+        ),
+    };
+
+    final updatedReactionGroups = {
+      ...reactionGroups,
+      reaction.type: reactionGroup.increment(reaction.createdAt),
+    };
+
+    final updatedReactionCount = updatedReactionGroups.values.sumOf(
+      (group) => group.count,
+    );
+
+    return copyWith(
+      ownReactions: updatedOwnReactions,
+      latestReactions: updatedLatestReactions,
+      reactionGroups: updatedReactionGroups,
+      reactionCount: updatedReactionCount,
+    );
+  }
+
+  /// Removes a reaction from the comment, updating the latest reactions, reaction groups, reaction
+  /// count, and own reactions if applicable.
+  ///
+  /// @param reaction The reaction to remove.
+  /// @param currentUserId The ID of the current user, used to update own reactions.
+  /// @return A new [CommentData] instance with the updated reaction data.
+  CommentData removeReaction(
+    FeedsReactionData reaction,
+    String currentUserId,
+  ) {
+    final updatedOwnReactions = switch (reaction.user.id == currentUserId) {
+      true => ownReactions.where((it) => it.id != reaction.id).toList(),
+      false => ownReactions,
+    };
+
+    final updatedLatestReactions = latestReactions.where((it) {
+      return it.id != reaction.id;
+    }).toList(growable: false);
+
+    final updatedReactionGroups = {...reactionGroups};
+    final reactionGroup = updatedReactionGroups.remove(reaction.type);
+
+    if (reactionGroup == null) {
+      // If there is no reaction group for this type, just update latest and own reactions.
+      // Note: This is only a hypothetical case, as we should always have a reaction group.
+      return copyWith(
+        latestReactions: updatedLatestReactions,
+        ownReactions: updatedOwnReactions,
+      );
+    }
+
+    final updatedReactionGroup = reactionGroup.decrement(reaction.createdAt);
+    if (updatedReactionGroup.count > 0) {
+      updatedReactionGroups[reaction.type] = updatedReactionGroup;
+    }
+
+    final updatedReactionCount = updatedReactionGroups.values.sumOf(
+      (group) => group.count,
+    );
+
+    return copyWith(
+      ownReactions: updatedOwnReactions,
+      latestReactions: updatedLatestReactions,
+      reactionGroups: updatedReactionGroups,
+      reactionCount: updatedReactionCount,
+    );
+  }
+}

packages/stream_feeds/lib/src/models/feeds_reaction_data.dart

@@ -26,6 +26,8 @@ class FeedsReactionData with _$FeedsReactionData {
   @override
   final String activityId;
 
+  /// The ID of the comment this reaction is associated with.
+  @override
   final String? commentId;
 
   /// The date and time when the reaction was created.

packages/stream_feeds/lib/src/models/feeds_reaction_data.freezed.dart

@@ -25,9 +25,11 @@ class CommentList extends Disposable {
     required this.query,
     required this.commentsRepository,
     required this.eventsEmitter,
+    required this.currentUserId,
   }) {
     _stateNotifier = CommentListStateNotifier(
       initialState: const CommentListState(),
+      currentUserId: currentUserId,
     );
 
     // Attach event handlers for real-time updates
@@ -41,6 +43,7 @@ class CommentList extends Disposable {
 
   final CommentsQuery query;
   final CommentsRepository commentsRepository;
+  final String currentUserId;
 
   CommentListState get state => stateNotifier.value;
   StateNotifier<CommentListState> get notifier => stateNotifier;

packages/stream_feeds/lib/src/state/comment_list.dart

@@ -3,6 +3,7 @@ import 'package:state_notifier/state_notifier.dart';
 import 'package:stream_core/stream_core.dart';
 
 import '../models/comment_data.dart';
+import '../models/feeds_reaction_data.dart';
 import '../models/pagination_data.dart';
 import 'query/comments_query.dart';
 
@@ -15,8 +16,10 @@ part 'comment_list_state.freezed.dart';
 class CommentListStateNotifier extends StateNotifier<CommentListState> {
   CommentListStateNotifier({
     required CommentListState initialState,
+    required this.currentUserId,
   }) : super(initialState);
 
+  final String currentUserId;
   ({Filter? filter, CommentsSort? sort})? _queryConfig;
   CommentsSort get commentSort => _queryConfig?.sort ?? CommentsSort.last;
 
@@ -59,6 +62,32 @@ class CommentListStateNotifier extends StateNotifier<CommentListState> {
 
     state = state.copyWith(comments: updatedComments);
   }
+
+  /// Handles the addition of a reaction to a comment.
+  void onCommentReactionAdded(String commentId, FeedsReactionData reaction) {
+    final updatedComments = state.comments.updateNested(
+      (comment) => comment.id == commentId,
+      children: (it) => it.replies ?? [],
+      update: (found) => found.addReaction(reaction, currentUserId),
+      updateChildren: (parent, replies) => parent.copyWith(replies: replies),
+      compare: commentSort.compare,
+    );
+
+    state = state.copyWith(comments: updatedComments);
+  }
+
+  /// Handles the removal of a reaction from a comment.
+  void onCommentReactionRemoved(String commentId, FeedsReactionData reaction) {
+    final updatedComments = state.comments.updateNested(
+      (comment) => comment.id == commentId,
+      children: (it) => it.replies ?? [],
+      update: (found) => found.removeReaction(reaction, currentUserId),
+      updateChildren: (parent, replies) => parent.copyWith(replies: replies),
+      compare: commentSort.compare,
+    );
+
+    state = state.copyWith(comments: updatedComments);
+  }
 }
 
 /// An observable state object that manages the current state of a comment list.

packages/stream_feeds/lib/src/state/comment_list_state.dart

@@ -3,6 +3,7 @@ import 'package:stream_core/stream_core.dart';
 import '../../generated/api/models.dart' as api;
 
 import '../../models/comment_data.dart';
+import '../../models/feeds_reaction_data.dart';
 import '../comment_list_state.dart';
 import '../query/comments_query.dart';
 import 'state_event_handler.dart';
@@ -41,6 +42,20 @@ class CommentListEventHandler implements StateEventHandler {
       return state.onCommentRemoved(event.comment.id);
     }
 
+    if (event is api.CommentReactionAddedEvent) {
+      return state.onCommentReactionAdded(
+        event.comment.id,
+        event.reaction.toModel(),
+      );
+    }
+
+    if (event is api.CommentReactionDeletedEvent) {
+      return state.onCommentReactionRemoved(
+        event.comment.id,
+        event.reaction.toModel(),
+      );
+    }
+
     // Handle other comment-related events if needed
   }
 }