using RimWorld;
using Verse;
using System.Collections.Generic;

namespace WulaFallenEmpire
{
    public class HediffGiver_NonPlayerFaction : HediffGiver
    {
        // 显式重新定义这些字段，确保 XML 解析器能找到它们
        public float mtbDays;
        public new HediffDef hediff;
        
        // 新增：目标派系列表
        public List<FactionDef> targetFactions;
        
        // 新增：是否排除玩家派系（默认为true）
        public bool excludePlayerFaction = true;
        
        // 新增：是否排除囚犯（默认为true）
        public bool excludePrisoners = true;

        // 新增：记录已经处理过的 pawn，避免重复处理
        private HashSet<Pawn> processedPawns = new HashSet<Pawn>();

        public override void OnIntervalPassed(Pawn pawn, Hediff cause)
        {
            // 检查 pawn 是否已经处理过
            if (processedPawns.Contains(pawn))
            {
                // 如果已经处理过，检查 hediff 是否还存在
                Hediff existingHediff = pawn.health?.hediffSet?.GetFirstHediffOfDef(this.hediff);
                if (existingHediff != null)
                {
                    // hediff 还存在，不需要再次处理
                    return;
                }
                else
                {
                    // hediff 被移除了，从记录中移除这个 pawn
                    processedPawns.Remove(pawn);
                }
            }

            // 检查派系条件
            if (ShouldHaveHediff(pawn))
            {
                // 检查是否已经有这个 hediff
                Hediff existing = pawn.health?.hediffSet?.GetFirstHediffOfDef(this.hediff);
                if (existing == null)
                {
                    // 给予 hediff
                    HealthUtility.AdjustSeverity(pawn, this.hediff, 1.0f);
                    // 标记为已处理
                    processedPawns.Add(pawn);
                    Log.Message($"Added hediff {this.hediff.defName} to pawn {pawn.Label}");
                }
            }
            else
            {
                // 移除 hediff
                if (RemoveHediffIfExists(pawn))
                {
                    // 如果成功移除了 hediff，也从记录中移除
                    processedPawns.Remove(pawn);
                }
            }
        }

        private bool ShouldHaveHediff(Pawn pawn)
        {
            // 检查派系是否存在
            if (pawn.Faction == null)
                return false;
                
            // 检查是否排除玩家派系
            if (excludePlayerFaction && pawn.Faction == Faction.OfPlayer)
                return false;
                
            // 检查是否排除囚犯
            if (excludePrisoners && pawn.IsPrisonerOfColony)
                return false;
            
            // 检查目标派系
            if (targetFactions != null && targetFactions.Count > 0)
            {
                // 如果指定了目标派系，只给这些派系添加 hediff
                return targetFactions.Contains(pawn.Faction.def);
            }
            else
            {
                // 如果没有指定目标派系，保持原来的行为：给所有非玩家派系添加
                return pawn.Faction != Faction.OfPlayer;
            }
        }

        private bool RemoveHediffIfExists(Pawn pawn)
        {
            Hediff existing = pawn.health?.hediffSet?.GetFirstHediffOfDef(this.hediff);
            if (existing != null)
            {
                pawn.health.RemoveHediff(existing);
                Log.Message($"Removed hediff {this.hediff.defName} from pawn {pawn.Label}");
                return true;
            }
            return false;
        }

        // 新增：在 pawn 死亡或被销毁时清理记录
        public override void Notify_PawnDied(Pawn pawn, DamageInfo? dinfo)
        {
            base.Notify_PawnDied(pawn, dinfo);
            processedPawns.Remove(pawn);
        }

        public override void Notify_PawnDespawned(Pawn pawn, Map map)
        {
            base.Notify_PawnDespawned(pawn, map);
            processedPawns.Remove(pawn);
        }
    }
}