add: base system + light module

2024-03-18 01:25:09 +05:00
parent 0377fc7091
commit 407427b31b
7 changed files with 172 additions and 1 deletions
--- a/Content.Server/_White/WeaponModules/WeaponModulesSystem.cs
+++ b/Content.Server/_White/WeaponModules/WeaponModulesSystem.cs
@@ -0,0 +1,96 @@
+using Content.Server.Light.Events;
+using Content.Server.Lightning;
+using Robust.Server.GameObjects;
+using Robust.Shared.Containers;
+
+namespace Content.Server._White.WeaponModules;
+
+public sealed class WeaponModulesSystem : EntitySystem
+{
+    protected const string ModulesSlot = "gun_modules";
+    [Dependency] private readonly SharedPointLightSystem _lightSystem = default!;
+    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 weapon = Prototype(args.Entity)!.ID;
+        EntityUid target = args.Container.Owner;
+
+        insertModules(weapon, comp);
+        moduleEffect(weapon, target);
+    }
+
+    private void OnEject(EntityUid uid, WeaponModulesComponent comp, EntRemovedFromContainerMessage args)
+    {
+        if (ModulesSlot != args.Container.ID)
+            return;
+
+        string weapon = Prototype(args.Entity)!.ID;
+        EntityUid target = args.Container.Owner;
+
+        removeModules(weapon, comp);
+        removeModuleEffect(weapon, target);
+    }
+
+    private void insertModules(string module, WeaponModulesComponent comp)
+    {
+        switch (module)
+        {
+            case "LightModule":
+                if(comp.Modules.Contains("LightModule")) break;
+                comp.Modules.Add("LightModule");
+                break;
+        }
+    }
+
+    private void moduleEffect(string module, EntityUid target)
+    {
+        switch (module)
+        {
+            case "LightModule":
+                if (HasComp<PointLightComponent>(target))
+                {
+                    _lightSystem.SetEnabled(target, true);
+                        break;
+                }
+
+                _lightSystem.TryGetLight(target, out var light);
+                _lightSystem.EnsureLight(target);
+                _lightSystem.SetRadius(target, 2F, light);
+                break;
+        }
+    }
+
+    private void removeModules(string module, WeaponModulesComponent comp)
+    {
+        switch (module)
+        {
+            case "LightModule":
+                comp.Modules.Remove("LightModule");
+                break;
+        }
+    }
+
+    private void removeModuleEffect(string module, EntityUid target)
+    {
+        switch (module)
+        {
+            case "LightModule":
+                if(!HasComp<PointLightComponent>(target))
+                    break;
+
+                _lightSystem.TryGetLight(target, out var light);
+                _lightSystem.SetEnabled(target, false, light);
+                break;
+        }
+    }
+}