NotificationSystem.cs

using System;
using System.Collections.Generic;
using System.IO;
using System.Reflection;
using MelonLoader;
using UnhollowerRuntimeLib;
using UnityEngine;
using Logger = VRC.Core.Logger;
using Object = UnityEngine.Object;

namespace ReMod.Core.Notification
{
    public class NotificationSystem
    {
        public static Sprite DefaultIcon;
        public static Color DefaultColour = new Color(0.1764f, 0.2549f, .3333f, 1f);
        public static MelonPreferences_Entry<float> NotificationAlpha;
        public static MelonPreferences_Entry<string> NotificationAlignment;
        public static bool UseVRChatNotificationSystem;
        
        //AssetBundle Parts
        private static AssetBundle _notifBundle;
        private static GameObject _notificationPrefab;
        
        private static GameObject _hudContent;
        private static GameObject _notificationGO;
        private static RectTransform _notificationRect;
        private static NotificationController _controllerInstance;
        
        public static void SetupNotifications()
        {
            _hudContent = GameObject.Find("/UserInterface/UnscaledUI/HudContent_Old");

            var notificationTransform = _hudContent.transform.Find("Notification (Clone)");
            if (notificationTransform != null)
            {
                //Notification system already initialized
                _notificationGO = notificationTransform.gameObject;
                _controllerInstance = _notificationGO.GetComponent<NotificationController>();
            }

            LoadAssets();
            
            NotificationController.RegisterSafe();

            MelonPreferences.CreateCategory("ReModCore", "ReMod.Core");
            NotificationAlpha = MelonPreferences.CreateEntry("ReModCore", "NotificationAlpha", .7f, "Notification Alpha", "Controls transparency of the notification system.");
            NotificationAlignment = MelonPreferences.CreateEntry("ReModCore", "NotificationAlignment", "centerMiddle", "Notification Alignment");

            NotificationAlignment.OnValueChanged += UpdateNotificationAlignment;
            
            //Create UIX settings enum
            RegSettingsEnum("ReModCore", "NotificationAlignment", new[] {("centerMiddle", "Middle Centered"), ("topCenter", "Top Centered"), ("topLeft", "Top Left"), ("topRight", "Top Right"), ("bottomCenter", "Bottom Centered"), ("bottomLeft", "Bottom Left"), ("bottomRight", "Bottom Right")});

            if (_notificationPrefab == null)
                throw new Exception("NotificationSystem failed to load, prefab missing!");

            //Instantiate prefab and let NotificationController setup!
            _notificationGO = Object.Instantiate(_notificationPrefab, _hudContent.transform);
            _controllerInstance = _notificationGO.AddComponent<NotificationController>();
            //Get the RectTransform for us to set the alignment
            _notificationRect = _notificationGO.GetComponent<RectTransform>();
            UpdateNotificationAlignment(null, null);

            _controllerInstance.defaultSprite = DefaultIcon;
        }

        /// <summary>
        /// Enqueue a new notification
        /// </summary>
        /// <param name="title">Title shown in the top of the notification</param>
        /// <param name="description">Main description, scales based on size</param>
        /// <param name="displayLength">How long in seconds you want it shown</param>
        /// <param name="icon">Optional icon sprite, defaults to Megaphone</param>
        public static void EnqueueNotification(string title, string description, float displayLength = 5f, Sprite icon = null)
        {
            EnqueueNotification(title, description, DefaultColour, displayLength, icon);
        }

        /// <summary>
        /// Enqueue a new notification
        /// </summary>
        /// <param name="title">Title shown in the top of the notification</param>
        /// <param name="description">Main description, scales based on size</param>
        /// <param name="backgroundColour">Background colour of the notification</param>
        /// <param name="displayLength">How long in seconds you want it shown</param>
        /// <param name="icon">Optional icon sprite, defaults to Megaphone</param>
        public static void EnqueueNotification(string title, string description, Color backgroundColour, float displayLength = 5f, Sprite icon = null)
        {
            var notif = new NotificationObject(title, description, icon, displayLength, backgroundColour);
            
            if(_controllerInstance == null)
                SetupNotifications();
            
            _controllerInstance.EnqueueNotification(notif);
        }

        public static void ClearNotification()
        {
            if(_controllerInstance == null)
                SetupNotifications();
            
            _controllerInstance.ClearNotifications();
        }

