Makes tools and welders ECS, add ToolQualityPrototype. (#4741)
This commit is contained in:
Vera Aguilera Puerto
committed by
GitHub
GitHub
parent
f2760d0002
commit
365c7da4dc
@@ -1,328 +1,52 @@
|
||||
using System;
|
||||
using System.Threading.Tasks;
|
||||
using Content.Server.Act;
|
||||
using Content.Server.Atmos.EntitySystems;
|
||||
using Content.Server.Chat.Managers;
|
||||
using Content.Server.Chemistry.Components;
|
||||
using Content.Server.Explosion;
|
||||
using Content.Server.Items;
|
||||
using Content.Server.Popups;
|
||||
using Content.Shared.Audio;
|
||||
using Content.Shared.Chemistry.Components;
|
||||
using Content.Shared.Chemistry.Components.SolutionManager;
|
||||
using Content.Shared.Chemistry.EntitySystems;
|
||||
using Content.Shared.Chemistry.Reagent;
|
||||
using Content.Shared.Interaction;
|
||||
using Content.Shared.Popups;
|
||||
using Content.Shared.Sound;
|
||||
using Content.Shared.Tool;
|
||||
using Robust.Server.GameObjects;
|
||||
using Robust.Shared.Audio;
|
||||
using Content.Shared.Tools.Components;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.GameStates;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Localization;
|
||||
using Robust.Shared.Player;
|
||||
using Robust.Shared.Players;
|
||||
using Robust.Shared.Serialization.Manager.Attributes;
|
||||
using Robust.Shared.ViewVariables;
|
||||
using Robust.Shared.Serialization.TypeSerializers.Implementations.Custom.Prototype;
|
||||
|
||||
namespace Content.Server.Tools.Components
|
||||
{
|
||||
[RegisterComponent]
|
||||
[ComponentReference(typeof(ToolComponent))]
|
||||
[ComponentReference(typeof(IToolComponent))]
|
||||
[NetworkedComponent()]
|
||||
public class WelderComponent : ToolComponent, IUse, ISuicideAct, IAfterInteract
|
||||
public class WelderComponent : SharedWelderComponent
|
||||
{
|
||||
[Dependency] private readonly IEntitySystemManager _entitySystemManager = default!;
|
||||
|
||||
public override string Name => "Welder";
|
||||
|
||||
public const string SolutionName = "welder";
|
||||
/// <summary>
|
||||
/// Solution on the entity that contains the fuel.
|
||||
/// </summary>
|
||||
[DataField("fuelSolution")]
|
||||
public string FuelSolution { get; } = "Welder";
|
||||
|
||||
/// <summary>
|
||||
/// Default Cost of using the welder fuel for an action
|
||||
/// Reagent that will be used as fuel for welding.
|
||||
/// </summary>
|
||||
public const float DefaultFuelCost = 10;
|
||||
[DataField("fuelReagent", customTypeSerializer:typeof(PrototypeIdSerializer<ReagentPrototype>))]
|
||||
public string FuelReagent { get; } = "WeldingFuel";
|
||||
|
||||
/// <summary>
|
||||
/// Rate at which we expunge fuel from ourselves when activated
|
||||
/// Fuel consumption per second, while the welder is active.
|
||||
/// </summary>
|
||||
public const float FuelLossRate = 0.5f;
|
||||
[DataField("fuelConsumption")]
|
||||
public ReagentUnit FuelConsumption { get; } = ReagentUnit.New(0.05f);
|
||||
|
||||
private bool _welderLit;
|
||||
private WelderSystem _welderSystem = default!;
|
||||
private SpriteComponent? _spriteComponent;
|
||||
private PointLightComponent? _pointLightComponent;
|
||||
|
||||
[DataField("weldSounds")]
|
||||
private SoundSpecifier WeldSounds { get; set; } = new SoundCollectionSpecifier("Welder");
|
||||
/// <summary>
|
||||
/// A fuel amount to be consumed when the welder goes from being unlit to being lit.
|
||||
/// </summary>
|
||||
[DataField("welderOnConsume")]
|
||||
public ReagentUnit FuelLitCost { get; } = ReagentUnit.New(0.5f);
|
||||
|
||||
/// <summary>
|
||||
/// Sound played when the welder is turned off.
|
||||
/// </summary>
|
||||
[DataField("welderOffSounds")]
|
||||
private SoundSpecifier WelderOffSounds { get; set; } = new SoundCollectionSpecifier("WelderOff");
|
||||
public SoundSpecifier WelderOffSounds { get; } = new SoundCollectionSpecifier("WelderOff");
|
||||
|
||||
/// <summary>
|
||||
/// Sound played when the tool is turned on.
|
||||
/// </summary>
|
||||
[DataField("welderOnSounds")]
|
||||
private SoundSpecifier WelderOnSounds { get; set; } = new SoundCollectionSpecifier("WelderOn");
|
||||
public SoundSpecifier WelderOnSounds { get; } = new SoundCollectionSpecifier("WelderOn");
|
||||
|
||||
[DataField("welderRefill")]
|
||||
private SoundSpecifier WelderRefill { get; set; } = new SoundPathSpecifier("/Audio/Effects/refill.ogg");
|
||||
|
||||
[ViewVariables] public float Fuel => WelderSolution?.GetReagentQuantity("WeldingFuel").Float() ?? 0f;
|
||||
|
||||
[ViewVariables] public float FuelCapacity => WelderSolution?.MaxVolume.Float() ?? 0f;
|
||||
|
||||
private Solution? WelderSolution
|
||||
{
|
||||
get
|
||||
{
|
||||
Owner.EntityManager.EntitySysManager.GetEntitySystem<SolutionContainerSystem>()
|
||||
.TryGetSolution(Owner, SolutionName, out var solution);
|
||||
return solution;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Status of welder, whether it is ignited
|
||||
/// </summary>
|
||||
[ViewVariables]
|
||||
public bool WelderLit
|
||||
{
|
||||
get => _welderLit;
|
||||
private set
|
||||
{
|
||||
_welderLit = value;
|
||||
Dirty();
|
||||
}
|
||||
}
|
||||
|
||||
protected override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
AddQuality(ToolQuality.Welding);
|
||||
|
||||
_welderSystem = _entitySystemManager.GetEntitySystem<WelderSystem>();
|
||||
|
||||
Owner.EnsureComponent<SolutionContainerManagerComponent>();
|
||||
Owner.TryGetComponent(out _spriteComponent);
|
||||
Owner.TryGetComponent(out _pointLightComponent);
|
||||
EntitySystem.Get<SolutionContainerSystem>().EnsureSolution(Owner, "welder");
|
||||
}
|
||||
|
||||
public override ComponentState GetComponentState(ICommonSession player)
|
||||
{
|
||||
return new WelderComponentState(FuelCapacity, Fuel, WelderLit);
|
||||
}
|
||||
|
||||
public override async Task<bool> UseTool(IEntity user, IEntity? target, float doAfterDelay,
|
||||
ToolQuality toolQualityNeeded, Func<bool>? doAfterCheck = null)
|
||||
{
|
||||
bool ExtraCheck()
|
||||
{
|
||||
var extraCheck = doAfterCheck?.Invoke() ?? true;
|
||||
|
||||
if (!CanWeld(DefaultFuelCost))
|
||||
{
|
||||
target?.PopupMessage(user, "Can't weld!");
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
return extraCheck;
|
||||
}
|
||||
|
||||
var canUse = await base.UseTool(user, target, doAfterDelay, toolQualityNeeded, ExtraCheck);
|
||||
|
||||
return toolQualityNeeded.HasFlag(ToolQuality.Welding) ? canUse && TryWeld(DefaultFuelCost, user) : canUse;
|
||||
}
|
||||
|
||||
public async Task<bool> UseTool(IEntity user, IEntity target, float doAfterDelay, ToolQuality toolQualityNeeded,
|
||||
float fuelConsumed, Func<bool>? doAfterCheck = null)
|
||||
{
|
||||
bool ExtraCheck()
|
||||
{
|
||||
var extraCheck = doAfterCheck?.Invoke() ?? true;
|
||||
|
||||
return extraCheck && CanWeld(fuelConsumed);
|
||||
}
|
||||
|
||||
return await base.UseTool(user, target, doAfterDelay, toolQualityNeeded, ExtraCheck) &&
|
||||
TryWeld(fuelConsumed, user);
|
||||
}
|
||||
|
||||
private bool TryWeld(float value, IEntity? user = null, bool silent = false)
|
||||
{
|
||||
if (!WelderLit)
|
||||
{
|
||||
if (!silent && user != null)
|
||||
Owner.PopupMessage(user, Loc.GetString("welder-component-welder-not-lit-message"));
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!CanWeld(value))
|
||||
{
|
||||
if (!silent && user != null)
|
||||
Owner.PopupMessage(user, Loc.GetString("welder-component-cannot-weld-message"));
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
if (WelderSolution == null)
|
||||
return false;
|
||||
|
||||
var succeeded = EntitySystem.Get<SolutionContainerSystem>()
|
||||
.TryRemoveReagent(Owner.Uid, WelderSolution, "WeldingFuel", ReagentUnit.New(value));
|
||||
|
||||
if (succeeded && !silent)
|
||||
{
|
||||
PlaySound(WeldSounds);
|
||||
}
|
||||
|
||||
return succeeded;
|
||||
}
|
||||
|
||||
private bool CanWeld(float value)
|
||||
{
|
||||
return Fuel > value || Qualities != ToolQuality.Welding;
|
||||
}
|
||||
|
||||
private bool CanLitWelder()
|
||||
{
|
||||
return Fuel > 0 || Qualities != ToolQuality.Welding;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// Deactivates welding tool if active, activates welding tool if possible
|
||||
/// </summary>
|
||||
private bool ToggleWelderStatus(IEntity? user = null)
|
||||
{
|
||||
var item = Owner.GetComponent<ItemComponent>();
|
||||
|
||||
if (WelderLit)
|
||||
{
|
||||
WelderLit = false;
|
||||
// Layer 1 is the flame.
|
||||
item.EquippedPrefix = "off";
|
||||
_spriteComponent?.LayerSetVisible(1, false);
|
||||
|
||||
if (_pointLightComponent != null) _pointLightComponent.Enabled = false;
|
||||
|
||||
PlaySound(WelderOffSounds, -5);
|
||||
_welderSystem.Unsubscribe(this);
|
||||
return true;
|
||||
}
|
||||
|
||||
if (!CanLitWelder() && user != null)
|
||||
{
|
||||
Owner.PopupMessage(user, Loc.GetString("welder-component-no-fuel-message"));
|
||||
return false;
|
||||
}
|
||||
|
||||
WelderLit = true;
|
||||
item.EquippedPrefix = "on";
|
||||
_spriteComponent?.LayerSetVisible(1, true);
|
||||
|
||||
if (_pointLightComponent != null) _pointLightComponent.Enabled = true;
|
||||
|
||||
PlaySound(WelderOnSounds, -5);
|
||||
_welderSystem.Subscribe(this);
|
||||
|
||||
EntitySystem.Get<AtmosphereSystem>().HotspotExpose(Owner.Transform.Coordinates, 700, 50, true);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
bool IUse.UseEntity(UseEntityEventArgs eventArgs)
|
||||
{
|
||||
return ToggleWelderStatus(eventArgs.User);
|
||||
}
|
||||
|
||||
protected override void Shutdown()
|
||||
{
|
||||
base.Shutdown();
|
||||
_welderSystem.Unsubscribe(this);
|
||||
}
|
||||
|
||||
public void OnUpdate(float frameTime)
|
||||
{
|
||||
if (!HasQuality(ToolQuality.Welding) || !WelderLit || Owner.Deleted)
|
||||
return;
|
||||
|
||||
EntitySystem.Get<SolutionContainerSystem>().TryRemoveReagent(Owner.Uid, WelderSolution, "WeldingFuel",
|
||||
ReagentUnit.New(FuelLossRate * frameTime));
|
||||
|
||||
EntitySystem.Get<AtmosphereSystem>().HotspotExpose(Owner.Transform.Coordinates, 700, 50, true);
|
||||
|
||||
if (Fuel == 0)
|
||||
ToggleWelderStatus();
|
||||
}
|
||||
|
||||
SuicideKind ISuicideAct.Suicide(IEntity victim, IChatManager chat)
|
||||
{
|
||||
string othersMessage;
|
||||
string selfMessage;
|
||||
|
||||
if (TryWeld(5, victim, silent: true))
|
||||
{
|
||||
PlaySound(WeldSounds);
|
||||
|
||||
othersMessage =
|
||||
Loc.GetString("welder-component-suicide-lit-others-message",
|
||||
("victim", victim));
|
||||
victim.PopupMessageOtherClients(othersMessage);
|
||||
|
||||
selfMessage = Loc.GetString("welder-component-suicide-lit-message");
|
||||
victim.PopupMessage(selfMessage);
|
||||
|
||||
return SuicideKind.Heat;
|
||||
}
|
||||
|
||||
othersMessage = Loc.GetString("welder-component-suicide-unlit-others-message", ("victim", victim));
|
||||
victim.PopupMessageOtherClients(othersMessage);
|
||||
|
||||
selfMessage = Loc.GetString("welder-component-suicide-unlit-message");
|
||||
victim.PopupMessage(selfMessage);
|
||||
|
||||
return SuicideKind.Blunt;
|
||||
}
|
||||
|
||||
async Task<bool> IAfterInteract.AfterInteract(AfterInteractEventArgs eventArgs)
|
||||
{
|
||||
if (eventArgs.Target == null || !eventArgs.CanReach)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
if (eventArgs.Target.TryGetComponent(out ReagentTankComponent? tank)
|
||||
&& tank.TankType == ReagentTankType.Fuel
|
||||
&& EntitySystem.Get<SolutionContainerSystem>()
|
||||
.TryGetDrainableSolution(eventArgs.Target.Uid, out var targetSolution)
|
||||
&& WelderSolution != null)
|
||||
{
|
||||
var trans = ReagentUnit.Min(WelderSolution.AvailableVolume, targetSolution.DrainAvailable);
|
||||
if (trans > 0)
|
||||
{
|
||||
var drained = EntitySystem.Get<SolutionContainerSystem>().Drain(eventArgs.Target.Uid, targetSolution, trans);
|
||||
EntitySystem.Get<SolutionContainerSystem>().TryAddSolution(Owner.Uid, WelderSolution, drained);
|
||||
SoundSystem.Play(Filter.Pvs(Owner), WelderRefill.GetSound(), Owner);
|
||||
eventArgs.Target.PopupMessage(eventArgs.User,
|
||||
Loc.GetString("welder-component-after-interact-refueled-message"));
|
||||
}
|
||||
else
|
||||
{
|
||||
eventArgs.Target.PopupMessage(eventArgs.User,
|
||||
Loc.GetString("welder-component-no-fuel-in-tank", ("owner", eventArgs.Target)));
|
||||
}
|
||||
}
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
private void PlaySound(SoundSpecifier sound, float volume = -5f)
|
||||
{
|
||||
SoundSystem.Play(Filter.Pvs(Owner), sound.GetSound(), Owner, AudioHelpers.WithVariation(0.15f).WithVolume(volume));
|
||||
}
|
||||
public SoundSpecifier WelderRefill { get; } = new SoundPathSpecifier("/Audio/Effects/refill.ogg");
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user