tommy/tbd-station-14
2
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add arguments to part and mechanism event methods (#2293)

Browse Source
This commit is contained in:
DrSmugleaf
2020-10-19 15:23:59 +02:00
committed by GitHub
parent 19d32eb4ce
commit 7ad46ddabf
9 changed files with 175 additions and 146 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
Content.Client/GameObjects/Components/Body/Mechanism/MechanismComponent.cs
View File

@@ -1,7 +1,5 @@
#nullable enable #nullable enable
using Content.Shared.GameObjects.Components.Body.Mechanism; using Content.Shared.GameObjects.Components.Body.Mechanism;
using Content.Shared.GameObjects.Components.Body.Part;
using Robust.Client.Interfaces.GameObjects.Components;
using Robust.Shared.GameObjects; using Robust.Shared.GameObjects;
namespace Content.Client.GameObjects.Components.Body.Mechanism namespace Content.Client.GameObjects.Components.Body.Mechanism
@@ -9,26 +7,5 @@ namespace Content.Client.GameObjects.Components.Body.Mechanism
[RegisterComponent] [RegisterComponent]
[ComponentReference(typeof(SharedMechanismComponent))] [ComponentReference(typeof(SharedMechanismComponent))]
[ComponentReference(typeof(IMechanism))] [ComponentReference(typeof(IMechanism))]
public class MechanismComponent : SharedMechanismComponent public class MechanismComponent : SharedMechanismComponent { }
{
protected override void OnAddedToPart()
{
base.OnAddedToPart();
if (Owner.TryGetComponent(out ISpriteComponent? sprite))
{
sprite.Visible = false;
}
}
protected override void OnRemovedFromPart(IBodyPart old)
{
base.OnRemovedFromPart(old);
if (Owner.TryGetComponent(out ISpriteComponent? sprite))
{
sprite.Visible = true;
}
}
}
} }

24
Content.IntegrationTests/Tests/Body/MechanismBehaviorEventsTest.cs
View File

