AdButtonToolkit.cs

using UnityEngine;
using UnityEngine.UIElements;
 
namespace SCG.SpaceAdMobile.UI
{
    [DisallowMultipleComponent]
    [AddComponentMenu("SCG/SpaceAdMobile/" + nameof(AdButtonToolkit))]
    public class AdButtonToolkit : MonoBehaviour
    {
        #region Serialized fields

        [Tooltip("Invoke AdUnit.Load() when clicked.")]
        [SerializeField] private bool load;

        [Tooltip("Invoke AdUnit.Show() when clicked.")]
        [SerializeField] private bool show;

        [Tooltip("Invoke Hide() on the controller if it implements IConcealableAd.")]
        [SerializeField] private bool hide;

        [Tooltip("Ad controller that hosts the current AdUnit (auto-filled in Editor).")]
        [SerializeField] private AdControllerBase unitController;

        [Tooltip("Specific AdUnit handler; if null, will be taken from controller.Ad (auto-filled in Editor).")]
        [SerializeField] private AdUnitHandlerBase adUnit;

        [Tooltip("UIDocument containing the target button; auto-filled from this GameObject if missing.")]
        [SerializeField] private UIDocument uiDocument;
 
#if UNITY_6000_5_OR_NEWER
        [Tooltip("PanelRenderer containing the target button; used when no UIDocument is available.")]
        [SerializeField] private PanelRenderer panelRenderer;
#endif

        [Tooltip("Name of the UI Toolkit Button in the visual tree (Q<Button> by name).")]
        [SerializeField] private string buttonName = "AdButton";
 
        #endregion
 
        #region Non-serialized state

        private Button _button;
 
#if UNITY_6000_5_OR_NEWER
        private PanelRenderer registeredPanelRenderer;
        private VisualElement currentRoot;
        private int lastPanelReloadVersion = -1;
#endif
 
        #endregion
 
        #region Unity lifecycle
 
#if UNITY_EDITOR
        protected virtual void OnValidate()
        {
#pragma warning disable UNT0039 // Use RequireComponent attribute when self-invoking GetComponent
            unitController = unitController ? unitController : GetComponent<AdControllerBase>();
            adUnit = adUnit ? adUnit : GetComponent<AdUnitHandlerBase>();
            uiDocument = uiDocument ? uiDocument : GetComponent<UIDocument>();
#if UNITY_6000_5_OR_NEWER
            panelRenderer = panelRenderer ? panelRenderer : GetComponent<PanelRenderer>();
#endif
#pragma warning restore UNT0039 // Use RequireComponent attribute when self-invoking GetComponent
        }
#endif

        protected virtual void OnEnable()
        {
#if UNITY_6000_5_OR_NEWER
            SynchronizeSource();
#else
#pragma warning disable UNT0039 // Use RequireComponent attribute when self-invoking GetComponent
            uiDocument = uiDocument ? uiDocument : GetComponent<UIDocument>();
#pragma warning restore UNT0039 // Use RequireComponent attribute when self-invoking GetComponent
#endif
 
            if (uiDocument != null)
            {
                RebindButton(uiDocument.rootVisualElement);
            }
#if UNITY_6000_5_OR_NEWER
            else if (registeredPanelRenderer == null)
                Debug.LogWarning($"{nameof(AdButtonToolkit)}: UIDocument or PanelRenderer not found.");
#else
            else
            {
                Debug.LogWarning($"{nameof(AdButtonToolkit)}: UIDocument not found.");
            }
#endif
 
        }

        protected virtual void OnDisable()
        {
            ClearButton();
 
#if UNITY_6000_5_OR_NEWER
            UnregisterPanelRenderer();
#endif
        }
 
        #endregion
 
        #region Handlers

        public void Rebind()
        {
            if (!isActiveAndEnabled)
                return;
 
#if UNITY_6000_5_OR_NEWER
            SynchronizeSource();
#else
#pragma warning disable UNT0039 // Use RequireComponent attribute when self-invoking GetComponent
            uiDocument = uiDocument ? uiDocument : GetComponent<UIDocument>();
#pragma warning restore UNT0039 // Use RequireComponent attribute when self-invoking GetComponent
#endif
 
            // ReSharper disable once InvertIf
            if (uiDocument != null)
            {
                RebindButton(uiDocument.rootVisualElement);
                // ReSharper disable once RedundantJumpStatement
                return;
            }
 
#if UNITY_6000_5_OR_NEWER
            RebindButton(currentRoot);
#endif
        }

        public void OnButtonClick()
        {
            if (adUnit == null && unitController != null)
                adUnit = unitController.Ad as AdUnitHandlerBase;
 
            if (adUnit == null)
                throw new System.NullReferenceException(nameof(adUnit));
 
            if (load) adUnit.Load();
            if (show) adUnit.Show();
 
            // ReSharper disable once SuspiciousTypeConversion.Global
            if (hide) ((IConcealableAd)unitController)?.Hide();
        }
 
        #endregion
 
        #region Helpers

        private void RebindButton(VisualElement root)
        {
            ClearButton();
 
            if (root == null)
            {
                Debug.LogWarning($"{nameof(AdButtonToolkit)}: rootVisualElement is null.");
                return;
            }
 
            _button = root.Q<Button>(buttonName);
            if (_button == null)
            {
                Debug.LogWarning($"{nameof(AdButtonToolkit)}: Button '{buttonName}' not found.");
                return;
            }
 
            _button.clicked += OnButtonClick;
        }
 
        private void ClearButton()
        {
            if (_button != null)
                _button.clicked -= OnButtonClick;
 
            _button = null;
        }
 
#if UNITY_6000_5_OR_NEWER
        private void SynchronizeSource()
        {
            var resolvedDocument = uiDocument;
            if (resolvedDocument == null)
                TryGetComponent(out resolvedDocument);
 
            var resolvedRenderer = resolvedDocument == null ? panelRenderer : null;
            if (resolvedDocument == null && resolvedRenderer == null)
                TryGetComponent(out resolvedRenderer);
 
            var sourceChanged = uiDocument != resolvedDocument || registeredPanelRenderer != resolvedRenderer;
            if (sourceChanged)
            {
                ClearButton();
                currentRoot = null;
                lastPanelReloadVersion = -1;
            }
 
            uiDocument = resolvedDocument;
 
            if (uiDocument != null)
            {
                UnregisterPanelRenderer();
                return;
            }
 
            if (registeredPanelRenderer == resolvedRenderer)
                return;
 
            UnregisterPanelRenderer();
            registeredPanelRenderer = resolvedRenderer;
            if (registeredPanelRenderer == null) return;
            registeredPanelRenderer.RegisterUIReloadCallback(OnUIReload);
            ((IPanelComponent)registeredPanelRenderer).PerformUpdate();
        }
 
        private void UnregisterPanelRenderer()
        {
            if (registeredPanelRenderer == null)
                return;
 
            registeredPanelRenderer.UnregisterUIReloadCallback(OnUIReload);
            registeredPanelRenderer = null;
            currentRoot = null;
        }
#endif
 
#if UNITY_6000_5_OR_NEWER
        private void OnUIReload(PanelRenderer renderer, VisualElement root, int version)
        {
            if (!isActiveAndEnabled)
                return;
 
            SynchronizeSource();
            if (uiDocument != null)
            {
                RebindButton(uiDocument.rootVisualElement);
                return;
            }
 
            if (renderer != registeredPanelRenderer || version == lastPanelReloadVersion)
                return;
 
            lastPanelReloadVersion = version;
            currentRoot = root;
            RebindButton(currentRoot);
        }
#endif
 
        #endregion
    }
}