DSP_Mods/UXAssist/UI/MyCheckButton.cs

using System;
using BepInEx.Configuration;
using UnityEngine;
using UnityEngine.UI;

namespace UXAssist.UI;

// MyCheckButton modified from LSTM: https://github.com/hetima/DSP_LSTM/blob/main/LSTM/MyCheckButton.cs
public class MyCheckButton : MonoBehaviour
{
    public RectTransform rectTrans;
    public UIButton uiButton;
    public Image icon;
    public Text labelText;
    public event Action OnChecked;
    private bool _checked;
    private float _iconWidth = 28f;

    private static GameObject _baseObject;

	private static Color openMouseOverColor;
	private static Color openPressColor;
	private static Color openNormalColor;
	private static Color closeMouseOverColor;
	private static Color closePressColor;
	private static Color closeNormalColor;

    public static void InitBaseObject()
    {
        if (_baseObject) return;
        var tankWindow = UIRoot.instance.uiGame.tankWindow;
        openMouseOverColor = tankWindow.openMouseOverColor;
        openPressColor = tankWindow.openPressColor;
        openNormalColor = tankWindow.openNormalColor;
        closeMouseOverColor = tankWindow.closeMouseOverColor;
        closePressColor = tankWindow.closePressColor;
        closeNormalColor = tankWindow.closeNormalColor;

        var go = Instantiate(UIRoot.instance.uiGame.beltWindow.reverseButton.gameObject);
        go.name = "my-checkbutton";
        go.SetActive(false);
        var comp = go.transform.Find("text");
        if (comp)
        {
            var txt = comp.GetComponent<Text>();
            if (txt)
            {
                txt.text = "";
                txt.rectTransform.pivot = new Vector2(0.5f, 0.5f);
            }
            var localizer = comp.GetComponent<Localizer>();
            if (localizer) DestroyImmediate(localizer);
        }
        _baseObject = go;
    }

    protected void OnDestroy()
    {
        if (_config != null) _config.SettingChanged -= _configChanged;
    }

    public static MyCheckButton CreateCheckButton(float x, float y, RectTransform parent, ConfigEntry<bool> config, string label = "", int fontSize = 15)
    {
        return CreateCheckButton(x, y, parent, config.Value, label, fontSize).WithConfigEntry(config);
    }

    public static MyCheckButton CreateCheckButton(float x, float y, RectTransform parent, bool check, string label = "", int fontSize = 15)
    {
        return CreateCheckButton(x, y, parent, fontSize).WithCheck(check).WithLabelText(label);
    }

    public static MyCheckButton CreateCheckButton(float x, float y, RectTransform parent, int fontSize = 15)
    {
        var go = Instantiate(_baseObject);
        go.name = "my-checkbutton";
        go.SetActive(true);
        var cb = go.AddComponent<MyCheckButton>();
        var rect = Util.NormalizeRectWithTopLeft(cb, x, y, parent);

        cb.rectTrans = rect;
        cb.uiButton = go.GetComponent<UIButton>();

        var child = go.transform.Find("text");
        if (child != null)
        {
            cb.labelText = child.GetComponent<Text>();
            if (cb.labelText)
            {
                cb.labelText.text = "";
                cb.labelText.fontSize = fontSize;
                cb.UpdateLabelTextWidth();
            }
        }

        cb.uiButton.onClick += cb.OnClick;
        cb._iconWidth = Mathf.Min(cb._iconWidth > 0f ? cb._iconWidth : 28f, rect.sizeDelta.y);
        cb.UpdateCheckColor();
        return cb;
    }

    private void UpdateLabelTextWidth()
    {
        if (labelText) labelText.rectTransform.sizeDelta = new Vector2(labelText.preferredWidth, labelText.rectTransform.sizeDelta.y);
    }

    public bool Checked
    {
        get => _checked;
        set
        {
            _checked = value;
            UpdateCheckColor();
        }
    }

    public void SetLabelText(string val)
    {
        if (labelText != null)
        {
            labelText.text = val.Translate();
            UpdateLabelTextWidth();
        }
    }

