using Content.Shared.Mobs;
using Content.Shared.Mobs.Components;
using Robust.Shared.Prototypes;

namespace Content.Shared.EntityEffects.EffectConditions;

public sealed partial class MobStateCondition : EntityEffectCondition
{
    [DataField]
    public MobState Mobstate = MobState.Alive;

    public override bool Condition(EntityEffectBaseArgs args)
    {
        if (args.EntityManager.TryGetComponent(args.TargetEntity, out MobStateComponent? mobState))
        {
            if (mobState.CurrentState == Mobstate)
                return true;
        }
        // Begin Offbrand
        if (Mobstate == MobState.Critical)
        {
            if (args.EntityManager.System<Content.Shared._Offbrand.Wounds.HealthRankingSystem>()
                .IsCritical(args.TargetEntity))
            {
                return true;
            }
        }
        // End Offbrand

        return false;
    }

    public override string GuidebookExplanation(IPrototypeManager prototype)
    {
        return Loc.GetString("reagent-effect-condition-guidebook-mob-state-condition", ("state", Mobstate));
    }
}