diff --git a/lib/view_models/home_view_model.dart b/lib/view_models/home_view_model.dart index d9a3481..6c1f93d 100644 --- a/lib/view_models/home_view_model.dart +++ b/lib/view_models/home_view_model.dart @@ -234,7 +234,7 @@ class HomeViewModel with ChangeNotifier { notifyListeners(); } - void serUnreadNotificationFalse() { + void setUnreadNotificationFalse() { _hasUnreadNotification = false; notifyListeners(); } diff --git a/lib/view_models/notification_view_model.dart b/lib/view_models/notification_view_model.dart index 5be00ca..65a65ae 100644 --- a/lib/view_models/notification_view_model.dart +++ b/lib/view_models/notification_view_model.dart @@ -300,8 +300,7 @@ class NotificationViewModel with ChangeNotifier { NotificationStrings.allNotificationMarkedAsRead, ); } - final homeViewModel = context.read(); - homeViewModel.serUnreadNotificationFalse(); + setNotificationFalse(context); } else { _errorMessage = message; if (context.mounted) { @@ -340,4 +339,14 @@ class NotificationViewModel with ChangeNotifier { return '${difference.inDays} ${NotificationStrings.daysAgo}'; } } + + void setNotificationFalse(BuildContext context) { + final homeViewModel = context.read(); + homeViewModel.setUnreadNotificationFalse(); + } + + void checkNotifications(BuildContext context) { + final homeViewModel = context.read(); + homeViewModel.checkUnreadNotifications(); + } } diff --git a/lib/views/notification/widgets/notification_card.dart b/lib/views/notification/widgets/notification_card.dart index a64976d..b3d95fd 100644 --- a/lib/views/notification/widgets/notification_card.dart +++ b/lib/views/notification/widgets/notification_card.dart @@ -131,7 +131,11 @@ class NotificationCard extends StatelessWidget { viewModel: viewModel, ); }, - ); + ).then((value) { + if (context.mounted) { + viewModel.checkNotifications(context); + } + }); }, child: Row( children: [