Update UnityExceptionTraceHook.cs

Author: ifBars

S1API/Internal/Logging/UnityExceptionTraceHook.cs

@@ -1,6 +1,9 @@
 using System;
 using S1API.Logging;
 using UnityEngine;
+#if IL2CPPMELON
+using Il2CppInterop.Runtime;
+#endif
 
 namespace S1API.Internal.Diagnostics
 {
@@ -11,6 +14,10 @@ internal static class UnityExceptionTraceHook
     {
         private static readonly Log Logger = new Log("UnityExceptionTrace");
         private static readonly object Sync = new object();
+#if IL2CPPMELON
+        private static readonly System.Action<string, string, LogType> ManagedCallback = OnUnityLogMessageReceived;
+        private static readonly Application.LogCallback Callback = DelegateSupport.ConvertDelegate<Application.LogCallback>(ManagedCallback);
+#endif
 
         private static string _lastExceptionSignature;
         private static DateTime _lastExceptionAtUtc;
@@ -23,7 +30,11 @@ internal static void Install()
                 return;
             }
 
+#if IL2CPPMELON
+            Application.add_logMessageReceivedThreaded(Callback);
+#else
             Application.logMessageReceivedThreaded += OnUnityLogMessageReceived;
+#endif
             _installed = true;
         }
 
@@ -34,8 +45,21 @@ internal static void Remove()
                 return;
             }
 
-            Application.logMessageReceivedThreaded -= OnUnityLogMessageReceived;
+#if IL2CPPMELON
             _installed = false;
+            return;
+#else
+            try
+            {
+                Application.logMessageReceivedThreaded -= OnUnityLogMessageReceived;
+            }
+            catch (Exception ex)
+            {
+                Logger.Warning($"[Unity] Failed to remove threaded log callback during shutdown: {ex.Message}");
+            }
+
+            _installed = false;
+#endif
         }
 
         private static void OnUnityLogMessageReceived(string condition, string stackTrace, LogType type)