using System; using System.Collections.Generic; using Content.Shared.GameObjects.Components.Storage; using Content.Client.Interfaces.GameObjects; using Robust.Client.Interfaces.GameObjects.Components; using Robust.Client.UserInterface; using Robust.Client.UserInterface.Controls; using Robust.Client.UserInterface.CustomControls; using Robust.Client.Player; using Robust.Shared.GameObjects; using Robust.Shared.Interfaces.GameObjects; using Robust.Shared.Interfaces.Network; using Robust.Shared.IoC; using Robust.Shared.Maths; using Robust.Shared.Players; namespace Content.Client.GameObjects.Components.Storage { /// /// Client version of item storage containers, contains a UI which displays stored entities and their size /// [RegisterComponent] public class ClientStorageComponent : SharedStorageComponent { private Dictionary StoredEntities { get; set; } = new Dictionary(); private int StorageSizeUsed; private int StorageCapacityMax; private StorageWindow Window; public override void OnAdd() { base.OnAdd(); Window = new StorageWindow() {StorageEntity = this};1 } public override void OnRemove() { Window.Dispose(); base.OnRemove(); } public override void HandleNetworkMessage(ComponentMessage message, INetChannel channel, ICommonSession session = null) { base.HandleNetworkMessage(message, channel, session); switch (message) { //Updates what we are storing for the UI case StorageHeldItemsMessage msg: HandleStorageMessage(msg); break; //Opens the UI case OpenStorageUIMessage _: OpenUI(); break; case CloseStorageUIMessage _: CloseUI(); break; } } /// /// Copies received values from server about contents of storage container /// /// private void HandleStorageMessage(StorageHeldItemsMessage storagestate) { StoredEntities = new Dictionary(storagestate.StoredEntities); StorageSizeUsed = storagestate.StorageSizeUsed; StorageCapacityMax = storagestate.StorageSizeMax; Window.BuildEntityList(); } /// /// Opens the storage UI /// private void OpenUI() { Window.Open(); } private void CloseUI() { Window.Close(); } /// /// Function for clicking one of the stored entity buttons in the UI, tells server to remove that entity /// /// private void Interact(EntityUid entityuid) { SendNetworkMessage(new RemoveEntityMessage(entityuid)); } /// /// GUI class for client storage component /// private class StorageWindow : SS14Window { private Control VSplitContainer; private VBoxContainer EntityList; private Label Information; public ClientStorageComponent StorageEntity; protected override Vector2? CustomSize => (180, 320); public StorageWindow() { Title = "Storage Item"; RectClipContent = true; var containerButton = new ContainerButton { SizeFlagsHorizontal = SizeFlags.Fill, SizeFlagsVertical = SizeFlags.Fill }; containerButton.OnPressed += args => { var controlledEntity = IoCManager.Resolve().LocalPlayer.ControlledEntity; if (controlledEntity.TryGetComponent(out IHandsComponent hands)) { StorageEntity.SendNetworkMessage(new InsertEntityMessage()); } }; VSplitContainer = new VBoxContainer(); containerButton.AddChild(VSplitContainer); Information = new Label { Text = "Items: 0 Volume: 0/0 Stuff", SizeFlagsVertical = SizeFlags.ShrinkCenter }; VSplitContainer.AddChild(Information); var listScrollContainer = new ScrollContainer { SizeFlagsVertical = SizeFlags.FillExpand, SizeFlagsHorizontal = SizeFlags.FillExpand, HScrollEnabled = true, VScrollEnabled = true }; EntityList = new VBoxContainer { SizeFlagsHorizontal = SizeFlags.FillExpand }; listScrollContainer.AddChild(EntityList); VSplitContainer.AddChild(listScrollContainer); Contents.AddChild(containerButton); } public override void Close() { StorageEntity.SendNetworkMessage(new CloseStorageUIMessage()); base.Close(); } /// /// Loops through stored entities creating buttons for each, updates information labels /// public void BuildEntityList() { EntityList.DisposeAllChildren(); var storagelist = StorageEntity.StoredEntities; foreach (var entityuid in storagelist) { var entity = IoCManager.Resolve().GetEntity(entityuid.Key); var button = new EntityButton() { EntityuID = entityuid.Key, MouseFilter = MouseFilterMode.Stop, }; button.ActualButton.OnToggled += OnItemButtonToggled; //Name and Size labels set button.EntityName.Text = entity.Name; button.EntitySize.Text = string.Format("{0}", entityuid.Value); //Gets entity sprite and assigns it to button texture if (entity.TryGetComponent(out ISpriteComponent sprite)) { button.EntitySpriteView.Sprite = sprite; } EntityList.AddChild(button); } //Sets information about entire storage container current capacity if (StorageEntity.StorageCapacityMax != 0) { Information.Text = String.Format("Items: {0}, Stored: {1}/{2}", storagelist.Count, StorageEntity.StorageSizeUsed, StorageEntity.StorageCapacityMax); } else { Information.Text = String.Format("Items: {0}", storagelist.Count); } } /// /// Function assigned to button toggle which removes the entity from storage /// /// private void OnItemButtonToggled(BaseButton.ButtonToggledEventArgs args) { var control = (EntityButton) args.Button.Parent; args.Button.Pressed = false; StorageEntity.Interact(control.EntityuID); } /// /// Function assigned to button that adds items to the storage entity. /// private void OnAddItemButtonPressed(BaseButton.ButtonEventArgs args) { var controlledEntity = IoCManager.Resolve().LocalPlayer.ControlledEntity; if (controlledEntity.TryGetComponent(out IHandsComponent hands)) { StorageEntity.SendNetworkMessage(new InsertEntityMessage()); } } } /// /// Button created for each entity that represents that item in the storage UI, with a texture, and name and size label /// private class EntityButton : PanelContainer { public EntityUid EntityuID { get; set; } public Button ActualButton { get; } public SpriteView EntitySpriteView { get; } public Control EntityControl { get; } public Label EntityName { get; } public Label EntitySize { get; } public EntityButton() { ActualButton = new Button { SizeFlagsHorizontal = SizeFlags.FillExpand, SizeFlagsVertical = SizeFlags.FillExpand, ToggleMode = true, MouseFilter = MouseFilterMode.Stop }; AddChild(ActualButton); var hBoxContainer = new HBoxContainer(); EntitySpriteView = new SpriteView { CustomMinimumSize = new Vector2(32.0f, 32.0f) }; EntityName = new Label { SizeFlagsVertical = SizeFlags.ShrinkCenter, Text = "Backpack", }; hBoxContainer.AddChild(EntitySpriteView); hBoxContainer.AddChild(EntityName); EntityControl = new Control { SizeFlagsHorizontal = SizeFlags.FillExpand }; EntitySize = new Label { SizeFlagsVertical = SizeFlags.ShrinkCenter, Text = "Size 6", Align = Label.AlignMode.Right, /*AnchorLeft = 1.0f, AnchorRight = 1.0f, AnchorBottom = 0.5f, AnchorTop = 0.5f, MarginLeft = -38.0f, MarginTop = -7.0f, MarginRight = -5.0f, MarginBottom = 7.0f*/ }; EntityControl.AddChild(EntitySize); hBoxContainer.AddChild(EntityControl); AddChild(hBoxContainer); } } } }