#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.EntitySystems.ActionBlocker;
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;
}
}
}