flutter/lib/screens/home_tabs_screen.dart

/*
 * This file is part of wger Workout Manager <https://github.com/wger-project>.
 * Copyright (c)  2026 wger Team
 *
 * wger Workout Manager is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

import 'package:flutter/material.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:font_awesome_flutter/font_awesome_flutter.dart';
import 'package:logging/logging.dart';
import 'package:provider/provider.dart';
import 'package:rive/rive.dart';
import 'package:wger/helpers/material.dart';
import 'package:wger/l10n/generated/app_localizations.dart';
import 'package:wger/providers/auth.dart';
import 'package:wger/providers/body_weight.dart';
import 'package:wger/providers/exercises.dart';
import 'package:wger/providers/gallery.dart';
import 'package:wger/providers/measurement.dart';
import 'package:wger/providers/nutrition.dart';
import 'package:wger/providers/routines.dart';
import 'package:wger/providers/trophies.dart';
import 'package:wger/providers/user.dart';
import 'package:wger/screens/dashboard.dart';
import 'package:wger/screens/gallery_screen.dart';
import 'package:wger/screens/nutritional_plans_screen.dart';
import 'package:wger/screens/routine_list_screen.dart';
import 'package:wger/screens/weight_screen.dart';

class HomeTabsScreen extends ConsumerStatefulWidget {
  final _logger = Logger('HomeTabsScreen');

  HomeTabsScreen();

  static const routeName = '/dashboard2';

  @override
  _HomeTabsScreenState createState() => _HomeTabsScreenState();
}

class _HomeTabsScreenState extends ConsumerState<HomeTabsScreen>
    with SingleTickerProviderStateMixin {
  Future<void>? _initialData;
  bool _errorHandled = false;
  int _selectedIndex = 0;
  bool _isWideScreen = false;

  @override
  void didChangeDependencies() {
    super.didChangeDependencies();

    final size = MediaQuery.sizeOf(context);
    _isWideScreen = size.width > MATERIAL_XS_BREAKPOINT;
    _initialData ??= _loadEntries();
  }

  void _onItemTapped(int index) {
    setState(() {
      _selectedIndex = index;
    });
  }

  final _screenList = [
    const DashboardScreen(),
    const RoutineListScreen(),
    const NutritionalPlansScreen(),
    const WeightScreen(),
    const GalleryScreen(),
  ];

  /// Load initial data from the server
  Future<void> _loadEntries() async {
    final languageCode = Localizations.localeOf(context).languageCode;
    final authProvider = context.read<AuthProvider>();
    final trophyNotifier = ref.read(trophyStateProvider.notifier);

    if (!authProvider.dataInit) {
      final routinesProvider = context.read<RoutinesProvider>();
      final nutritionPlansProvider = context.read<NutritionPlansProvider>();
      final exercisesProvider = context.read<ExercisesProvider>();
      final galleryProvider = context.read<GalleryProvider>();
      final weightProvider = context.read<BodyWeightProvider>();
      final measurementProvider = context.read<MeasurementProvider>();
      final userProvider = context.read<UserProvider>();

      //
      // Base data
      widget._logger.info('Loading base data');
      await Future.wait([
        authProvider.setServerVersion(),
        userProvider.fetchAndSetProfile(),
        routinesProvider.fetchAndSetUnits(),
        nutritionPlansProvider.fetchIngredientsFromCache(),
        exercisesProvider.fetchAndSetInitialData(),
      ]);
      exercisesProvider.fetchAndSetAllExercises();

      // Workaround for https://github.com/wger-project/flutter/issues/901
      // It seems that it can happen that sometimes the units were not loaded properly
      // so now we check and try again if necessary. We might need a better general
      // solution since this could potentially happen with other data as well.
      if (routinesProvider.repetitionUnits.isEmpty || routinesProvider.weightUnits.isEmpty) {
        widget._logger.info('Routine units are empty, fetching again');
        await routinesProvider.fetchAndSetUnits();
      }

      //
      // Plans, weight and gallery
      widget._logger.info('Loading routines, weight, measurements and gallery');
      await Future.wait([
        galleryProvider.fetchAndSetGallery(),
        nutritionPlansProvider.fetchAndSetAllPlansSparse(),
        routinesProvider.fetchAndSetAllRoutinesSparse(),
        // routinesProvider.fetchAndSetAllRoutinesFull(),
        weightProvider.fetchAndSetEntries(),
        measurementProvider.fetchAndSetAllCategoriesAndEntries(),
        trophyNotifier.fetchAll(language: languageCode),
      ]);

      //
      // Current nutritional plan
      widget._logger.info('Loading current nutritional plan');
      if (nutritionPlansProvider.currentPlan != null) {
        final plan = nutritionPlansProvider.currentPlan!;
        await nutritionPlansProvider.fetchAndSetPlanFull(plan.id!);
      }

      //
      // Current routine
      widget._logger.info('Loading current routine');
      if (routinesProvider.currentRoutine != null) {
        final planId = routinesProvider.currentRoutine!.id!;
        await routinesProvider.fetchAndSetRoutineFull(planId);
      }
    }

    authProvider.dataInit = true;
  }

  @override
  Widget build(BuildContext context) {
    final destinations = [
      NavigationDestination(
        icon: const Icon(Icons.home),
        label: AppLocalizations.of(context).labelDashboard,
      ),
      NavigationDestination(
        icon: const Icon(Icons.fitness_center),
        label: AppLocalizations.of(context).labelBottomNavWorkout,
      ),
      NavigationDestination(
        icon: const Icon(Icons.restaurant),
        label: AppLocalizations.of(context).labelBottomNavNutrition,
      ),
      NavigationDestination(
        icon: const FaIcon(FontAwesomeIcons.weightScale, size: 20),
        label: AppLocalizations.of(context).weight,
      ),
      NavigationDestination(
        icon: const Icon(Icons.photo_library),
        label: AppLocalizations.of(context).gallery,
      ),
    ];

    /// Navigation bar for narrow screens
    Widget getNavigationBar() {
      return NavigationBar(
        destinations: destinations,
        onDestinationSelected: _onItemTapped,
        selectedIndex: _selectedIndex,
        labelBehavior: NavigationDestinationLabelBehavior.alwaysHide,
      );
    }

    /// Navigation rail for wide screens
    Widget getNavigationRail() {
      return NavigationRail(
        selectedIndex: _selectedIndex,
        onDestinationSelected: _onItemTapped,
        labelType: NavigationRailLabelType.all,
        scrollable: true,
        destinations: destinations
            .map(
              (d) => NavigationRailDestination(
                icon: d.icon,
                label: Text(d.label),
              ),
            )
            .toList(),
      );
    }

    return FutureBuilder<void>(
      future: _initialData,
      builder: (context, snapshot) {
        if (snapshot.hasError) {
          // Throw the original error with the original stack trace, otherwise
          // the error will only point to these lines here
          if (!_errorHandled) {
            _errorHandled = true;
            final error = snapshot.error;
            final stackTrace = snapshot.stackTrace;

            WidgetsBinding.instance.addPostFrameCallback((_) {
              if (mounted) {
                if (error != null && stackTrace != null) {
                  throw Error.throwWithStackTrace(error, stackTrace);
                }
                throw error!;
              }
            });
          }

          // Note that we continue to show the app, even if there was an error.
          // return const Scaffold(body: LoadingWidget());
        }

        if (snapshot.connectionState != ConnectionState.done) {
          return const Scaffold(
            body: LoadingWidget(),
          );
        }

        return Scaffold(
          body: Row(
            children: [
              if (_isWideScreen) getNavigationRail(),
              Expanded(child: _screenList.elementAt(_selectedIndex)),
            ],
          ),
          bottomNavigationBar: _isWideScreen ? null : getNavigationBar(),
        );
      },
    );
  }
}

class LoadingWidget extends StatelessWidget {
  const LoadingWidget({
    super.key,
  });

  @override
  Widget build(BuildContext context) {
    return Center(
      child: Column(
        mainAxisAlignment: MainAxisAlignment.center,
        children: [
          const Center(
            child: SizedBox(
              height: 70,
              child: RiveAnimation.asset(
                'assets/animations/wger_logo.riv',
                animations: ['idle_loop2'],
              ),
            ),
          ),
          Text(
            AppLocalizations.of(context).loadingText,
            style: Theme.of(context).textTheme.headlineSmall,
          ),
        ],
      ),
    );
  }
}