lathe machine upgrading (#12032)

Content.Shared/Lathe/SharedLatheSystem.cs

@@ -1,7 +1,9 @@
 using Content.Shared.Materials;
 using Content.Shared.Research.Prototypes;
 using JetBrains.Annotations;
+using Robust.Shared.GameStates;
 using Robust.Shared.Prototypes;
+using Robust.Shared.Serialization;
 
 namespace Content.Shared.Lathe;
 
@@ -13,6 +15,26 @@ public abstract class SharedLatheSystem : EntitySystem
     [Dependency] private readonly IPrototypeManager _proto = default!;
     [Dependency] private readonly SharedMaterialStorageSystem _materialStorage = default!;
 
+    public override void Initialize()
+    {
+        base.Initialize();
+
+        SubscribeLocalEvent<LatheComponent, ComponentGetState>(OnGetState);
+        SubscribeLocalEvent<LatheComponent, ComponentHandleState>(OnHandleState);
+    }
+
+    private void OnGetState(EntityUid uid, LatheComponent component, ref ComponentGetState args)
+    {
+        args.State = new LatheComponentState(component.MaterialUseMultiplier);
+    }
+
+    private void OnHandleState(EntityUid uid, LatheComponent component, ref ComponentHandleState args)
+    {
+        if (args.Current is not LatheComponentState state)
+            return;
+        component.MaterialUseMultiplier = state.MaterialUseMultiplier;
+    }
+
     [PublicAPI]
     public bool CanProduce(EntityUid uid, string recipe, int amount = 1, LatheComponent? component = null)
     {
@@ -28,7 +50,7 @@ public abstract class SharedLatheSystem : EntitySystem
 
         foreach (var (material, needed) in recipe.RequiredMaterials)
         {
-            if (_materialStorage.GetMaterialAmount(component.Owner, material) < (amount * needed))
+            if (_materialStorage.GetMaterialAmount(component.Owner, material) < amount * needed * component.MaterialUseMultiplier)
                 return false;
         }
         return true;
@@ -36,3 +58,14 @@ public abstract class SharedLatheSystem : EntitySystem
 
     protected abstract bool HasRecipe(EntityUid uid, LatheRecipePrototype recipe, LatheComponent component);
 }
+
+[Serializable, NetSerializable]
+public sealed class LatheComponentState : ComponentState
+{
+    public float MaterialUseMultiplier;
+
+    public LatheComponentState(float materialUseMultiplier)
+    {
+        MaterialUseMultiplier = materialUseMultiplier;
+    }
+}