VigorModSystem.cs

using System;
using System.Collections.Concurrent;
using System.Collections.Generic;
using Vintagestory.API.Client;
using Vintagestory.API.Common;
using Vintagestory.API.Datastructures;
using Vintagestory.API.Server;
using Vigor.API;
using Vigor.Behaviors;
using Vigor.Config;
using Vigor.Hud;
using Vigor.Utils;
using Vigor.Core;
using Vintagestory.Client.NoObf;

namespace Vigor
{
    public class VigorModSystem : ModSystem
    {
        private const string NETWORK_CHANNEL = "vigor:statesync";
        private const string ConfigLibConfigSavedEvent = "configlib:vigor:config-saved";
        private const string ConfigLibConfigReloadEvent = "configlib:config-reload";
        
        private ICoreAPI _api;
        private ICoreClientAPI _capi;
        private ICoreServerAPI _sapi;
        private HudVigorBar _vigorHud;
        private HudVigorDebug _debugHud;
        private long _syncTimerId;
        private long _drainTickId;
        private long _diagnosticsAutoDumpListenerId;
        private bool _initialDiagnosticsSnapshotCaptured;
        private float _timeSinceLastAutoDiagnosticsSnapshot;
        private IClientNetworkChannel _clientNetworkChannel;
        private IServerNetworkChannel _serverNetworkChannel;
        private VigorAPI _activeStaminaDrainHandler;
        
        // Dictionary to store synchronized client-side stamina state by player UID
        private readonly ConcurrentDictionary<string, StaminaStatePacket> _clientStaminaState = new ConcurrentDictionary<string, StaminaStatePacket>();

        public event Action<StaminaStatePacket> LocalPlayerStaminaStateUpdated;
        
        public static VigorModSystem Instance { get; private set; }
        public VigorConfig CurrentConfig { get; private set; }
        public string ModId => Mod.Info.ModID;
        public ILogger Logger => Mod.Logger;
        
        // Public API instances that other mods can access
        public IVigorAPI API { get; private set; } // General API - maintains backward compatibility
        public IVigorAPI ClientAPI { get; private set; } // Client-specific API
        public IVigorAPI ServerAPI { get; private set; } // Server-specific API

        public override void Start(ICoreAPI api)
        {
            base.Start(api);
            _api = api;
            Instance = this;
            LoadConfig(api);
            RegisterConfigReloadListeners(api);
            
            // Create general API instance for backward compatibility
            // Note: This instance should be avoided in client/server contexts
            API = new VigorAPI(api);
            
            // Register behavior
            api.RegisterEntityBehaviorClass("vigor:vigorstamina", typeof(Behaviors.EntityBehaviorVigorStamina));
            
            // Register network channel for stamina state synchronization
            api.Network.RegisterChannel(NETWORK_CHANNEL)
                .RegisterMessageType<StaminaStatePacket>();

            if (CurrentConfig.DebugMode)
            {
                Logger.Debug($"[{ModId}] Registered network channel '{NETWORK_CHANNEL}'");
                Logger.Debug($"[{ModId}] {api.Side} API available via ModLoader.GetModSystem<VigorModSystem>().API");
            }
        }

        // Added property to check for HydrateOrDiedrate mod
        public bool IsHydrateOrDiedrateLoaded { get; private set; }
        
