Re-organize all projects (#4166)
This commit is contained in:
DrSmugleaf
261
Content.Server/Nutrition/Components/DrinkComponent.cs
Normal file
261
Content.Server/Nutrition/Components/DrinkComponent.cs
Normal file
@@ -0,0 +1,261 @@
|
||||
#nullable enable
|
||||
using System.Linq;
|
||||
using System.Threading.Tasks;
|
||||
using Content.Server.Body.Behavior;
|
||||
using Content.Server.Fluids.Components;
|
||||
using Content.Shared.Audio;
|
||||
using Content.Shared.Body.Components;
|
||||
using Content.Shared.Chemistry;
|
||||
using Content.Shared.Chemistry.Reagent;
|
||||
using Content.Shared.Chemistry.Solution;
|
||||
using Content.Shared.Chemistry.Solution.Components;
|
||||
using Content.Shared.Examine;
|
||||
using Content.Shared.Interaction;
|
||||
using Content.Shared.Interaction.Helpers;
|
||||
using Content.Shared.Notification;
|
||||
using Content.Shared.Nutrition.Components;
|
||||
using Content.Shared.Throwing;
|
||||
using JetBrains.Annotations;
|
||||
using Robust.Server.GameObjects;
|
||||
using Robust.Shared.Audio;
|
||||
using Robust.Shared.GameObjects;
|
||||
using Robust.Shared.IoC;
|
||||
using Robust.Shared.Localization;
|
||||
using Robust.Shared.Player;
|
||||
using Robust.Shared.Prototypes;
|
||||
using Robust.Shared.Random;
|
||||
using Robust.Shared.Serialization.Manager.Attributes;
|
||||
using Robust.Shared.Utility;
|
||||
using Robust.Shared.ViewVariables;
|
||||
|
||||
namespace Content.Server.Nutrition.Components
|
||||
{
|
||||
[RegisterComponent]
|
||||
public class DrinkComponent : Component, IUse, IAfterInteract, ISolutionChange, IExamine, ILand
|
||||
{
|
||||
[Dependency] private readonly IPrototypeManager _prototypeManager = default!;
|
||||
[Dependency] private readonly IRobustRandom _random = default!;
|
||||
|
||||
public override string Name => "Drink";
|
||||
|
||||
int IAfterInteract.Priority => 10;
|
||||
|
||||
[ViewVariables]
|
||||
private bool _opened;
|
||||
|
||||
[ViewVariables]
|
||||
[DataField("useSound")]
|
||||
private string _useSound = "/Audio/Items/drink.ogg";
|
||||
|
||||
[ViewVariables]
|
||||
[DataField("isOpen")]
|
||||
private bool _defaultToOpened;
|
||||
|
||||
[ViewVariables(VVAccess.ReadWrite)]
|
||||
public ReagentUnit TransferAmount { get; [UsedImplicitly] private set; } = ReagentUnit.New(5);
|
||||
|
||||
[ViewVariables(VVAccess.ReadWrite)]
|
||||
public bool Opened
|
||||
{
|
||||
get => _opened;
|
||||
set
|
||||
{
|
||||
if (_opened == value)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
_opened = value;
|
||||
OpenedChanged();
|
||||
}
|
||||
}
|
||||
|
||||
[ViewVariables]
|
||||
public bool Empty => Owner.GetComponentOrNull<ISolutionInteractionsComponent>()?.DrainAvailable <= 0;
|
||||
|
||||
[DataField("openSounds")]
|
||||
private string _soundCollection = "canOpenSounds";
|
||||
[DataField("pressurized")]
|
||||
private bool _pressurized = default;
|
||||
[DataField("burstSound")]
|
||||
private string _burstSound = "/Audio/Effects/flash_bang.ogg";
|
||||
|
||||
public override void Initialize()
|
||||
{
|
||||
base.Initialize();
|
||||
|
||||
Opened = _defaultToOpened;
|
||||
UpdateAppearance();
|
||||
}
|
||||
|
||||
private void OpenedChanged()
|
||||
{
|
||||
if (!Owner.TryGetComponent(out SharedSolutionContainerComponent? contents))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
if (Opened)
|
||||
{
|
||||
contents.Capabilities |= SolutionContainerCaps.Refillable | SolutionContainerCaps.Drainable;
|
||||
}
|
||||
else
|
||||
{
|
||||
contents.Capabilities &= ~(SolutionContainerCaps.Refillable | SolutionContainerCaps.Drainable);
|
||||
}
|
||||
}
|
||||
|
||||
void ISolutionChange.SolutionChanged(SolutionChangeEventArgs eventArgs)
|
||||
{
|
||||
UpdateAppearance();
|
||||
}
|
||||
|
||||
private void UpdateAppearance()
|
||||
{
|
||||
if (!Owner.TryGetComponent(out AppearanceComponent? appearance) ||
|
||||
!Owner.TryGetComponent(out ISolutionInteractionsComponent? contents))
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
appearance.SetData(SharedFoodComponent.FoodVisuals.Visual, contents.DrainAvailable.Float());
|
||||
}
|
||||
|
||||
bool IUse.UseEntity(UseEntityEventArgs args)
|
||||
{
|
||||
if (!Opened)
|
||||
{
|
||||
//Do the opening stuff like playing the sounds.
|
||||
var soundCollection = _prototypeManager.Index<SoundCollectionPrototype>(_soundCollection);
|
||||
var file = _random.Pick(soundCollection.PickFiles);
|
||||
|
||||
SoundSystem.Play(Filter.Pvs(args.User), file, args.User, AudioParams.Default);
|
||||
Opened = true;
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!Owner.TryGetComponent(out ISolutionInteractionsComponent? contents) ||
|
||||
contents.DrainAvailable <= 0)
|
||||
{
|
||||
args.User.PopupMessage(Loc.GetString("{0:theName} is empty!", Owner));
|
||||
return true;
|
||||
}
|
||||
|
||||
return TryUseDrink(args.User, args.User);
|
||||
}
|
||||
|
||||
//Force feeding a drink to someone.
|
||||
async Task<bool> IAfterInteract.AfterInteract(AfterInteractEventArgs eventArgs)
|
||||
{
|
||||
if (eventArgs.Target == null)
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
return TryUseDrink(eventArgs.User, eventArgs.Target, true);
|
||||
}
|
||||
|
||||
public void Examine(FormattedMessage message, bool inDetailsRange)
|
||||
{
|
||||
if (!Opened || !inDetailsRange)
|
||||
{
|
||||
return;
|
||||
}
|
||||
var color = Empty ? "gray" : "yellow";
|
||||
var openedText = Loc.GetString(Empty ? "Empty" : "Opened");
|
||||
message.AddMarkup(Loc.GetString("[color={0}]{1}[/color]", color, openedText));
|
||||
}
|
||||
|
||||
private bool TryUseDrink(IEntity user, IEntity target, bool forced = false)
|
||||
{
|
||||
if (!Opened)
|
||||
{
|
||||
target.PopupMessage(Loc.GetString("Open {0:theName} first!", Owner));
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!Owner.TryGetComponent(out ISolutionInteractionsComponent? interactions) ||
|
||||
!interactions.CanDrain ||
|
||||
interactions.DrainAvailable <= 0)
|
||||
{
|
||||
if (!forced)
|
||||
{
|
||||
target.PopupMessage(Loc.GetString("{0:theName} is empty!", Owner));
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!target.TryGetComponent(out IBody? body) ||
|
||||
!body.TryGetMechanismBehaviors<StomachBehavior>(out var stomachs))
|
||||
{
|
||||
target.PopupMessage(Loc.GetString("You can't drink {0:theName}!", Owner));
|
||||
return false;
|
||||
}
|
||||
|
||||
|
||||
if (user != target &&
|
||||
!user.InRangeUnobstructed(target, popup: true))
|
||||
{
|
||||
return false;
|
||||
}
|
||||
|
||||
var transferAmount = ReagentUnit.Min(TransferAmount, interactions.DrainAvailable);
|
||||
var drain = interactions.Drain(transferAmount);
|
||||
var firstStomach = stomachs.FirstOrDefault(stomach => stomach.CanTransferSolution(drain));
|
||||
|
||||
// All stomach are full or can't handle whatever solution we have.
|
||||
if (firstStomach == null)
|
||||
{
|
||||
target.PopupMessage(Loc.GetString("You've had enough {0:theName}!", Owner));
|
||||
|
||||
if (!interactions.CanRefill)
|
||||
{
|
||||
drain.SpillAt(target, "PuddleSmear");
|
||||
return false;
|
||||
}
|
||||
|
||||
interactions.Refill(drain);
|
||||
return false;
|
||||
}
|
||||
|
||||
if (!string.IsNullOrEmpty(_useSound))
|
||||
{
|
||||
SoundSystem.Play(Filter.Pvs(target), _useSound, target, AudioParams.Default.WithVolume(-2f));
|
||||
}
|
||||
|
||||
target.PopupMessage(Loc.GetString("Slurp"));
|
||||
UpdateAppearance();
|
||||
|
||||
// TODO: Account for partial transfer.
|
||||
|
||||
drain.DoEntityReaction(target, ReactionMethod.Ingestion);
|
||||
|
||||
firstStomach.TryTransferSolution(drain);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void ILand.Land(LandEventArgs eventArgs)
|
||||
{
|
||||
if (_pressurized &&
|
||||
!Opened &&
|
||||
_random.Prob(0.25f) &&
|
||||
Owner.TryGetComponent(out ISolutionInteractionsComponent? interactions))
|
||||
{
|
||||
Opened = true;
|
||||
|
||||
if (!interactions.CanDrain)
|
||||
{
|
||||
return;
|
||||
}
|
||||
|
||||
var solution = interactions.Drain(interactions.DrainAvailable);
|
||||
solution.SpillAt(Owner, "PuddleSmear");
|
||||
|
||||
SoundSystem.Play(Filter.Pvs(Owner), _burstSound, Owner,
|
||||
AudioParams.Default.WithVolume(-4));
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user