Flatpacks and the Flatpacker 1001 (#23338)

* Flatpacker and flatpacks * ok that's good enough * convert solars/AME to flatpacks * mats, mats, we are the mats * basic mechanics are DONE * thing * final UI * sloth * rped jumpscare * rename
2024-01-03 01:16:02 -05:00
parent 04fc06e9d4
commit 1c11332fa4
40 changed files with 1066 additions and 101 deletions
--- a/Content.Client/Construction/UI/FlatpackCreatorMenu.xaml.cs
+++ b/Content.Client/Construction/UI/FlatpackCreatorMenu.xaml.cs
@@ -0,0 +1,147 @@
+using System.Linq;
+using Content.Client.Materials;
+using Content.Client.Message;
+using Content.Client.UserInterface.Controls;
+using Content.Shared.Construction.Components;
+using Content.Shared.Containers.ItemSlots;
+using Content.Shared.Materials;
+using Robust.Client.AutoGenerated;
+using Robust.Client.GameObjects;
+using Robust.Client.UserInterface.XAML;
+using Robust.Shared.Prototypes;
+using Robust.Shared.Timing;
+using Robust.Shared.Utility;
+
+namespace Content.Client.Construction.UI;
+
+[GenerateTypedNameReferences]
+public sealed partial class FlatpackCreatorMenu : FancyWindow
+{
+    [Dependency] private readonly IEntityManager _entityManager = default!;
+    [Dependency] private readonly IPrototypeManager _prototypeManager = default!;
+
+    private readonly ItemSlotsSystem _itemSlots;
+    private readonly FlatpackSystem _flatpack;
+    private readonly MaterialStorageSystem _materialStorage;
+    private readonly SpriteSystem _spriteSystem;
+
+    private readonly EntityUid _owner;
+
+    private EntityUid? _currentBoard = EntityUid.Invalid;
+    private EntityUid? _machinePreview;
+
+    public event Action? PackButtonPressed;
+
+    public FlatpackCreatorMenu(EntityUid uid)
+    {
+        RobustXamlLoader.Load(this);
+        IoCManager.InjectDependencies(this);
+
+        _itemSlots = _entityManager.System<ItemSlotsSystem>();
+        _flatpack = _entityManager.System<FlatpackSystem>();
+        _materialStorage = _entityManager.System<MaterialStorageSystem>();
+        _spriteSystem = _entityManager.System<SpriteSystem>();
+
+        _owner = uid;
+
+        PackButton.OnPressed += _ => PackButtonPressed?.Invoke();
+
+        MaterialStorageControl.SetOwner(uid);
+    }
+
+    protected override void FrameUpdate(FrameEventArgs args)
+    {
+        base.FrameUpdate(args);
+        
+        if (_machinePreview is not { } && _entityManager.Deleted(_machinePreview))
+        {
+            _machinePreview = null;
+            MachineSprite.SetEntity(_machinePreview);
+        }
+
+        if (!_entityManager.TryGetComponent<FlatpackCreatorComponent>(_owner, out var flatpacker) ||
+            !_itemSlots.TryGetSlot(_owner, flatpacker.SlotId, out var itemSlot))
+            return;
+
+        if (flatpacker.Packing)
+        {
+            PackButton.Disabled = true;
+        }
+        else if (_currentBoard != null)
+        {
+            //todo double trycomp is kinda stinky.
+            if (_entityManager.TryGetComponent<MachineBoardComponent>(_currentBoard, out var board) &&
+                board.Prototype != null)
+            {
+                var cost = _flatpack.GetFlatpackCreationCost((_owner, flatpacker),
+                    (_currentBoard.Value, board));
+                PackButton.Disabled = !_materialStorage.CanChangeMaterialAmount(_owner, cost);
+            }
+        }
+
+        if (_currentBoard == itemSlot.Item)
+            return;
+
+        if (_machinePreview != null)
+            _entityManager.DeleteEntity(_machinePreview);
+
+        _currentBoard = itemSlot.Item;
+        CostHeaderLabel.Visible = _currentBoard != null;
+
+        if (_currentBoard != null &&
+            _entityManager.TryGetComponent<MachineBoardComponent>(_currentBoard, out var machineBoard) &&
+            machineBoard.Prototype != null)
+        {
+            var proto = _prototypeManager.Index<EntityPrototype>(machineBoard.Prototype);
+            _machinePreview = _entityManager.Spawn(proto.ID);
+            _spriteSystem.ForceUpdate(_machinePreview.Value);
+            MachineNameLabel.SetMessage(proto.Name);
+            var cost = _flatpack.GetFlatpackCreationCost((_owner, flatpacker),
+                (_currentBoard.Value, machineBoard));
+            CostLabel.SetMarkup(GetCostString(cost));
+        }
+        else
+        {
+            _machinePreview = null;
+            MachineNameLabel.SetMessage(" ");
+            CostLabel.SetMessage(Loc.GetString("flatpacker-ui-no-board-label"));
+            PackButton.Disabled = true;
+        }
+
+        MachineSprite.SetEntity(_machinePreview);
+    }
+
+    //todo beautify
+    private string GetCostString(Dictionary<string, int> costs)
+    {
+        var orderedCosts = costs.OrderBy(p => p.Value);
+        var msg = new FormattedMessage();
+        foreach (var (mat, amount) in orderedCosts)
+        {
+            var matProto = _prototypeManager.Index<MaterialPrototype>(mat);
+
+            var sheetVolume = _materialStorage.GetSheetVolume(matProto);
+            var sheets = (float) -amount / sheetVolume;
+            var amountText = Loc.GetString("lathe-menu-material-amount",
+                ("amount", sheets),
+                ("unit", Loc.GetString(matProto.Unit)));
+            var text = Loc.GetString("lathe-menu-tooltip-display",
+                ("amount", amountText),
+                ("material", Loc.GetString(matProto.Name)));
+
+            msg.AddMarkup(text);
+            msg.PushNewline();
+        }
+        msg.Pop();
+
+        return msg.ToMarkup();
+    }
+
+    public override void Close()
+    {
+        base.Close();
+
+        _entityManager.DeleteEntity(_machinePreview);
+        _machinePreview = null;
+    }
+}