From 74c8d776b896af974bfce0da8c56e0ca5eb06f69 Mon Sep 17 00:00:00 2001
From: csqrb <56765288+CaptainSqrBeard@users.noreply.github.com>
Date: Tue, 19 Nov 2024 01:49:21 +0600
Subject: [PATCH] =?UTF-8?q?=D0=98=D0=B7=D0=B2=D0=BB=D0=B5=D0=BA=D0=B0?=
 =?UTF-8?q?=D0=B5=D0=BC=D1=8B=D0=B5=20=D1=87=D0=B5=D1=80=D1=82=D0=B5=D0=B6?=
 =?UTF-8?q?=D0=B8=20(#47)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

* ejectable blueprints

* // Corvax-Next-BlueprintEject

* // Corvax-Next-Something

Co-authored-by: FN <37689533+FireNameFN@users.noreply.github.com>

* Apply suggestions from code review

---------

Co-authored-by: FN <37689533+FireNameFN@users.noreply.github.com>
Co-authored-by: AwareFoxy <135021509+AwareFoxy@users.noreply.github.com>
---
 .../Lathe/UI/LatheBoundUserInterface.cs       |  8 +++++
 Content.Client/Lathe/UI/LatheMenu.xaml        |  8 +++++
 Content.Client/Lathe/UI/LatheMenu.xaml.cs     | 17 +++++++++++
 Content.Server/Lathe/LatheSystem.cs           | 29 ++++++++++++++++++-
 Content.Shared/Lathe/LatheMessages.cs         | 14 ++++++++-
 .../Research/Systems/BlueprintSystem.cs       |  7 +++++
 .../ru-RU/_corvaxnext/lathe/ui/lathe-menu.ftl |  1 +
 7 files changed, 82 insertions(+), 2 deletions(-)
 create mode 100644 Resources/Locale/ru-RU/_corvaxnext/lathe/ui/lathe-menu.ftl

diff --git a/Content.Client/Lathe/UI/LatheBoundUserInterface.cs b/Content.Client/Lathe/UI/LatheBoundUserInterface.cs
index a599f79152e..d4c21ad372d 100644
--- a/Content.Client/Lathe/UI/LatheBoundUserInterface.cs
+++ b/Content.Client/Lathe/UI/LatheBoundUserInterface.cs
@@ -27,6 +27,13 @@ protected override void Open()
                 SendMessage(new ConsoleServerSelectionMessage());
             };
 
+            // Corvax-Next-BlueprintEject-Start
+            _menu.OnBlueprintEjectButtonPressed += _ =>
+            {
+                SendMessage(new LatheBlueprintEjectMessage());
+            };
+            // Corvax-Next-BlueprintEject-End
+
             _menu.RecipeQueueAction += (recipe, amount) =>
             {
                 SendMessage(new LatheQueueRecipeMessage(recipe, amount));
@@ -42,6 +49,7 @@ protected override void UpdateState(BoundUserInterfaceState state)
                 case LatheUpdateState msg:
                     if (_menu != null)
                         _menu.Recipes = msg.Recipes;
+                    _menu?.UpdateBlueprintEjectButton(msg.HasAnyBlueprints); // Corvax-Next-BlueprintEject
                     _menu?.PopulateRecipes();
                     _menu?.UpdateCategories();
                     _menu?.PopulateQueueList(msg.Queue);
diff --git a/Content.Client/Lathe/UI/LatheMenu.xaml b/Content.Client/Lathe/UI/LatheMenu.xaml
index 5b21f0bae66..70458f3974b 100644
--- a/Content.Client/Lathe/UI/LatheMenu.xaml
+++ b/Content.Client/Lathe/UI/LatheMenu.xaml
@@ -74,6 +74,14 @@
                     Orientation="Horizontal"
                     Align="End"
                     HorizontalExpand="True">
+                    <!-- Corvax-Next-BlueprintEject-Start -->
+                    <Button
+                        Name="BlueprintEjectButton"
+                        Text="{Loc 'lathe-menu-blueprint-eject'}"
+                        TextAlign="Center"
+                        Mode="Press">
+                    </Button>
+                    <!-- Corvax-Next-BlueprintEject-End -->
                     <Button
                         Name="ServerListButton"
                         Text="{Loc 'lathe-menu-server-list'}"
diff --git a/Content.Client/Lathe/UI/LatheMenu.xaml.cs b/Content.Client/Lathe/UI/LatheMenu.xaml.cs
index 02464d22e12..b7ee435f2f8 100644
--- a/Content.Client/Lathe/UI/LatheMenu.xaml.cs
+++ b/Content.Client/Lathe/UI/LatheMenu.xaml.cs
@@ -4,6 +4,7 @@
 using Content.Shared.Lathe;
 using Content.Shared.Lathe.Prototypes;
 using Content.Shared.Research.Prototypes;
+using Content.Shared.Research.Components; // Corvax-Next-BlueprintEject
 using Robust.Client.AutoGenerated;
 using Robust.Client.GameObjects;
 using Robust.Client.UserInterface;
@@ -25,6 +26,7 @@ public sealed partial class LatheMenu : DefaultWindow
     private readonly MaterialStorageSystem _materialStorage;
 
     public event Action<BaseButton.ButtonEventArgs>? OnServerListButtonPressed;
+    public event Action<BaseButton.ButtonEventArgs>? OnBlueprintEjectButtonPressed; // Corvax-Next-BlueprintEject
     public event Action<string, int>? RecipeQueueAction;
 
     public List<ProtoId<LatheRecipePrototype>> Recipes = new();
@@ -56,6 +58,7 @@ public LatheMenu()
         FilterOption.OnItemSelected += OnItemSelected;
 
         ServerListButton.OnPressed += a => OnServerListButtonPressed?.Invoke(a);
+        BlueprintEjectButton.OnPressed += a => OnBlueprintEjectButtonPressed?.Invoke(a); // Corvax-Next-BlueprintEject
     }
 
     public void SetEntity(EntityUid uid)
