Cargo economy balance (#11123)

Co-authored-by: Visne <39844191+Visne@users.noreply.github.com>
2022-09-15 11:53:17 +10:00
parent f4c38d74e1
commit ad7a851e27
113 changed files with 615 additions and 58 deletions
--- a/Content.Server/UserInterface/StatValuesCommand.cs
+++ b/Content.Server/UserInterface/StatValuesCommand.cs
@@ -0,0 +1,144 @@
+using Content.Server.Administration;
+using Content.Server.Cargo.Systems;
+using Content.Server.EUI;
+using Content.Shared.Administration;
+using Content.Shared.Materials;
+using Content.Shared.Research.Prototypes;
+using Content.Shared.UserInterface;
+using Robust.Server.Player;
+using Robust.Shared.Console;
+using Robust.Shared.Prototypes;
+
+namespace Content.Server.UserInterface;
+
+[AdminCommand(AdminFlags.Debug)]
+public sealed class StatValuesCommand : IConsoleCommand
+{
+    public string Command => "showvalues";
+    public string Description => "Dumps all stats for a particular category into a table.";
+    public string Help => $"{Command} <cargosell / lathsell>";
+    public void Execute(IConsoleShell shell, string argStr, string[] args)
+    {
+        if (shell.Player is not IPlayerSession pSession)
+        {
+            shell.WriteError($"{Command} can't be run on server!");
+            return;
+        }
+
+        if (args.Length != 1)
+        {
+            shell.WriteError($"Invalid number of args, need 1");
+            return;
+        }
+
+        StatValuesEuiMessage message;
+
+        switch (args[0])
+        {
+            case "cargosell":
+                message = GetCargo();
+                break;
+            case "lathesell":
+                message = GetLatheMessage();
+                break;
+            default:
+                shell.WriteError($"{args[0]} is not a valid stat!");
+                return;
+        }
+
+        var euiManager = IoCManager.Resolve<EuiManager>();
+        var eui = new StatValuesEui();
+        euiManager.OpenEui(eui, pSession);
+        eui.SendMessage(message);
+    }
+
+    private StatValuesEuiMessage GetCargo()
+    {
+        // Okay so there's no easy way to do this with how pricing works
+        // So we'll just get the first value for each prototype ID which is probably good enough for the majority.
+
+        var values = new List<string[]>();
+        var entManager = IoCManager.Resolve<IEntityManager>();
+        var priceSystem = IoCManager.Resolve<IEntitySystemManager>().GetEntitySystem<PricingSystem>();
+        var metaQuery = entManager.GetEntityQuery<MetaDataComponent>();
+        var prices = new HashSet<string>(256);
+
+        foreach (var entity in entManager.GetEntities())
+        {
+            if (!metaQuery.TryGetComponent(entity, out var meta))
+                continue;
+
+            var id = meta.EntityPrototype?.ID;
+
+            // We'll add it even if we don't have it so we don't have to raise the event again because this is probably faster.
+            if (id == null || !prices.Add(id))
+                continue;
+
+            var price = priceSystem.GetPrice(entity);
+
+            if (price == 0)
+                continue;
+
+            values.Add(new string[]
+            {
+                id,
+                $"{price:0}",
+            });
+        }
+
+        var state = new StatValuesEuiMessage()
+        {
+            Title = "Cargo sell prices",
+            Headers = new List<string>()
+            {
+                "ID",
+                "Price",
+            },
+            Values = values,
+        };
+
+        return state;
+    }
+
+    private StatValuesEuiMessage GetLatheMessage()
+    {
+        var values = new List<string[]>();
+        var protoManager = IoCManager.Resolve<IPrototypeManager>();
+        var factory = IoCManager.Resolve<IComponentFactory>();
+        var priceSystem = IoCManager.Resolve<IEntitySystemManager>().GetEntitySystem<PricingSystem>();
+
+        foreach (var proto in protoManager.EnumeratePrototypes<LatheRecipePrototype>())
+        {
+            var cost = 0.0;
+
+            foreach (var (material, count) in proto.RequiredMaterials)
+            {
+                var materialPrice = protoManager.Index<MaterialPrototype>(material).Price;
+                cost += materialPrice * count;
+            }
+
+            var sell = priceSystem.GetEstimatedPrice(protoManager.Index<EntityPrototype>(proto.Result), factory);
+
+            values.Add(new[]
+            {
+                proto.ID,
+                $"{cost:0}",
+                $"{sell:0}",
+            });
+        }
+
+        var state = new StatValuesEuiMessage()
+        {
+            Title = "Lathe sell prices",
+            Headers = new List<string>()
+            {
+                "ID",
+                "Cost",
+                "Sell price",
+            },
+            Values = values,
+        };
+
+        return state;
+    }
+}