wwdpublic/Content.Shared/_Shitcode/Wizard/Projectiles/LifeStealOnProjectileHitSystem.cs

// SPDX-FileCopyrightText: 2025 Aiden <28298836+Aidenkrz@users.noreply.github.com>
// SPDX-FileCopyrightText: 2025 Aviu00 <93730715+Aviu00@users.noreply.github.com>
// SPDX-FileCopyrightText: 2025 Misandry <mary@thughunt.ing>
// SPDX-FileCopyrightText: 2025 gus <august.eymann@gmail.com>
//
// SPDX-License-Identifier: AGPL-3.0-or-later

using Content.Shared._Shitcode.Wizard.SanguineStrike;
using Content.Shared.Mobs.Systems;
using Content.Shared.Projectiles;
using Content.Shared.Whitelist;

namespace Content.Shared._Shitcode.Wizard.Projectiles;

public sealed class LifeStealOnProjectileHitSystem : EntitySystem
{
    [Dependency] private readonly EntityWhitelistSystem _whitelist = default!;
    [Dependency] private readonly SharedSanguineStrikeSystem _sanguine = default!;
    [Dependency] private readonly MobStateSystem _mobState = default!;

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

        SubscribeLocalEvent<LifeStealOnProjectileHitComponent, ProjectileHitEvent>(OnHit);
    }

    private void OnHit(Entity<LifeStealOnProjectileHitComponent> ent, ref ProjectileHitEvent args)
    {
        if (args.Shooter == null || args.Shooter.Value == args.Target)
            return;

        var (_, comp) = ent;

        if (!_whitelist.IsValid(comp.Whitelist, args.Target))
            return;

        if (_mobState.IsDead(args.Target))
            return;

        _sanguine.LifeSteal(args.Shooter.Value, comp.LifeStealAmount);
        List<EntityUid> target = new() { args.Target };
        _sanguine.BloodSteal(args.Shooter.Value, target, comp.BloodStealAmount, null);
        _sanguine.ParticleEffects(args.Shooter.Value, target, comp.Effect);
    }
}