using System.Linq;
using System.Text.Json.Serialization;
using Content.Shared.Chemistry;
using Content.Shared.Chemistry.Components;
using Content.Shared.Database;
using Content.Shared.FixedPoint;
using Content.Shared.Localizations;
using JetBrains.Annotations;
using Robust.Shared.Prototypes;
using Robust.Shared.Random;
using Content.Shared.Chemistry.Reagent;
using Robust.Shared.Toolshed.TypeParsers;
namespace Content.Shared.EntityEffects;
///
/// Entity effects describe behavior that occurs on different kinds of triggers, e.g. when a reagent is ingested and metabolized by some
/// organ. They only trigger when all of are satisfied.
///
[ImplicitDataDefinitionForInheritors]
[MeansImplicitUse]
public abstract partial class EntityEffect
{
private protected string _id => this.GetType().Name;
///
/// The list of conditions required for the effect to activate. Not required.
///
[DataField("conditions")]
public EntityEffectCondition[]? Conditions;
public virtual string ReagentEffectFormat => "guidebook-reagent-effect-description";
protected abstract string? ReagentEffectGuidebookText(IPrototypeManager prototype, IEntitySystemManager entSys);
///
/// What's the chance, from 0 to 1, that this effect will occur?
///
[DataField("probability")]
public float Probability = 1.0f;
public virtual LogImpact LogImpact { get; private set; } = LogImpact.Low;
///
/// Should this entity effect log at all?
///
public virtual bool ShouldLog { get; private set; } = false;
public abstract void Effect(EntityEffectBaseArgs args);
///
/// Produces a localized, bbcode'd guidebook description for this effect.
///
///
public string? GuidebookEffectDescription(IPrototypeManager prototype, IEntitySystemManager entSys)
{
var effect = ReagentEffectGuidebookText(prototype, entSys);
if (effect is null)
return null;
return Loc.GetString(ReagentEffectFormat, ("effect", effect), ("chance", Probability),
("conditionCount", Conditions?.Length ?? 0),
("conditions",
ContentLocalizationManager.FormatList(Conditions?.Select(x => x.GuidebookExplanation(prototype)).ToList() ??
new List())));
}
}
public static class EntityEffectExt
{
public static bool ShouldApply(this EntityEffect effect, EntityEffectBaseArgs args,
IRobustRandom? random = null)
{
if (random == null)
random = IoCManager.Resolve();
if (effect.Probability < 1.0f && !random.Prob(effect.Probability))
return false;
if (effect.Conditions != null)
{
foreach (var cond in effect.Conditions)
{
if (!cond.Condition(args))
return false;
}
}
return true;
}
}
[ByRefEvent]
public struct ExecuteEntityEffectEvent where T : EntityEffect
{
public T Effect;
public EntityEffectBaseArgs Args;
public ExecuteEntityEffectEvent(T effect, EntityEffectBaseArgs args)
{
Effect = effect;
Args = args;
}
}
///
/// EntityEffectBaseArgs only contains the target of an effect.
/// If a trigger wants to include more info (e.g. the quantity of the chemical triggering the effect), it can be extended (see EntityEffectReagentArgs).
///
public record class EntityEffectBaseArgs
{
public EntityUid TargetEntity;
public IEntityManager EntityManager = default!;
public EntityEffectBaseArgs(EntityUid targetEntity, IEntityManager entityManager)
{
TargetEntity = targetEntity;
EntityManager = entityManager;
}
}
public record class EntityEffectReagentArgs : EntityEffectBaseArgs
{
public EntityUid? OrganEntity;
public Solution? Source;
public FixedPoint2 Quantity;
public ReagentPrototype? Reagent;
public ReactionMethod? Method;
public FixedPoint2 Scale;
public EntityEffectReagentArgs(EntityUid targetEntity, IEntityManager entityManager, EntityUid? organEntity, Solution? source, FixedPoint2 quantity, ReagentPrototype? reagent, ReactionMethod? method, FixedPoint2 scale) : base(targetEntity, entityManager)
{
OrganEntity = organEntity;
Source = source;
Quantity = quantity;
Reagent = reagent;
Method = method;
Scale = scale;
}
}