        public override void StartClientSide(ICoreClientAPI api)
        {
            base.StartClientSide(api);
            _capi = api;
            
            // Create client-specific API instance
            ClientAPI = new VigorAPI(api);
            
            // Set up client network channel and register packet handler
            _clientNetworkChannel = api.Network.GetChannel(NETWORK_CHANNEL);
            _clientNetworkChannel.SetMessageHandler<StaminaStatePacket>(OnStaminaStatePacket);
            if (CurrentConfig.DebugMode)
            {
                Logger.Debug($"[{ModId}] Client API initialized");
                Logger.Debug($"[{ModId}] Client stamina sync handler registered");
            }
            
            // Check if HydrateOrDiedrate mod is installed
            IsHydrateOrDiedrateLoaded = api.ModLoader.IsModEnabled("hydrateordiedrate");
            
            if (IsHydrateOrDiedrateLoaded && CurrentConfig.DebugMode)
            {
                Logger.Notification($"[{ModId}] HydrateOrDiedrate mod detected, will adjust HUD position.");
            }

            _vigorHud = new HudVigorBar(api);
            _debugHud = new HudVigorDebug(api);
            api.Gui.RegisterDialog(_vigorHud);
            api.Gui.RegisterDialog(_debugHud);

            api.Input.RegisterHotKey("vigordebug", "Vigor: Toggle Debug Info", GlKeys.F8, HotkeyType.GUIOrOtherControls);
            api.Input.SetHotKeyHandler("vigordebug", OnToggleDebugHud);
            api.Input.RegisterHotKey("vigordiagdump", "Vigor: Dump Diagnostics Snapshot", GlKeys.F6, HotkeyType.GUIOrOtherControls);
            api.Input.SetHotKeyHandler("vigordiagdump", OnDumpDiagnostics);

            if (CurrentConfig.EnableDiagnosticsSnapshots)
            {
                _diagnosticsAutoDumpListenerId = api.Event.RegisterGameTickListener(OnDiagnosticsTick, 1000);
            }

            if (CurrentConfig.EnableDiagnosticsSnapshots)
            {
                if (CurrentConfig.EnableAutomaticDiagnosticsSnapshots)
                {
                    Logger.Notification($"Stamina systems ready. Diagnostics enabled. Press F6 for a manual snapshot. Automatic snapshots every {Math.Max(1, CurrentConfig.DiagnosticsSnapshotIntervalMinutes)} minutes.");
                }
                else
                {
                    Logger.Notification("Stamina systems ready. Diagnostics enabled. Press F6 for a manual snapshot.");
                }
            }
            else
            {
                Logger.Notification("Stamina systems ready. Diagnostics disabled in vigor.json.");
            }
        }

        private void RegisterConfigReloadListeners(ICoreAPI api)
        {
            api.Event.RegisterEventBusListener(OnConfigLibConfigSaved, filterByEventName: ConfigLibConfigSavedEvent);
            api.Event.RegisterEventBusListener(OnConfigLibConfigReload, filterByEventName: ConfigLibConfigReloadEvent);
        }

        private void OnConfigLibConfigSaved(string eventName, ref EnumHandling handling, IAttribute data)
        {
            if (!IsVigorConfigEvent(data))
            {
                return;
            }

            ReloadConfigAndApplyRuntimeChanges();
        }

        private void OnConfigLibConfigReload(string eventName, ref EnumHandling handling, IAttribute data)
        {
            if (!IsVigorConfigEvent(data))
            {
                return;
            }

            ReloadConfigAndApplyRuntimeChanges();
        }

        private bool IsVigorConfigEvent(IAttribute data)
        {
            return (data as ITreeAttribute)?.GetAsString("domain") == ModId;
        }

        private void ReloadConfigAndApplyRuntimeChanges()
        {
            if (_api == null)
            {
                return;
            }

            LoadConfig(_api);

            if (_sapi != null)
            {
                RefreshServerSyncTimer();
            }

            if (_capi != null)
            {
                RefreshClientRuntimeConfigState();
            }
        }

        private void RefreshServerSyncTimer()
        {
            if (_sapi == null)
            {
                return;
            }

            if (_syncTimerId != 0)
            {
                _sapi.Event.UnregisterGameTickListener(_syncTimerId);
                _syncTimerId = 0;
            }

            float syncIntervalSeconds = Math.Max(0.03f, CurrentConfig.StaminaSyncIntervalSeconds);
            _syncTimerId = _sapi.Event.RegisterGameTickListener(SyncStaminaState, (int)(syncIntervalSeconds * 1000));
        }

        private void RefreshClientRuntimeConfigState()
        {
            if (_capi == null)
            {
                return;
            }

            RefreshClientDiagnosticsListener();
            RecreateClientHudElements();
        }

