ADOFAIRenderer/Data/Input/IntegerInputValidator.cs

using System.Collections.Generic;
using System.Linq;
using MelonLoader;
using UnityEngine.UIElements;

namespace ADOFAIRenderer.Data.Input;

internal class IntegerInputValidator
{
    private struct FieldEntry
    {
        public List<IntegerLimit> Limits;
    }

    private readonly VisualElement _rootVisualElement;
    private readonly Dictionary<string, FieldEntry> _fields = new();

    public IntegerInputValidator(VisualElement rootVisualElement)
    {
        _rootVisualElement = rootVisualElement;
    }

    /// <summary>
    /// Registers one or more <see cref="IntegerLimit"/>s for the field with the given name.
    /// Multiple limits can be added to the same field — they are all checked in order.
    /// </summary>
    /// <param name="fieldName">The name of the IntegerField in the UXML.</param>
    /// <param name="limits">The limits to apply.</param>
    public IntegerInputValidator AddLimits(string fieldName, params IntegerLimit[] limits)
    {
        if (!_fields.TryGetValue(fieldName, out var entry))
        {
            entry = new FieldEntry
            {
                Limits = new List<IntegerLimit>()
            };
        }

        entry.Limits.AddRange(limits);
        _fields[fieldName] = entry;
        return this;
    }

    /// <summary>
    /// Queries all registered fields from the root VisualElement and
    /// registers value-changed callbacks that rollback invalid values.
    /// </summary>
    public void SetupValidation()
    {
        foreach (var (fieldName, entry) in _fields)
        {
            var field = _rootVisualElement.Q<IntegerField>(fieldName);
            if (field == null)
            {
                Melon<RenderMod>.Logger.Warning(
                    $"IntegerField '{fieldName}'을(를) 찾을 수 없습니다. 유효성 검사를 건너뜁니다.");
                continue;
            }

            var limits = entry.Limits;

            field.RegisterValueChangedCallback(evt =>
            {
                if (string.IsNullOrEmpty(field.text))
                {
                    return;
                }

                if (limits.Any(limit => !limit.Validate(evt.newValue, out _)))
                {
                    field.SetValueWithoutNotify(evt.previousValue);
                }
            });
        }
    }
}