GuideReagentEmbed.xaml.cs

using Content.Client.Chemistry.EntitySystems;
using Content.Client.Guidebook.Richtext;
using Content.Client.Message;
using Content.Client.UserInterface.ControlExtensions;
using Content.Shared.CCVar;
using Content.Shared.Chemistry.Reaction;
using Content.Shared.Chemistry.Reagent;
using Content.Shared.Contraband;
using Content.Shared.Localizations;
using Content.Shared.Metabolism;
using JetBrains.Annotations;
using Robust.Client.AutoGenerated;
using Robust.Client.Graphics;
using Robust.Client.UserInterface;
using Robust.Client.UserInterface.Controls;
using Robust.Client.UserInterface.XAML;
using Robust.Shared.Configuration;
using Robust.Shared.Prototypes;
using Robust.Shared.Utility;
using System.Diagnostics.CodeAnalysis;
using System.Linq;

namespace Content.Client.Guidebook.Controls;

/// <summary>
///     Control for embedding a reagent into a guidebook.
/// </summary>
[UsedImplicitly, GenerateTypedNameReferences]
public sealed partial class GuideReagentEmbed : BoxContainer, IDocumentTag, ISearchableControl, IPrototypeRepresentationControl
{
    [Dependency] private readonly IEntitySystemManager _systemManager = default!;
    [Dependency] private readonly ILogManager _logManager = default!;
    [Dependency] private readonly IPrototypeManager _prototype = default!;
    [Dependency] private readonly IConfigurationManager _config = default!;

    private readonly ChemistryGuideDataSystem _chemistryGuideData;
    private readonly ContrabandSystem _contraband;
    private readonly ISawmill _sawmill;

    public IPrototype? RepresentedPrototype { get; private set; }

    public GuideReagentEmbed()
    {
        RobustXamlLoader.Load(this);
        IoCManager.InjectDependencies(this);
        _sawmill = _logManager.GetSawmill("guidebook.reagent");
        _chemistryGuideData = _systemManager.GetEntitySystem<ChemistryGuideDataSystem>();
        _contraband = _systemManager.GetEntitySystem<ContrabandSystem>();
        MouseFilter = MouseFilterMode.Stop;
    }

    public GuideReagentEmbed(string reagent) : this()
    {
        GenerateControl(_prototype.Index<ReagentPrototype>(reagent));
    }

    public GuideReagentEmbed(ReagentPrototype reagent) : this()
    {
        GenerateControl(reagent);
    }

    public GuideReagentEmbed(ReactionPrototype reagent) : this()
    {
        GenerateControl(reagent);
    }

    public bool CheckMatchesSearch(string query)
    {
        return this.ChildrenContainText(query);
    }

    public void SetHiddenState(bool state, string query)
    {
        Visible = CheckMatchesSearch(query) ? state : !state;
    }

    public bool TryParseTag(Dictionary<string, string> args, [NotNullWhen(true)] out Control? control)
    {
        control = null;
        if (!args.TryGetValue("Reagent", out var id))
        {
            _sawmill.Error("Reagent embed tag is missing reagent prototype argument");
            return false;
        }

        if (!_prototype.TryIndex<ReagentPrototype>(id, out var reagent))
        {
            _sawmill.Error($"Specified reagent prototype \"{id}\" is not a valid reagent prototype");
            return false;
        }

        GenerateControl(reagent);

        control = this;
        return true;
    }

