using Content.Shared._DV.CCVars;
using Content.Shared.Damage;
using Content.Shared.Damage.Prototypes;
using Content.Shared.Standing;
using Content.Shared.Weapons.Melee.Events;
using Robust.Shared.Configuration;
using Robust.Shared.Prototypes;

namespace Content.Shared._DV.Standing;

/// <summary>
/// Prevents shooting and makes melee weaker while you are laying down (R)
/// </summary>
public sealed class LayingDownCombatSystem : EntitySystem
{
    [Dependency] private readonly IConfigurationManager _cfg = default!;
    [Dependency] private readonly IPrototypeManager _proto = default!;
    [Dependency] private readonly StandingStateSystem _standing = default!;

    private DamageModifierSet _meleeMod = new();

    public override void Initialize()
    {
        base.Initialize();

        // subscribe to LayingDownComponent instead of StandingState so it only applies to mobs that can lie down on keypress
        SubscribeLocalEvent<StandingStateComponent, GetMeleeDamageEvent>(OnGetMeleeDamage);

        Subs.CVar(_cfg, DCCVars.LayingDownMeleeMod, mod =>
        {
            _meleeMod.Coefficients.Clear();
            foreach (var proto in _proto.EnumeratePrototypes<DamageTypePrototype>())
            {
                _meleeMod.Coefficients.Add(proto.ID, mod);
            }
        }, true);
    }

    private void OnGetMeleeDamage(Entity<StandingStateComponent> ent, ref GetMeleeDamageEvent args)
    {
        if (!_standing.IsDown((ent, ent.Comp)))
            return;

        args.Modifiers.Add(_meleeMod);
    }
}