diff --git a/compass_app/app/lib/routing/router.dart b/compass_app/app/lib/routing/router.dart index 284706850..2d3030085 100644 --- a/compass_app/app/lib/routing/router.dart +++ b/compass_app/app/lib/routing/router.dart @@ -14,8 +14,7 @@ import '../ui/auth/login/widgets/login_screen.dart'; import '../ui/auth/logout/view_models/logout_viewmodel.dart'; import '../ui/booking/view_models/booking_viewmodel.dart'; import '../ui/booking/widgets/booking_screen.dart'; -import '../ui/home/view_models/home_viewmodel.dart'; -import '../ui/home/widgets/home_screen.dart'; +import '../ui/home/widgets/home_screen_container.dart'; import '../ui/results/view_models/results_viewmodel.dart'; import '../ui/results/widgets/results_screen.dart'; import '../ui/search_form/view_models/search_form_viewmodel.dart'; @@ -43,16 +42,11 @@ GoRouter router(AuthRepository authRepository) => GoRouter( GoRoute( path: Routes.home, builder: (context, state) { - final viewModel = HomeViewModel( - bookingRepository: context.read(), - userRepository: context.read(), - ); final logoutViewModel = LogoutViewModel( authRepository: context.read(), itineraryConfigRepository: context.read(), ); - return HomeScreen( - viewModel: viewModel, + return HomeScreenContainer( logoutViewModel: logoutViewModel, ); }, diff --git a/compass_app/app/lib/ui/home/widgets/home_screen_container.dart b/compass_app/app/lib/ui/home/widgets/home_screen_container.dart index c62fbc3c1..31608b1a5 100644 --- a/compass_app/app/lib/ui/home/widgets/home_screen_container.dart +++ b/compass_app/app/lib/ui/home/widgets/home_screen_container.dart @@ -3,11 +3,17 @@ import 'package:provider/provider.dart'; import '../../../data/repositories/booking/booking_repository.dart'; import '../../../data/repositories/user/user_repository.dart'; +import '../../auth/logout/view_models/logout_viewmodel.dart'; import '../view_models/home_viewmodel.dart'; import 'home_screen.dart'; class HomeScreenContainer extends StatefulWidget { - const HomeScreenContainer({super.key}); + const HomeScreenContainer({ + super.key, + required this.logoutViewModel, + }); + + final LogoutViewModel logoutViewModel; @override State createState() => _HomeScreenContainerState(); @@ -28,7 +34,10 @@ class _HomeScreenContainerState extends State { @override Widget build(BuildContext context) { - return HomeScreen(viewModel: _viewModel); + return HomeScreen( + viewModel: _viewModel, + logoutViewModel: widget.logoutViewModel, + ); } @override