xtremflow/lib/core/router/app_router.dart

import 'package:flutter/material.dart';
import 'package:flutter_riverpod/flutter_riverpod.dart';
import 'package:go_router/go_router.dart';
import '../../features/auth/providers/auth_provider.dart';
import '../../features/auth/screens/login_screen.dart';
import '../../features/iptv/screens/playlist_selection_screen.dart';
import '../../features/iptv/screens/dashboard_screen.dart';
import '../../features/admin/screens/admin_panel.dart';
import '../../mobile/features/iptv/screens/mobile_playlist_selection_screen.dart';
import '../../mobile/features/iptv/screens/mobile_dashboard_screen.dart';
import '../../mobile/features/auth/screens/mobile_login_screen.dart';
import '../widgets/themed_loading_screen.dart';
import '../../features/iptv/providers/xtream_provider.dart';

final routerProvider = Provider<GoRouter>((ref) {
  return GoRouter(
    initialLocation: '/login',
    refreshListenable: RouterRefreshNotifier(ref),
    redirect: (context, state) {
      final authState = ref.read(authProvider);
      final isLoggedIn = authState.isAuthenticated;
      final isLoginRoute = state.matchedLocation == '/login';

      // Wait for initial auth check to complete
      if (!authState.isInitialized) {
        return null;
      }

      // Redirect to login if not authenticated
      if (!isLoggedIn && !isLoginRoute) {
        return '/login';
      }

      // Redirect to playlists if already logged in and trying to access login
      if (isLoggedIn && isLoginRoute) {
        return '/playlists';
      }

      // Admin guard for /admin route
      if (state.matchedLocation == '/admin' && !authState.isAdmin) {
        return '/playlists';
      }

      // If we are on dashboard but have no playlist selected, try to restore it
      if (state.matchedLocation == '/dashboard') {
        final playlist = ref.read(selectedPlaylistProvider);
        if (playlist == null) {
          return '/playlists';
        }
      }

      return null;
    },
    routes: [
      GoRoute(
        path: '/login',
        builder: (context, state) {
          if (MediaQuery.of(context).size.width < 768) {
            return const MobileLoginScreen();
          }
          return const LoginScreen();
        },
      ),
      GoRoute(
        path: '/playlists',
        builder: (context, state) {
          if (MediaQuery.of(context).size.width < 768) {
            return const MobilePlaylistSelectionScreen();
          }
          return const PlaylistSelectionScreen();
        },
      ),
      GoRoute(
        path: '/dashboard',
        // Le `ref` du routeur ne doit pas surveiller la playlist : un
        // `ref.watch` ici rend `routerProvider` dépendant de
        // `selectedPlaylistProvider`, si bien que changer d'IPTV depuis le
        // tableau de bord recréait le GoRouter, reparti de `/login` puis
        // redirigé vers `/playlists`. Le `Consumer` limite la reconstruction
        // à cette seule route.
        builder: (context, state) => Consumer(
          builder: (context, ref, _) {
            final playlist = ref.watch(selectedPlaylistProvider);
            if (playlist == null) {
              return const Scaffold(
                body: ThemedLoading(),
              );
            }

            // Une clé par IPTV : les onglets chargent leur catalogue dans
            // `initState` et ignorent un changement de `playlist`. Sans elle,
            // le tableau de bord garderait le contenu de l'IPTV précédente.
            final key = ValueKey(playlist.id);

            // Adaptive layout without forcing a router rebuild on every pixel change
            final isMobile = MediaQuery.sizeOf(context).width < 768;
            if (isMobile) {
              return MobileDashboardScreen(key: key, playlist: playlist);
            }
            return DashboardScreen(key: key, playlist: playlist);
          },
        ),
      ),
      GoRoute(
        path: '/admin',
        builder: (context, state) => const AdminPanel(),
      ),
    ],
  );
});

/// Notifier that triggers router refresh when auth state or selected playlist changes
class RouterRefreshNotifier extends ChangeNotifier {
  RouterRefreshNotifier(this._ref) {
    _ref.listen(authProvider, (_, __) {
      notifyListeners();
    });
    _ref.listen(selectedPlaylistProvider, (_, __) {
      notifyListeners();
    });
  }

  final Ref _ref;
}