        private void RefreshClientDiagnosticsListener()
        {
            if (_capi == null)
            {
                return;
            }

            if (_diagnosticsAutoDumpListenerId != 0)
            {
                _capi.Event.UnregisterGameTickListener(_diagnosticsAutoDumpListenerId);
                _diagnosticsAutoDumpListenerId = 0;
            }

            _initialDiagnosticsSnapshotCaptured = false;
            _timeSinceLastAutoDiagnosticsSnapshot = 0f;

            if (CurrentConfig.EnableDiagnosticsSnapshots)
            {
                _diagnosticsAutoDumpListenerId = _capi.Event.RegisterGameTickListener(OnDiagnosticsTick, 1000);
            }
        }

        private void RecreateClientHudElements()
        {
            if (_capi == null)
            {
                return;
            }

            DisposeClientDialog(_vigorHud);
            DisposeClientDialog(_debugHud);

            _vigorHud = new HudVigorBar(_capi);
            _debugHud = new HudVigorDebug(_capi);
            _capi.Gui.RegisterDialog(_vigorHud, _debugHud);
        }

        private void DisposeClientDialog(GuiDialog dialog)
        {
            if (dialog == null || _capi == null)
            {
                return;
            }

            dialog.TryClose();
            if (_capi.World is ClientMain clientMain)
            {
                clientMain.UnregisterDialog(dialog);
            }

            dialog.Dispose();
        }

        private bool OnToggleDebugHud(KeyCombination comb)
        {
            _debugHud?.Toggle();
            return true;
        }

        private bool OnDumpDiagnostics(KeyCombination comb)
        {
            try
            {
                if (!CurrentConfig.EnableDiagnosticsSnapshots)
                {
                    _capi?.Logger.Notification("Diagnostics snapshots are disabled in the Vigor config.");
                    return true;
                }

                var filePath = DumpDiagnosticsSnapshot("manual");
                _capi?.Logger.Notification($"Diagnostics snapshot written to {filePath}");
            }
            catch (Exception ex)
            {
                _capi?.Logger.Error($"Failed to dump diagnostics snapshot: {ex}");
            }

            return true;
        }

        private void OnDiagnosticsTick(float dt)
        {
            try
            {
                if (_capi?.World?.Player?.Entity == null) return;
                if (!CurrentConfig.EnableDiagnosticsSnapshots) return;

                if (!_initialDiagnosticsSnapshotCaptured && CurrentConfig.CaptureInitialDiagnosticsSnapshot)
                {
                    var initialPath = DumpDiagnosticsSnapshot("initial");
                    _initialDiagnosticsSnapshotCaptured = true;
                    _timeSinceLastAutoDiagnosticsSnapshot = 0f;
                    _capi.Logger.Notification($"Initial diagnostics snapshot written to {initialPath}");
                    return;
                }

                if (!CurrentConfig.EnableAutomaticDiagnosticsSnapshots) return;

                _timeSinceLastAutoDiagnosticsSnapshot += dt;
                float intervalSeconds = Math.Max(1, CurrentConfig.DiagnosticsSnapshotIntervalMinutes) * 60f;
                if (_timeSinceLastAutoDiagnosticsSnapshot < intervalSeconds) return;

                _timeSinceLastAutoDiagnosticsSnapshot = 0f;
                var filePath = DumpDiagnosticsSnapshot("auto");
                _capi.Logger.Notification($"Automatic diagnostics snapshot written to {filePath}");
            }
            catch (Exception ex)
            {
                _capi?.Logger.Error($"Failed to write automatic diagnostics snapshot: {ex}");
            }
        }

        private string DumpDiagnosticsSnapshot(string reason)
        {
            VigorDiagnostics.SetGauge("clientStaminaState.count", _clientStaminaState.Count);
            return VigorDiagnostics.DumpSnapshotToFile(ModId, reason);
        }
        
