From 4a730c59108550da1e14eef21dddb8b7fc9ae8fd Mon Sep 17 00:00:00 2001
From: Nemanja <98561806+EmoGarbage404@users.noreply.github.com>
Date: Sat, 4 Nov 2023 09:19:23 -0400
Subject: [PATCH] Fix double counting bounties (#21399)

---
 .../StationCargoBountyDatabaseComponent.cs        |  9 ++++++++-
 .../Cargo/Systems/CargoSystem.Bounty.cs           | 15 +++++++++++----
 2 files changed, 19 insertions(+), 5 deletions(-)

diff --git a/Content.Server/Cargo/Components/StationCargoBountyDatabaseComponent.cs b/Content.Server/Cargo/Components/StationCargoBountyDatabaseComponent.cs
index d94c2995e3..68517ea144 100644
--- a/Content.Server/Cargo/Components/StationCargoBountyDatabaseComponent.cs
+++ b/Content.Server/Cargo/Components/StationCargoBountyDatabaseComponent.cs
@@ -33,8 +33,15 @@ public sealed partial class StationCargoBountyDatabaseComponent : Component
     public float MinBountyTime = 600f;
 
     /// <summary>
-    /// The maxmium amount of time the bounty lasts before being removed.
+    /// The maximum amount of time the bounty lasts before being removed.
     /// </summary>
     [DataField("maxBountyTime"), ViewVariables(VVAccess.ReadWrite)]
     public float MaxBountyTime = 905f;
+
+    /// <summary>
+    /// A list of bounty IDs that have been checked this tick.
+    /// Used to prevent multiplying bounty prices.
+    /// </summary>
+    [DataField]
+    public HashSet<int> CheckedBounties = new();
 }
diff --git a/Content.Server/Cargo/Systems/CargoSystem.Bounty.cs b/Content.Server/Cargo/Systems/CargoSystem.Bounty.cs
index be8f54dfee..f502b7f92f 100644
--- a/Content.Server/Cargo/Systems/CargoSystem.Bounty.cs
+++ b/Content.Server/Cargo/Systems/CargoSystem.Bounty.cs
@@ -89,18 +89,24 @@ public sealed partial class CargoSystem
         if (!_container.TryGetContainingContainer(uid, out var container) || container.ID != LabelSystem.ContainerName)
             return;
 
-        if (_station.GetOwningStation(uid) is not { } station)
+        if (_station.GetOwningStation(uid) is not { } station || !TryComp<StationCargoBountyDatabaseComponent>(station, out var database))
+            return;
+
+        if (database.CheckedBounties.Contains(component.Id))
             return;
 
-        if (!TryGetBountyFromId(station, component.Id, out var bounty))
+        if (!TryGetBountyFromId(station, component.Id, out var bounty, database))
             return;
 
-        if (!_protoMan.TryIndex<CargoBountyPrototype>(bounty.Value.Bounty, out var bountyProtoype) ||!IsBountyComplete(container.Owner, bountyProtoype))
+        if (!_protoMan.TryIndex<CargoBountyPrototype>(bounty.Value.Bounty, out var bountyPrototype) ||
+            !IsBountyComplete(container.Owner, bountyPrototype))
             return;
+
+        database.CheckedBounties.Add(component.Id);
         args.Handled = true;
 
         component.Calculating = true;
-        args.Price = bountyProtoype.Reward - _pricing.GetPrice(container.Owner);
+        args.Price = bountyPrototype.Reward - _pricing.GetPrice(container.Owner);
         component.Calculating = false;
     }
 
@@ -329,6 +335,7 @@ public sealed partial class CargoSystem
         var query = EntityQueryEnumerator<StationCargoBountyDatabaseComponent>();
         while (query.MoveNext(out var uid, out var bountyDatabase))
         {
+            bountyDatabase.CheckedBounties.Clear();
             var bounties = new ValueList<CargoBountyData>(bountyDatabase.Bounties);
             foreach (var bounty in bounties)
             {
-- 
2.51.2