Colosseum/Assets/_Game/Scripts/Player/PlayerNetworkController.cs

using System;
using UnityEngine;
using Unity.Netcode;

using Colosseum.Abnormalities;
using Colosseum.Stats;
using Colosseum.Combat;
using Colosseum.Skills;

namespace Colosseum.Player
{
    /// <summary>
    /// 플레이어 네트워크 상태 관리 (HP, MP 등)
    /// </summary>
    public class PlayerNetworkController : NetworkBehaviour, IDamageable
    {
        [Header("References")]
        [Tooltip("CharacterStats 컴포넌트 (없으면 자동 검색)")]
        [SerializeField] private CharacterStats characterStats;

        [Tooltip("이상상태 관리자 (없으면 자동 검색)")]
        [SerializeField] private AbnormalityManager abnormalityManager;

        [Header("Shield")]
        [Tooltip("보호막 타입이 지정되지 않았을 때 사용할 기본 보호막 이상상태 데이터")]
        [SerializeField] private AbnormalityData shieldStateAbnormality;

        // 네트워크 동기화 변수
        private NetworkVariable<float> currentHealth = new NetworkVariable<float>(100f);
        private NetworkVariable<float> currentMana = new NetworkVariable<float>(50f);
        private NetworkVariable<bool> isDead = new NetworkVariable<bool>(false);
        private NetworkVariable<float> currentShield = new NetworkVariable<float>(0f);
        private readonly ShieldCollection shieldCollection = new ShieldCollection();

        public float Health => currentHealth.Value;
        public float Mana => currentMana.Value;
        public float Shield => currentShield.Value;
        public float MaxHealth => characterStats != null ? characterStats.MaxHealth : 100f;
        public float MaxMana => characterStats != null ? characterStats.MaxMana : 50f;
        public CharacterStats Stats => characterStats;

        // 체력/마나 변경 이벤트
        public event Action<float, float> OnHealthChanged; // (oldValue, newValue)
        public event Action<float, float> OnManaChanged; // (oldValue, newValue)
        public event Action<float, float> OnShieldChanged; // (oldValue, newValue)

        // 사망 이벤트
        public event Action<PlayerNetworkController> OnDeath;
        public event Action<bool> OnDeathStateChanged; // (isDead)
        public event Action<PlayerNetworkController> OnRespawned;

        // IDamageable 구현
        public float CurrentHealth => currentHealth.Value;
        public bool IsDead => isDead.Value;

        public override void OnNetworkSpawn()
        {
            // CharacterStats 참조 확인
            if (characterStats == null)
            {
                characterStats = GetComponent<CharacterStats>();
            }

            if (abnormalityManager == null)
            {
                abnormalityManager = GetComponent<AbnormalityManager>();
            }

            // 네트워크 변수 변경 콜백 등록
            currentHealth.OnValueChanged += HandleHealthChanged;
            currentMana.OnValueChanged += HandleManaChanged;
            currentShield.OnValueChanged += HandleShieldChanged;
            isDead.OnValueChanged += HandleDeathStateChanged;

            // 초기화
            if (IsServer)
            {
                currentHealth.Value = MaxHealth;
                currentMana.Value = MaxMana;
                isDead.Value = false;
                RefreshShieldState();
            }
        }

        private void Update()
        {
            if (!IsServer || isDead.Value)
                return;

            if (shieldCollection.Tick(Time.deltaTime))
            {
                RefreshShieldState();
            }
        }

        public override void OnNetworkDespawn()
        {
            // 콜백 해제
            currentHealth.OnValueChanged -= HandleHealthChanged;
            currentMana.OnValueChanged -= HandleManaChanged;
            currentShield.OnValueChanged -= HandleShieldChanged;
            isDead.OnValueChanged -= HandleDeathStateChanged;
        }

        private void HandleHealthChanged(float oldValue, float newValue)
        {
            OnHealthChanged?.Invoke(oldValue, newValue);
        }

