gopeed/ui/flutter/lib/app/router/mobile_exit_guard.dart

import 'dart:async';

import 'package:flutter/foundation.dart';
import 'package:flutter/widgets.dart';

import '../../shared/navigation/app_exit_confirmation_controller.dart';

/// Confirms app exit only on the home route; child routes pop normally.
class MobileExitGuard extends StatefulWidget {
  const MobileExitGuard({super.key, required this.child});

  final Widget child;

  @override
  State<MobileExitGuard> createState() => _MobileExitGuardState();
}

class _MobileExitGuardState extends State<MobileExitGuard> with WidgetsBindingObserver {
  final _controller = AppExitConfirmationController();

  @override
  void initState() {
    super.initState();
    WidgetsBinding.instance.addObserver(this);
  }

  @override
  void didChangeAppLifecycleState(AppLifecycleState state) {
    if (state != AppLifecycleState.resumed) _controller.dispose();
  }

  @override
  void dispose() {
    WidgetsBinding.instance.removeObserver(this);
    _controller.dispose();
    super.dispose();
  }

  @override
  Widget build(BuildContext context) {
    if (kIsWeb || defaultTargetPlatform != TargetPlatform.android) return widget.child;

    return PopScope(
      canPop: false,
      onPopInvokedWithResult: (didPop, _) {
        if (!didPop) unawaited(_controller.handleBack(context));
      },
      child: widget.child,
    );
  }
}