91 lines
3.1 KiB
C#
91 lines
3.1 KiB
C#
using Content.Client._Horizon.RCD;
|
|
using Content.Shared.Hands.Components;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.RCD;
|
|
using Content.Shared.RCD.Components;
|
|
using Content.Shared.RCD.Systems;
|
|
using Robust.Client.Placement;
|
|
using Robust.Client.Player;
|
|
using Robust.Shared.Enums;
|
|
using Robust.Shared.Prototypes;
|
|
|
|
namespace Content.Client.RCD;
|
|
|
|
public sealed class RCDConstructionGhostSystem : EntitySystem
|
|
{
|
|
[Dependency] private readonly IPlayerManager _playerManager = default!;
|
|
[Dependency] private readonly IPlacementManager _placementManager = default!;
|
|
[Dependency] private readonly IPrototypeManager _protoManager = default!;
|
|
|
|
private string _placementMode = typeof(AlignRCDConstruction).Name;
|
|
private Direction _placementDirection = default;
|
|
|
|
public override void Update(float frameTime)
|
|
{
|
|
base.Update(frameTime);
|
|
|
|
// Get current placer data
|
|
var placerEntity = _placementManager.CurrentPermission?.MobUid;
|
|
var placerProto = _placementManager.CurrentPermission?.EntityType;
|
|
var placerIsRCD = HasComp<RCDComponent>(placerEntity);
|
|
|
|
// Exit if erasing or the current placer is not an RCD (build mode is active)
|
|
if (_placementManager.Eraser || (placerEntity != null && !placerIsRCD))
|
|
return;
|
|
|
|
// Determine if player is carrying an RCD in their active hand
|
|
var player = _playerManager.LocalSession?.AttachedEntity;
|
|
|
|
if (!TryComp<HandsComponent>(player, out var hands))
|
|
return;
|
|
|
|
var heldEntity = hands.ActiveHand?.HeldEntity;
|
|
|
|
// Horizon start
|
|
if (player.HasValue)
|
|
{
|
|
var ev = new GetRCDEntityEvent();
|
|
RaiseLocalEvent(player.Value, ref ev);
|
|
|
|
if (ev.Entity.HasValue)
|
|
heldEntity = ev.Entity.Value;
|
|
}
|
|
// Horizon end
|
|
|
|
if (!TryComp<RCDComponent>(heldEntity, out var rcd))
|
|
{
|
|
// If the player was holding an RCD, but is no longer, cancel placement
|
|
if (placerIsRCD)
|
|
_placementManager.Clear();
|
|
|
|
return;
|
|
}
|
|
var prototype = _protoManager.Index(rcd.ProtoId);
|
|
|
|
// Update the direction the RCD prototype based on the placer direction
|
|
if (_placementDirection != _placementManager.Direction)
|
|
{
|
|
_placementDirection = _placementManager.Direction;
|
|
RaiseNetworkEvent(new RCDConstructionGhostRotationEvent(GetNetEntity(heldEntity.Value), _placementDirection));
|
|
}
|
|
|
|
// If the placer has not changed, exit
|
|
if (heldEntity == placerEntity && prototype.Prototype == placerProto)
|
|
return;
|
|
|
|
// Create a new placer
|
|
var newObjInfo = new PlacementInformation
|
|
{
|
|
MobUid = heldEntity.Value,
|
|
PlacementOption = _placementMode,
|
|
EntityType = prototype.Prototype,
|
|
Range = (int) Math.Ceiling(rcd.Range), // Horizon - custom rcd range
|
|
IsTile = (prototype.Mode == RcdMode.ConstructTile),
|
|
UseEditorContext = false,
|
|
};
|
|
|
|
_placementManager.Clear();
|
|
_placementManager.BeginPlacing(newObjInfo);
|
|
}
|
|
}
|