Implanter draw rework (#32136)
* Initial commit * Clean-up * Fix ftl, new damage * ftl fix for real * Updates based on feedback * Child implant fix * Make the UI only open when implanter is in draw mode * Review fixes * shunting
This commit is contained in:
SlamBamActionman
@@ -1,14 +1,18 @@
|
||||
using System.Diagnostics.CodeAnalysis;
|
||||
using System.Linq;
|
||||
using Content.Shared.Containers.ItemSlots;
|
||||
using Content.Shared.Damage;
|
||||
using Content.Shared.DoAfter;
|
||||
using Content.Shared.Examine;
|
||||
using Content.Shared.Forensics;
|
||||
using Content.Shared.IdentityManagement;
|
||||
using Content.Shared.Implants.Components;
|
||||
using Content.Shared.Interaction.Events;
|
||||
using Content.Shared.Popups;
|
||||
using Content.Shared.Verbs;
|
||||
using Content.Shared.Whitelist;
|
||||
using Robust.Shared.Containers;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Serialization;
|
||||
using Robust.Shared.Utility;
|
||||
|
||||
@@ -21,6 +25,9 @@ public abstract class SharedImplanterSystem : EntitySystem
|
||||
[Dependency] private readonly SharedAppearanceSystem _appearance = default!;
|
||||
[Dependency] private readonly SharedPopupSystem _popup = default!;
|
||||
[Dependency] private readonly EntityWhitelistSystem _whitelistSystem = default!;
|
||||
[Dependency] private readonly DamageableSystem _damageableSystem = default!;
|
||||
[Dependency] private readonly SharedUserInterfaceSystem _uiSystem = default!;
|
||||
[Dependency] private readonly IPrototypeManager _proto = default!;
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
@@ -29,6 +36,10 @@ public abstract class SharedImplanterSystem : EntitySystem
|
||||
SubscribeLocalEvent<ImplanterComponent, ComponentInit>(OnImplanterInit);
|
||||
SubscribeLocalEvent<ImplanterComponent, EntInsertedIntoContainerMessage>(OnEntInserted);
|
||||
SubscribeLocalEvent<ImplanterComponent, ExaminedEvent>(OnExamine);
|
||||
|
||||
SubscribeLocalEvent<ImplanterComponent, UseInHandEvent>(OnUseInHand);
|
||||
SubscribeLocalEvent<ImplanterComponent, GetVerbsEvent<InteractionVerb>>(OnVerb);
|
||||
SubscribeLocalEvent<ImplanterComponent, DeimplantChangeVerbMessage>(OnSelected);
|
||||
}
|
||||
|
||||
private void OnImplanterInit(EntityUid uid, ImplanterComponent component, ComponentInit args)
|
||||
@@ -37,6 +48,10 @@ public abstract class SharedImplanterSystem : EntitySystem
|
||||
component.ImplanterSlot.StartingItem = component.Implant;
|
||||
|
||||
_itemSlots.AddItemSlot(uid, ImplanterComponent.ImplanterSlotId, component.ImplanterSlot);
|
||||
|
||||
component.DeimplantChosen ??= component.DeimplantWhitelist.FirstOrNull();
|
||||
|
||||
Dirty(uid, component);
|
||||
}
|
||||
|
||||
private void OnEntInserted(EntityUid uid, ImplanterComponent component, EntInsertedIntoContainerMessage args)
|
||||
@@ -56,10 +71,49 @@ public abstract class SharedImplanterSystem : EntitySystem
|
||||
{
|
||||
if (!TryComp<ImplantedComponent>(target, out var implanted))
|
||||
return false;
|
||||
|
||||
var implantPrototype = Prototype(implant);
|
||||
return implanted.ImplantContainer.ContainedEntities.Any(entity => Prototype(entity) == implantPrototype);
|
||||
}
|
||||
|
||||
private void OnVerb(EntityUid uid, ImplanterComponent component, GetVerbsEvent<InteractionVerb> args)
|
||||
{
|
||||
if (!args.CanAccess || !args.CanInteract)
|
||||
return;
|
||||
|
||||
if (component.CurrentMode == ImplanterToggleMode.Draw)
|
||||
{
|
||||
args.Verbs.Add(new InteractionVerb()
|
||||
{
|
||||
Text = Loc.GetString("implanter-set-draw-verb"),
|
||||
Act = () => TryOpenUi(uid, args.User, component)
|
||||
});
|
||||
}
|
||||
}
|
||||
|
||||
private void OnUseInHand(EntityUid uid, ImplanterComponent? component, UseInHandEvent args)
|
||||
{
|
||||
if (!Resolve(uid, ref component))
|
||||
return;
|
||||
|
||||
if (component.CurrentMode == ImplanterToggleMode.Draw)
|
||||
TryOpenUi(uid, args.User, component);
|
||||
}
|
||||
|
||||
private void OnSelected(EntityUid uid, ImplanterComponent component, DeimplantChangeVerbMessage args)
|
||||
{
|
||||
component.DeimplantChosen = args.Implant;
|
||||
SetSelectedDeimplant(uid, args.Implant, component: component);
|
||||
}
|
||||
|
||||
private void TryOpenUi(EntityUid uid, EntityUid user, ImplanterComponent? component = null)
|
||||
{
|
||||
if (!Resolve(uid, ref component))
|
||||
return;
|
||||
_uiSystem.TryToggleUi(uid, DeimplantUiKey.Key, user);
|
||||
component.DeimplantChosen ??= component.DeimplantWhitelist.FirstOrNull();
|
||||
Dirty(uid, component);
|
||||
}
|
||||
|
||||
//Instantly implant something and add all necessary components and containers.
|
||||
//Set to draw mode if not implant only
|
||||
public void Implant(EntityUid user, EntityUid target, EntityUid implanter, ImplanterComponent component)
|
||||
@@ -142,40 +196,103 @@ public abstract class SharedImplanterSystem : EntitySystem
|
||||
{
|
||||
var implantCompQuery = GetEntityQuery<SubdermalImplantComponent>();
|
||||
|
||||
foreach (var implant in implantContainer.ContainedEntities)
|
||||
if (component.AllowDeimplantAll)
|
||||
{
|
||||
if (!implantCompQuery.TryGetComponent(implant, out var implantComp))
|
||||
continue;
|
||||
|
||||
//Don't remove a permanent implant and look for the next that can be drawn
|
||||
if (!_container.CanRemove(implant, implantContainer))
|
||||
foreach (var implant in implantContainer.ContainedEntities)
|
||||
{
|
||||
var implantName = Identity.Entity(implant, EntityManager);
|
||||
var targetName = Identity.Entity(target, EntityManager);
|
||||
var failedPermanentMessage = Loc.GetString("implanter-draw-failed-permanent",
|
||||
("implant", implantName), ("target", targetName));
|
||||
_popup.PopupEntity(failedPermanentMessage, target, user);
|
||||
if (!implantCompQuery.TryGetComponent(implant, out var implantComp))
|
||||
continue;
|
||||
|
||||
//Don't remove a permanent implant and look for the next that can be drawn
|
||||
if (!_container.CanRemove(implant, implantContainer))
|
||||
{
|
||||
DrawPermanentFailurePopup(implant, target, user);
|
||||
permanentFound = implantComp.Permanent;
|
||||
continue;
|
||||
}
|
||||
|
||||
DrawImplantIntoImplanter(implanter, target, implant, implantContainer, implanterContainer, implantComp);
|
||||
permanentFound = implantComp.Permanent;
|
||||
continue;
|
||||
|
||||
//Break so only one implant is drawn
|
||||
break;
|
||||
}
|
||||
|
||||
_container.Remove(implant, implantContainer);
|
||||
implantComp.ImplantedEntity = null;
|
||||
_container.Insert(implant, implanterContainer);
|
||||
permanentFound = implantComp.Permanent;
|
||||
if (component.CurrentMode == ImplanterToggleMode.Draw && !component.ImplantOnly && !permanentFound)
|
||||
ImplantMode(implanter, component);
|
||||
}
|
||||
else
|
||||
{
|
||||
EntityUid? implant = null;
|
||||
var implants = implantContainer.ContainedEntities;
|
||||
foreach (var implantEntity in implants)
|
||||
{
|
||||
if (TryComp<SubdermalImplantComponent>(implantEntity, out var subdermalComp))
|
||||
{
|
||||
if (component.DeimplantChosen == subdermalComp.DrawableProtoIdOverride ||
|
||||
(Prototype(implantEntity) != null && component.DeimplantChosen == Prototype(implantEntity)!))
|
||||
implant = implantEntity;
|
||||
}
|
||||
}
|
||||
|
||||
var ev = new TransferDnaEvent { Donor = target, Recipient = implanter };
|
||||
RaiseLocalEvent(target, ref ev);
|
||||
if (implant != null && implantCompQuery.TryGetComponent(implant, out var implantComp))
|
||||
{
|
||||
//Don't remove a permanent implant
|
||||
if (!_container.CanRemove(implant.Value, implantContainer))
|
||||
{
|
||||
DrawPermanentFailurePopup(implant.Value, target, user);
|
||||
permanentFound = implantComp.Permanent;
|
||||
|
||||
//Break so only one implant is drawn
|
||||
break;
|
||||
}
|
||||
else
|
||||
{
|
||||
DrawImplantIntoImplanter(implanter, target, implant.Value, implantContainer, implanterContainer, implantComp);
|
||||
permanentFound = implantComp.Permanent;
|
||||
}
|
||||
|
||||
if (component.CurrentMode == ImplanterToggleMode.Draw && !component.ImplantOnly && !permanentFound)
|
||||
ImplantMode(implanter, component);
|
||||
}
|
||||
else
|
||||
{
|
||||
DrawCatastrophicFailure(implanter, component, user);
|
||||
}
|
||||
}
|
||||
|
||||
if (component.CurrentMode == ImplanterToggleMode.Draw && !component.ImplantOnly && !permanentFound)
|
||||
ImplantMode(implanter, component);
|
||||
|
||||
Dirty(implanter, component);
|
||||
|
||||
}
|
||||
else
|
||||
{
|
||||
DrawCatastrophicFailure(implanter, component, user);
|
||||
}
|
||||
}
|
||||
|
||||
private void DrawPermanentFailurePopup(EntityUid implant, EntityUid target, EntityUid user)
|
||||
{
|
||||
var implantName = Identity.Entity(implant, EntityManager);
|
||||
var targetName = Identity.Entity(target, EntityManager);
|
||||
var failedPermanentMessage = Loc.GetString("implanter-draw-failed-permanent",
|
||||
("implant", implantName), ("target", targetName));
|
||||
_popup.PopupEntity(failedPermanentMessage, target, user);
|
||||
}
|
||||
|
||||
private void DrawImplantIntoImplanter(EntityUid implanter, EntityUid target, EntityUid implant, BaseContainer implantContainer, ContainerSlot implanterContainer, SubdermalImplantComponent implantComp)
|
||||
{
|
||||
_container.Remove(implant, implantContainer);
|
||||
implantComp.ImplantedEntity = null;
|
||||
_container.Insert(implant, implanterContainer);
|
||||
|
||||
var ev = new TransferDnaEvent { Donor = target, Recipient = implanter };
|
||||
RaiseLocalEvent(target, ref ev);
|
||||
}
|
||||
|
||||
private void DrawCatastrophicFailure(EntityUid implanter, ImplanterComponent component, EntityUid user)
|
||||
{
|
||||
_damageableSystem.TryChangeDamage(user, component.DeimplantFailureDamage, ignoreResistances: true, origin: implanter);
|
||||
var userName = Identity.Entity(user, EntityManager);
|
||||
var failedCatastrophicallyMessage = Loc.GetString("implanter-draw-failed-catastrophically", ("user", userName));
|
||||
_popup.PopupEntity(failedCatastrophicallyMessage, user, PopupType.MediumCaution);
|
||||
}
|
||||
|
||||
private void ImplantMode(EntityUid uid, ImplanterComponent component)
|
||||
@@ -216,6 +333,17 @@ public abstract class SharedImplanterSystem : EntitySystem
|
||||
else
|
||||
_appearance.SetData(uid, ImplanterVisuals.Full, implantFound, appearance);
|
||||
}
|
||||
|
||||
public void SetSelectedDeimplant(EntityUid uid, string? implant, ImplanterComponent? component = null)
|
||||
{
|
||||
if (!Resolve(uid, ref component, false))
|
||||
return;
|
||||
|
||||
if (implant != null && _proto.TryIndex(implant, out EntityPrototype? proto))
|
||||
component.DeimplantChosen = proto;
|
||||
|
||||
Dirty(uid, component);
|
||||
}
|
||||
}
|
||||
|
||||
[Serializable, NetSerializable]
|
||||
@@ -243,3 +371,39 @@ public sealed class AddImplantAttemptEvent : CancellableEntityEventArgs
|
||||
Implanter = implanter;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
[Serializable, NetSerializable]
|
||||
public sealed class DeimplantBuiState : BoundUserInterfaceState
|
||||
{
|
||||
public readonly string? Implant;
|
||||
|
||||
public Dictionary<string, string> ImplantList;
|
||||
|
||||
public DeimplantBuiState(string? implant, Dictionary<string, string> implantList)
|
||||
{
|
||||
Implant = implant;
|
||||
ImplantList = implantList;
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// Change the chosen implanter in the UI.
|
||||
/// </summary>
|
||||
[Serializable, NetSerializable]
|
||||
public sealed class DeimplantChangeVerbMessage : BoundUserInterfaceMessage
|
||||
{
|
||||
public readonly string? Implant;
|
||||
|
||||
public DeimplantChangeVerbMessage(string? implant)
|
||||
{
|
||||
Implant = implant;
|
||||
}
|
||||
}
|
||||
|
||||
[Serializable, NetSerializable]
|
||||
public enum DeimplantUiKey : byte
|
||||
{
|
||||
Key
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user