    private EventHandler _configChanged;
    private Action _checkedChanged;
    private ConfigEntry<bool> _config;
    public void SetConfigEntry(ConfigEntry<bool> config)
    {
        if (_checkedChanged != null) OnChecked -= _checkedChanged;
        if (_configChanged != null) config.SettingChanged -= _configChanged;

        _config = config;
        _checkedChanged = () => config.Value = !config.Value;
        OnChecked += _checkedChanged;
        _configChanged = (_, _) => Checked = config.Value;
        config.SettingChanged += _configChanged;
    }

    public MyCheckButton WithLabelText(string val)
    {
        SetLabelText(val);
        return this;
    }

    private void UpdateSize()
    {
        var width = rectTrans.sizeDelta.x;
        var height = rectTrans.sizeDelta.y;
        labelText.rectTransform.sizeDelta = new Vector2(icon != null ? width - _iconWidth : width, height);
        if (icon != null)
        {
            icon.transform.localPosition = new Vector3((-width - _iconWidth) * 0.5f, 0f, 0f);
        }
    }

    public MyCheckButton WithSize(float width, float height)
    {
        rectTrans.sizeDelta = new Vector2(width, height);
        if (height > _iconWidth) _iconWidth = height;
        UpdateSize();
        return this;
    }

    public MyCheckButton WithIconWidth(float width)
    {
        if (_iconWidth == width) return this;
        var height = rectTrans.sizeDelta.y;
        if (width > height)
        {
            width = height;
            if (_iconWidth == width) return this;
        }
        _iconWidth = width;
        if (icon != null) UpdateSize();
        return this;
    }

    public MyCheckButton WithIcon(Sprite sprite = null)
    {
        var trans = labelText.gameObject.transform;
        if (icon == null)
        {
            var iconGo = new GameObject("icon");
            var rect = iconGo.AddComponent<RectTransform>();
            (icon = iconGo.AddComponent<Image>()).sprite = sprite;
            iconGo.transform.SetParent(trans);
            rect.localPosition = new Vector3(0f, 0f, 0f);
            rect.sizeDelta = new Vector2(_iconWidth, _iconWidth);
            rect.localScale = new Vector3(1f, 1f, 1f);
            rect.anchorMax = new Vector2(0f, 1f);
            rect.anchorMin = new Vector2(0f, 1f);
            rect.pivot = new Vector2(0f, 0.5f);
            iconGo.SetActive(sprite != null);

            var width = rectTrans.sizeDelta.x;
            var originPosition = labelText.rectTransform.localPosition;
            labelText.rectTransform.localPosition = new Vector3(originPosition.x + _iconWidth * 0.5f, originPosition.y, originPosition.z);
            labelText.rectTransform.sizeDelta = new Vector2(width - _iconWidth, labelText.rectTransform.sizeDelta.y);
        }
        else
        {
            SetIcon(sprite);
        }
        return this;
    }

    public void SetIcon(Sprite sprite = null)
    {
        icon.sprite = sprite;
        icon.transform.localPosition = new Vector3((-rectTrans.sizeDelta.x - _iconWidth) * 0.5f, 0f, 0f);
        icon.gameObject.SetActive(sprite != null);
    }

    public MyCheckButton WithCheck(bool check)
    {
        Checked = check;
        return this;
    }

    public MyCheckButton WithConfigEntry(ConfigEntry<bool> config)
    {
        SetConfigEntry(config);
        return this;
    }

    public void OnClick(int obj)
    {
        _checked = !_checked;
        UpdateCheckColor();
        OnChecked?.Invoke();
    }

    public float Width => rectTrans.sizeDelta.x + labelText.rectTransform.sizeDelta.x;
    public float Height => Math.Max(rectTrans.sizeDelta.y, labelText.rectTransform.sizeDelta.y);

    private void UpdateCheckColor()
    {
        if (_checked)
        {
            uiButton.transitions[0].mouseoverColor = openMouseOverColor;
            uiButton.transitions[0].pressedColor = openPressColor;
            uiButton.transitions[0].normalColor = openNormalColor;
        }
        else
        {
            uiButton.transitions[0].mouseoverColor = closeMouseOverColor;
            uiButton.transitions[0].pressedColor = closePressColor;
            uiButton.transitions[0].normalColor = closeNormalColor;
        }
    }
}