using Content.Server.Body.Components; using Content.Server.Body.Systems; using Content.Server.EntityEffects.Effects; using Content.Server.Fluids.EntitySystems; using Content.Server.Forensics; using Content.Server.Inventory; using Content.Server.Popups; using Content.Shared.Administration.Logs; using Content.Shared.Body.Components; using Content.Shared.Chemistry; using Content.Shared.Chemistry.Components; using Content.Shared.Chemistry.Components.SolutionManager; using Content.Shared.Chemistry.EntitySystems; using Content.Shared.Chemistry.Reagent; using Content.Shared.Database; using Content.Shared.DoAfter; using Content.Shared.EntityEffects; using Content.Shared.FixedPoint; using Content.Shared.Hands.EntitySystems; using Content.Shared.IdentityManagement; using Content.Shared.Interaction; using Content.Shared.Interaction.Events; using Content.Shared.Mobs.Systems; using Content.Shared.Nutrition; using Content.Shared.Nutrition.Components; using Content.Shared.Nutrition.EntitySystems; using Content.Shared.Verbs; using Robust.Shared.Audio; using Robust.Shared.Audio.Systems; using Robust.Shared.Player; using Robust.Shared.Prototypes; using Robust.Shared.Utility; namespace Content.Server.Nutrition.EntitySystems; public sealed class DrinkSystem : SharedDrinkSystem { [Dependency] private readonly BodySystem _body = default!; [Dependency] private readonly FlavorProfileSystem _flavorProfile = default!; [Dependency] private readonly FoodSystem _food = default!; [Dependency] private readonly IPrototypeManager _proto = default!; [Dependency] private readonly ISharedAdminLogManager _adminLogger = default!; [Dependency] private readonly MobStateSystem _mobState = default!; [Dependency] private readonly OpenableSystem _openable = default!; [Dependency] private readonly PopupSystem _popup = default!; [Dependency] private readonly PuddleSystem _puddle = default!; [Dependency] private readonly ReactiveSystem _reaction = default!; [Dependency] private readonly SharedAppearanceSystem _appearance = default!; [Dependency] private readonly SharedAudioSystem _audio = default!; [Dependency] private readonly SharedDoAfterSystem _doAfter = default!; [Dependency] private readonly SharedHandsSystem _hands = default!; [Dependency] private readonly SharedInteractionSystem _interaction = default!; [Dependency] private readonly SharedSolutionContainerSystem _solutionContainer = default!; [Dependency] private readonly StomachSystem _stomach = default!; [Dependency] private readonly ForensicsSystem _forensics = default!; public override void Initialize() { base.Initialize(); // TODO add InteractNoHandEvent for entities like mice. SubscribeLocalEvent(OnSolutionChange); SubscribeLocalEvent(OnDrinkInit); // run before inventory so for bucket it always tries to drink before equipping (when empty) // run after openable so its always open -> drink SubscribeLocalEvent(OnUse, before: [typeof(ServerInventorySystem)], after: [typeof(OpenableSystem)]); SubscribeLocalEvent(AfterInteract); SubscribeLocalEvent>(AddDrinkVerb); SubscribeLocalEvent(OnDoAfter); } /// /// Get the total hydration factor contained in a drink's solution. /// public float TotalHydration(EntityUid uid, DrinkComponent? comp = null) { if (!Resolve(uid, ref comp)) return 0f; if (!_solutionContainer.TryGetSolution(uid, comp.Solution, out _, out var solution)) return 0f; var total = 0f; foreach (var quantity in solution.Contents) { var reagent = _proto.Index(quantity.Reagent.Prototype); if (reagent.Metabolisms == null) continue; foreach (var entry in reagent.Metabolisms.Values) { foreach (var effect in entry.Effects) { // ignores any effect conditions, just cares about how much it can hydrate if (effect is SatiateThirst thirst) { total += thirst.HydrationFactor * quantity.Quantity.Float(); } } } } return total; } private void AfterInteract(Entity entity, ref AfterInteractEvent args) { if (args.Handled || args.Target == null || !args.CanReach) return; args.Handled = TryDrink(args.User, args.Target.Value, entity.Comp, entity); } private void OnUse(Entity entity, ref UseInHandEvent args) { if (args.Handled) return; args.Handled = TryDrink(args.User, args.User, entity.Comp, entity); } private void OnDrinkInit(Entity entity, ref ComponentInit args) { if (TryComp(entity, out var existingDrainable)) { // Beakers have Drink component but they should use the existing Drainable entity.Comp.Solution = existingDrainable.Solution; } else { _solutionContainer.EnsureSolution(entity.Owner, entity.Comp.Solution, out _); } UpdateAppearance(entity, entity.Comp); if (TryComp(entity, out RefillableSolutionComponent? refillComp)) refillComp.Solution = entity.Comp.Solution; if (TryComp(entity, out DrainableSolutionComponent? drainComp)) drainComp.Solution = entity.Comp.Solution; } private void OnSolutionChange(Entity entity, ref SolutionContainerChangedEvent args) { UpdateAppearance(entity, entity.Comp); } public void UpdateAppearance(EntityUid uid, DrinkComponent component) { if (!TryComp(uid, out var appearance) || !HasComp(uid)) { return; } var drainAvailable = DrinkVolume(uid, component); _appearance.SetData(uid, FoodVisuals.Visual, drainAvailable.Float(), appearance); } /// /// Tries to feed the drink item to the target entity /// private bool TryDrink(EntityUid user, EntityUid target, DrinkComponent drink, EntityUid item) { if (!HasComp(target)) return false; if (!_body.TryGetBodyOrganEntityComps(target, out var stomachs)) return false; if (_openable.IsClosed(item, user)) return true; if (!_solutionContainer.TryGetSolution(item, drink.Solution, out _, out var drinkSolution) || drinkSolution.Volume <= 0) { if (drink.IgnoreEmpty) return false; _popup.PopupEntity(Loc.GetString("drink-component-try-use-drink-is-empty", ("entity", item)), item, user); return true; } if (_food.IsMouthBlocked(target, user)) return true; if (!_interaction.InRangeUnobstructed(user, item, popup: true)) return true; var forceDrink = user != target; if (forceDrink) { var userName = Identity.Entity(user, EntityManager); _popup.PopupEntity(Loc.GetString("drink-component-force-feed", ("user", userName)), user, target); // logging _adminLogger.Add(LogType.ForceFeed, LogImpact.High, $"{ToPrettyString(user):user} is forcing {ToPrettyString(target):target} to drink {ToPrettyString(item):drink} {SharedSolutionContainerSystem.ToPrettyString(drinkSolution)}"); } else { // log voluntary drinking _adminLogger.Add(LogType.Ingestion, LogImpact.Low, $"{ToPrettyString(target):target} is drinking {ToPrettyString(item):drink} {SharedSolutionContainerSystem.ToPrettyString(drinkSolution)}"); } var flavors = _flavorProfile.GetLocalizedFlavorsMessage(user, drinkSolution); var doAfterEventArgs = new DoAfterArgs(EntityManager, user, forceDrink ? drink.ForceFeedDelay : drink.Delay, new ConsumeDoAfterEvent(drink.Solution, flavors), eventTarget: item, target: target, used: item) { BreakOnHandChange = false, BreakOnMove = forceDrink, BreakOnDamage = true, MovementThreshold = 0.01f, DistanceThreshold = 1.0f, // do-after will stop if item is dropped when trying to feed someone else // or if the item started out in the user's own hands NeedHand = forceDrink || _hands.IsHolding(user, item), }; _doAfter.TryStartDoAfter(doAfterEventArgs); return true; } /// /// Raised directed at a victim when someone has force fed them a drink. /// private void OnDoAfter(Entity entity, ref ConsumeDoAfterEvent args) { if (args.Handled || args.Cancelled || entity.Comp.Deleted) return; if (!TryComp(args.Target, out var body)) return; if (args.Used is null || !_solutionContainer.TryGetSolution(args.Used.Value, args.Solution, out var soln, out var solution)) return; if (_openable.IsClosed(args.Used.Value, args.Target.Value)) return; // TODO this should really be checked every tick. if (_food.IsMouthBlocked(args.Target.Value)) return; // TODO this should really be checked every tick. if (!_interaction.InRangeUnobstructed(args.User, args.Target.Value)) return; var transferAmount = FixedPoint2.Min(entity.Comp.TransferAmount, solution.Volume); var drained = _solutionContainer.SplitSolution(soln.Value, transferAmount); var forceDrink = args.User != args.Target; args.Handled = true; if (transferAmount <= 0) return; if (!_body.TryGetBodyOrganEntityComps((args.Target.Value, body), out var stomachs)) { _popup.PopupEntity(Loc.GetString(forceDrink ? "drink-component-try-use-drink-cannot-drink-other" : "drink-component-try-use-drink-had-enough"), args.Target.Value, args.User); if (HasComp(args.Target.Value)) { _puddle.TrySpillAt(args.User, drained, out _); return; } _solutionContainer.Refill(args.Target.Value, soln.Value, drained); return; } var firstStomach = stomachs.FirstOrNull(stomach => _stomach.CanTransferSolution(stomach.Owner, drained, stomach.Comp1)); //All stomachs are full or can't handle whatever solution we have. if (firstStomach == null) { _popup.PopupEntity(Loc.GetString("drink-component-try-use-drink-had-enough"), args.Target.Value, args.Target.Value); if (forceDrink) { _popup.PopupEntity(Loc.GetString("drink-component-try-use-drink-had-enough-other"), args.Target.Value, args.User); _puddle.TrySpillAt(args.Target.Value, drained, out _); } else _solutionContainer.TryAddSolution(soln.Value, drained); return; } var flavors = args.FlavorMessage; if (forceDrink) { var targetName = Identity.Entity(args.Target.Value, EntityManager); var userName = Identity.Entity(args.User, EntityManager); _popup.PopupEntity(Loc.GetString("drink-component-force-feed-success", ("user", userName), ("flavors", flavors)), args.Target.Value, args.Target.Value); _popup.PopupEntity( Loc.GetString("drink-component-force-feed-success-user", ("target", targetName)), args.User, args.User); // log successful forced drinking _adminLogger.Add(LogType.ForceFeed, LogImpact.Medium, $"{ToPrettyString(entity.Owner):user} forced {ToPrettyString(args.User):target} to drink {ToPrettyString(entity.Owner):drink}"); } else { _popup.PopupEntity( Loc.GetString("drink-component-try-use-drink-success-slurp-taste", ("flavors", flavors)), args.User, args.User); _popup.PopupEntity( Loc.GetString("drink-component-try-use-drink-success-slurp"), args.User, Filter.PvsExcept(args.User), true); // log successful voluntary drinking _adminLogger.Add(LogType.Ingestion, LogImpact.Low, $"{ToPrettyString(args.User):target} drank {ToPrettyString(entity.Owner):drink}"); } _audio.PlayPvs(entity.Comp.UseSound, args.Target.Value, AudioParams.Default.WithVolume(-2f).WithVariation(0.25f)); _reaction.DoEntityReaction(args.Target.Value, solution, ReactionMethod.Ingestion); _stomach.TryTransferSolution(firstStomach.Value.Owner, drained, firstStomach.Value.Comp1); _forensics.TransferDna(entity, args.Target.Value); if (!forceDrink && solution.Volume > 0) args.Repeat = true; } private void AddDrinkVerb(Entity entity, ref GetVerbsEvent ev) { if (entity.Owner == ev.User || !ev.CanInteract || !ev.CanAccess || !TryComp(ev.User, out var body) || !_body.TryGetBodyOrganEntityComps((ev.User, body), out var stomachs)) return; // Make sure the solution exists if (!_solutionContainer.TryGetSolution(entity.Owner, entity.Comp.Solution, out var solution)) return; // no drinking from living drinks, have to kill them first. if (_mobState.IsAlive(entity)) return; var user = ev.User; AlternativeVerb verb = new() { Act = () => { TryDrink(user, user, entity.Comp, entity); }, Icon = new SpriteSpecifier.Texture(new("/Textures/Interface/VerbIcons/drink.svg.192dpi.png")), Text = Loc.GetString("drink-system-verb-drink"), Priority = 2 }; ev.Verbs.Add(verb); } }