        /// <summary>
        /// Client-side handler for stamina state packet
        /// </summary>
        private void OnStaminaStatePacket(StaminaStatePacket packet)
        {
            if (_capi == null)
            {
                Logger.Error($"[{ModId}] OnStaminaStatePacket: _capi is null, cannot process packet!");
                return;
            }
            
            try
            {
                // Always log first packet received for own player (critical for debugging sync issues)
                bool isFirstPacket = !_clientStaminaState.ContainsKey(packet.PlayerUID);
                bool isOwnPlayer = _capi.World.Player != null && _capi.World.Player.PlayerUID == packet.PlayerUID;
                
                // Store the updated state in our client-side dictionary
                _clientStaminaState[packet.PlayerUID] = packet;
                VigorDiagnostics.Increment("network.packetReceived");
                VigorDiagnostics.SetGauge("clientStaminaState.count", _clientStaminaState.Count);
                
                if (isFirstPacket && isOwnPlayer)
                {
                    // Use Error level for the first packet to ensure it's visible in logs
                    _capi.Logger.Error($"[{ModId}] FIRST stamina state packet received for own player: {packet.CurrentStamina}/{packet.MaxStamina}, Exhausted: {packet.IsExhausted}");
                }
                else if (CurrentConfig.DebugMode && isOwnPlayer)
                {
                    // Use regular debug for subsequent packets
                    _capi.Logger.Debug($"[{ModId}] Received stamina state: {packet.CurrentStamina}/{packet.MaxStamina}, Exhausted: {packet.IsExhausted}");
                }
                
                if (isOwnPlayer)
                {
                    VigorDiagnostics.Increment("network.packetLocalPlayerCached");
                    LocalPlayerStaminaStateUpdated?.Invoke(packet);

                    if (CurrentConfig.DebugMode)
                    {
                        _capi.Logger.Debug($"[{ModId}] Cached local stamina packet without mirroring to WatchedAttributes");
                    }
                }
            }
            catch (Exception ex)
            {
                _capi.Logger.Error($"[{ModId}] Error processing stamina state packet: {ex.Message}\nStack trace: {ex.StackTrace}");
            }
        }
        
        /// <summary>
        /// Gets the synchronized client-side stamina state for a player
        /// </summary>
        /// <param name="playerUID">The player UID</param>
        /// <returns>The stamina state packet, or null if not found</returns>
        public StaminaStatePacket GetClientStaminaState(string playerUID)
        {
            if (string.IsNullOrEmpty(playerUID)) return null;
            
            if (_clientStaminaState.TryGetValue(playerUID, out var state))
            {
                return state;
            }
            
            return null;
        }
        
