Fix implementation and reorganize statements

NIAEFEUP · Dec 8, 2023 · 46de6ed · 46de6ed
1 parent e56e915
commit 46de6ed
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/uni/lib/view/restaurant/widgets/restaurant_page_card.dart b/uni/lib/view/restaurant/widgets/restaurant_page_card.dart
@@ -47,17 +47,22 @@ class CardFavoriteButton extends StatelessWidget {
       builder: (context, restaurantProvider) {
         final isFavorite =
             restaurantProvider.favoriteRestaurants.contains(restaurant.name);
-        final favoriteCardTypes =
-            Provider.of<HomePageProvider>(context).favoriteCards;
         return IconButton(
           icon: isFavorite ? Icon(MdiIcons.heart) : Icon(MdiIcons.heartOutline),
-          onPressed: () => {
+          onPressed: () {
             restaurantProvider.toggleFavoriteRestaurant(
               restaurant.name,
-            ),
+            );
+            final favoriteCardTypes =
+                context.read<HomePageProvider>().favoriteCards;
             if (!isFavorite &&
-                !favoriteCardTypes.contains(FavoriteWidgetType.restaurant))
-              showRestaurantCardHomeDialog(context, favoriteCardTypes),
+                !favoriteCardTypes.contains(FavoriteWidgetType.restaurant)) {
+              showRestaurantCardHomeDialog(context, favoriteCardTypes,
+                  (newFavoriteCards) {
+                Provider.of<HomePageProvider>(context, listen: false)
+                    .setFavoriteCards(newFavoriteCards);
+              });
+            }
           },
         );
       },
@@ -67,6 +72,7 @@ class CardFavoriteButton extends StatelessWidget {
   void showRestaurantCardHomeDialog(
     BuildContext context,
     List<FavoriteWidgetType> favoriteCardTypes,
+    void Function(List<FavoriteWidgetType>) updateHomePage,
   ) {
     showDialog<void>(
       context: context,
@@ -81,10 +87,8 @@ class CardFavoriteButton extends StatelessWidget {
           ),
           ElevatedButton(
             onPressed: () {
-              Provider.of<HomePageProvider>(context, listen: false)
-                  .setFavoriteCards(
-                favoriteCardTypes + [FavoriteWidgetType.restaurant],
-              );
+              updateHomePage(
+                  favoriteCardTypes + [FavoriteWidgetType.restaurant],);
               Navigator.of(context).pop();
             },
             child: Text(S.of(context).yes),