@@ -70,6 +73,13 @@ public void SetEntity(EntityUid uid)
             }
         }
 
+        // Corvax-Next-BlueprintEject-Start
+        if (!_entityManager.HasComponent<BlueprintReceiverComponent>(Entity))
+        {
+            BlueprintEjectButton.Visible = false;
+        }
+        // Corvax-Next-BlueprintEject-End
+
         MaterialsList.SetOwner(Entity);
     }
 
@@ -280,4 +290,11 @@ private void OnItemSelected(OptionButton.ItemSelectedEventArgs obj)
         }
         PopulateRecipes();
     }
+
+    // Corvax-Next-BlueprintEject-Start
+    public void UpdateBlueprintEjectButton(bool hasAnyBlueprints)
+    {
+        BlueprintEjectButton.Disabled = !hasAnyBlueprints;
+    }
+    // Corvax-Next-BlueprintEject-End
 }
diff --git a/Content.Server/Lathe/LatheSystem.cs b/Content.Server/Lathe/LatheSystem.cs
index 18f246dcef4..9f574fb3f02 100644
--- a/Content.Server/Lathe/LatheSystem.cs
+++ b/Content.Server/Lathe/LatheSystem.cs
@@ -18,6 +18,7 @@
 using Content.Shared.Emag.Components;
 using Content.Shared.Examine;
 using Content.Shared.Lathe;
+using Content.Shared.Research.Systems; // Corvax-Next-BlueprintEject
 using Content.Shared.Materials;
 using Content.Shared.Power;
 using Content.Shared.ReagentSpeed;
@@ -50,6 +51,7 @@ public sealed class LatheSystem : SharedLatheSystem
         [Dependency] private readonly SharedSolutionContainerSystem _solution = default!;
         [Dependency] private readonly StackSystem _stack = default!;
         [Dependency] private readonly TransformSystem _transform = default!;
+        [Dependency] private readonly BlueprintSystem _blueprints = default!; // Corvax-Next-BlueprintEject
 
         /// <summary>
         /// Per-tick cache
@@ -73,6 +75,7 @@ public override void Initialize()
             SubscribeLocalEvent<TechnologyDatabaseComponent, LatheGetRecipesEvent>(OnGetRecipes);
             SubscribeLocalEvent<EmagLatheRecipesComponent, LatheGetRecipesEvent>(GetEmagLatheRecipes);
             SubscribeLocalEvent<LatheHeatProducingComponent, LatheStartPrintingEvent>(OnHeatStartPrinting);