        private static void UpdateNotificationAlignment(string sender, string args)
        {
            if (_notificationRect == null) return;

            switch (NotificationAlignment.Value)
            {
                case "centerMiddle":
                    _notificationRect.anchorMin = new Vector2(0.5f, 0.5f);
                    _notificationRect.anchorMax = new Vector2(0.5f, 0.5f);
                    _notificationRect.pivot = new Vector2(0.5f, 0.5f);
                    break;
                case "topCenter":
                    _notificationRect.anchorMin = new Vector2(0.5f, 1f);
                    _notificationRect.anchorMax = new Vector2(0.5f, 1f);
                    _notificationRect.pivot = new Vector2(0.5f, 1f);
                    break;
                case "topLeft":
                    _notificationRect.anchorMin = new Vector2(0f, 1f);
                    _notificationRect.anchorMax = new Vector2(0f, 1f);
                    _notificationRect.pivot = new Vector2(0f, 1f);
                    break;
                case "topRight":
                    _notificationRect.anchorMin = new Vector2(1f, 1f);
                    _notificationRect.anchorMax = new Vector2(1f, 1f);
                    _notificationRect.pivot = new Vector2(1f, 1f);
                    break;
                case "bottomCenter":
                    _notificationRect.anchorMin = new Vector2(0.5f, 0f);
                    _notificationRect.anchorMax = new Vector2(0.5f, 0f);
                    _notificationRect.pivot = new Vector2(0.5f, 0f);
                    break;
                case "bottomLeft":
                    _notificationRect.anchorMin = new Vector2(0f, 0f);
                    _notificationRect.anchorMax = new Vector2(0f, 0f);
                    _notificationRect.pivot = new Vector2(0f, 0f);
                    break;
                case "bottomRight":
                    _notificationRect.anchorMin = new Vector2(1f, 0f);
                    _notificationRect.anchorMax = new Vector2(1f, 0f);
                    _notificationRect.pivot = new Vector2(1f, 0f);
                    break;
            }
        }

        private static void LoadAssets()
        {
            using (var assetStream = Assembly.GetExecutingAssembly().GetManifestResourceStream("ReMod.Core.Notification.notification"))
            {
                if (assetStream != null)
                {
                    using var tempStream = new MemoryStream((int) assetStream.Length);
                    assetStream.CopyTo(tempStream);

                    _notifBundle = AssetBundle.LoadFromMemory_Internal(tempStream.ToArray(), 0);
                    _notifBundle.hideFlags |= HideFlags.DontUnloadUnusedAsset;
                }
            }

            if (_notifBundle != null)
            {
                //Load Sprites
                DefaultIcon = _notifBundle.LoadAsset_Internal("Megaphone", Il2CppType.Of<Sprite>()).Cast<Sprite>();
                DefaultIcon.hideFlags |= HideFlags.DontUnloadUnusedAsset;

                //Prefab
                _notificationPrefab = _notifBundle.LoadAsset_Internal("Notification", Il2CppType.Of<GameObject>()).Cast<GameObject>();
                _notificationPrefab.hideFlags |= HideFlags.DontUnloadUnusedAsset;
            }
        }
        
        #region UIXAdapter
        
        private static bool? _uixAvailable;
        private static MethodInfo _regSettingEnum;
        private static bool _methodsGetRan;

        private static bool IsUIXAvailable()
        {
            _uixAvailable ??= MelonHandler.IsModAlreadyLoaded("UI Expansion Kit");
            return _uixAvailable.Value;
        }

        private static bool GetUIXMethods()
        {
            if (_methodsGetRan) return true;

            Type expandedMenu = Type.GetType("UIExpansionKit.API.ExpansionKitApi, UIExpansionKit");

            if (expandedMenu == null) return false;
            
            _regSettingEnum = expandedMenu.GetMethod("RegisterSettingAsStringEnum", BindingFlags.Public | BindingFlags.Static);
            
            _methodsGetRan = true;

            return true;
        }

        private static bool RegSettingsEnum(string settingsCat, string settingsName, IList<(string value, string desc)> values)
        {
            if (!IsUIXAvailable()) return false;
            if (!GetUIXMethods()) return false;

            _regSettingEnum.Invoke(null, new object[] { settingsCat, settingsName, values });

            return true;
        }
        
        #endregion
    }
}