    private void GenerateControl(ReagentPrototype reagent)
    {
        RepresentedPrototype = reagent;

        NameBackground.PanelOverride = new StyleBoxFlat
        {
            BackgroundColor = reagent.SubstanceColor
        };

        var r = reagent.SubstanceColor.R;
        var g = reagent.SubstanceColor.G;
        var b = reagent.SubstanceColor.B;

        var textColor = 0.2126f * r + 0.7152f * g + 0.0722f * b > 0.5
            ? Color.Black
            : Color.White;

        ReagentName.SetMarkup(Loc.GetString("guidebook-reagent-name",
            ("color", textColor), ("name", reagent.LocalizedName)));

        #region Recipe
        var reactions = _prototype.EnumeratePrototypes<ReactionPrototype>()
            .Where(p => !p.Source && p.Products.ContainsKey(reagent.ID))
            .OrderBy(p => p.Priority)
            .ThenBy(p => p.Products.Count)
            .ToList();

        if (reactions.Any())
        {
            foreach (var reactionPrototype in reactions)
            {
                RecipesDescriptionContainer.AddChild(new GuideReagentReaction(reactionPrototype, _prototype, _systemManager));
            }
        }
        else
        {
            RecipesContainer.Visible = false;
        }
        #endregion

        #region Effects
        if (_chemistryGuideData.ReagentGuideRegistry.TryGetValue(reagent.ID, out var guideEntryRegistry) &&
            guideEntryRegistry.GuideEntries != null &&
            guideEntryRegistry.GuideEntries.Values.Any(pair => pair.EffectDescriptions.Any() || pair.Metabolites?.Any() == true))
        {
            EffectsDescriptionContainer.Children.Clear();
            foreach (var (stage, effect) in guideEntryRegistry.GuideEntries)
            {
                var hasMetabolites = effect.Metabolites?.Any() == true;
                if (!effect.EffectDescriptions.Any() && !hasMetabolites)
                    continue;

                var groupLabel = new RichTextLabel();
                groupLabel.SetMarkup(Loc.GetString("guidebook-reagent-effects-metabolism-stage-rate",
                    ("stage", _prototype.Index<MetabolismStagePrototype>(stage).LocalizedName), ("rate", effect.MetabolismRate)));
                var descriptionLabel = new RichTextLabel
                {
                    Margin = new Thickness(25, 0, 10, 0)
                };

                var descMsg = new FormattedMessage();
                var descriptionsCount = effect.EffectDescriptions.Length;
                var i = 0;
                foreach (var effectString in effect.EffectDescriptions)
                {
                    descMsg.AddMarkupOrThrow(effectString);
                    i++;
                    if (i < descriptionsCount || hasMetabolites)
                        descMsg.PushNewline();
                }
                if (hasMetabolites)
                {
                    var metabolites = new List<string>();
                    foreach (var (metabolite, ratio) in effect.Metabolites!)
                    {
                        metabolites.Add(Loc.GetString("guidebook-reagent-effects-metabolite-item", ("rate", (double)ratio), ("reagent", _prototype.Index(metabolite).LocalizedName)));
                    }
                    metabolites.Sort();

                    descMsg.AddMarkupOrThrow(Loc.GetString("guidebook-reagent-effects-metabolites", ("items", ContentLocalizationManager.FormatList(metabolites))));
                }
                descriptionLabel.SetMessage(descMsg);

                EffectsDescriptionContainer.AddChild(groupLabel);
                EffectsDescriptionContainer.AddChild(descriptionLabel);
            }
        }
        else
        {
            EffectsContainer.Visible = false;
        }
        #endregion

        #region PlantMetabolisms
        if (_chemistryGuideData.ReagentGuideRegistry.TryGetValue(reagent.ID, out var guideEntryRegistryPlant) &&
            guideEntryRegistryPlant.PlantMetabolisms != null &&
            guideEntryRegistryPlant.PlantMetabolisms.Count > 0)
        {
            PlantMetabolismsDescriptionContainer.Children.Clear();
            var metabolismLabel = new RichTextLabel();
            metabolismLabel.SetMarkup(Loc.GetString("guidebook-reagent-plant-metabolisms-rate"));
            var descriptionLabel = new RichTextLabel
            {
                Margin = new Thickness(25, 0, 10, 0)
            };
            var descMsg = new FormattedMessage();
            var descriptionsCount = guideEntryRegistryPlant.PlantMetabolisms.Count;
            var i = 0;
            foreach (var effectString in guideEntryRegistryPlant.PlantMetabolisms)
            {
                descMsg.AddMarkupOrThrow(effectString);
                i++;
                if (i < descriptionsCount)
                    descMsg.PushNewline();
            }
            descriptionLabel.SetMessage(descMsg);

            PlantMetabolismsDescriptionContainer.AddChild(metabolismLabel);
            PlantMetabolismsDescriptionContainer.AddChild(descriptionLabel);
        }
        else
        {
            PlantMetabolismsContainer.Visible = false;
        }
        #endregion

        GenerateSources(reagent);

        FormattedMessage description = new();
        description.AddText(reagent.LocalizedDescription);
        description.PushNewline();
        description.AddMarkupOrThrow(Loc.GetString("guidebook-reagent-physical-description",
            ("description", reagent.LocalizedPhysicalDescription)));

        if (_config.GetCVar(CCVars.ContrabandExamine))
        {
            // Department-restricted text
            if (reagent.AllowedJobs.Count > 0 || reagent.AllowedDepartments.Count > 0)
            {
                description.PushNewline();
                description.AddMarkupPermissive(
                    _contraband.GenerateDepartmentExamineMessage(reagent.AllowedDepartments, reagent.AllowedJobs, ContrabandItemType.Reagent));
            }
            // Other contraband text
            else if (reagent.ContrabandSeverity != null &&
                     _prototype.Resolve(reagent.ContrabandSeverity.Value, out var severity))
            {
                description.PushNewline();
                description.AddMarkupPermissive(Loc.GetString(severity.ExamineText, ("type", ContrabandItemType.Reagent)));
            }
        }

        ReagentDescription.SetMessage(description);
    }

