using Content.Shared.Administration.Logs;
using Content.Shared.Chemistry.Components;
using Content.Shared.CombatMode;
using Content.Shared.DoAfter;
using Content.Shared.FixedPoint;
using Content.Shared.Interaction.Events;
using Content.Shared.Mobs.Systems;
using Content.Shared.Popups;
using Content.Shared.Verbs;
using Robust.Shared.Player;
namespace Content.Shared.Chemistry.EntitySystems;
public abstract class SharedInjectorSystem : EntitySystem
{
///
/// Default transfer amounts for the set-transfer verb.
///
public static readonly FixedPoint2[] TransferAmounts = { 1, 5, 10, 15 };
[Dependency] protected readonly SharedPopupSystem Popup = default!;
[Dependency] protected readonly SharedSolutionContainerSystem SolutionContainers = default!;
[Dependency] protected readonly MobStateSystem MobState = default!;
[Dependency] protected readonly SharedCombatModeSystem Combat = default!;
[Dependency] protected readonly SharedDoAfterSystem DoAfter = default!;
[Dependency] protected readonly ISharedAdminLogManager AdminLogger = default!;
public override void Initialize()
{
SubscribeLocalEvent>(AddSetTransferVerbs);
SubscribeLocalEvent(OnInjectorStartup);
SubscribeLocalEvent(OnInjectorUse);
}
private void AddSetTransferVerbs(Entity entity, ref GetVerbsEvent args)
{
if (!args.CanAccess || !args.CanInteract || args.Hands == null)
return;
var user = args.User;
var (_, component) = entity;
var min = component.MinimumTransferAmount;
var max = component.MaximumTransferAmount;
var cur = component.TransferAmount;
var toggleAmount = cur == max ? min : max;
var priority = 0;
AlternativeVerb toggleVerb = new()
{
Text = Loc.GetString("comp-solution-transfer-verb-toggle", ("amount", toggleAmount)),
Category = VerbCategory.SetTransferAmount,
Act = () =>
{
component.TransferAmount = toggleAmount;
Popup.PopupClient(Loc.GetString("comp-solution-transfer-set-amount", ("amount", toggleAmount)), user, user);
Dirty(entity);
},
Priority = priority
};
args.Verbs.Add(toggleVerb);
priority -= 1;
// Add specific transfer verbs according to the container's size
foreach (var amount in TransferAmounts)
{
if (amount < component.MinimumTransferAmount || amount > component.MaximumTransferAmount)
continue;
AlternativeVerb verb = new()
{
Text = Loc.GetString("comp-solution-transfer-verb-amount", ("amount", amount)),
Category = VerbCategory.SetTransferAmount,
Act = () =>
{
component.TransferAmount = amount;
Popup.PopupClient(Loc.GetString("comp-solution-transfer-set-amount", ("amount", amount)), user, user);
Dirty(entity);
},
// we want to sort by size, not alphabetically by the verb text.
Priority = priority
};
priority -= 1;
args.Verbs.Add(verb);
}
}
private void OnInjectorStartup(Entity entity, ref ComponentStartup args)
{
// ???? why ?????
Dirty(entity);
}
private void OnInjectorUse(Entity entity, ref UseInHandEvent args)
{
if (args.Handled)
return;
Toggle(entity, args.User);
args.Handled = true;
}
///
/// Toggle between draw/inject state if applicable
///
private void Toggle(Entity injector, EntityUid user)
{
if (injector.Comp.InjectOnly)
return;
if (!SolutionContainers.TryGetSolution(injector.Owner, injector.Comp.SolutionName, out var solEnt, out var solution))
return;
string msg;
switch (injector.Comp.ToggleState)
{
case InjectorToggleMode.Inject:
if (solution.AvailableVolume > 0) // If solution has empty space to fill up, allow toggling to draw
{
SetMode(injector, InjectorToggleMode.Draw);
msg = "injector-component-drawing-text";
}
else
{
msg = "injector-component-cannot-toggle-draw-message";
}
break;
case InjectorToggleMode.Draw:
if (solution.Volume > 0) // If solution has anything in it, allow toggling to inject
{
SetMode(injector, InjectorToggleMode.Inject);
msg = "injector-component-injecting-text";
}
else
{
msg = "injector-component-cannot-toggle-inject-message";
}
break;
default:
throw new ArgumentOutOfRangeException();
}
Popup.PopupClient(Loc.GetString(msg), injector, user);
}
public void SetMode(Entity injector, InjectorToggleMode mode)
{
injector.Comp.ToggleState = mode;
Dirty(injector);
}
}