using Content.Server.Electrocution;
using Content.Shared.Chemistry.Reagent;

namespace Content.Server.Chemistry.ReagentEffects;

public sealed class Electrocute : ReagentEffect
{
    [DataField("electrocuteTime")] public int ElectrocuteTime = 2;

    [DataField("electrocuteDamageScale")] public int ElectrocuteDamageScale = 5;

    /// <remarks>
    ///     true - refresh electrocute time,  false - accumulate electrocute time
    /// </remarks>
    [DataField("refresh")] public bool Refresh = true;

    public override bool ShouldLog => true;

    public override void Effect(ReagentEffectArgs args)
    {
        EntitySystem.Get<ElectrocutionSystem>().TryDoElectrocution(args.SolutionEntity, null,
            Math.Max((args.Quantity * ElectrocuteDamageScale).Int(), 1), TimeSpan.FromSeconds(ElectrocuteTime), Refresh);

        args.Source?.RemoveReagent(args.Reagent.ID, args.Quantity);
    }
}