    private void GenerateControl(ReactionPrototype reagent)
    {
        RepresentedPrototype = reagent;

        NameBackground.PanelOverride = new StyleBoxFlat
        {
            BackgroundColor = Color.Black
        };

        if (reagent.Effects.Length < 1)
            return;
        IEnumerable<Shared.EntityEffects.Effects.EntitySpawning.SpawnEntity> spawnEvents = reagent.Effects.OfType<Shared.EntityEffects.Effects.EntitySpawning.SpawnEntity>();
        if (spawnEvents.Count() == 0)
            return;

        var spawnId = spawnEvents.First().Entity;
        ReagentName.SetMarkup(Loc.GetString("guidebook-reagent-name",
            ("color", Color.White), ("name", _prototype.Index<EntityPrototype>(spawnId).Name)));

        _sawmill.Info($"{reagent.ID}");
        #region Recipe
        var reactions = _prototype.EnumeratePrototypes<ReactionPrototype>()
                .Where(p => !p.Source && p.Effects.Length >= 1)
                .OrderBy(p => p.Priority)
                .ThenBy(p => p.Products.Count)
                .ToList();

        if (reactions.Any())
        {
            foreach (var reactionPrototype in reactions)
            {
                IEnumerable<Shared.EntityEffects.Effects.EntitySpawning.SpawnEntity> stuff = reactionPrototype.Effects.OfType<Shared.EntityEffects.Effects.EntitySpawning.SpawnEntity>();
                if (stuff.Count() >= 1 && stuff.First().Entity == spawnId)
                {
                    RecipesDescriptionContainer.AddChild(new GuideReagentReaction(reactionPrototype, _prototype, _systemManager, spawnId));
                }
            }
        }
        else
        {
            RecipesContainer.Visible = false;
        }
        #endregion

        #region Effects
        if (_chemistryGuideData.ReagentGuideRegistry.TryGetValue(reagent.ID, out var guideEntryRegistry) &&
            guideEntryRegistry.GuideEntries != null &&
            guideEntryRegistry.GuideEntries.Values.Any(pair => pair.EffectDescriptions.Any() || pair.Metabolites?.Any() == true))
        {
            EffectsDescriptionContainer.Children.Clear();
            foreach (var (stage, effect) in guideEntryRegistry.GuideEntries)
            {
                var hasMetabolites = effect.Metabolites?.Any() == true;
                if (!effect.EffectDescriptions.Any() && !hasMetabolites)
                    continue;

                var groupLabel = new RichTextLabel();
                groupLabel.SetMarkup(Loc.GetString("guidebook-reagent-effects-metabolism-stage-rate",
                    ("stage", _prototype.Index<MetabolismStagePrototype>(stage).LocalizedName), ("rate", effect.MetabolismRate)));
                var descriptionLabel = new RichTextLabel
                {
                    Margin = new Thickness(25, 0, 10, 0)
                };

                var descMsg = new FormattedMessage();
                var descriptionsCount = effect.EffectDescriptions.Length;
                var i = 0;
                foreach (var effectString in effect.EffectDescriptions)
                {
                    descMsg.AddMarkupOrThrow(effectString);
                    i++;
                    if (i < descriptionsCount || hasMetabolites)
                        descMsg.PushNewline();
                }
                if (hasMetabolites)
                {
                    var metabolites = new List<string>();
                    foreach (var (metabolite, ratio) in effect.Metabolites!)
                    {
                        metabolites.Add(Loc.GetString("guidebook-reagent-effects-metabolite-item", ("rate", (double)ratio), ("reagent", _prototype.Index(metabolite).LocalizedName)));
                    }
                    metabolites.Sort();

                    descMsg.AddMarkupOrThrow(Loc.GetString("guidebook-reagent-effects-metabolites", ("items", ContentLocalizationManager.FormatList(metabolites))));
                }
                descriptionLabel.SetMessage(descMsg);

                EffectsDescriptionContainer.AddChild(groupLabel);
                EffectsDescriptionContainer.AddChild(descriptionLabel);
            }
        }
        else
        {
            EffectsContainer.Visible = false;
        }
        #endregion

        #region PlantMetabolisms
        if (_chemistryGuideData.ReagentGuideRegistry.TryGetValue(reagent.ID, out var guideEntryRegistryPlant) &&
            guideEntryRegistryPlant.PlantMetabolisms != null &&
            guideEntryRegistryPlant.PlantMetabolisms.Count > 0)
        {
            PlantMetabolismsDescriptionContainer.Children.Clear();
            var metabolismLabel = new RichTextLabel();
            metabolismLabel.SetMarkup(Loc.GetString("guidebook-reagent-plant-metabolisms-rate"));
            var descriptionLabel = new RichTextLabel
            {
                Margin = new Thickness(25, 0, 10, 0)
            };
            var descMsg = new FormattedMessage();
            var descriptionsCount = guideEntryRegistryPlant.PlantMetabolisms.Count;
            var i = 0;
            foreach (var effectString in guideEntryRegistryPlant.PlantMetabolisms)
            {
                descMsg.AddMarkupOrThrow(effectString);
                i++;
                if (i < descriptionsCount)
                    descMsg.PushNewline();
            }
            descriptionLabel.SetMessage(descMsg);

            PlantMetabolismsDescriptionContainer.AddChild(metabolismLabel);
            PlantMetabolismsDescriptionContainer.AddChild(descriptionLabel);
        }
        else
        {
            PlantMetabolismsContainer.Visible = false;
        }
        #endregion
        SourcesContainer.Visible = false;

        FormattedMessage description = new();
        description.AddText(_prototype.Index<EntityPrototype>(spawnId).Description);


        ReagentDescription.SetMessage(description);
    }