        private void HandleManaChanged(float oldValue, float newValue)
        {
            OnManaChanged?.Invoke(oldValue, newValue);
        }

        private void HandleShieldChanged(float oldValue, float newValue)
        {
            OnShieldChanged?.Invoke(oldValue, newValue);
        }

        private void HandleDeathStateChanged(bool oldValue, bool newValue)
        {
            OnDeathStateChanged?.Invoke(newValue);
        }

        /// <summary>
        /// 대미지 적용 (서버에서만 실행)
        /// </summary>
        [Rpc(SendTo.Server)]
        public void TakeDamageRpc(float damage)
        {
            ApplyDamageInternal(damage, null);
        }

        /// <summary>
        /// 마나 소모 (서버에서만 실행)
        /// </summary>
        [Rpc(SendTo.Server)]
        public void UseManaRpc(float amount)
        {
            if (isDead.Value) return;

            currentMana.Value = Mathf.Max(0f, currentMana.Value - amount);
        }

        /// <summary>
        /// 체력 회복 (서버에서만 실행)
        /// </summary>
        [Rpc(SendTo.Server)]
        public void RestoreHealthRpc(float amount)
        {
            if (isDead.Value) return;

            currentHealth.Value = Mathf.Min(MaxHealth, currentHealth.Value + amount);
        }

        /// <summary>
        /// 마나 회복 (서버에서만 실행)
        /// </summary>
        [Rpc(SendTo.Server)]
        public void RestoreManaRpc(float amount)
        {
            if (isDead.Value) return;

            currentMana.Value = Mathf.Min(MaxMana, currentMana.Value + amount);
        }

        /// <summary>
        /// 사망 애니메이션 재생 (모든 클라이언트에서 실행)
        /// </summary>
        [Rpc(SendTo.Everyone)]
        private void PlayDeathAnimationRpc()
        {
            var animator = GetComponentInChildren<Animator>();
            if (animator != null)
            {
                animator.SetTrigger("Die");
            }
        }

        /// <summary>
        /// 사망 처리 (서버에서만 실행)
        /// </summary>
        private void HandleDeath()
        {
            if (isDead.Value) return;

            isDead.Value = true;
            shieldCollection.Clear();
            RefreshShieldState();

            // 사망 시 활성 이상 상태를 정리해 리스폰 시 잔존하지 않게 합니다.
            if (abnormalityManager != null)
            {
                abnormalityManager.RemoveAllAbnormalities();
            }

            // 이동 비활성화
            var movement = GetComponent<PlayerMovement>();
            if (movement != null)
            {
                movement.ClearForcedMovement();
                movement.enabled = false;
            }

            var hitReactionController = GetComponent<HitReactionController>();
            if (hitReactionController != null)
            {
                hitReactionController.ClearHitReactionState();
            }

            var threatController = GetComponent<ThreatController>();
            if (threatController != null)
            {
                threatController.ClearThreatModifiers();
            }

            // 스킬 입력 비활성화
            var skillInput = GetComponent<PlayerSkillInput>();
            if (skillInput != null)
            {
                skillInput.enabled = false;
            }

            // 실행 중인 스킬 즉시 취소
            var skillController = GetComponent<SkillController>();
            if (skillController != null)
            {
                skillController.CancelSkill(SkillCancelReason.Death);
            }

            // 모든 클라이언트에서 사망 애니메이션 재생
            PlayDeathAnimationRpc();

            // 사망 이벤트 발생
            OnDeath?.Invoke(this);

            Debug.Log($"[Player] Player {OwnerClientId} died!");
        }

