From a405cd82b58ae1275b6bdce9100eac66369906eb Mon Sep 17 00:00:00 2001
From: CMDR-Bill-Doors <Billyanghk@qq.com>
Date: Tue, 17 Sep 2024 04:46:53 +0100
Subject: [PATCH] Create Harmony_StartingPawnUtility.cs

---
 .../Harmony/Harmony_StartingPawnUtility.cs    | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 Source/CombatExtended/Harmony/Harmony_StartingPawnUtility.cs

diff --git a/Source/CombatExtended/Harmony/Harmony_StartingPawnUtility.cs b/Source/CombatExtended/Harmony/Harmony_StartingPawnUtility.cs
new file mode 100644
index 0000000000..a2e75e08c0
--- /dev/null
+++ b/Source/CombatExtended/Harmony/Harmony_StartingPawnUtility.cs
@@ -0,0 +1,40 @@
+using HarmonyLib;
+using Verse;
+using System.Linq;
+using System.Collections.Generic;
+
+namespace CombatExtended.HarmonyCE
+{
+    [HarmonyPatch(typeof(StartingPawnUtility), nameof(StartingPawnUtility.GeneratePossessions))]
+    internal static class StartingPawnUtility_GenerateAmmo
+    {
+        static IntRange magRange = new IntRange(3, 5);
+
+        internal static void Postfix(Pawn pawn)
+        {
+            var startingPossessions = Find.GameInitData.startingPossessions;
+            if (startingPossessions.ContainsKey(pawn))
+            {
+                List<ThingDefCount> ammoList = new List<ThingDefCount>();
+                foreach (var possession in startingPossessions[pawn])
+                {
+                    if (possession.thingDef.GetCompProperties<CompProperties_AmmoUser>() is CompProperties_AmmoUser ammoUser && ammoUser.ammoSet != null)
+                    {
+                        int count = ammoUser.AmmoGenPerMagOverride;
+                        if (count <= 0)
+                        {
+                            count = ammoUser.magazineSize;
+                        }
+                        count *= magRange.RandomInRange;
+
+                        ammoList.Add(new ThingDefCount(ammoUser.ammoSet.ammoTypes.First().ammo, count));
+                    }
+                }
+                foreach (var ammo in ammoList)
+                {
+                    startingPossessions[pawn].Add(ammo);
+                }
+            }
+        }
+    }
+}