Add Multiphase energy gun

Content.Client/SS220/Weapons/Ranged/Systems/BatteryWeaponFireModesSystem.cs

@@ -0,0 +1,26 @@
+// © SS220, An EULA/CLA with a hosting restriction, full text: https://raw.githubusercontent.com/SerbiaStrong-220/space-station-14/master/CLA.txt
+using Content.Client.Weapons.Ranged.Systems;
+using Content.Shared.Weapons.Ranged.Components;
+using Content.Shared.Weapons.Ranged.Systems;
+
+namespace Content.Client.SS220.Weapons.Ranged.Systems;
+
+public sealed partial class BatteryWeaponFireModesSystem : EntitySystem
+{
+    [Dependency] private readonly GunSystem _gunSystem = default!;
+
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<BatteryWeaponFireModesComponent, ChangeFireModeEvent>(OnFireModeChange);
+    }
+
+    private void OnFireModeChange(Entity<BatteryWeaponFireModesComponent> ent, ref ChangeFireModeEvent args)
+    {
+        var fireMode = ent.Comp.FireModes[args.Index];
+
+        if (fireMode.MagState is not null)
+            _gunSystem.SetMagState(ent, fireMode.MagState);
+    }
+}

Content.Client/Weapons/Ranged/Systems/GunSystem.Battery.cs

@@ -1,4 +1,5 @@
 using Content.Shared.Weapons.Ranged.Components;
+using Robust.Client.GameObjects;
 
 namespace Content.Client.Weapons.Ranged.Systems;
 
@@ -10,10 +11,12 @@ protected override void InitializeBattery()
         // Hitscan
         SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, AmmoCounterControlEvent>(OnControl);
         SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, UpdateAmmoCounterEvent>(OnAmmoCountUpdate);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, AppearanceChangeEvent>(OnHitscanAppearanceChange); //SS220 Add Multifaze gun
 
         // Projectile
         SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, AmmoCounterControlEvent>(OnControl);
         SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, UpdateAmmoCounterEvent>(OnAmmoCountUpdate);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, AppearanceChangeEvent>(OnProjectileAppearanceChange); //SS220 Add Multifaze gun
     }
 
     private void OnAmmoCountUpdate(EntityUid uid, BatteryAmmoProviderComponent component, UpdateAmmoCounterEvent args)
@@ -27,4 +30,16 @@ private void OnControl(EntityUid uid, BatteryAmmoProviderComponent component, Am
     {
         args.Control = new BoxesStatusControl();
     }
+
+    //SS220 Add Multifaze gun begin
+    private void OnHitscanAppearanceChange(Entity<HitscanBatteryAmmoProviderComponent> ent, ref AppearanceChangeEvent args)
+    {
+        UpdateAmmoCount(ent);
+    }
+
+    private void OnProjectileAppearanceChange(Entity<ProjectileBatteryAmmoProviderComponent> ent, ref AppearanceChangeEvent args)
+    {
+        UpdateAmmoCount(ent);
+    }
+    //SS220 Add Multifaze gun end
 }

Content.Client/Weapons/Ranged/Systems/GunSystem.cs

@@ -387,4 +387,14 @@ protected override void CreateEffect(EntityUid gunUid, MuzzleFlashEvent message,
         _animPlayer.Stop(gunUid, uidPlayer, "muzzle-flash-light");
         _animPlayer.Play((gunUid, uidPlayer), animTwo, "muzzle-flash-light");
     }
+
+    //SS220 Add Multifaze gun begin
+    public void SetMagState(EntityUid uid, string magState, MagazineVisualsComponent? component = null)
+    {
+        if (!Resolve(uid, ref component))
+            return;
+
+        component.MagState = magState;
+    }
+    //SS220 Add Multifaze gun end
 }

Content.Server/Weapons/Ranged/Systems/GunSystem.Battery.cs

@@ -5,6 +5,7 @@
 using Content.Shared.Projectiles;
 using Content.Shared.Weapons.Ranged;
 using Content.Shared.Weapons.Ranged.Components;
+using Content.Shared.Weapons.Ranged.Systems;
 using Robust.Shared.Prototypes;
 
 namespace Content.Server.Weapons.Ranged.Systems;
@@ -19,11 +20,13 @@ protected override void InitializeBattery()
         SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, ComponentStartup>(OnBatteryStartup);
         SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, ChargeChangedEvent>(OnBatteryChargeChange);
         SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, DamageExamineEvent>(OnBatteryDamageExamine);
+        SubscribeLocalEvent<HitscanBatteryAmmoProviderComponent, ChangeFireModeEvent>(OnHitscanFireModeChange); //SS220 Add Multifaze gun
 
         // Projectile
         SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, ComponentStartup>(OnBatteryStartup);
         SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, ChargeChangedEvent>(OnBatteryChargeChange);
         SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, DamageExamineEvent>(OnBatteryDamageExamine);