        /// <summary>
        /// 리스폰 (서버에서만 실행)
        /// </summary>
        public void Respawn()
        {
            if (!IsServer) return;

            if (abnormalityManager != null)
            {
                abnormalityManager.RemoveAllAbnormalities();
            }

            isDead.Value = false;
            currentHealth.Value = MaxHealth;
            currentMana.Value = MaxMana;
            shieldCollection.Clear();
            RefreshShieldState();

            // 이동 재활성화
            var movement = GetComponent<PlayerMovement>();
            if (movement != null)
            {
                movement.ClearForcedMovement();
                movement.enabled = true;
            }

            var hitReactionController = GetComponent<HitReactionController>();
            if (hitReactionController != null)
            {
                hitReactionController.ClearHitReactionState();
            }

            var threatController = GetComponent<ThreatController>();
            if (threatController != null)
            {
                threatController.ClearThreatModifiers();
            }

            // 스킬 입력 재활성화
            var skillInput = GetComponent<PlayerSkillInput>();
            if (skillInput != null)
            {
                skillInput.enabled = true;
            }

            // 애니메이션 리셋
            var animator = GetComponentInChildren<Animator>();
            if (animator != null)
            {
                animator.Rebind();
            }

            var skillController = GetComponent<SkillController>();
            if (skillController != null)
            {
                skillController.CancelSkill(SkillCancelReason.Respawn);
            }

            OnRespawned?.Invoke(this);

            Debug.Log($"[Player] Player {OwnerClientId} respawned!");
        }

        #region IDamageable
        /// <summary>
        /// 대미지 적용 (서버에서만 호출)
        /// </summary>
        public float TakeDamage(float damage, object source = null)
        {
            return ApplyDamageInternal(damage, source);
        }

        /// <summary>
        /// 체력 회복 (서버에서만 호출)
        /// </summary>
        public float Heal(float amount)
        {
            if (!IsServer || isDead.Value) return 0f;

            float actualHeal = Mathf.Min(amount, MaxHealth - currentHealth.Value);
            currentHealth.Value = Mathf.Min(MaxHealth, currentHealth.Value + amount);

            return actualHeal;
        }

        /// <summary>
        /// 보호막을 적용합니다.
        /// </summary>
        public float ApplyShield(float amount, float duration, AbnormalityData shieldAbnormality = null, GameObject source = null)
        {
            if (!IsServer || isDead.Value || amount <= 0f)
                return 0f;

            AbnormalityData shieldType = shieldAbnormality != null ? shieldAbnormality : shieldStateAbnormality;
            float actualAppliedShield = shieldCollection.ApplyShield(shieldType, amount, duration, source);
            RefreshShieldState();
            return actualAppliedShield;
        }

        private bool IsDamageImmune()
        {
            return abnormalityManager != null && abnormalityManager.IsInvincible;
        }

        private float GetIncomingDamageMultiplier()
        {
            if (abnormalityManager == null)
                return 1f;

            return Mathf.Max(0f, abnormalityManager.IncomingDamageMultiplier);
        }

        private float ConsumeShield(float incomingDamage)
        {
            if (incomingDamage <= 0f || currentShield.Value <= 0f)
                return incomingDamage;

            float remainingDamage = shieldCollection.ConsumeDamage(incomingDamage);
            RefreshShieldState();
            return remainingDamage;
        }

        private float ApplyDamageInternal(float damage, object source)
        {
            if (!IsServer || isDead.Value || IsDamageImmune())
                return 0f;

            float finalDamage = damage * GetIncomingDamageMultiplier();
            float mitigatedDamage = ConsumeShield(finalDamage);
            float actualDamage = Mathf.Min(mitigatedDamage, currentHealth.Value);
            currentHealth.Value = Mathf.Max(0f, currentHealth.Value - actualDamage);

            CombatBalanceTracker.RecordDamage(source as GameObject, gameObject, actualDamage);

            if (currentHealth.Value <= 0f)
            {
                HandleDeath();
            }

            return actualDamage;
        }

        private void RefreshShieldState()
        {
            currentShield.Value = shieldCollection.TotalAmount;

            ShieldAbnormalityUtility.SyncShieldAbnormalities(
                abnormalityManager,
                shieldCollection.ActiveShields,
                gameObject);
        }
        #endregion
    }
}