    private void GenerateSources(ReagentPrototype reagent)
    {
        var sources = _chemistryGuideData.GetReagentSources(reagent.ID);
        if (sources.Count == 0)
        {
            SourcesContainer.Visible = false;
            return;
        }
        SourcesContainer.Visible = true;

        var orderedSources = sources
            .OrderBy(o => o.OutputCount)
            .ThenBy(o => o.IdentifierString);
        foreach (var source in orderedSources)
        {
            if (source is ReagentEntitySourceData entitySourceData)
            {
                SourcesDescriptionContainer.AddChild(new GuideReagentReaction(
                    entitySourceData.SourceEntProto,
                    entitySourceData.Solution,
                    entitySourceData.MixingType,
                    _prototype,
                    _systemManager));
            }
            else if (source is ReagentReactionSourceData reactionSourceData)
            {
                SourcesDescriptionContainer.AddChild(new GuideReagentReaction(
                    reactionSourceData.ReactionPrototype,
                    _prototype,
                    _systemManager));
            }
            else if (source is ReagentGasSourceData gasSourceData)
            {
                SourcesDescriptionContainer.AddChild(new GuideReagentReaction(
                    gasSourceData.GasPrototype,
                    gasSourceData.MixingType,
                    _prototype,
                    _systemManager));
            }
        }
    }
}