@@ -35,16 +35,6 @@ namespace Content.IntegrationTests.Tests.Body
public override void Update(float frameTime) { } public override void Update(float frameTime) { }
public bool AllAdded()
{
return WasAddedToBody && WasAddedToPart && WasAddedToPartInBody;
}
public bool AllRemoved()
{
return WasRemovedFromBody && WasRemovedFromPart && WasRemovedFromPartInBody;
}
public bool NoAdded() public bool NoAdded()
{ {
return !WasAddedToBody && !WasAddedToPart && !WasAddedToPartInBody; return !WasAddedToBody && !WasAddedToPart && !WasAddedToPartInBody;
@@ -75,23 +65,23 @@ namespace Content.IntegrationTests.Tests.Body
ResetRemoved(); ResetRemoved();
} }
protected override void OnAddedToBody() protected override void OnAddedToBody(IBody body)
{ {
base.OnAddedToBody(); base.OnAddedToBody(body);
WasAddedToBody = true; WasAddedToBody = true;
} }
protected override void OnAddedToPart() protected override void OnAddedToPart(IBodyPart part)
{ {
base.OnAddedToPart(); base.OnAddedToPart(part);
WasAddedToPart = true; WasAddedToPart = true;
} }
protected override void OnAddedToPartInBody() protected override void OnAddedToPartInBody(IBody body, IBodyPart part)
{ {
base.OnAddedToPartInBody(); base.OnAddedToPartInBody(body, part);
WasAddedToPartInBody = true; WasAddedToPartInBody = true;
} }
@@ -110,7 +100,7 @@ namespace Content.IntegrationTests.Tests.Body
WasRemovedFromPart = true; WasRemovedFromPart = true;
} }
protected override void OnRemovedFromPartInBody(IBody? oldBody, IBodyPart? oldPart) protected override void OnRemovedFromPartInBody(IBody oldBody, IBodyPart oldPart)
{ {
base.OnRemovedFromPartInBody(oldBody, oldPart); base.OnRemovedFromPartInBody(oldBody, oldPart);

24
Content.Server/GameObjects/Components/Body/Behavior/BrainBehaviorComponent.cs
View File

@@ -17,25 +17,25 @@ namespace Content.Server.GameObjects.Components.Body.Behavior
{ {
public override string Name => "Brain"; public override string Name => "Brain";
protected override void OnAddedToBody() protected override void OnAddedToBody(IBody body)
{ {
base.OnAddedToBody(); base.OnAddedToBody(body);
HandleMind(Body!.Owner, Owner); HandleMind(body.Owner, Owner);
} }
protected override void OnAddedToPart() protected override void OnAddedToPart(IBodyPart part)
{ {
base.OnAddedToPart(); base.OnAddedToPart(part);
HandleMind(Part!.Owner, Owner); HandleMind(part.Owner, Owner);
} }
protected override void OnAddedToPartInBody() protected override void OnAddedToPartInBody(IBody body, IBodyPart part)
{ {
base.OnAddedToPartInBody(); base.OnAddedToPartInBody(body, part);
HandleMind(Body!.Owner, Owner); HandleMind(body.Owner, Owner);
} }
protected override void OnRemovedFromBody(IBody old) protected override void OnRemovedFromBody(IBody old)
@@ -52,16 +52,16 @@ namespace Content.Server.GameObjects.Components.Body.Behavior
HandleMind(Owner, old.Owner); HandleMind(Owner, old.Owner);
} }
protected override void OnRemovedFromPartInBody(IBody? oldBody, IBodyPart? oldPart) protected override void OnRemovedFromPartInBody(IBody oldBody, IBodyPart oldPart)
{ {
base.OnRemovedFromPartInBody(oldBody, oldPart); base.OnRemovedFromPartInBody(oldBody, oldPart);
HandleMind(oldBody!.Owner, Owner); HandleMind(oldBody.Owner, Owner);
} }
private void HandleMind(IEntity newEntity, IEntity oldEntity) private void HandleMind(IEntity newEntity, IEntity oldEntity)
{ {
var newMind = newEntity.EnsureComponent<MindComponent>(); newEntity.EnsureComponent<MindComponent>();
var oldMind = oldEntity.EnsureComponent<MindComponent>(); var oldMind = oldEntity.EnsureComponent<MindComponent>();
oldMind.Mind?.TransferTo(newEntity); oldMind.Mind?.TransferTo(newEntity);

20
Content.Server/GameObjects/Components/Body/MechanismComponent.cs
View File

@@ -157,25 +157,5 @@ namespace Content.Server.GameObjects.Components.Body
break; break;
} }
} }
protected override void OnAddedToPart()
{
base.OnAddedToPart();
if (Owner.TryGetComponent(out SpriteComponent? sprite))
{
sprite.Visible = false;
}
}
protected override void OnRemovedFromPart(IBodyPart old)
{
base.OnRemovedFromPart(old);
if (Owner.TryGetComponent(out SpriteComponent? sprite))
{
sprite.Visible = true;
}
}
} }
} }

32
Content.Shared/GameObjects/Components/Body/Behavior/IMechanismBehavior.cs
View File