        /// <summary>
        /// Server-side method to sync stamina state to clients
        /// </summary>
        private void SyncStaminaState(float dt)
        {
            if (_sapi == null) 
            {
                Logger.Error($"[{ModId}] SyncStaminaState: _sapi is null! Stamina sync to clients will not work.");
                return;
            }
            
            try
            {
                // Log channel status
                if (_serverNetworkChannel == null) 
                {
                    Logger.Error($"[{ModId}] SyncStaminaState: _serverNetworkChannel is null! Network not initialized properly.");
                    return;
                }
                
                if (CurrentConfig.DebugMode)
        {
            Logger.Debug($"[{ModId}] SyncStaminaState: Syncing stamina data for {_sapi.World.AllOnlinePlayers.Length} online players");
        }
                
                // Sync stamina state for all online players
                foreach (var player in _sapi.World.AllOnlinePlayers)
                {
                    // Get player entity
                    var playerEntity = player.Entity;
                    if (playerEntity == null) 
                    {
                        Logger.Warning($"[{ModId}] SyncStaminaState: Entity null for player {player.PlayerName}");
                        continue;
                    }
                    
                    // Skip very-early entities not fully initialized
                    if (playerEntity.Api == null || playerEntity.World == null)
                    {
                        if (CurrentConfig.DebugMode)
                        {
                            Logger.Debug($"[{ModId}] SyncStaminaState: Entity API/World not ready for player {player.PlayerName}, skipping this tick");
                        }
                        continue;
                    }

                    // Get stamina behavior (guard against rare NRE during engine init)
                    EntityBehaviorVigorStamina staminaBehavior = null;
                    try
                    {
                        staminaBehavior = playerEntity.GetBehavior<EntityBehaviorVigorStamina>();
                    }
                    catch (Exception ex)
                    {
                        if (CurrentConfig.DebugMode)
                        {
                            Logger.Debug($"[{ModId}] SyncStaminaState: GetBehavior threw early for {player.PlayerName}: {ex.Message}");
                        }
                        continue;
                    }

                    if (staminaBehavior == null)
                    {
                        if (CurrentConfig.DebugMode)
                        {
                            Logger.Debug($"[{ModId}] SyncStaminaState: Stamina behavior not yet attached for {player.PlayerName}");
                        }
                        continue;
                    }
                    
                    // Create packet with current state
                    var packet = new StaminaStatePacket
                    {
                        PlayerUID = player.PlayerUID,
                        CurrentStamina = staminaBehavior.CurrentStamina,
                        MaxStamina = staminaBehavior.MaxStamina,
                        IsExhausted = staminaBehavior.IsExhausted
                    };
                    
                    // Send to player (cast to IServerPlayer is safe since we're in server-side code)
                    var serverPlayer = player as IServerPlayer;
                    if (serverPlayer == null)
                    {
                        Logger.Warning($"[{ModId}] SyncStaminaState: Failed to cast player {player.PlayerName} to IServerPlayer");
                        continue;
                    }
                    
                    _serverNetworkChannel.SendPacket(packet, serverPlayer);
                    VigorDiagnostics.Increment("network.packetSent");
                    
                    // Log more frequently when in debug mode to identify synchronization issues
                    if (CurrentConfig.DebugMode && (_sapi.World.Rand.NextDouble() < 0.2)) // Increased log frequency to 20%
                    {
                        _sapi.Logger.Debug($"[{ModId}] SyncStaminaState: Sent packet to {player.PlayerName}: {packet.CurrentStamina}/{packet.MaxStamina}, Exhausted: {packet.IsExhausted}");
                    }
                }
            }
            catch (Exception ex)
            {
                _sapi.Logger.Error($"[{ModId}] Error syncing stamina state: {ex.Message}\nStack trace: {ex.StackTrace}");
            }
        }
        
        /// <summary>
        /// Cleanup when player disconnects
        /// </summary>
        private void OnPlayerDisconnect(IServerPlayer player)
        {
            if (player != null)
            {
                // No need to sync data for disconnected players
                _sapi?.Logger.Debug($"[{ModId}] Player {player.PlayerName} disconnected, cleaning up sync state");
            }
        }

        public override void StartServerSide(ICoreServerAPI api)
        {
            base.StartServerSide(api);
            _sapi = api;
            
            // Create server-specific API instance
            ServerAPI = new VigorAPI(api);
            
            // Set up server network channel
            _serverNetworkChannel = api.Network.GetChannel(NETWORK_CHANNEL);
            if (_serverNetworkChannel == null)
            {
                Logger.Error($"[{ModId}] Failed to get network channel for stamina state sync");
            }
            else if (CurrentConfig.DebugMode)
            {
                Logger.Debug($"[{ModId}] Server API initialized");
                Logger.Debug($"[{ModId}] Server network channel ready");
            }
            
            // Register periodic sync timer
            float syncIntervalSeconds = Math.Max(0.03f, CurrentConfig.StaminaSyncIntervalSeconds);
            _syncTimerId = api.Event.RegisterGameTickListener(SyncStaminaState, (int)(syncIntervalSeconds * 1000));
            if (CurrentConfig.DebugMode)
            {
                Logger.Debug($"[{ModId}] Registered stamina state sync timer ({syncIntervalSeconds}s)");
            }
            
            // Listen for player disconnect to clean up sync state
            api.Event.PlayerDisconnect += OnPlayerDisconnect;
            
            // Listen for player join to attach behavior
            api.Event.PlayerJoin += OnPlayerJoin;
            
            Logger.Notification($"[{ModId}] Server stamina synchronization ready.");
        }

