action refactor proper ecs edition (#27422)
This commit is contained in:
@@ -1,18 +1,23 @@
|
||||
using System.IO;
|
||||
using System.Linq;
|
||||
using Content.Shared.Actions;
|
||||
using Content.Shared.Actions.Components;
|
||||
using Content.Shared.Charges.Systems;
|
||||
using Content.Shared.Mapping;
|
||||
using Content.Shared.Maps;
|
||||
using JetBrains.Annotations;
|
||||
using Robust.Client.Player;
|
||||
using Robust.Shared.ContentPack;
|
||||
using Robust.Shared.GameStates;
|
||||
using Robust.Shared.Input.Binding;
|
||||
using Robust.Shared.Player;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Serialization.Manager;
|
||||
using Robust.Shared.Serialization.Markdown;
|
||||
using Robust.Shared.Serialization.Markdown.Mapping;
|
||||
using Robust.Shared.Serialization.Markdown.Sequence;
|
||||
using Robust.Shared.Serialization.Markdown.Value;
|
||||
using Robust.Shared.Timing;
|
||||
using Robust.Shared.Utility;
|
||||
using YamlDotNet.RepresentationModel;
|
||||
|
||||
@@ -25,6 +30,7 @@ namespace Content.Client.Actions
|
||||
|
||||
[Dependency] private readonly SharedChargesSystem _sharedCharges = default!;
|
||||
[Dependency] private readonly IPlayerManager _playerManager = default!;
|
||||
[Dependency] private readonly IPrototypeManager _proto = default!;
|
||||
[Dependency] private readonly IResourceManager _resources = default!;
|
||||
[Dependency] private readonly ISerializationManager _serialization = default!;
|
||||
[Dependency] private readonly MetaDataSystem _metaData = default!;
|
||||
@@ -38,131 +44,67 @@ namespace Content.Client.Actions
|
||||
public event Action<List<SlotAssignment>>? AssignSlot;
|
||||
|
||||
private readonly List<EntityUid> _removed = new();
|
||||
private readonly List<(EntityUid, BaseActionComponent?)> _added = new();
|
||||
private readonly List<Entity<ActionComponent>> _added = new();
|
||||
|
||||
public static readonly EntProtoId MappingEntityAction = "BaseMappingEntityAction";
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
SubscribeLocalEvent<ActionsComponent, LocalPlayerAttachedEvent>(OnPlayerAttached);
|
||||
SubscribeLocalEvent<ActionsComponent, LocalPlayerDetachedEvent>(OnPlayerDetached);
|
||||
SubscribeLocalEvent<ActionsComponent, ComponentHandleState>(HandleComponentState);
|
||||
SubscribeLocalEvent<ActionsComponent, ComponentHandleState>(OnHandleState);
|
||||
|
||||
SubscribeLocalEvent<InstantActionComponent, ComponentHandleState>(OnInstantHandleState);
|
||||
SubscribeLocalEvent<EntityTargetActionComponent, ComponentHandleState>(OnEntityTargetHandleState);
|
||||
SubscribeLocalEvent<WorldTargetActionComponent, ComponentHandleState>(OnWorldTargetHandleState);
|
||||
SubscribeLocalEvent<EntityWorldTargetActionComponent, ComponentHandleState>(OnEntityWorldTargetHandleState);
|
||||
SubscribeLocalEvent<ActionComponent, AfterAutoHandleStateEvent>(OnActionAutoHandleState);
|
||||
|
||||
SubscribeLocalEvent<EntityTargetActionComponent, ActionTargetAttemptEvent>(OnEntityTargetAttempt);
|
||||
SubscribeLocalEvent<WorldTargetActionComponent, ActionTargetAttemptEvent>(OnWorldTargetAttempt);
|
||||
}
|
||||
|
||||
private void OnInstantHandleState(EntityUid uid, InstantActionComponent component, ref ComponentHandleState args)
|
||||
{
|
||||
if (args.Current is not InstantActionComponentState state)
|
||||
return;
|
||||
|
||||
BaseHandleState<InstantActionComponent>(uid, component, state);
|
||||
private void OnActionAutoHandleState(Entity<ActionComponent> ent, ref AfterAutoHandleStateEvent args)
|
||||
{
|
||||
UpdateAction(ent);
|
||||
}
|
||||
|
||||
private void OnEntityTargetHandleState(EntityUid uid, EntityTargetActionComponent component, ref ComponentHandleState args)
|
||||
public override void UpdateAction(Entity<ActionComponent> ent)
|
||||
{
|
||||
if (args.Current is not EntityTargetActionComponentState state)
|
||||
return;
|
||||
|
||||
component.Whitelist = state.Whitelist;
|
||||
component.Blacklist = state.Blacklist;
|
||||
component.CanTargetSelf = state.CanTargetSelf;
|
||||
BaseHandleState<EntityTargetActionComponent>(uid, component, state);
|
||||
}
|
||||
|
||||
private void OnWorldTargetHandleState(EntityUid uid, WorldTargetActionComponent component, ref ComponentHandleState args)
|
||||
{
|
||||
if (args.Current is not WorldTargetActionComponentState state)
|
||||
return;
|
||||
|
||||
BaseHandleState<WorldTargetActionComponent>(uid, component, state);
|
||||
}
|
||||
|
||||
private void OnEntityWorldTargetHandleState(EntityUid uid,
|
||||
EntityWorldTargetActionComponent component,
|
||||
ref ComponentHandleState args)
|
||||
{
|
||||
if (args.Current is not EntityWorldTargetActionComponentState state)
|
||||
return;
|
||||
|
||||
component.Whitelist = state.Whitelist;
|
||||
component.CanTargetSelf = state.CanTargetSelf;
|
||||
BaseHandleState<EntityWorldTargetActionComponent>(uid, component, state);
|
||||
}
|
||||
|
||||
private void BaseHandleState<T>(EntityUid uid, BaseActionComponent component, BaseActionComponentState state) where T : BaseActionComponent
|
||||
{
|
||||
// TODO ACTIONS use auto comp states
|
||||
component.Icon = state.Icon;
|
||||
component.IconOn = state.IconOn;
|
||||
component.IconColor = state.IconColor;
|
||||
component.OriginalIconColor = state.OriginalIconColor;
|
||||
component.DisabledIconColor = state.DisabledIconColor;
|
||||
component.Keywords.Clear();
|
||||
component.Keywords.UnionWith(state.Keywords);
|
||||
component.Enabled = state.Enabled;
|
||||
component.Toggled = state.Toggled;
|
||||
component.Cooldown = state.Cooldown;
|
||||
component.UseDelay = state.UseDelay;
|
||||
component.Container = EnsureEntity<T>(state.Container, uid);
|
||||
component.EntityIcon = EnsureEntity<T>(state.EntityIcon, uid);
|
||||
component.CheckCanInteract = state.CheckCanInteract;
|
||||
component.CheckConsciousness = state.CheckConsciousness;
|
||||
component.ClientExclusive = state.ClientExclusive;
|
||||
component.Priority = state.Priority;
|
||||
component.AttachedEntity = EnsureEntity<T>(state.AttachedEntity, uid);
|
||||
component.RaiseOnUser = state.RaiseOnUser;
|
||||
component.RaiseOnAction = state.RaiseOnAction;
|
||||
component.AutoPopulate = state.AutoPopulate;
|
||||
component.Temporary = state.Temporary;
|
||||
component.ItemIconStyle = state.ItemIconStyle;
|
||||
component.Sound = state.Sound;
|
||||
|
||||
UpdateAction(uid, component);
|
||||
}
|
||||
|
||||
public override void UpdateAction(EntityUid? actionId, BaseActionComponent? action = null)
|
||||
{
|
||||
if (!ResolveActionData(actionId, ref action))
|
||||
return;
|
||||
|
||||
// TODO: Decouple this.
|
||||
action.IconColor = _sharedCharges.GetCurrentCharges(actionId.Value) == 0 ? action.DisabledIconColor : action.OriginalIconColor;
|
||||
|
||||
base.UpdateAction(actionId, action);
|
||||
if (_playerManager.LocalEntity != action.AttachedEntity)
|
||||
ent.Comp.IconColor = _sharedCharges.GetCurrentCharges(ent.Owner) == 0 ? ent.Comp.DisabledIconColor : ent.Comp.OriginalIconColor;
|
||||
base.UpdateAction(ent);
|
||||
if (_playerManager.LocalEntity != ent.Comp.AttachedEntity)
|
||||
return;
|
||||
|
||||
ActionsUpdated?.Invoke();
|
||||
}
|
||||
|
||||
private void HandleComponentState(EntityUid uid, ActionsComponent component, ref ComponentHandleState args)
|
||||
private void OnHandleState(Entity<ActionsComponent> ent, ref ComponentHandleState args)
|
||||
{
|
||||
if (args.Current is not ActionsComponentState state)
|
||||
return;
|
||||
|
||||
var (uid, comp) = ent;
|
||||
_added.Clear();
|
||||
_removed.Clear();
|
||||
var stateEnts = EnsureEntitySet<ActionsComponent>(state.Actions, uid);
|
||||
foreach (var act in component.Actions)
|
||||
foreach (var act in comp.Actions)
|
||||
{
|
||||
if (!stateEnts.Contains(act) && !IsClientSide(act))
|
||||
_removed.Add(act);
|
||||
}
|
||||
component.Actions.ExceptWith(_removed);
|
||||
comp.Actions.ExceptWith(_removed);
|
||||
|
||||
foreach (var actionId in stateEnts)
|
||||
{
|
||||
if (!actionId.IsValid())
|
||||
continue;
|
||||
|
||||
if (!component.Actions.Add(actionId))
|
||||
if (!comp.Actions.Add(actionId))
|
||||
continue;
|
||||
|
||||
TryGetActionData(actionId, out var action);
|
||||
_added.Add((actionId, action));
|
||||
if (GetAction(actionId) is {} action)
|
||||
_added.Add(action);
|
||||
}
|
||||
|
||||
if (_playerManager.LocalEntity != uid)
|
||||
@@ -177,47 +119,46 @@ namespace Content.Client.Actions
|
||||
|
||||
foreach (var action in _added)
|
||||
{
|
||||
OnActionAdded?.Invoke(action.Item1);
|
||||
OnActionAdded?.Invoke(action);
|
||||
}
|
||||
|
||||
ActionsUpdated?.Invoke();
|
||||
}
|
||||
|
||||
public static int ActionComparer((EntityUid, BaseActionComponent?) a, (EntityUid, BaseActionComponent?) b)
|
||||
public static int ActionComparer(Entity<ActionComponent> a, Entity<ActionComponent> b)
|
||||
{
|
||||
var priorityA = a.Item2?.Priority ?? 0;
|
||||
var priorityB = b.Item2?.Priority ?? 0;
|
||||
var priorityA = a.Comp?.Priority ?? 0;
|
||||
var priorityB = b.Comp?.Priority ?? 0;
|
||||
if (priorityA != priorityB)
|
||||
return priorityA - priorityB;
|
||||
|
||||
priorityA = a.Item2?.Container?.Id ?? 0;
|
||||
priorityB = b.Item2?.Container?.Id ?? 0;
|
||||
priorityA = a.Comp?.Container?.Id ?? 0;
|
||||
priorityB = b.Comp?.Container?.Id ?? 0;
|
||||
return priorityA - priorityB;
|
||||
}
|
||||
|
||||
protected override void ActionAdded(EntityUid performer, EntityUid actionId, ActionsComponent comp,
|
||||
BaseActionComponent action)
|
||||
protected override void ActionAdded(Entity<ActionsComponent> performer, Entity<ActionComponent> action)
|
||||
{
|
||||
if (_playerManager.LocalEntity != performer)
|
||||
if (_playerManager.LocalEntity != performer.Owner)
|
||||
return;
|
||||
|
||||
OnActionAdded?.Invoke(actionId);
|
||||
OnActionAdded?.Invoke(action);
|
||||
ActionsUpdated?.Invoke();
|
||||
}
|
||||
|
||||
protected override void ActionRemoved(EntityUid performer, EntityUid actionId, ActionsComponent comp, BaseActionComponent action)
|
||||
protected override void ActionRemoved(Entity<ActionsComponent> performer, Entity<ActionComponent> action)
|
||||
{
|
||||
if (_playerManager.LocalEntity != performer)
|
||||
if (_playerManager.LocalEntity != performer.Owner)
|
||||
return;
|
||||
|
||||
OnActionRemoved?.Invoke(actionId);
|
||||
OnActionRemoved?.Invoke(action);
|
||||
ActionsUpdated?.Invoke();
|
||||
}
|
||||
|
||||
public IEnumerable<(EntityUid Id, BaseActionComponent Comp)> GetClientActions()
|
||||
public IEnumerable<Entity<ActionComponent>> GetClientActions()
|
||||
{
|
||||
if (_playerManager.LocalEntity is not { } user)
|
||||
return Enumerable.Empty<(EntityUid, BaseActionComponent)>();
|
||||
return Enumerable.Empty<Entity<ActionComponent>>();
|
||||
|
||||
return GetActions(user);
|
||||
}
|
||||
@@ -254,24 +195,23 @@ namespace Content.Client.Actions
|
||||
CommandBinds.Unregister<ActionsSystem>();
|
||||
}
|
||||
|
||||
public void TriggerAction(EntityUid actionId, BaseActionComponent action)
|
||||
public void TriggerAction(Entity<ActionComponent> action)
|
||||
{
|
||||
if (_playerManager.LocalEntity is not { } user ||
|
||||
!TryComp(user, out ActionsComponent? actions))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (action is not InstantActionComponent instantAction)
|
||||
if (_playerManager.LocalEntity is not { } user)
|
||||
return;
|
||||
|
||||
if (action.ClientExclusive)
|
||||
// TODO: unhardcode this somehow
|
||||
|
||||
if (!HasComp<InstantActionComponent>(action))
|
||||
return;
|
||||
|
||||
if (action.Comp.ClientExclusive)
|
||||
{
|
||||
PerformAction(user, actions, actionId, instantAction, instantAction.Event, GameTiming.CurTime);
|
||||
PerformAction(user, action);
|
||||
}
|
||||
else
|
||||
{
|
||||
var request = new RequestPerformActionEvent(GetNetEntity(actionId));
|
||||
var request = new RequestPerformActionEvent(GetNetEntity(action));
|
||||
EntityManager.RaisePredictiveEvent(request);
|
||||
}
|
||||
}
|
||||
@@ -295,39 +235,137 @@ namespace Content.Client.Actions
|
||||
if (yamlStream.Documents[0].RootNode.ToDataNode() is not SequenceDataNode sequence)
|
||||
return;
|
||||
|
||||
var actions = EnsureComp<ActionsComponent>(user);
|
||||
|
||||
ClearAssignments?.Invoke();
|
||||
|
||||
var assignments = new List<SlotAssignment>();
|
||||
|
||||
foreach (var entry in sequence.Sequence)
|
||||
{
|
||||
if (entry is not MappingDataNode map)
|
||||
continue;
|
||||
|
||||
if (!map.TryGet("action", out var actionNode))
|
||||
continue;
|
||||
|
||||
var action = _serialization.Read<BaseActionComponent>(actionNode, notNullableOverride: true);
|
||||
var actionId = Spawn();
|
||||
AddComp(actionId, action);
|
||||
AddActionDirect(user, actionId);
|
||||
|
||||
if (map.TryGet<ValueDataNode>("name", out var nameNode))
|
||||
_metaData.SetEntityName(actionId, nameNode.Value);
|
||||
|
||||
if (!map.TryGet("assignments", out var assignmentNode))
|
||||
continue;
|
||||
|
||||
var nodeAssignments = _serialization.Read<List<(byte Hotbar, byte Slot)>>(assignmentNode, notNullableOverride: true);
|
||||
|
||||
foreach (var index in nodeAssignments)
|
||||
var actionId = EntityUid.Invalid;
|
||||
if (map.TryGet<ValueDataNode>("action", out var actionNode))
|
||||
{
|
||||
var assignment = new SlotAssignment(index.Hotbar, index.Slot, actionId);
|
||||
assignments.Add(assignment);
|
||||
var id = new EntProtoId(actionNode.Value);
|
||||
actionId = Spawn(id);
|
||||
}
|
||||
else if (map.TryGet<ValueDataNode>("entity", out var entityNode))
|
||||
{
|
||||
var id = new EntProtoId(entityNode.Value);
|
||||
var proto = _proto.Index(id);
|
||||
actionId = Spawn(MappingEntityAction);
|
||||
SetIcon(actionId, new SpriteSpecifier.EntityPrototype(id));
|
||||
SetEvent(actionId, new StartPlacementActionEvent()
|
||||
{
|
||||
PlacementOption = "SnapgridCenter",
|
||||
EntityType = id
|
||||
});
|
||||
_metaData.SetEntityName(actionId, proto.Name);
|
||||
}
|
||||
else if (map.TryGet<ValueDataNode>("tileId", out var tileNode))
|
||||
{
|
||||
var id = new ProtoId<ContentTileDefinition>(tileNode.Value);
|
||||
var proto = _proto.Index(id);
|
||||
actionId = Spawn(MappingEntityAction);
|
||||
if (proto.Sprite is {} sprite)
|
||||
SetIcon(actionId, new SpriteSpecifier.Texture(sprite));
|
||||
SetEvent(actionId, new StartPlacementActionEvent()
|
||||
{
|
||||
PlacementOption = "AlignTileAny",
|
||||
TileId = id
|
||||
});
|
||||
_metaData.SetEntityName(actionId, Loc.GetString(proto.Name));
|
||||
}
|
||||
else
|
||||
{
|
||||
Log.Error($"Mapping actions from {path} had unknown action data!");
|
||||
continue;
|
||||
}
|
||||
|
||||
AddActionDirect((user, actions), actionId);
|
||||
}
|
||||
}
|
||||
|
||||
private void OnWorldTargetAttempt(Entity<WorldTargetActionComponent> ent, ref ActionTargetAttemptEvent args)
|
||||
{
|
||||
if (args.Handled)
|
||||
return;
|
||||
|
||||
args.Handled = true;
|
||||
|
||||
var (uid, comp) = ent;
|
||||
var action = args.Action;
|
||||
var coords = args.Input.Coordinates;
|
||||
var user = args.User;
|
||||
|
||||
if (!ValidateWorldTarget(user, coords, ent))
|
||||
return;
|
||||
|
||||
// optionally send the clicked entity too, if it matches its whitelist etc
|
||||
// this is the actual entity-world targeting magic
|
||||
EntityUid? targetEnt = null;
|
||||
if (TryComp<EntityTargetActionComponent>(ent, out var entity) &&
|
||||
args.Input.EntityUid != null &&
|
||||
ValidateEntityTarget(user, args.Input.EntityUid, (uid, entity)))
|
||||
{
|
||||
targetEnt = args.Input.EntityUid;
|
||||
}
|
||||
|
||||
AssignSlot?.Invoke(assignments);
|
||||
if (action.ClientExclusive)
|
||||
{
|
||||
// TODO: abstract away from single event or maybe just RaiseLocalEvent?
|
||||
if (comp.Event is {} ev)
|
||||
{
|
||||
ev.Target = coords;
|
||||
ev.Entity = targetEnt;
|
||||
}
|
||||
|
||||
PerformAction((user, user.Comp), (uid, action));
|
||||
}
|
||||
else
|
||||
RaisePredictiveEvent(new RequestPerformActionEvent(GetNetEntity(uid), GetNetEntity(targetEnt), GetNetCoordinates(coords)));
|
||||
|
||||
args.FoundTarget = true;
|
||||
}
|
||||
|
||||
private void OnEntityTargetAttempt(Entity<EntityTargetActionComponent> ent, ref ActionTargetAttemptEvent args)
|
||||
{
|
||||
if (args.Handled || args.Input.EntityUid is not { Valid: true } entity)
|
||||
return;
|
||||
|
||||
// let world target component handle it
|
||||
var (uid, comp) = ent;
|
||||
if (comp.Event is not {} ev)
|
||||
{
|
||||
DebugTools.Assert(HasComp<WorldTargetActionComponent>(ent), $"Action {ToPrettyString(ent)} requires WorldTargetActionComponent for entity-world targeting");
|
||||
return;
|
||||
}
|
||||
|
||||
args.Handled = true;
|
||||
|
||||
var action = args.Action;
|
||||
var user = args.User;
|
||||
|
||||
if (!ValidateEntityTarget(user, entity, ent))
|
||||
return;
|
||||
|
||||
if (action.ClientExclusive)
|
||||
{
|
||||
ev.Target = entity;
|
||||
|
||||
PerformAction((user, user.Comp), (uid, action));
|
||||
}
|
||||
else
|
||||
{
|
||||
RaisePredictiveEvent(new RequestPerformActionEvent(GetNetEntity(uid), GetNetEntity(entity)));
|
||||
}
|
||||
|
||||
args.FoundTarget = true;
|
||||
}
|
||||
|
||||
public record struct SlotAssignment(byte Hotbar, byte Slot, EntityUid ActionId);
|
||||
|
||||
Reference in New Issue
Block a user