95 lines
2.9 KiB
C#
95 lines
2.9 KiB
C#
using Content.Client.UserInterface.Systems.Gameplay;
|
|
using Content.Client.UserInterface.Systems.Hands;
|
|
using Content.Client.UserInterface.Systems.Hands.Controls;
|
|
using Content.Client.UserInterface.Systems.Hotbar.Widgets;
|
|
using Content.Client.UserInterface.Systems.Inventory;
|
|
using Content.Client.UserInterface.Systems.Inventory.Controls;
|
|
using Content.Client.UserInterface.Systems.Inventory.Widgets;
|
|
using Content.Client.UserInterface.Systems.Storage;
|
|
using Content.Client.UserInterface.Systems.Storage.Controls;
|
|
using Robust.Client.UserInterface;
|
|
using Robust.Client.UserInterface.Controllers;
|
|
|
|
namespace Content.Client.UserInterface.Systems.Hotbar;
|
|
|
|
public sealed class HotbarUIController : UIController
|
|
{
|
|
private InventoryUIController? _inventory;
|
|
private HandsUIController? _hands;
|
|
private StorageUIController? _storage;
|
|
|
|
public override void Initialize()
|
|
{
|
|
base.Initialize();
|
|
|
|
var gameplayStateLoad = UIManager.GetUIController<GameplayStateLoadController>();
|
|
gameplayStateLoad.OnScreenLoad += OnScreenLoad;
|
|
}
|
|
|
|
private void OnScreenLoad()
|
|
{
|
|
ReloadHotbar();
|
|
}
|
|
|
|
public void Setup(HandsContainer handsContainer, StorageContainer storageContainer)
|
|
{
|
|
_inventory = UIManager.GetUIController<InventoryUIController>();
|
|
_hands = UIManager.GetUIController<HandsUIController>();
|
|
_storage = UIManager.GetUIController<StorageUIController>();
|
|
_hands.RegisterHandContainer(handsContainer);
|
|
_storage.RegisterStorageContainer(storageContainer);
|
|
}
|
|
|
|
public void ReloadHotbar()
|
|
{
|
|
if (UIManager.ActiveScreen == null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
if (UIManager.ActiveScreen.GetWidget<HotbarGui>() is { } hotbar)
|
|
{
|
|
foreach (var container in GetAllItemSlotContainers(hotbar))
|
|
{
|
|
// Yes, this is dirty.
|
|
container.SlotGroup = container.SlotGroup;
|
|
}
|
|
}
|
|
|
|
_hands?.ReloadHands();
|
|
_inventory?.ReloadSlots();
|
|
|
|
//todo move this over to its own hellhole
|
|
var inventory = UIManager.ActiveScreen.GetWidget<InventoryGui>();
|
|
if (inventory == null)
|
|
{
|
|
return;
|
|
}
|
|
|
|
foreach (var container in GetAllItemSlotContainers(inventory))
|
|
{
|
|
// Yes, this is dirty.
|
|
container.SlotGroup = container.SlotGroup;
|
|
}
|
|
|
|
_inventory?.RegisterInventoryBarContainer(inventory.InventoryHotbar);
|
|
}
|
|
|
|
private static IEnumerable<ItemSlotButtonContainer> GetAllItemSlotContainers(Control gui)
|
|
{
|
|
var result = new List<ItemSlotButtonContainer>();
|
|
|
|
foreach (var child in gui.Children)
|
|
{
|
|
if (child is ItemSlotButtonContainer container)
|
|
{
|
|
result.Add(container);
|
|
}
|
|
|
|
result.AddRange(GetAllItemSlotContainers(child));
|
|
}
|
|
|
|
return result;
|
|
}
|
|
}
|