@@ -23,7 +23,10 @@ namespace Content.Shared.GameObjects.Components.Body.Behavior
/// For instance, attaching a head with a brain inside to a body. /// For instance, attaching a head with a brain inside to a body.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
void AddedToBody(); /// <param name="body">
/// The body that the containing <see cref="IMechanism"/> was added to.
/// </param>
void AddedToBody(IBody body);
/// <summary> /// <summary>
/// Called when the parent <see cref="IMechanism"/> is /// Called when the parent <see cref="IMechanism"/> is
@@ -32,7 +35,10 @@ namespace Content.Shared.GameObjects.Components.Body.Behavior
/// For instance, adding a brain to a dismembered head. /// For instance, adding a brain to a dismembered head.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
void AddedToPart(); /// <param name="part">
/// The part that the containing <see cref="IMechanism"/> was added to.
/// </param>
void AddedToPart(IBodyPart part);
/// <summary> /// <summary>
/// Called when the parent <see cref="IMechanism"/> is added to a /// Called when the parent <see cref="IMechanism"/> is added to a
@@ -40,7 +46,13 @@ namespace Content.Shared.GameObjects.Components.Body.Behavior
/// For instance, adding a brain to a head that is attached to a body. /// For instance, adding a brain to a head that is attached to a body.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
void AddedToPartInBody(); /// <param name="body">
/// The body that the containing <see cref="IMechanism"/> was added to.
/// </param>
/// <param name="part">
/// The part that the containing <see cref="IMechanism"/> was added to.
/// </param>
void AddedToPartInBody(IBody body, IBodyPart part);
/// <summary> /// <summary>
/// Called when the parent <see cref="IBodyPart"/> is removed from a /// Called when the parent <see cref="IBodyPart"/> is removed from a
@@ -48,6 +60,9 @@ namespace Content.Shared.GameObjects.Components.Body.Behavior
/// For instance, removing a head with a brain inside from a body. /// For instance, removing a head with a brain inside from a body.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
/// <param name="old">
/// The body that the containing <see cref="IMechanism"/> was removed from.
/// </param>
void RemovedFromBody(IBody old); void RemovedFromBody(IBody old);
/// <summary> /// <summary>
@@ -57,6 +72,9 @@ namespace Content.Shared.GameObjects.Components.Body.Behavior
/// For instance, removing a brain from a dismembered head. /// For instance, removing a brain from a dismembered head.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
/// <param name="old">
/// The part that the containing <see cref="IMechanism"/> was removed from.
/// </param>
void RemovedFromPart(IBodyPart old); void RemovedFromPart(IBodyPart old);
/// <summary> /// <summary>
@@ -65,6 +83,12 @@ namespace Content.Shared.GameObjects.Components.Body.Behavior
/// For instance, removing a brain from a head that is attached to a body. /// For instance, removing a brain from a head that is attached to a body.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
void RemovedFromPartInBody(IBody? oldBody, IBodyPart? oldPart); /// <param name="oldBody">
/// The body that the containing <see cref="IMechanism"/> was removed from.
/// </param>
/// <param name="oldPart">
/// The part that the containing <see cref="IMechanism"/> was removed from.
/// </param>
void RemovedFromPartInBody(IBody oldBody, IBodyPart oldPart);
} }
} }

55
Content.Shared/GameObjects/Components/Body/Behavior/MechanismBehaviorComponent.cs
View File

