ReVoltVPN/lib/components/connect_button.dart

import 'package:flutter/material.dart';
import 'package:flutter/services.dart';
import 'package:provider/provider.dart';
import 'package:paladinvpn/logic/vpn_connection.dart';
import 'package:paladinvpn/logic/session_timer.dart';
import 'package:paladinvpn/logic/ad_manager.dart';
import 'package:paladinvpn/components/clock_display.dart';

class ConnectButton extends StatefulWidget {
  const ConnectButton({super.key});

  @override
  State<ConnectButton> createState() => _ConnectButtonState();
}

class _ConnectButtonState extends State<ConnectButton>
    with SingleTickerProviderStateMixin {
  late AnimationController _pulseController;
  late Animation<double> _pulseAnim;

  static const Color _cyanGlow = Color(0xFF00E5FF);
  static const Color _bgDark = Color(0xFF1A202C);

  bool _inCooldown = false;

  @override
  void initState() {
    super.initState();
    _pulseController = AnimationController(
      vsync: this,
      duration: const Duration(milliseconds: 2000),
    );
    _pulseAnim = Tween<double>(begin: 0.0, end: 1.0).animate(
      CurvedAnimation(parent: _pulseController, curve: Curves.easeInOut),
    );

    // Listen to VpnConnection changes to drive the pulse animation
    // from outside build(), keeping build() pure.
    WidgetsBinding.instance.addPostFrameCallback((_) {
      context.read<VpnConnection>().addListener(_onVpnStateChanged);
    });
  }

  @override
  void dispose() {
    context.read<VpnConnection>().removeListener(_onVpnStateChanged);
    _pulseController.dispose();
    super.dispose();
  }

  void _onVpnStateChanged() {
    if (!mounted) return;
    final isConnected = context.read<VpnConnection>().status == VpnStatus.connected;
    if (isConnected && !_pulseController.isAnimating) {
      _pulseController.repeat(reverse: true);
    } else if (!isConnected && _pulseController.isAnimating) {
      _pulseController.stop();
      _pulseController.reset();
    }
  }

  void _handleTap() async {
    if (_inCooldown) return;

    final vpn = context.read<VpnConnection>();
    final timer = context.read<SessionTimer>();
    final ad = context.read<AdManager>();

    if (vpn.status == VpnStatus.connected) {
      // Disconnect — apply a short cooldown to prevent rapid toggle spam
      HapticFeedback.lightImpact();
      vpn.disconnect();
      timer.stop();
      setState(() => _inCooldown = true);
      Future.delayed(const Duration(seconds: 2), () {
        if (mounted) setState(() => _inCooldown = false);
      });
    } else if (vpn.status == VpnStatus.disconnected ||
               vpn.status == VpnStatus.error) {
      // Only attempt connect if an ad is loaded (or we're in dev/web mode)
      if (!ad.isAdLoaded && !ad.isAdLoading) {
        ad.preloadAd(); // trigger a reload in background
        ScaffoldMessenger.of(context).showSnackBar(
          const SnackBar(
            content: Text('Ad is not ready yet. Please wait a moment.'),
            duration: Duration(seconds: 2),
          ),
        );
        return;
      }
      HapticFeedback.lightImpact();
      final success = await vpn.connect();
      if (success) {
        timer.start('main_ad');
        // Only apply cooldown on success — failures clear immediately
        setState(() => _inCooldown = true);
        Future.delayed(const Duration(seconds: 5), () {
          if (mounted) setState(() => _inCooldown = false);
        });
      }
      // On failure the button stays active — the error message in
      // StatusText tells the user what went wrong.
    }
  }

  @override
  Widget build(BuildContext context) {
    return Consumer3<VpnConnection, SessionTimer, AdManager>(
      builder: (context, vpn, timer, ad, _) {
        final isConnected = vpn.status == VpnStatus.connected;
        final isConnecting = vpn.status == VpnStatus.connecting;

        return GestureDetector(
          onTap: (isConnecting || _inCooldown) ? null : _handleTap,
          child: AnimatedBuilder(
            animation: _pulseAnim,
            builder: (context, child) {
              final glowRadius = isConnected ? 20.0 + (_pulseAnim.value * 15.0) : 0.0;
              final glowAlpha = isConnected
                  ? ((0.3 + _pulseAnim.value * 0.2) * 255).round()
                  : 0;

              return Container(
                width: 260,
                height: 260,
                decoration: BoxDecoration(
                  shape: BoxShape.circle,
                  color: _bgDark,
                  border: Border.all(
                    color: _inCooldown
                        ? Colors.grey.withValues(alpha: 0.5)
                        : isConnected
                            ? _cyanGlow
                            : isConnecting
                                ? const Color.fromRGBO(0, 229, 255, 0.5)
                                : const Color.fromRGBO(74, 85, 104, 0.5),
                    width: isConnected && !_inCooldown ? 3 : 2,
                  ),
                  boxShadow: isConnected
                      ? [
                          BoxShadow(
                            color: Color.fromARGB(glowAlpha, 0, 229, 255),
                            blurRadius: glowRadius,
                            spreadRadius: glowRadius * 0.3,
                          ),
                        ]
                      : const [],
                ),
                child: Center(
                  child: isConnected
                    ? const ClockDisplay()
                    : _buildDisconnectedIcon(vpn.status, _inCooldown, ad),
                ),
              );
            },
          ),
        );
      },
    );
  }

  Widget _buildDisconnectedIcon(VpnStatus status, bool inCooldown, AdManager ad) {
    if (status == VpnStatus.connecting || status == VpnStatus.disconnecting) {
      return const SizedBox(
        width: 48,
        height: 48,
        child: CircularProgressIndicator(
          strokeWidth: 2.5,
          color: Color.fromRGBO(0, 229, 255, 0.7),
        ),
      );
    }
    // Determine the right label for the disconnected state
    String label;
    IconData icon;
    Color tint;

    if (inCooldown) {
      label = 'PLEASE WAIT';
      icon  = Icons.hourglass_bottom;
      tint  = Colors.grey;
    } else if (status == VpnStatus.error) {
      label = 'TRY AGAIN';
      icon  = Icons.refresh;
      tint  = const Color(0xFFEF5350); // soft red — something went wrong
    } else if (!ad.isAdLoaded && !ad.isAdLoading) {
      label = 'LOADING AD…';
      icon  = Icons.hourglass_empty;
      tint  = const Color.fromRGBO(74, 85, 104, 0.7);
    } else {
      label = 'TAP TO CONNECT';
      icon  = Icons.power_settings_new;
      tint  = const Color.fromRGBO(74, 85, 104, 0.7);
    }

    return Column(
      mainAxisSize: MainAxisSize.min,
      children: [
        Icon(icon, size: 64, color: tint),
        const SizedBox(height: 12),
        Text(
          label,
          style: TextStyle(
            color: tint,
            fontWeight: FontWeight.bold,
            letterSpacing: 1.5,
          ),
        ),
      ],
    );
  }
}