#nullable enable
using System;
using System.Threading.Tasks;
using Content.Server.GameObjects.Components.Interactable;
using Content.Server.GameObjects.Components.Pulling;
using Content.Server.Utility;
using Content.Shared.GameObjects.Components.Interactable;
using Content.Shared.Interfaces.GameObjects.Components;
using Robust.Shared.GameObjects;
using Robust.Shared.Serialization.Manager.Attributes;
using Robust.Shared.Physics;
using Robust.Shared.ViewVariables;
namespace Content.Server.GameObjects.Components
{
// TODO: Move this component's logic to an EntitySystem.
[RegisterComponent]
public class AnchorableComponent : Component, IInteractUsing
{
public override string Name => "Anchorable";
[ComponentDependency] private PhysicsComponent? _physicsComponent = default!;
[ViewVariables]
[DataField("tool")]
public ToolQuality Tool { get; private set; } = ToolQuality.Anchoring;
[ViewVariables]
int IInteractUsing.Priority => 1;
[ViewVariables(VVAccess.ReadWrite)]
[DataField("snap")]
public bool Snap { get; private set; } = true;
///
/// Checks if a tool can change the anchored status.
///
/// The user doing the action
/// The tool being used
/// True if we're anchoring, and false if we're unanchoring.
/// true if it is valid, false otherwise
private async Task Valid(IEntity user, IEntity utilizing, bool anchoring)
{
if (!Owner.HasComponent())
{
return false;
}
BaseAnchoredAttemptEvent attempt =
anchoring ? new AnchorAttemptEvent(user, utilizing) : new UnanchorAttemptEvent(user, utilizing);
Owner.EntityManager.EventBus.RaiseLocalEvent(Owner.Uid, attempt, false);
if (attempt.Cancelled)
return false;
return utilizing.TryGetComponent(out ToolComponent? tool) && await tool.UseTool(user, Owner, 0.5f, Tool);
}
///
/// Tries to anchor the owner of this component.
///
/// The entity doing the anchoring
/// The tool being used
/// true if anchored, false otherwise
public async Task TryAnchor(IEntity user, IEntity utilizing)
{
if (!(await Valid(user, utilizing, true)))
{
return false;
}
if (_physicsComponent == null)
return false;
// Snap rotation to cardinal (multiple of 90)
var rot = Owner.Transform.LocalRotation;
Owner.Transform.LocalRotation = Math.Round(rot / (Math.PI / 2)) * (Math.PI / 2);
if (Owner.TryGetComponent(out PullableComponent? pullableComponent))
{
if (pullableComponent.Puller != null)
{
pullableComponent.TryStopPull();
}
}
if (Snap)
Owner.SnapToGrid(Owner.EntityManager);
_physicsComponent.BodyType = BodyType.Static;
Owner.EntityManager.EventBus.RaiseLocalEvent(Owner.Uid, new AnchoredEvent(user, utilizing), false);
return true;
}
///
/// Tries to unanchor the owner of this component.
///
/// The entity doing the unanchoring
/// The tool being used, if any
/// Whether or not to ignore valid tool checks
/// true if unanchored, false otherwise
public async Task TryUnAnchor(IEntity user, IEntity utilizing)
{
if (!(await Valid(user, utilizing, false)))
{
return false;
}
if (_physicsComponent == null)
return false;
_physicsComponent.BodyType = BodyType.Dynamic;
Owner.EntityManager.EventBus.RaiseLocalEvent(Owner.Uid, new UnanchoredEvent(user, utilizing), false);
return true;
}
///
/// Tries to toggle the anchored status of this component's owner.
///
/// The entity doing the unanchoring
/// The tool being used
/// true if toggled, false otherwise
public async Task TryToggleAnchor(IEntity user, IEntity utilizing)
{
if (_physicsComponent == null)
return false;
return _physicsComponent.BodyType == BodyType.Static ?
await TryUnAnchor(user, utilizing) :
await TryAnchor(user, utilizing);
}
async Task IInteractUsing.InteractUsing(InteractUsingEventArgs eventArgs)
{
return await TryToggleAnchor(eventArgs.User, eventArgs.Using);
}
}
public abstract class BaseAnchoredAttemptEvent : CancellableEntityEventArgs
{
public IEntity User { get; }
public IEntity Tool { get; }
protected BaseAnchoredAttemptEvent(IEntity user, IEntity tool)
{
User = user;
Tool = tool;
}
}
public class AnchorAttemptEvent : BaseAnchoredAttemptEvent
{
public AnchorAttemptEvent(IEntity user, IEntity tool) : base(user, tool) { }
}
public class UnanchorAttemptEvent : BaseAnchoredAttemptEvent
{
public UnanchorAttemptEvent(IEntity user, IEntity tool) : base(user, tool) { }
}
public abstract class BaseAnchoredEvent : EntityEventArgs
{
public IEntity User { get; }
public IEntity Tool { get; }
protected BaseAnchoredEvent(IEntity user, IEntity tool)
{
User = user;
Tool = tool;
}
}
public class AnchoredEvent : BaseAnchoredEvent
{
public AnchoredEvent(IEntity user, IEntity tool) : base(user, tool) { }
}
public class UnanchoredEvent : BaseAnchoredEvent
{
public UnanchoredEvent(IEntity user, IEntity tool) : base(user, tool) { }
}
}