@@ -2,6 +2,7 @@
using Content.Shared.GameObjects.Components.Body.Mechanism; using Content.Shared.GameObjects.Components.Body.Mechanism;
using Content.Shared.GameObjects.Components.Body.Part; using Content.Shared.GameObjects.Components.Body.Part;
using Robust.Shared.GameObjects; using Robust.Shared.GameObjects;
using Robust.Shared.Utility;
namespace Content.Shared.GameObjects.Components.Body.Behavior namespace Content.Shared.GameObjects.Components.Body.Behavior
{ {
@@ -24,56 +25,78 @@ namespace Content.Shared.GameObjects.Components.Body.Behavior
if (Body == null) if (Body == null)
{ {
AddedToPart(); AddedToPart(Part);
} }
else else
{ {
AddedToPartInBody(); AddedToPartInBody(Body, Part);
} }
} }
public abstract void Update(float frameTime); public abstract void Update(float frameTime);
public void AddedToBody() public void AddedToBody(IBody body)
{ {
OnAddedToBody(); DebugTools.AssertNotNull(Body);
DebugTools.AssertNotNull(body);
OnAddedToBody(body);
} }
public void AddedToPart() public void AddedToPart(IBodyPart part)
{ {
OnAddedToPart(); DebugTools.AssertNotNull(Part);
DebugTools.AssertNotNull(part);
OnAddedToPart(part);
}
public void AddedToPartInBody(IBody body, IBodyPart part)
{
DebugTools.AssertNotNull(Body);
DebugTools.AssertNotNull(body);
DebugTools.AssertNotNull(Part);
DebugTools.AssertNotNull(part);
OnAddedToPartInBody(body, part);
} }
public void RemovedFromBody(IBody old) public void RemovedFromBody(IBody old)
{ {
DebugTools.AssertNull(Body);
DebugTools.AssertNotNull(old);
OnRemovedFromBody(old); OnRemovedFromBody(old);
} }
public void RemovedFromPart(IBodyPart old) public void RemovedFromPart(IBodyPart old)
{ {
DebugTools.AssertNull(Part);
DebugTools.AssertNotNull(old);
OnRemovedFromPart(old); OnRemovedFromPart(old);
} }
public void AddedToPartInBody() public void RemovedFromPartInBody(IBody oldBody, IBodyPart oldPart)
{ {
OnAddedToPartInBody(); DebugTools.AssertNull(Body);
} DebugTools.AssertNull(Part);
DebugTools.AssertNotNull(oldBody);
DebugTools.AssertNotNull(oldPart);
public void RemovedFromPartInBody(IBody? oldBody, IBodyPart? oldPart)
{
OnRemovedFromPartInBody(oldBody, oldPart); OnRemovedFromPartInBody(oldBody, oldPart);
} }
protected virtual void OnAddedToBody() { } protected virtual void OnAddedToBody(IBody body) { }
protected virtual void OnAddedToPart() { } protected virtual void OnAddedToPart(IBodyPart part) { }
protected virtual void OnAddedToPartInBody(IBody body, IBodyPart part) { }
protected virtual void OnRemovedFromBody(IBody old) { } protected virtual void OnRemovedFromBody(IBody old) { }
protected virtual void OnRemovedFromPart(IBodyPart old) { } protected virtual void OnRemovedFromPart(IBodyPart old) { }
protected virtual void OnAddedToPartInBody() { } protected virtual void OnRemovedFromPartInBody(IBody oldBody, IBodyPart oldPart) { }
protected virtual void OnRemovedFromPartInBody(IBody? oldBody, IBodyPart? oldPart) { }
} }
} }

32
Content.Shared/GameObjects/Components/Body/Mechanism/IMechanism.cs
View File

@@ -62,7 +62,10 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
/// For instance, attaching a head with a brain inside to a body. /// For instance, attaching a head with a brain inside to a body.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
void AddedToBody(); /// <param name="body">
/// The body that this <see cref="IMechanism"/> was added to.
/// </param>
void AddedToBody(IBody body);
/// <summary> /// <summary>
/// Called when the parent <see cref="IMechanism"/> is /// Called when the parent <see cref="IMechanism"/> is
@@ -71,7 +74,10 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
/// For instance, adding a brain to a dismembered head. /// For instance, adding a brain to a dismembered head.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
void AddedToPart(); /// <param name="part">
/// The part that this <see cref="IMechanism"/> was added to.
/// </param>
void AddedToPart(IBodyPart part);
/// <summary> /// <summary>
/// Called when the parent <see cref="IMechanism"/> is added to a /// Called when the parent <see cref="IMechanism"/> is added to a
@@ -79,7 +85,13 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
/// For instance, adding a brain to a head that is attached to a body. /// For instance, adding a brain to a head that is attached to a body.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
void AddedToPartInBody(); /// <param name="body">
/// The body that this <see cref="IMechanism"/> was added to.
/// </param>
/// <param name="part">
/// The part that this <see cref="IMechanism"/> was added to.
/// </param>
void AddedToPartInBody(IBody body, IBodyPart part);
/// <summary> /// <summary>
/// Called when the parent <see cref="IBodyPart"/> is removed from a /// Called when the parent <see cref="IBodyPart"/> is removed from a
@@ -87,6 +99,9 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
/// For instance, removing a head with a brain inside from a body. /// For instance, removing a head with a brain inside from a body.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
/// <param name="old">
/// The body that this <see cref="IMechanism"/> was removed from.
/// </param>
void RemovedFromBody(IBody old); void RemovedFromBody(IBody old);
/// <summary> /// <summary>
@@ -96,6 +111,9 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
/// For instance, removing a brain from a dismembered head. /// For instance, removing a brain from a dismembered head.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
/// <param name="old">
/// The part that this <see cref="IMechanism"/> was removed from.
/// </param>
void RemovedFromPart(IBodyPart old); void RemovedFromPart(IBodyPart old);
/// <summary> /// <summary>
@@ -104,6 +122,12 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
/// For instance, removing a brain from a head that is attached to a body. /// For instance, removing a brain from a head that is attached to a body.
/// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE! /// DO NOT CALL THIS DIRECTLY FROM OUTSIDE BODY SYSTEM CODE!
/// </summary> /// </summary>
void RemovedFromPartInBody(IBody? oldBody, IBodyPart? oldPart); /// <param name="oldBody">
/// The body that this <see cref="IMechanism"/> was removed from.
/// </param>
/// <param name="oldPart">
/// The part that this <see cref="IMechanism"/> was removed from.
/// </param>
void RemovedFromPartInBody(IBody oldBody, IBodyPart oldPart);
} }
} }

