import 'dart:async';
import 'package:flutter/material.dart';
import 'package:go_router/go_router.dart';
import 'core/auth/auth_bloc.dart';
import 'features/splash/splash_screen.dart';
import 'features/auth/screens/login_screen.dart';
import 'features/auth/screens/otp_screen.dart';
import 'features/home/home_screen.dart';
import 'features/chat/screens/active_sessions_screen.dart';
import 'features/chat/screens/mitra_chat_screen.dart';
import 'features/chat/screens/chat_history_screen.dart';
import 'features/chat/screens/chat_transcript_screen.dart';

class _BlocRefreshNotifier extends ChangeNotifier {
  late final StreamSubscription _subscription;

  _BlocRefreshNotifier(AuthBloc bloc) {
    _subscription = bloc.stream.listen((_) => notifyListeners());
  }

  @override
  void dispose() {
    _subscription.cancel();
    super.dispose();
  }
}

GoRouter buildRouter(AuthBloc authBloc) {
  return GoRouter(
    initialLocation: '/splash',
    refreshListenable: _BlocRefreshNotifier(authBloc),
    redirect: (context, state) {
      final authState = authBloc.state;
      final isSplash = state.matchedLocation == '/splash';
      final isAuthRoute = state.matchedLocation.startsWith('/login') ||
          state.matchedLocation.startsWith('/otp');

      // Show splash while loading
      if (authState is AuthLoading) return isSplash ? null : '/splash';

      if (authState is AuthAuthenticated) {
        return (isSplash || isAuthRoute) ? '/home' : null;
      }
      if (!isAuthRoute && !isSplash) return '/login';
      if (isSplash) return '/login';
      return null;
    },
    routes: [
      GoRoute(path: '/splash', builder: (_, __) => const SplashScreen()),
      GoRoute(path: '/login', builder: (_, __) => const LoginScreen()),
      GoRoute(path: '/otp', builder: (context, state) => OtpScreen(phone: state.extra as String)),
      GoRoute(path: '/home', builder: (_, __) => const HomeScreen()),
      GoRoute(path: '/sessions', builder: (_, __) => const ActiveSessionsScreen()),
      GoRoute(path: '/chat/session/:sessionId', builder: (context, state) {
        final extra = state.extra as Map<String, dynamic>?;
        return MitraChatScreen(
          sessionId: state.pathParameters['sessionId']!,
          customerName: extra?['customerName'] as String? ?? 'Customer',
        );
      }),
      GoRoute(path: '/chat/history', builder: (_, __) => const MitraChatHistoryScreen()),
      GoRoute(path: '/chat/history/:sessionId', builder: (context, state) {
        return MitraChatTranscriptScreen(sessionId: state.pathParameters['sessionId']!);
      }),
    ],
  );
}