+            SubscribeLocalEvent<LatheComponent, LatheBlueprintEjectMessage>(OnBlueprintEjectMessage); // Corvax-Next-BlueprintEject
         }
         public override void Update(float frameTime)
         {
@@ -271,7 +274,9 @@ public void UpdateUserInterfaceState(EntityUid uid, LatheComponent? component =
 
             var producing = component.CurrentRecipe ?? component.Queue.FirstOrDefault();
 
-            var state = new LatheUpdateState(GetAvailableRecipes(uid, component), component.Queue, producing);
+            var hasAnyBlueprints = TryComp<BlueprintReceiverComponent>(uid, out var blueprintReceiver) && _blueprints.HasAnyBlueprints(uid, blueprintReceiver); //Corvax-Next-BlueprintEject
+
+            var state = new LatheUpdateState(GetAvailableRecipes(uid, component), component.Queue, producing, hasAnyBlueprints); //Corvax-Next-BlueprintEject
             _uiSys.SetUiState(uid, LatheUiKey.Key, state);
         }
 
@@ -395,6 +400,28 @@ private void OnLatheSyncRequestMessage(EntityUid uid, LatheComponent component,
         {
             UpdateUserInterfaceState(uid, component);
         }
+
+        // Corvax-Next-BlueprintEject-Start
+        private void OnBlueprintEjectMessage(EntityUid uid, LatheComponent component, LatheBlueprintEjectMessage args)
+        {
+            if (!TryComp<BlueprintReceiverComponent>(uid, out var blueprintReceiver))
+                return;
+
+            var toRemove = new HashSet<EntityUid>();
+            var container = _container.GetContainer(uid, blueprintReceiver.ContainerId);
+            foreach (var entity in container.ContainedEntities)
+            {   
+                toRemove.Add(entity);
+            }
+
+            foreach (var entity in toRemove)
+            {
+                _container.Remove(entity, container);
+            }
+
+            UpdateUserInterfaceState(uid, component);
+        }
+        // Corvax-Next-BlueprintEject-End
         #endregion
     }
 }
diff --git a/Content.Shared/Lathe/LatheMessages.cs b/Content.Shared/Lathe/LatheMessages.cs
index 820c496d30c..d8921d164cd 100644
--- a/Content.Shared/Lathe/LatheMessages.cs
+++ b/Content.Shared/Lathe/LatheMessages.cs
@@ -12,15 +12,27 @@ public sealed class LatheUpdateState : BoundUserInterfaceState
     public List<LatheRecipePrototype> Queue;
 
     public LatheRecipePrototype? CurrentlyProducing;
+    public bool HasAnyBlueprints; // Corvax-Next-BlueprintEject
 
-    public LatheUpdateState(List<ProtoId<LatheRecipePrototype>> recipes, List<LatheRecipePrototype> queue, LatheRecipePrototype? currentlyProducing = null)
+    public LatheUpdateState(List<ProtoId<LatheRecipePrototype>> recipes, List<LatheRecipePrototype> queue, LatheRecipePrototype? currentlyProducing = null, bool hasAnyBlueprints = false) // Corvax-Next-BlueprintEject
     {
         Recipes = recipes;
         Queue = queue;
         CurrentlyProducing = currentlyProducing;
+        HasAnyBlueprints = hasAnyBlueprints; // Corvax-Next-BlueprintEject
     }
 }
 
+// Corvax-Next-BlueprintEject-start
+
+/// <summary>
+///     Sent to the server to eject blueprints from the lathe.
+/// </summary>
+[Serializable, NetSerializable]
+public sealed class LatheBlueprintEjectMessage : BoundUserInterfaceMessage;
+
+// Corvax-Next-BlueprintEject-end
+
 /// <summary>
 ///     Sent to the server to sync material storage and the recipe queue.
 /// </summary>
diff --git a/Content.Shared/Research/Systems/BlueprintSystem.cs b/Content.Shared/Research/Systems/BlueprintSystem.cs
index 237ff703002..2dfef3be33b 100644
--- a/Content.Shared/Research/Systems/BlueprintSystem.cs
+++ b/Content.Shared/Research/Systems/BlueprintSystem.cs
@@ -111,4 +111,11 @@ public HashSet<ProtoId<LatheRecipePrototype>> GetBlueprintRecipes(Entity<Bluepri
 
         return recipes;
     }
+
+    // Corvax-Next-BlueprintEject-Start
+    public bool HasAnyBlueprints(EntityUid ent, BlueprintReceiverComponent comp)
+    {
+        return _container.GetContainer(ent, comp.ContainerId).ContainedEntities.Count > 0;
+    }
+    // Corvax-Next-BlueprintEject-End
 }
diff --git a/Resources/Locale/ru-RU/_corvaxnext/lathe/ui/lathe-menu.ftl b/Resources/Locale/ru-RU/_corvaxnext/lathe/ui/lathe-menu.ftl
new file mode 100644
index 00000000000..218c3af8d15
--- /dev/null
+++ b/Resources/Locale/ru-RU/_corvaxnext/lathe/ui/lathe-menu.ftl
@@ -0,0 +1 @@
+lathe-menu-blueprint-eject = Извлечь чертежи
\ No newline at end of file