Also check total selections

Simple-Station · Dec 19, 2024 · 92a1c73 · 92a1c73
1 parent 09741da
commit 92a1c73
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/Content.Server/Traits/TraitSystem.cs b/Content.Server/Traits/TraitSystem.cs
@@ -46,6 +46,7 @@ public override void Initialize()
     private void OnPlayerSpawnComplete(PlayerSpawnCompleteEvent args)
     {
         var pointsTotal = _configuration.GetCVar(CCVars.GameTraitsDefaultPoints);
+        var traitSelections = _configuration.GetCVar(CCVars.GameTraitsMax);
 
         foreach (var traitId in args.Profile.TraitPreferences)
         {
@@ -65,11 +66,12 @@ private void OnPlayerSpawnComplete(PlayerSpawnCompleteEvent args)
 
             // To check for cheaters. :FaridaBirb.png:
             pointsTotal += traitPrototype.Points;
+            --traitSelections;
 
             AddTrait(args.Mob, traitPrototype);
         }
 
-        if (pointsTotal < 0)
+        if (pointsTotal < 0 || traitSelections < 0)
             PunishCheater(args.Mob);
     }