using HarmonyLib;
using RimWorld;
using Verse;

namespace WulaFallenEmpire
{
    public class CompProperties_GiveHediffOnShot : CompProperties
    {
        public HediffDef hediffDef;
        public float severityToAdd = 0.1f;

        public CompProperties_GiveHediffOnShot()
        {
            compClass = typeof(CompGiveHediffOnShot);
        }
    }

    public class CompGiveHediffOnShot : ThingComp
    {
        public CompProperties_GiveHediffOnShot Props => (CompProperties_GiveHediffOnShot)props;
    }

    // Patch 1: For all standard projectile verbs.
    [HarmonyPatch(typeof(Verb_LaunchProjectile), "TryCastShot")]
    public static class Patch_Verb_LaunchProjectile_TryCastShot
    {
        public static void Postfix(Verb_LaunchProjectile __instance, bool __result)
        {
            if (!__result) return;
            if (__instance.CasterPawn == null || __instance.EquipmentSource == null) return;

            CompGiveHediffOnShot comp = __instance.EquipmentSource.GetComp<CompGiveHediffOnShot>();
            if (comp == null || comp.Props.hediffDef == null) return;

            Hediff hediff = __instance.CasterPawn.health.GetOrAddHediff(comp.Props.hediffDef);
            hediff.Severity += comp.Props.severityToAdd;

            var disappearsComp = hediff.TryGetComp<HediffComp_Disappears>();
            disappearsComp?.ResetElapsedTicks();
        }
    }

    // Patch 2: Specifically for Verb_ShootWithOffset.
    [HarmonyPatch(typeof(Verb_ShootWithOffset), "TryCastShot")]
    public static class Patch_Verb_ShootWithOffset_TryCastShot
    {
        public static void Postfix(Verb_ShootWithOffset __instance, bool __result)
        {
            if (!__result) return;
            if (__instance.CasterPawn == null || __instance.EquipmentSource == null) return;

            CompGiveHediffOnShot comp = __instance.EquipmentSource.GetComp<CompGiveHediffOnShot>();
            if (comp == null || comp.Props.hediffDef == null) return;

            Hediff hediff = __instance.CasterPawn.health.GetOrAddHediff(comp.Props.hediffDef);
            hediff.Severity += comp.Props.severityToAdd;

            var disappearsComp = hediff.TryGetComp<HediffComp_Disappears>();
            disappearsComp?.ResetElapsedTicks();
        }
    }

    // Patch 3: Specifically for our new Verb_ShootShotgunWithOffset.
    [HarmonyPatch(typeof(Verb_ShootShotgunWithOffset), "TryCastShot")]
    public static class Patch_Verb_ShootShotgunWithOffset_TryCastShot
    {
        public static void Postfix(Verb_ShootShotgunWithOffset __instance, bool __result)
        {
            if (!__result) return;
            if (__instance.CasterPawn == null || __instance.EquipmentSource == null) return;

            CompGiveHediffOnShot comp = __instance.EquipmentSource.GetComp<CompGiveHediffOnShot>();
            if (comp == null || comp.Props.hediffDef == null) return;

            Hediff hediff = __instance.CasterPawn.health.GetOrAddHediff(comp.Props.hediffDef);
            hediff.Severity += comp.Props.severityToAdd;

            var disappearsComp = hediff.TryGetComp<HediffComp_Disappears>();
            disappearsComp?.ResetElapsedTicks();
        }
    }

    // 新增补丁：为 Verb_ShootBeamExplosive 添加支持
    [HarmonyPatch(typeof(Verb_ShootBeamExplosive), "TryCastShot")]
    public static class Patch_Verb_ShootBeamExplosive_TryCastShot
    {
        public static void Postfix(Verb_ShootBeamExplosive __instance, bool __result)
        {
            if (!__result) return;
            if (__instance.CasterPawn == null || __instance.EquipmentSource == null) return;

            CompGiveHediffOnShot comp = __instance.EquipmentSource.GetComp<CompGiveHediffOnShot>();
            if (comp == null || comp.Props.hediffDef == null) return;

            Hediff hediff = __instance.CasterPawn.health.GetOrAddHediff(comp.Props.hediffDef);
            hediff.Severity += comp.Props.severityToAdd;

            var disappearsComp = hediff.TryGetComp<HediffComp_Disappears>();
            disappearsComp?.ResetElapsedTicks();
        }
    }
}