OldThink/Content.Server/_White/WeaponModules/WeaponModulesSystem.cs

using System.Numerics;
using Content.Shared.Weapons.Ranged.Components;
using Content.Shared.Weapons.Ranged.Systems;
using Robust.Server.GameObjects;
using Robust.Shared.Audio;
using Robust.Shared.Containers;

namespace Content.Server._White.WeaponModules;

public sealed class WeaponModulesSystem : EntitySystem
{
    protected const string ModulesSlot = "gun_modules";
    [Dependency] private readonly PointLightSystem _lightSystem = default!;
    [Dependency] private readonly SharedAppearanceSystem Appearance = default!;
    [Dependency] private readonly SharedGunSystem _gunSystem = default!;

    SoundSpecifier? oldSoundGunshot;

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

        SubscribeLocalEvent<WeaponModulesComponent, EntInsertedIntoContainerMessage>(OnInsert);
        SubscribeLocalEvent<WeaponModulesComponent, EntRemovedFromContainerMessage>(OnEject);
    }

    private void OnInsert(EntityUid uid, WeaponModulesComponent comp, EntInsertedIntoContainerMessage args)
    {
        if (ModulesSlot != args.Container.ID)
            return;

        string module = Prototype(args.Entity)!.ID;
        EntityUid weapon = args.Container.Owner;

        TryComp<GunComponent>(weapon, out var gunComp);
        oldSoundGunshot = gunComp!.SoundGunshot;

        insertModules(module, comp);
        moduleEffect(module, weapon);
    }

    private void OnEject(EntityUid uid, WeaponModulesComponent comp, EntRemovedFromContainerMessage args)
    {
        if (ModulesSlot != args.Container.ID)
            return;

        string module = Prototype(args.Entity)!.ID;
        EntityUid weapon = args.Container.Owner;

        removeModules(module, comp);
        removeModuleEffect(module, weapon);
    }

    private void insertModules(string module, WeaponModulesComponent comp)
    {
        switch (module)
        {
            case "LightModule":
                if(comp.Modules.Contains("LightModule")) break;
                comp.Modules.Add("LightModule");
                break;

            case "LaserModule":
                if(comp.Modules.Contains("LaserModule")) break;
                comp.Modules.Add("LaserModule");
                break;

            case "FlameHiderModule":
                if(comp.Modules.Contains("FlameHiderModule")) break;
                comp.Modules.Add("FlameHiderModule");
                break;

            case "SilencerModule":
                if(comp.Modules.Contains("SilencerModule")) break;
                comp.Modules.Add("SilencerModule");
                break;
        }
    }

    private void moduleEffect(string module, EntityUid weapon)
    {
        switch (module)
        {
            case "LightModule" when HasComp<PointLightComponent>(weapon):
            {
                _lightSystem.SetEnabled(weapon, true);
                break;
            }

            case "LightModule":

                _lightSystem.TryGetLight(weapon, out var light);
                _lightSystem.EnsureLight(weapon).Offset = new Vector2(0, -1);
                _lightSystem.SetRadius(weapon, 2F, light);
                break;

            case "LaserModule":
                _gunSystem.setProjectileSpeed(weapon, 40F);
                break;

            case "FlameHiderModule":
                _gunSystem.setUseEffect(weapon, true);
                break;

            case "SilencerModule":
                _gunSystem.setUseEffect(weapon, true);
                _gunSystem.setSound(weapon, new SoundPathSpecifier("/Audio/White/Weapons/Modules/silence.ogg"));
                break;
        }
    }

    private void removeModules(string module, WeaponModulesComponent comp)
    {
        switch (module)
        {
            case "LightModule":
                comp.Modules.Remove("LightModule");
                break;

            case "LaserModule":
                comp.Modules.Remove("LaserModule");
                break;

            case "FlameHiderModule":
                comp.Modules.Remove("FlameHiderModule");
                break;

            case "SilencerModule":
                comp.Modules.Remove("SilencerModule");
                break;
        }
    }

    private void removeModuleEffect(string module, EntityUid weapon)
    {
        switch (module)
        {
            case "LightModule":
                if (!HasComp<PointLightComponent>(weapon))
                    break;

                _lightSystem.TryGetLight(weapon, out var light);
                _lightSystem.SetEnabled(weapon, false, light);
                break;

            case "LaserModule":
                if (!HasComp<GunComponent>(weapon))
                    break;
                _gunSystem.setProjectileSpeed(weapon, 25F);
                break;

            case "FlameHiderModule":
                _gunSystem.setUseEffect(weapon, false);
                break;

            case "SilencerModule":
                _gunSystem.setUseEffect(weapon, false);
                _gunSystem.setSound(weapon, oldSoundGunshot!);
                break;
        }
    }
}