Added Ammo Config

AlmightyLks-SCP · Feb 7, 2021 · e03786d · e03786d
1 parent 40b0ee4
commit e03786d
Show file tree

Hide file tree

Showing 7 changed files with 50 additions and 8 deletions.
diff --git a/CustomLoadout/Config/PluginConfig.cs b/CustomLoadout/Config/PluginConfig.cs
@@ -1,10 +1,11 @@
-using Synapse.Config;
+using CustomLoadout.Config.Types;
+using Synapse.Config;
 using System.Collections.Generic;
 using System.ComponentModel;
 
 namespace CustomLoadout
 {
-    public class PluginConfig : AbstractConfigSection
+    public sealed class PluginConfig : AbstractConfigSection
     {
         [Description("Loadout configuration")]
         public List<Loadout> RoleInventory { get; set; } = new List<Loadout>()

diff --git a/CustomLoadout/Config/Types/AmmoConfig.cs b/CustomLoadout/Config/Types/AmmoConfig.cs
@@ -0,0 +1,26 @@
+using System;
+using System.Collections.Generic;
+using System.ComponentModel;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace CustomLoadout.Config.Types
+{
+    public sealed class AmmoConfig
+    {
+        [Description("Should the class' amount of ammo be replaced?")]
+        public bool ReplaceAmmo { get; set; }
+        public uint Ammo5 { get; set; }
+        public uint Ammo7 { get; set; }
+        public uint Ammo9 { get; set; }
+
+        public AmmoConfig()
+        {
+            ReplaceAmmo = false;
+            Ammo5 = 0;
+            Ammo7 = 0;
+            Ammo9 = 0;
+        }
+    }
+}
diff --git a/CustomLoadout/Config/ItemChance.cs → CustomLoadout/Config/Types/ItemChance.cs b/CustomLoadout/Config/ItemChance.cs → CustomLoadout/Config/Types/ItemChance.cs
@@ -1,8 +1,8 @@
 using Synapse.Config;
 
-namespace CustomLoadout.Config
+namespace CustomLoadout.Config.Types
 {
-    public class ItemChance
+    public sealed class ItemChance
     {
         public SerializedItem Item { get; set; }
         public float Chance { get; set; }

diff --git a/CustomLoadout/Config/Loadout.cs → CustomLoadout/Config/Types/Loadout.cs b/CustomLoadout/Config/Loadout.cs → CustomLoadout/Config/Types/Loadout.cs
@@ -2,9 +2,9 @@
 using System.Collections.Generic;
 using System.ComponentModel;
 
-namespace CustomLoadout
+namespace CustomLoadout.Config.Types
 {
-    public class Loadout
+    public sealed class Loadout
     {
         public int RoleID { get; set; }
 
@@ -13,11 +13,16 @@ public class Loadout
 
         [Description("Items & their Chance to spawn")]
         public List<ItemChance> Items { get; set; }
+
+        [Description("Configure the class' ammunition")]
+        public AmmoConfig AmmoConfig { get; set; }
+
         public Loadout()
         {
             RoleID = -1;
             ReplaceDefault = false;
             Items = new List<ItemChance>() { new ItemChance() };
+            AmmoConfig = new AmmoConfig();
         }
     }
 }
diff --git a/CustomLoadout/CustomLoadout.cs b/CustomLoadout/CustomLoadout.cs
@@ -6,6 +6,7 @@ namespace CustomLoadout
         Author = "AlmightyLks",
         Description = "Configure roles' inventories",
         Name = "CustomLoadout",
+        LoadPriority = int.MinValue,
         SynapseMajor = 2,
         SynapseMinor = 1,
         SynapsePatch = 0,

diff --git a/CustomLoadout/CustomLoadout.csproj b/CustomLoadout/CustomLoadout.csproj
@@ -72,9 +72,10 @@
     </Reference>
   </ItemGroup>
   <ItemGroup>
-    <Compile Include="Config\ItemChance.cs" />
+    <Compile Include="Config\Types\AmmoConfig.cs" />
+    <Compile Include="Config\Types\ItemChance.cs" />
     <Compile Include="CustomLoadout.cs" />
-    <Compile Include="Config\Loadout.cs" />
+    <Compile Include="Config\Types\Loadout.cs" />
     <Compile Include="Config\PluginConfig.cs" />
     <Compile Include="Handler\PluginEventHandler.cs" />
     <Compile Include="Properties\AssemblyInfo.cs" />

diff --git a/CustomLoadout/Handler/PluginEventHandler.cs b/CustomLoadout/Handler/PluginEventHandler.cs
@@ -19,13 +19,21 @@ private void Player_PlayerSetClassEvent(PlayerSetClassEventArgs ev)
                     if (loadout is null)
                         return;
 
+                    uint ammo5 = loadout.AmmoConfig.ReplaceAmmo ? loadout.AmmoConfig.Ammo5 : ev.Player.Ammo5;
+                    uint ammo7 = loadout.AmmoConfig.ReplaceAmmo ? loadout.AmmoConfig.Ammo7 : ev.Player.Ammo7;
+                    uint ammo9 = loadout.AmmoConfig.ReplaceAmmo ? loadout.AmmoConfig.Ammo9 : ev.Player.Ammo9;
+
                     if (loadout.ReplaceDefault)
                         ev.Player.Inventory.Clear();
 
                     foreach (var item in loadout.Items)
                     {
                         if (UnityEngine.Random.Range(0f, 100f) <= item.Chance)
                             ev.Player.Inventory.AddItem(item.Item.Parse());
+
+                        ev.Player.Ammo5 = ammo5;
+                        ev.Player.Ammo7 = ammo7;
+                        ev.Player.Ammo9 = ammo9;
                     }
                 }
             });