333 lines
12 KiB
C#
333 lines
12 KiB
C#
using System.Collections.Generic;
|
|
using System.Linq;
|
|
using Content.Server.Chemistry.Components;
|
|
using Content.Server.Items;
|
|
using Content.Server.Tools.Components;
|
|
using Content.Shared.Audio;
|
|
using Content.Shared.Chemistry.Components.SolutionManager;
|
|
using Content.Shared.Chemistry.EntitySystems;
|
|
using Content.Shared.Chemistry.Reagent;
|
|
using Content.Shared.Examine;
|
|
using Content.Shared.Interaction;
|
|
using Content.Shared.Popups;
|
|
using Content.Shared.Temperature;
|
|
using Content.Shared.Tools.Components;
|
|
using Robust.Server.GameObjects;
|
|
using Robust.Shared.Audio;
|
|
using Robust.Shared.GameObjects;
|
|
using Robust.Shared.GameStates;
|
|
using Robust.Shared.Localization;
|
|
using Robust.Shared.Player;
|
|
|
|
namespace Content.Server.Tools
|
|
{
|
|
public partial class ToolSystem
|
|
{
|
|
private readonly HashSet<EntityUid> _activeWelders = new();
|
|
|
|
private const float WelderUpdateTimer = 1f;
|
|
private float _welderTimer = 0f;
|
|
|
|
public void InitializeWelders()
|
|
{
|
|
SubscribeLocalEvent<WelderComponent, ComponentStartup>(OnWelderStartup);
|
|
SubscribeLocalEvent<WelderComponent, IsHotEvent>(OnWelderIsHotEvent);
|
|
SubscribeLocalEvent<WelderComponent, ExaminedEvent>(OnWelderExamine);
|
|
SubscribeLocalEvent<WelderComponent, SolutionChangedEvent>(OnWelderSolutionChange);
|
|
SubscribeLocalEvent<WelderComponent, UseInHandEvent>(OnWelderUseInHand);
|
|
SubscribeLocalEvent<WelderComponent, ActivateInWorldEvent>(OnWelderActivate);
|
|
SubscribeLocalEvent<WelderComponent, AfterInteractEvent>(OnWelderAfterInteract);
|
|
SubscribeLocalEvent<WelderComponent, ToolUseAttemptEvent>(OnWelderToolUseAttempt);
|
|
SubscribeLocalEvent<WelderComponent, ToolUseFinishAttemptEvent>(OnWelderToolUseFinishAttempt);
|
|
SubscribeLocalEvent<WelderComponent, ComponentShutdown>(OnWelderShutdown);
|
|
SubscribeLocalEvent<WelderComponent, ComponentGetState>(OnWelderGetState);
|
|
}
|
|
|
|
public (ReagentUnit fuel, ReagentUnit capacity) GetWelderFuelAndCapacity(EntityUid uid, WelderComponent? welder = null, SolutionContainerManagerComponent? solutionContainer = null)
|
|
{
|
|
if (!Resolve(uid, ref welder, ref solutionContainer)
|
|
|| !_solutionContainerSystem.TryGetSolution(uid, welder.FuelSolution, out var fuelSolution, solutionContainer))
|
|
return (ReagentUnit.Zero, ReagentUnit.Zero);
|
|
|
|
return (_solutionContainerSystem.GetReagentQuantity(uid, welder.FuelReagent), fuelSolution.MaxVolume);
|
|
}
|
|
|
|
public bool TryToggleWelder(EntityUid uid, EntityUid? user,
|
|
WelderComponent? welder = null,
|
|
SolutionContainerManagerComponent? solutionContainer = null,
|
|
ItemComponent? item = null,
|
|
PointLightComponent? light = null,
|
|
SpriteComponent? sprite = null)
|
|
{
|
|
// Right now, we only need the welder.
|
|
// So let's not unnecessarily resolve components
|
|
if (!Resolve(uid, ref welder))
|
|
return false;
|
|
|
|
return !welder.Lit
|
|
? TryTurnWelderOn(uid, user, welder, solutionContainer, item, light, sprite)
|
|
: TryTurnWelderOff(uid, user, welder, item, light, sprite);
|
|
}
|
|
|
|
public bool TryTurnWelderOn(EntityUid uid, EntityUid? user,
|
|
WelderComponent? welder = null,
|
|
SolutionContainerManagerComponent? solutionContainer = null,
|
|
ItemComponent? item = null,
|
|
PointLightComponent? light = null,
|
|
SpriteComponent? sprite = null)
|
|
{
|
|
if (!Resolve(uid, ref welder, ref solutionContainer))
|
|
return false;
|
|
|
|
// Optional components.
|
|
Resolve(uid, ref item, ref light, ref sprite);
|
|
|
|
if (!_solutionContainerSystem.TryGetSolution(uid, welder.FuelSolution, out var solution, solutionContainer))
|
|
return false;
|
|
|
|
var fuel = solution.GetReagentQuantity(welder.FuelReagent);
|
|
|
|
// Not enough fuel to lit welder.
|
|
if (fuel == ReagentUnit.Zero || fuel < welder.FuelLitCost)
|
|
{
|
|
if(user != null)
|
|
_popupSystem.PopupEntity(Loc.GetString("welder-component-no-fuel-message"), uid, Filter.Entities(user.Value));
|
|
return false;
|
|
}
|
|
|
|
if (user != null && !_actionBlockerSystem.CanInteract(user.Value))
|
|
return false;
|
|
|
|
solution.RemoveReagent(welder.FuelReagent, welder.FuelLitCost);
|
|
|
|
welder.Lit = true;
|
|
|
|
if(item != null)
|
|
item.EquippedPrefix = "on";
|
|
|
|
sprite?.LayerSetVisible(1, true);
|
|
|
|
if (light != null)
|
|
light.Enabled = true;
|
|
|
|
SoundSystem.Play(Filter.Pvs(uid), welder.WelderOnSounds.GetSound(), uid, AudioHelpers.WithVariation(0.125f).WithVolume(-5f));
|
|
|
|
// TODO: Use ITransformComponent directly.
|
|
_atmosphereSystem.HotspotExpose(welder.Owner.Transform.Coordinates, 700, 50, true);
|
|
|
|
welder.Dirty();
|
|
|
|
_activeWelders.Add(uid);
|
|
return true;
|
|
}
|
|
|
|
public bool TryTurnWelderOff(EntityUid uid, EntityUid? user,
|
|
WelderComponent? welder = null,
|
|
ItemComponent? item = null,
|
|
PointLightComponent? light = null,
|
|
SpriteComponent? sprite = null)
|
|
{
|
|
if (!Resolve(uid, ref welder))
|
|
return false;
|
|
|
|
// Optional components.
|
|
Resolve(uid, ref item, ref light, ref sprite);
|
|
|
|
if (user != null && !_actionBlockerSystem.CanInteract(user.Value))
|
|
return false;
|
|
|
|
welder.Lit = false;
|
|
|
|
// TODO: Make all this use visualizers.
|
|
if (item != null)
|
|
item.EquippedPrefix = "off";
|
|
|
|
// Layer 1 is the flame.
|
|
sprite?.LayerSetVisible(1, false);
|
|
|
|
if (light != null)
|
|
light.Enabled = false;
|
|
|
|
SoundSystem.Play(Filter.Pvs(uid), welder.WelderOffSounds.GetSound(), uid, AudioHelpers.WithVariation(0.125f).WithVolume(-5f));
|
|
|
|
welder.Dirty();
|
|
|
|
_activeWelders.Remove(uid);
|
|
return true;
|
|
}
|
|
|
|
private void OnWelderStartup(EntityUid uid, WelderComponent component, ComponentStartup args)
|
|
{
|
|
component.Dirty();
|
|
}
|
|
|
|
private void OnWelderIsHotEvent(EntityUid uid, WelderComponent welder, IsHotEvent args)
|
|
{
|
|
args.IsHot = welder.Lit;
|
|
}
|
|
|
|
private void OnWelderExamine(EntityUid uid, WelderComponent welder, ExaminedEvent args)
|
|
{
|
|
if (welder.Lit)
|
|
{
|
|
args.PushMarkup(Loc.GetString("welder-component-on-examine-welder-lit-message"));
|
|
}
|
|
else
|
|
{
|
|
args.PushMarkup(Loc.GetString("welder-component-on-examine-welder-not-lit-message"));
|
|
}
|
|
|
|
if (args.IsInDetailsRange)
|
|
{
|
|
var (fuel, capacity) = GetWelderFuelAndCapacity(uid, welder);
|
|
|
|
args.PushMarkup(Loc.GetString("welder-component-on-examine-detailed-message",
|
|
("colorName", fuel < capacity / ReagentUnit.New(4f) ? "darkorange" : "orange"),
|
|
("fuelLeft", fuel),
|
|
("fuelCapacity", capacity)));
|
|
}
|
|
}
|
|
|
|
private void OnWelderSolutionChange(EntityUid uid, WelderComponent welder, SolutionChangedEvent args)
|
|
{
|
|
welder.Dirty();
|
|
}
|
|
|
|
private void OnWelderActivate(EntityUid uid, WelderComponent welder, ActivateInWorldEvent args)
|
|
{
|
|
args.Handled = TryToggleWelder(uid, args.User.Uid, welder);
|
|
}
|
|
|
|
private void OnWelderAfterInteract(EntityUid uid, WelderComponent welder, AfterInteractEvent args)
|
|
{
|
|
if (args.Handled)
|
|
return;
|
|
|
|
if (args.Target == null || !args.CanReach)
|
|
return;
|
|
|
|
// TODO: Clean up this inherited oldcode.
|
|
|
|
if (args.Target.TryGetComponent(out ReagentTankComponent? tank)
|
|
&& tank.TankType == ReagentTankType.Fuel
|
|
&& _solutionContainerSystem.TryGetDrainableSolution(args.Target.Uid, out var targetSolution)
|
|
&& _solutionContainerSystem.TryGetSolution(uid, welder.FuelSolution, out var welderSolution))
|
|
{
|
|
var trans = ReagentUnit.Min(welderSolution.AvailableVolume, targetSolution.DrainAvailable);
|
|
if (trans > 0)
|
|
{
|
|
var drained = _solutionContainerSystem.Drain(args.Target.Uid, targetSolution, trans);
|
|
_solutionContainerSystem.TryAddSolution(uid, welderSolution, drained);
|
|
SoundSystem.Play(Filter.Pvs(uid), welder.WelderRefill.GetSound(), uid);
|
|
args.Target.PopupMessage(args.User, Loc.GetString("welder-component-after-interact-refueled-message"));
|
|
}
|
|
else
|
|
{
|
|
args.Target.PopupMessage(args.User, Loc.GetString("welder-component-no-fuel-in-tank", ("owner", args.Target)));
|
|
}
|
|
}
|
|
|
|
args.Handled = true;
|
|
}
|
|
|
|
private void OnWelderUseInHand(EntityUid uid, WelderComponent welder, UseInHandEvent args)
|
|
{
|
|
args.Handled = TryToggleWelder(uid, args.User.Uid, welder);
|
|
}
|
|
|
|
private void OnWelderToolUseAttempt(EntityUid uid, WelderComponent welder, ToolUseAttemptEvent args)
|
|
{
|
|
if (args.Cancelled)
|
|
return;
|
|
|
|
if (!welder.Lit)
|
|
{
|
|
_popupSystem.PopupEntity(Loc.GetString("welder-component-welder-not-lit-message"), uid, Filter.Entities(args.User));
|
|
args.Cancel();
|
|
return;
|
|
}
|
|
|
|
var (fuel, _) = GetWelderFuelAndCapacity(uid, welder);
|
|
|
|
if (ReagentUnit.New(args.Fuel) > fuel)
|
|
{
|
|
_popupSystem.PopupEntity(Loc.GetString("welder-component-cannot-weld-message"), uid, Filter.Entities(args.User));
|
|
args.Cancel();
|
|
return;
|
|
}
|
|
}
|
|
|
|
private void OnWelderToolUseFinishAttempt(EntityUid uid, WelderComponent welder, ToolUseFinishAttemptEvent args)
|
|
{
|
|
if (args.Cancelled)
|
|
return;
|
|
|
|
if (!welder.Lit)
|
|
{
|
|
_popupSystem.PopupEntity(Loc.GetString("welder-component-welder-not-lit-message"), uid, Filter.Entities(args.User));
|
|
args.Cancel();
|
|
return;
|
|
}
|
|
|
|
var (fuel, _) = GetWelderFuelAndCapacity(uid, welder);
|
|
|
|
var neededFuel = ReagentUnit.New(args.Fuel);
|
|
|
|
if (neededFuel > fuel)
|
|
{
|
|
args.Cancel();
|
|
}
|
|
|
|
if (!_solutionContainerSystem.TryGetSolution(uid, welder.FuelSolution, out var solution))
|
|
{
|
|
args.Cancel();
|
|
return;
|
|
}
|
|
|
|
solution.RemoveReagent(welder.FuelReagent, neededFuel);
|
|
welder.Dirty();
|
|
}
|
|
|
|
private void OnWelderShutdown(EntityUid uid, WelderComponent welder, ComponentShutdown args)
|
|
{
|
|
_activeWelders.Remove(uid);
|
|
}
|
|
|
|
private void OnWelderGetState(EntityUid uid, WelderComponent welder, ref ComponentGetState args)
|
|
{
|
|
var (fuel, capacity) = GetWelderFuelAndCapacity(uid, welder);
|
|
args.State = new WelderComponentState(capacity.Float(), fuel.Float(), welder.Lit);
|
|
}
|
|
|
|
private void UpdateWelders(float frameTime)
|
|
{
|
|
_welderTimer += frameTime;
|
|
|
|
if (_welderTimer < WelderUpdateTimer)
|
|
return;
|
|
|
|
foreach (var tool in _activeWelders.ToArray())
|
|
{
|
|
if (!EntityManager.TryGetComponent(tool, out WelderComponent? welder)
|
|
|| !EntityManager.TryGetComponent(tool, out SolutionContainerManagerComponent? solutionContainer))
|
|
continue;
|
|
|
|
if (!_solutionContainerSystem.TryGetSolution(tool, welder.FuelSolution, out var solution, solutionContainer))
|
|
continue;
|
|
|
|
// TODO: Use ITransformComponent directly.
|
|
_atmosphereSystem.HotspotExpose(welder.Owner.Transform.Coordinates, 700, 50, true);
|
|
|
|
solution.RemoveReagent(welder.FuelReagent, welder.FuelConsumption * _welderTimer);
|
|
|
|
if (solution.GetReagentQuantity(welder.FuelReagent) <= ReagentUnit.Zero)
|
|
TryTurnWelderOff(tool, null, welder);
|
|
|
|
welder.Dirty();
|
|
}
|
|
|
|
_welderTimer -= WelderUpdateTimer;
|
|
}
|
|
}
|
|
}
|