        /// <summary>
        /// Called when a player joins the server, attaches the stamina behavior to them
        /// </summary>
        private void OnPlayerJoin(IServerPlayer player)
        {
            try
            {
                // Delay attachment slightly to ensure entity is fully initialized
                _sapi?.Event.RegisterCallback(dt =>
                {
                    try
                    {
                        if (player?.Entity == null) return;
                        var entity = player.Entity;
                        if (entity.Api == null || entity.World == null) return;

                        if (entity.GetBehavior<EntityBehaviorVigorStamina>() == null)
                        {
                            entity.AddBehavior(new EntityBehaviorVigorStamina(entity));
                            Logger.Event($"[{ModId}] Attached vigor stamina behavior to player {player.PlayerName}");
                        }
                    }
                    catch (Exception ex)
                    {
                        Logger.Error($"[{ModId}] Delayed attach error: {ex.Message}\n{ex.StackTrace}");
                    }
                }, 100);
            }
            catch (Exception ex)
            {
                Logger.Error($"[{ModId}] Error attaching stamina behavior to player: {ex.Message}\n{ex.StackTrace}");
            }
        }

        /// <summary>
        /// Process continuous stamina drain for all active drains
        /// </summary>
        private void ProcessStaminaDrainTick(float dt)
        {
            if (_sapi == null || _activeStaminaDrainHandler == null) return;
            
            try
            {
                // Process all active stamina drains
                _activeStaminaDrainHandler.ProcessActiveStaminaDrains(dt);
            }
            catch (Exception ex)
            {
                Logger.Error($"[{ModId}] Error processing stamina drains: {ex}");
            }
        }
        
        /// <summary>
        /// Checks if the continuous stamina drain is active
        /// </summary>
        /// <returns>True if the drain is currently active</returns>
        public bool IsStaminaDrainActive()
        {
            Logger.Debug($"[{ModId}] IsStaminaDrainActive called, current _drainTickId: {_drainTickId}");
            return _drainTickId != 0;
        }
        
        /// <summary>
        /// Starts the continuous stamina drain tick processing
        /// </summary>
        /// <param name="handler">The VigorAPI instance that will handle the drain processing</param>
        public void StartStaminaDrainTick(VigorAPI handler)
        {
            Logger.Debug($"[{ModId}] StartStaminaDrainTick called with handler: {handler}");
            
            if (_sapi == null)
            {
                Logger.Error($"[{ModId}] Cannot start stamina drain tick - _sapi is null!");
                return;
            }
            
            // Don't register multiple times
            if (IsStaminaDrainActive())
            {
                Logger.Debug($"[{ModId}] Drain tick already active, updating handler only");
                _activeStaminaDrainHandler = handler; // Update handler if needed
                return;
            }
            
            _activeStaminaDrainHandler = handler;
            
            // Register tick processor at 4 times per second (250ms)
            Logger.Debug($"[{ModId}] About to register game tick listener for stamina drain");
            _drainTickId = _sapi.Event.RegisterGameTickListener(dt => ProcessStaminaDrainTick(dt), 250);
            Logger.Event($"[{ModId}] Started continuous stamina drain processing with tickId: {_drainTickId}");
        }
        
        /// <summary>
        /// Stops the continuous stamina drain tick processing
        /// </summary>
        public void StopStaminaDrainTick()
        {
            if (_sapi == null || !IsStaminaDrainActive()) return;
            
            _sapi.Event.UnregisterGameTickListener(_drainTickId);
            _drainTickId = 0;
            _activeStaminaDrainHandler = null;
            Logger.Event($"[{ModId}] Stopped continuous stamina drain processing");
        }
        
        public void LoadConfig(ICoreAPI api)
        {
            try
            {
                CurrentConfig = api.LoadModConfig<VigorConfig>($"{ModId}.json");

                if (CurrentConfig == null)
                {
                    Logger.Notification($"[{ModId}] No config found, creating default.");
                    CurrentConfig = new VigorConfig();
                    api.StoreModConfig(CurrentConfig, $"{ModId}.json");
                }
                else
                {
                    // Use a loud, unconditional log level to ensure this message always appears.
                if (CurrentConfig.DebugMode)
                {
                    Logger.Notification("Debug mode enabled.");
                }
                }
            }
            catch (Exception e)
            {
                Logger.Error($"[{ModId}] Failed to load or create config: {e}");
                CurrentConfig = new VigorConfig();
            }
        }
    }
}