99
Content.Shared/GameObjects/Components/Body/Mechanism/SharedMechanismComponent.cs
View File

@@ -15,13 +15,9 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
public override string Name => "Mechanism"; public override string Name => "Mechanism";
private IBodyPart? _part; private IBodyPart? _part;
protected readonly Dictionary<int, object> OptionsCache = new Dictionary<int, object>(); protected readonly Dictionary<int, object> OptionsCache = new Dictionary<int, object>();
protected IBody? BodyCache; protected IBody? BodyCache;
protected int IdHash; protected int IdHash;
protected IEntity? PerformerCache; protected IEntity? PerformerCache;
public IBody? Body => Part?.Body; public IBody? Body => Part?.Body;
@@ -55,11 +51,11 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
{ {
if (value.Body == null) if (value.Body == null)
{ {
AddedToPart(); AddedToPart(value);
} }
else else
{ {
AddedToPartInBody(); AddedToPartInBody(value.Body, value);
} }
} }
} }
@@ -104,20 +100,54 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
serializer.DataField(this, m => m.Compatibility, "compatibility", BodyPartCompatibility.Universal); serializer.DataField(this, m => m.Compatibility, "compatibility", BodyPartCompatibility.Universal);
} }
public void AddedToBody() public void AddedToBody(IBody body)
{ {
DebugTools.AssertNotNull(Body); DebugTools.AssertNotNull(Body);
DebugTools.AssertNotNull(body);
OnAddedToBody(); OnAddedToBody(body);
foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>()) foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>())
{ {
behavior.AddedToBody(); behavior.AddedToBody(body);
}
}
public void AddedToPart(IBodyPart part)
{
DebugTools.AssertNotNull(Part);
DebugTools.AssertNotNull(part);
Owner.Transform.AttachParent(part.Owner);
OnAddedToPart(part);
foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>().ToArray())
{
behavior.AddedToPart(part);
}
}
public void AddedToPartInBody(IBody body, IBodyPart part)
{
DebugTools.AssertNotNull(Body);
DebugTools.AssertNotNull(body);
DebugTools.AssertNotNull(Part);
DebugTools.AssertNotNull(part);
Owner.Transform.AttachParent(part.Owner);
OnAddedToPartInBody(body, part);
foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>())
{
behavior.AddedToPartInBody(body, part);
} }
} }
public void RemovedFromBody(IBody old) public void RemovedFromBody(IBody old)
{ {
DebugTools.AssertNull(Body);
DebugTools.AssertNotNull(old);
OnRemovedFromBody(old); OnRemovedFromBody(old);
foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>()) foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>())
@@ -126,35 +156,11 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
} }
} }
public void AddedToPart()
{
DebugTools.AssertNotNull(Part);
Owner.Transform.AttachParent(Part!.Owner);
OnAddedToPart();
foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>().ToArray())
{
behavior.AddedToPart();
}
}
public void AddedToPartInBody()
{
DebugTools.AssertNotNull(Body);
DebugTools.AssertNotNull(Part);
Owner.Transform.AttachParent(Part!.Owner);
OnAddedToPartInBody();
foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>())
{
behavior.AddedToPartInBody();
}
}
public void RemovedFromPart(IBodyPart old) public void RemovedFromPart(IBodyPart old)
{ {
DebugTools.AssertNull(Part);
DebugTools.AssertNotNull(old);
Owner.Transform.AttachToGridOrMap(); Owner.Transform.AttachToGridOrMap();
OnRemovedFromPart(old); OnRemovedFromPart(old);
@@ -164,10 +170,15 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
} }
} }
public void RemovedFromPartInBody(IBody? oldBody, IBodyPart? oldPart) public void RemovedFromPartInBody(IBody oldBody, IBodyPart oldPart)
{ {
DebugTools.AssertNull(Body);
DebugTools.AssertNotNull(oldBody);
DebugTools.AssertNull(Part);
DebugTools.AssertNotNull(oldPart);
Owner.Transform.AttachToGridOrMap(); Owner.Transform.AttachToGridOrMap();
OnRemovedFromPartInBody(); OnRemovedFromPartInBody(oldBody, oldPart);
foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>()) foreach (var behavior in Owner.GetAllComponents<IMechanismBehavior>())
{ {
@@ -175,16 +186,16 @@ namespace Content.Shared.GameObjects.Components.Body.Mechanism
} }
} }
protected virtual void OnAddedToBody() { } protected virtual void OnAddedToBody(IBody body) { }
protected virtual void OnAddedToPart(IBodyPart part) { }
protected virtual void OnAddedToPartInBody(IBody body, IBodyPart part) { }
protected virtual void OnRemovedFromBody(IBody old) { } protected virtual void OnRemovedFromBody(IBody old) { }
protected virtual void OnAddedToPart() { }
protected virtual void OnRemovedFromPart(IBodyPart old) { } protected virtual void OnRemovedFromPart(IBodyPart old) { }
protected virtual void OnAddedToPartInBody() { } protected virtual void OnRemovedFromPartInBody(IBody oldBody, IBodyPart oldPart) { }
protected virtual void OnRemovedFromPartInBody() { }
} }
} }

10
Content.Shared/GameObjects/Components/Body/Part/SharedBodyPartComponent.cs
View File

@@ -50,7 +50,7 @@ namespace Content.Shared.GameObjects.Components.Body.Part
if (value != null) if (value != null)
{ {
AddedToBody(); AddedToBody(value);
} }
} }
} }
@@ -284,14 +284,14 @@ namespace Content.Shared.GameObjects.Components.Body.Part
return true; return true;
} }
private void AddedToBody() private void AddedToBody(IBody body)
{ {
Owner.Transform.AttachParent(Body!.Owner); Owner.Transform.AttachParent(Body!.Owner);
OnAddedToBody(); OnAddedToBody(body);
foreach (var mechanism in _mechanisms) foreach (var mechanism in _mechanisms)
{ {
mechanism.AddedToBody(); mechanism.AddedToBody(body);
} }
} }
@@ -310,7 +310,7 @@ namespace Content.Shared.GameObjects.Components.Body.Part
} }
} }
protected virtual void OnAddedToBody() { } protected virtual void OnAddedToBody(IBody body) { }
protected virtual void OnRemovedFromBody(IBody old) { } protected virtual void OnRemovedFromBody(IBody old) { }
} }