#nullable enable using System; using Content.Server.GameObjects.Components.GUI; using Content.Server.GameObjects.Components.Items.Storage; using Content.Shared.Audio; using Content.Shared.GameObjects.EntitySystems; using Content.Shared.GameObjects.Verbs; using Robust.Server.GameObjects.Components.Container; using Robust.Server.GameObjects.EntitySystems; using Robust.Server.Interfaces.GameObjects; using Robust.Shared.GameObjects; using Robust.Shared.GameObjects.Systems; using Robust.Shared.Interfaces.GameObjects; using Robust.Shared.Localization; using Robust.Shared.Serialization; using Robust.Shared.Utility; using Robust.Shared.ViewVariables; namespace Content.Server.GameObjects.Components.Power { /// /// Provides a "battery compartment" that can contain a of the matching /// . Intended to supplement other components, not very useful by itself. /// [RegisterComponent] public class PowerCellSlotComponent : Component, IExamine, IMapInit { public override string Name => "PowerCellSlot"; /// /// What size of cell fits into this component. /// [ViewVariables(VVAccess.ReadWrite)] public PowerCellSize SlotSize { get; set; } = PowerCellSize.Small; /// /// Can the cell be removed ? /// [ViewVariables(VVAccess.ReadWrite)] public bool CanRemoveCell { get; set; } = true; /// /// Should the "Remove cell" verb be displayed on this component? /// [ViewVariables(VVAccess.ReadWrite)] public bool ShowVerb { get; set; } = true; /// /// String passed to String.Format when showing the description text for this item. /// String.Format is given a single parameter which is the size letter (S/M/L) of the cells this component uses. /// Use null to show no text. /// [ViewVariables(VVAccess.ReadWrite)] public string? DescFormatString { get; set; } /// /// File path to a sound file that should be played when the cell is removed. /// /// "/Audio/Items/pistol_magout.ogg" [ViewVariables(VVAccess.ReadWrite)] public string? CellRemoveSound { get; set; } /// /// File path to a sound file that should be played when a cell is inserted. /// /// "/Audio/Items/pistol_magin.ogg" [ViewVariables(VVAccess.ReadWrite)] public string? CellInsertSound { get; set; } [ViewVariables] private ContainerSlot _cellContainer = default!; [ViewVariables] public PowerCellComponent? Cell { get { if (_cellContainer.ContainedEntity == null) return null; return _cellContainer.ContainedEntity.TryGetComponent(out PowerCellComponent? cell) ? cell : null; } } [ViewVariables] public bool HasCell => Cell != null; /// /// True if we don't want a cell inserted during map init. /// private bool _startEmpty = false; /// /// If not null, this cell type will be inserted at MapInit instead of the default Standard cell. /// private string? _startingCellType = null; public override void ExposeData(ObjectSerializer serializer) { base.ExposeData(serializer); serializer.DataField(this, x => x.SlotSize, "slotSize", PowerCellSize.Small); serializer.DataField(this, x => x.CanRemoveCell, "canRemoveCell", true); serializer.DataField(this, x => x.ShowVerb, "showVerb", true); serializer.DataField(ref _startEmpty, "startEmpty", false); serializer.DataField(ref _startingCellType, "startingCellType", null); serializer.DataField(this, x => x.CellRemoveSound, "cellRemoveSound", "/Audio/Items/pistol_magin.ogg"); serializer.DataField(this, x => x.CellInsertSound, "cellInsertSound", "/Audio/Items/pistol_magout.ogg"); serializer.DataField(this, x => x.DescFormatString, "descFormatString", "It uses size {0} power cells."); } public override void Initialize() { base.Initialize(); _cellContainer = ContainerManagerComponent.Ensure("cellslot_cell_container", Owner, out _); } void IExamine.Examine(FormattedMessage message, bool inDetailsRange) { if (!inDetailsRange) return; string sizeLetter = SlotSize switch { PowerCellSize.Small => Loc.GetString("S"), PowerCellSize.Medium => Loc.GetString("M"), PowerCellSize.Large => Loc.GetString("L"), _ => "???" }; if (DescFormatString != null) message.AddMarkup(string.Format(DescFormatString, sizeLetter)); } /// /// Remove the cell from this component. If a user is specified, the cell will be put in their hands /// or failing that, at their feet. If no user is specified the cell will be put at the location of /// the parent of this component. /// /// (optional) the user to give the removed cell to. /// Should be played upon removal? /// The cell component of the entity that was removed, or null if removal failed. public PowerCellComponent? EjectCell(IEntity? user, bool playSound = true) { var cell = Cell; if (cell == null || !CanRemoveCell) return null; if (!_cellContainer.Remove(cell.Owner)) return null; //Dirty(); if (user != null) { if (!user.TryGetComponent(out HandsComponent? hands) || !hands.PutInHand(cell.Owner.GetComponent())) { cell.Owner.Transform.Coordinates = user.Transform.Coordinates; } } else { cell.Owner.Transform.Coordinates = Owner.Transform.Coordinates; } if (playSound && CellRemoveSound != null) { EntitySystem.Get().PlayFromEntity(CellRemoveSound, Owner, AudioHelpers.WithVariation(0.125f)); } SendMessage(new PowerCellChangedMessage(true)); return cell; } /// /// Tries to insert the given cell into this component. The cell will be put into the container of this component. /// /// The cell to insert. /// Should be played upon insertion? /// True if insertion succeeded; false otherwise. public bool InsertCell(IEntity cell, bool playSound = true) { if (Cell != null) return false; if (!cell.TryGetComponent(out var _)) return false; if (!cell.TryGetComponent(out var cellComponent)) return false; if (cellComponent.CellSize != SlotSize) return false; if (!_cellContainer.Insert(cell)) return false; //Dirty(); if (playSound && CellInsertSound != null) { EntitySystem.Get().PlayFromEntity(CellInsertSound, Owner, AudioHelpers.WithVariation(0.125f)); } SendMessage(new PowerCellChangedMessage(false)); return true; } [Verb] public sealed class EjectCellVerb : Verb { protected override void GetData(IEntity user, PowerCellSlotComponent component, VerbData data) { if (!component.ShowVerb || !ActionBlockerSystem.CanInteract(user)) { data.Visibility = VerbVisibility.Invisible; return; } if (component.Cell == null) { data.Text = Loc.GetString("Eject cell (cell missing)"); } else { data.Text = Loc.GetString("Eject cell"); } if (component.Cell == null || !component.CanRemoveCell) { data.Visibility = VerbVisibility.Disabled; } } protected override void Activate(IEntity user, PowerCellSlotComponent component) { component.EjectCell(user); } } void IMapInit.MapInit() { if (_startEmpty || _cellContainer.ContainedEntity != null) { return; } string type; if (_startingCellType != null) { type = _startingCellType; } else { type = SlotSize switch { PowerCellSize.Small => "PowerCellSmallStandard", PowerCellSize.Medium => "PowerCellMediumStandard", PowerCellSize.Large => "PowerCellLargeStandard", _ => throw new ArgumentOutOfRangeException() }; } var cell = Owner.EntityManager.SpawnEntity(type, Owner.Transform.Coordinates); _cellContainer.Insert(cell); } } public class PowerCellChangedMessage : ComponentMessage { /// /// If true, the cell was ejected; if false, it was inserted. /// public bool Ejected { get; } public PowerCellChangedMessage(bool ejected) { Ejected = ejected; } } }