+        SubscribeLocalEvent<ProjectileBatteryAmmoProviderComponent, ChangeFireModeEvent>(OnProjectileFireModeChange); //SS220 Add Multifaze gun
     }
 
     private void OnBatteryStartup(EntityUid uid, BatteryAmmoProviderComponent component, ComponentStartup args)
@@ -107,4 +110,18 @@ protected override void TakeCharge(EntityUid uid, BatteryAmmoProviderComponent c
         // Will raise ChargeChangedEvent
         _battery.UseCharge(uid, component.FireCost);
     }
+
+    //SS220 Add Multifaze gun begin
+    private void OnHitscanFireModeChange(Entity<HitscanBatteryAmmoProviderComponent> ent, ref ChangeFireModeEvent args)
+    {
+        UpdateShots(ent, ent.Comp);
+        Dirty(ent, ent.Comp);
+    }
+
+    private void OnProjectileFireModeChange(Entity<ProjectileBatteryAmmoProviderComponent> ent, ref ChangeFireModeEvent args)
+    {
+        UpdateShots(ent, ent.Comp);
+        Dirty(ent, ent.Comp);
+    }
+    //SS220 Add Multifaze gun end
 }

Content.Shared/Weapons/Ranged/Components/BatteryWeaponFireModesComponent.cs

@@ -35,7 +35,27 @@ public sealed partial class BatteryWeaponFireMode
     /// The projectile prototype associated with this firing mode
     /// </summary>
     [DataField("proto", required: true)]
-    public EntProtoId Prototype = default!;
+    public string Prototype = default!; //SS220 Add Multifaze gun
+
+    //SS220 Add Multifaze gun begin
+    /// <summary>
+    /// Name of the fire mode
+    /// </summary>
+    [DataField("name")]
+    public string? FireModeName;
+
+    /// <summary>
+    /// Sound of a gunshot that is used in the selected fire mode
+    /// </summary>
+    [DataField]
+    public string? SoundGunshot;
+
+    /// <summary>
+    /// Sprite of the remaining charge that is used in the selected fire mode
+    /// </summary>
+    [DataField]
+    public string? MagState;
+    //SS220 Add Multifaze gun end
 
     /// <summary>
     /// The battery cost to fire the projectile associated with this firing mode

Content.Shared/Weapons/Ranged/Systems/BatteryWeaponFireModesSystem.cs

@@ -5,14 +5,19 @@
 using Content.Shared.Popups;
 using Content.Shared.Verbs;
 using Content.Shared.Weapons.Ranged.Components;
+using Content.Shared.Weapons.Ranged.Events;
+using Robust.Shared.Audio;
 using Robust.Shared.Prototypes;
