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

Shuffle body container subs slightly (#21084)

Browse Source
This commit is contained in:
metalgearsloth
2023-12-11 21:35:24 +11:00
committed by GitHub
parent c2df96a879
commit eb49ad11ba
4 changed files with 118 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

48
Content.Shared/Body/Systems/SharedBodySystem.Organs.cs
View File

@@ -9,26 +9,50 @@ namespace Content.Shared.Body.Systems;
public partial class SharedBodySystem
{
private void AddOrgan(EntityUid uid, EntityUid bodyUid, EntityUid parentPartUid, OrganComponent component)
private void InitializeOrgans()
{
component.Body = bodyUid;
RaiseLocalEvent(uid, new AddedToPartEvent(parentPartUid));
if (component.Body != null)
RaiseLocalEvent(uid, new AddedToPartInBodyEvent(component.Body.Value, parentPartUid));
Dirty(uid, component);
SubscribeLocalEvent<OrganComponent, EntGotInsertedIntoContainerMessage>(OnOrganInserted);
SubscribeLocalEvent<OrganComponent, EntGotRemovedFromContainerMessage>(OnOrganRemoved);
}
private void RemoveOrgan(EntityUid uid, EntityUid parentPartUid, OrganComponent component)
private void OnOrganInserted(EntityUid uid, OrganComponent component, EntGotInsertedIntoContainerMessage args)
{
RaiseLocalEvent(uid, new RemovedFromPartEvent(parentPartUid));
// No recursive updates for these as you can't insert organs into organsTM.
var parentUid = args.Container.Owner;
if (component.Body != null)
if (HasComp<BodyComponent>(parentUid))
{
RaiseLocalEvent(uid, new RemovedFromPartInBodyEvent(component.Body.Value, parentPartUid));
component.Body = parentUid;
Dirty(uid, component);
}
// Organ inserted into body part.
if (TryComp(parentUid, out BodyPartComponent? partComp))
{
RaiseLocalEvent(uid, new AddedToPartEvent(parentUid));
if (partComp.Body != null)
{
component.Body = partComp.Body;
RaiseLocalEvent(uid, new AddedToPartInBodyEvent(partComp.Body.Value, parentUid));
Dirty(uid, component);
}
}
}
private void OnOrganRemoved(EntityUid uid, OrganComponent component, EntGotRemovedFromContainerMessage args)
{
// Lifestage shenanigans.
if (component.Body != null && TerminatingOrDeleted(component.Body.Value))
return;
if (HasComp<BodyPartComponent>(args.Container.Owner))
RaiseLocalEvent(uid, new RemovedFromPartEvent(args.Container.Owner));
if (component.Body == null)
return;
RaiseLocalEvent(uid, new RemovedFromPartInBodyEvent(component.Body.Value, args.Container.Owner));
component.Body = null;
Dirty(uid, component);
}