+using Robust.Shared.Timing;
 
 namespace Content.Shared.Weapons.Ranged.Systems;
 
 public sealed class BatteryWeaponFireModesSystem : EntitySystem
 {
     [Dependency] private readonly IPrototypeManager _prototypeManager = default!;
     [Dependency] private readonly SharedPopupSystem _popupSystem = default!;
+    [Dependency] private readonly IGameTiming _gameTiming = default!; //SS220 Add Multifaze gun
+    [Dependency] private readonly SharedGunSystem _gunSystem = default!; //SS220 Add Multifaze gun
 
     public override void Initialize()
     {
@@ -21,6 +26,8 @@ public override void Initialize()
         SubscribeLocalEvent<BatteryWeaponFireModesComponent, ActivateInWorldEvent>(OnInteractHandEvent);
         SubscribeLocalEvent<BatteryWeaponFireModesComponent, GetVerbsEvent<Verb>>(OnGetVerb);
         SubscribeLocalEvent<BatteryWeaponFireModesComponent, ExaminedEvent>(OnExamined);
+        SubscribeLocalEvent<BatteryWeaponFireModesComponent, ComponentInit>(OnInit); //SS220 Add Multifaze gun
+        SubscribeLocalEvent<BatteryWeaponFireModesComponent, GunRefreshModifiersEvent>(OnRefreshModifiers); //SS220 Add Multifaze gun
     }
 
     private void OnExamined(EntityUid uid, BatteryWeaponFireModesComponent component, ExaminedEvent args)
@@ -29,11 +36,19 @@ private void OnExamined(EntityUid uid, BatteryWeaponFireModesComponent component
             return;
 
         var fireMode = GetMode(component);
+        var name = string.Empty; //SS220 Add Multifaze gun
 
-        if (!_prototypeManager.TryIndex<EntityPrototype>(fireMode.Prototype, out var proto))
-            return;
+        //SS220 Add Multifaze gun begin
+        //if (!_prototypeManager.TryIndex<EntityPrototype>(fireMode.Prototype, out var proto))
+        //    return;
+
+        if (fireMode.FireModeName != null)
+            name = fireMode.FireModeName;
+        else if (_prototypeManager.TryIndex<EntityPrototype>(fireMode.Prototype, out var entProto))
+            name = entProto.Name;
+        //SS220 Add Multifaze gun end
 
-        args.PushMarkup(Loc.GetString("gun-set-fire-mode", ("mode", proto.Name)));
+        args.PushMarkup(Loc.GetString("gun-set-fire-mode", ("mode", Loc.GetString(name)))); //SS220 Add Multifaze gun
     }
 
     private BatteryWeaponFireMode GetMode(BatteryWeaponFireModesComponent component)
@@ -52,14 +67,30 @@ private void OnGetVerb(EntityUid uid, BatteryWeaponFireModesComponent component,
         for (var i = 0; i < component.FireModes.Count; i++)
         {
             var fireMode = component.FireModes[i];
-            var entProto = _prototypeManager.Index<EntityPrototype>(fireMode.Prototype);
+            // var entProto = _prototypeManager.Index<EntityPrototype>(fireMode.Prototype); //SS220 Add Multifaze gun
             var index = i;
 
+            //SS220 Add Multifaze gun begin
+            var text = string.Empty;
+
+            if (_prototypeManager.TryIndex<EntityPrototype>(fireMode.Prototype, out var entProto))
+            {
+                if (fireMode.FireModeName is not null)
+                    text = fireMode.FireModeName;
+                else
+                    text = entProto.Name;
+            }
+            else if (_prototypeManager.TryIndex<HitscanPrototype>(fireMode.Prototype, out _))
+            {
+                text += fireMode.FireModeName;
+            }
+            //SS220 Add Multifaze gun end
+
             var v = new Verb
             {
                 Priority = 1,
                 Category = VerbCategory.SelectType,
-                Text = entProto.Name,
+                Text = Loc.GetString(text), //SS220 Add Multifaze gun
                 Disabled = i == component.CurrentFireMode,
                 Impact = LogImpact.Low,
                 DoContactInteraction = true,
@@ -99,18 +130,103 @@ private void SetFireMode(EntityUid uid, BatteryWeaponFireModesComponent componen
         component.CurrentFireMode = index;
         Dirty(uid, component);
 
-        if (TryComp(uid, out ProjectileBatteryAmmoProviderComponent? projectileBatteryAmmoProvider))
+        //SS220 Add Multifaze gun begin
+        var name = string.Empty;
+
+        //if (TryComp(uid, out ProjectileBatteryAmmoProviderComponent? projectileBatteryAmmoProvider))
+        //{
+        //    if (!_prototypeManager.TryIndex<EntityPrototype>(fireMode.Prototype, out var prototype))
+        //        return;
+
+        //    projectileBatteryAmmoProvider.Prototype = fireMode.Prototype;
+        //    projectileBatteryAmmoProvider.FireCost = fireMode.FireCost;
+        //}
+
+        if (_prototypeManager.TryIndex<EntityPrototype>(fireMode.Prototype, out var entProto))
+        {
+            if (HasComp<HitscanBatteryAmmoProviderComponent>(uid))
+                RemComp<HitscanBatteryAmmoProviderComponent>(uid);
+
+            if (!_gameTiming.ApplyingState)
+                EnsureComp<ProjectileBatteryAmmoProviderComponent>(uid);
+
+            if (TryComp(uid, out ProjectileBatteryAmmoProviderComponent? projectileBatteryAmmoProvider))
+            {
+
+                if (fireMode.FireModeName is not null)
+                    name = fireMode.FireModeName;
+                else
+                    name = entProto.Name;
+
+                projectileBatteryAmmoProvider.Prototype = fireMode.Prototype;
+                projectileBatteryAmmoProvider.FireCost = fireMode.FireCost;
+
+                Dirty(uid, projectileBatteryAmmoProvider);
+            }
+        }
+        else if (_prototypeManager.TryIndex<HitscanPrototype>(fireMode.Prototype, out _))
         {
-            if (!_prototypeManager.TryIndex<EntityPrototype>(fireMode.Prototype, out var prototype))
-                return;
+            if (HasComp<ProjectileBatteryAmmoProviderComponent>(uid))
+                RemComp<ProjectileBatteryAmmoProviderComponent>(uid);
 
-            projectileBatteryAmmoProvider.Prototype = fireMode.Prototype;
-            projectileBatteryAmmoProvider.FireCost = fireMode.FireCost;
+            if (!_gameTiming.ApplyingState)
+                EnsureComp<HitscanBatteryAmmoProviderComponent>(uid);
 
-            if (user != null)
+            if (TryComp(uid, out HitscanBatteryAmmoProviderComponent? hitscanBatteryAmmoProvider))
             {
-                _popupSystem.PopupClient(Loc.GetString("gun-set-fire-mode", ("mode", prototype.Name)), uid, user.Value);
+                name += fireMode.FireModeName;
+
+                hitscanBatteryAmmoProvider.Prototype = fireMode.Prototype;
+                hitscanBatteryAmmoProvider.FireCost = fireMode.FireCost;
+
+                Dirty(uid, hitscanBatteryAmmoProvider);
             }
         }
+        else
+        {
+            Log.Error($"{fireMode.Prototype} is not Entity or Hitscan prototype");
+            return;
+        }
+
+        _gunSystem.RefreshModifiers(uid);
+        //SS220 Add Multifaze gun end
+
+        if (user != null)
+        {
+            _popupSystem.PopupClient(Loc.GetString("gun-set-fire-mode", ("mode", Loc.GetString(name))), uid, user.Value); //SS220 Add Multifaze gun
+        }
+
+        //SS220 Add Multifaze gun begin
+        var ev = new ChangeFireModeEvent(index);
+        RaiseLocalEvent(uid, ref ev);
+        //SS220 Add Multifaze gun end
     }
+
+    //SS220 Add Multifaze gun begin
+    private void OnInit(Entity<BatteryWeaponFireModesComponent> ent, ref ComponentInit args)
+    {
+        if (ent.Comp.FireModes.Count <= 0)
+            return;
+
+        var index = ent.Comp.CurrentFireMode % ent.Comp.FireModes.Count;
+        SetFireMode(ent, ent.Comp, index);
+    }
+
+    private void OnRefreshModifiers(Entity<BatteryWeaponFireModesComponent> ent, ref GunRefreshModifiersEvent args)
+    {
+        var firemode = GetMode(ent.Comp);
+
+        if (firemode.SoundGunshot is not null)
+            args.SoundGunshot = new SoundPathSpecifier(firemode.SoundGunshot);
+    }
+    //SS220 Add Multifaze gun end
 }
+
+//SS220 Add Multifaze gun begin
+/// <summary>
+/// The event that rises when the fire mode is selected
+/// </summary>
+/// <param name="Index"></param>
+[ByRefEvent]
+public record struct ChangeFireModeEvent(int Index);
+//SS220 Add Multifaze gun end

Content.Shared/Weapons/Ranged/Systems/SharedGunSystem.Battery.cs

@@ -35,15 +35,32 @@ private void OnBatteryHandleState(EntityUid uid, BatteryAmmoProviderComponent co
         component.Shots = state.Shots;
         component.Capacity = state.MaxShots;
         component.FireCost = state.FireCost;
+
+        //SS220 Add Multifaze gun begin
+        if (component is ProjectileBatteryAmmoProviderComponent projectileComp)
+            projectileComp.Prototype = state.Prototype;
+        else if (component is HitscanBatteryAmmoProviderComponent hitscanComp)
+            hitscanComp.Prototype = state.Prototype;
+        //SS220 Add Multifaze gun end
     }
 
     private void OnBatteryGetState(EntityUid uid, BatteryAmmoProviderComponent component, ref ComponentGetState args)
     {
+        //SS220 Add Multifaze gun begin
+        string prototype = string.Empty;
+
+        if (component is ProjectileBatteryAmmoProviderComponent projectileComp)
+            prototype = projectileComp.Prototype;
+        else if (component is HitscanBatteryAmmoProviderComponent hitscanComp)
+            prototype = hitscanComp.Prototype;
+        //SS220 Add Multifaze gun end
+
         args.State = new BatteryAmmoProviderComponentState()
         {
             Shots = component.Shots,
             MaxShots = component.Capacity,
             FireCost = component.FireCost,
+            Prototype = prototype, //SS220 Add Multifaze gun
         };
     }
 
@@ -112,5 +129,6 @@ private sealed class BatteryAmmoProviderComponentState : ComponentState
         public int Shots;
         public int MaxShots;
         public float FireCost;
+        public string Prototype = string.Empty; //SS220 Add Multifaze gun
     }
 }

Resources/Locale/ru-RU/ss220/weapons/multiphase_energy_gun.ftl

@@ -0,0 +1,6 @@
+firemode-name-disabler = Станнер
+firemode-name-laser = Лазер
+firemode-name-emp = ЭМИ
+
+ent-WeaponMultiPhaseEnergyGun = мультифазовый энергетический пистолет
+    .desc = Это дорогая, современная версия антикварного лазерного пистолета. У этого оружия есть несколько уникальных режимов огня, но нет возможности самостоятельно перезаряжаться с течением времени.