feat(managed): OnStartupServer event

Author: samyycX

managed/src/SwiftlyS2.Core/Modules/Events/EventPublisher.cs

@@ -50,6 +50,7 @@ public static void Register()
             NativeEvents.RegisterOnClientProcessUsercmdsCallback((nint)(delegate* unmanaged< int, nint, int, byte, float, void >)&OnClientProcessUsercmds);
             NativeEvents.RegisterOnEntityTakeDamageCallback((nint)(delegate* unmanaged< nint, nint, nint, byte >)&OnEntityTakeDamage);
             NativeEvents.RegisterOnPrecacheResourceCallback((nint)(delegate* unmanaged< nint, void >)&OnPrecacheResource);
+            NativeEvents.RegisterOnStartupServerCallback((nint)(delegate* unmanaged< void >)&OnStartupServer);
             NativeConvars.AddConvarCreatedListener((nint)(delegate* unmanaged< nint, void >)&OnConVarCreated);
             NativeConvars.AddConCommandCreatedListener((nint)(delegate* unmanaged< nint, void >)&OnConCommandCreated);
             NativeConvars.AddGlobalChangeListener((nint)(delegate* unmanaged< nint, int, nint, nint, void >)&OnConVarValueChanged);
@@ -534,6 +535,24 @@ public static void OnPrecacheResource( nint pResourceManifest )
         }
     }
 
+    [UnmanagedCallersOnly]
+    public static void OnStartupServer( )
+    {
+        if (_subscribers.Count == 0) return;
+        try
+        {
+            foreach (var subscriber in _subscribers)
+            {
+                subscriber.InvokeOnStartupServer();
+            }
+        } 
+        catch (Exception e)
+        {
+            if (!GlobalExceptionHandler.Handle(e)) return;
+            AnsiConsole.WriteException(e);
+        }
+    }
+
     [Obsolete("InvokeOnEntityTouchHook is deprecated. Use InvokeOnEntityStartTouch, InvokeOnEntityTouch, or InvokeOnEntityEndTouch instead.")]
     public static void InvokeOnEntityTouchHook( OnEntityTouchHookEvent @event )
     {

managed/src/SwiftlyS2.Core/Modules/Events/EventSubscriber.cs

@@ -59,6 +59,7 @@ public EventSubscriber( CoreContext id, IContextedProfilerService profiler, ILog
     public event EventDelegates.OnMovementServicesRunCommandHook? OnMovementServicesRunCommandHook;
     public event EventDelegates.OnPlayerPawnPostThink? OnPlayerPawnPostThink;
     public event EventDelegates.OnEntityIdentityAcceptInputHook? OnEntityIdentityAcceptInputHook;
+    public event EventDelegates.OnStartupServer? OnStartupServer;
 
     public void Dispose()
     {
@@ -638,4 +639,22 @@ public void InvokeOnEntityIdentityAcceptInputHook( OnEntityIdentityAcceptInputHo
             _Profiler.StopRecording("Event::OnEntityIdentityAcceptInput");
         }
     }
+
+    public void InvokeOnStartupServer()
+    {
+        try
+        {
+            if (OnStartupServer == null) return;
+            _Profiler.StartRecording("Event::OnStartupServer");
+            OnStartupServer?.Invoke();
+        }
+        catch (Exception e)
+        {
+            if (GlobalExceptionHandler.Handle(e)) _Logger.LogError(e, "Error invoking OnStartupServer.");
+        }
+        finally
+        {
+            _Profiler.StopRecording("Event::OnStartupServer");
+        }
+    }
 }

managed/src/SwiftlyS2.Generated/Natives/Events.cs

@@ -163,4 +163,13 @@ public unsafe static void RegisterOnPrecacheResourceCallback(nint callback) {
   public unsafe static void RegisterOnPreworldUpdateCallback(nint callback) {
     _RegisterOnPreworldUpdateCallback(callback);
   }
+
+  private unsafe static delegate* unmanaged<nint, void> _RegisterOnStartupServerCallback;
+
+  /// <summary>
+  /// void
+  /// </summary>
+  public unsafe static void RegisterOnStartupServerCallback(nint callback) {
+    _RegisterOnStartupServerCallback(callback);
+  }
 }

managed/src/SwiftlyS2.Shared/Modules/Events/EventDelegates.cs

@@ -166,4 +166,9 @@ public class EventDelegates
     /// Called when an entity identity accept input hook is triggered.
     /// </summary>
     public delegate void OnEntityIdentityAcceptInputHook( IOnEntityIdentityAcceptInputHookEvent @event );
+
+    /// <summary>
+    /// Called when the server is started.
+    /// </summary>
+    public delegate void OnStartupServer();
 }

managed/src/SwiftlyS2.Shared/Modules/Events/IEventSubscriber.cs

@@ -166,4 +166,9 @@ public interface IEventSubscriber
     /// Called when an entity identity accept input hook is triggered.
     /// </summary>
     public event EventDelegates.OnEntityIdentityAcceptInputHook? OnEntityIdentityAcceptInputHook;
+
+    /// <summary>
+    /// Called when the server is started.
+    /// </summary>
+    public event EventDelegates.OnStartupServer? OnStartupServer;
 }

natives/engine/events.native

@@ -16,4 +16,5 @@ void RegisterOnMapLoadCallback = ptr callback // string mapname
 void RegisterOnMapUnloadCallback = ptr callback // string mapname
 void RegisterOnEntityTakeDamageCallback = ptr callback // CBaseEntity* entity, CTakeDamageInfo* info -> bool (true -> ignored, false -> supercede)
 void RegisterOnPrecacheResourceCallback = ptr callback // IEntityResourceManifest* pResourceManifest
-void RegisterOnPreworldUpdateCallback = ptr callback // bool simulating
+void RegisterOnPreworldUpdateCallback = ptr callback // bool simulating
+void RegisterOnStartupServerCallback = ptr callback // void

src/engine/entities/entitysystem.cpp

@@ -32,6 +32,7 @@
 #include <api/interfaces/manager.h>
 #include <s2binlib/s2binlib.h>
 
+
 typedef void (*CBaseEntity_DispatchSpawn)(void*, void*);
 typedef void (*UTIL_Remove)(void*);
 typedef void* (*UTIL_CreateEntityByName)(const char*, int);
@@ -44,6 +45,7 @@ void* g_pGameRules = nullptr;
 
 extern void* g_pOnEntityTakeDamageCallback;
 extern void* g_pTraceManager;
+extern void* g_pOnStartupServerCallback;
 
 IFunctionHook* g_pOnEntityTakeDamageHook = nullptr;
 IFunctionHook* g_pTraceShapeHook = nullptr;
@@ -134,6 +136,11 @@ void StartupServerHook(void* _this, const GameSessionConfiguration_t& config, IS
     g_pGameEntitySystem = entSystem;
     g_pGameEntitySystem->AddListenerEntity(&g_entityListener);
 
+    if (g_pOnStartupServerCallback)
+    {
+        reinterpret_cast<void(*)()>(g_pOnStartupServerCallback)();
+    }
+
     g_bDone = true;
 }
 

src/scripting/engine/events.cpp

@@ -36,6 +36,7 @@ void* g_pOnMapUnloadCallback = nullptr;
 void* g_pOnEntityTakeDamageCallback = nullptr;
 void* g_pOnPrecacheResourceCallback = nullptr;
 void* g_pOnPreworldUpdateCallback = nullptr;
+void* g_pOnStartupServerCallback = nullptr;
 
 void Bridge_Events_RegisterOnGameTickCallback(void* callback)
 {
@@ -122,6 +123,11 @@ void Bridge_Events_RegisterOnPreworldUpdateCallback(void* callback)
     g_pOnPreworldUpdateCallback = callback;
 }
 
+void Bridge_Events_RegisterOnStartupServerCallback(void* callback)
+{
+    g_pOnStartupServerCallback = callback;
+}
+
 DEFINE_NATIVE("Events.RegisterOnGameTickCallback", Bridge_Events_RegisterOnGameTickCallback);
 DEFINE_NATIVE("Events.RegisterOnClientConnectCallback", Bridge_Events_RegisterOnClientConnectCallback);
 DEFINE_NATIVE("Events.RegisterOnClientDisconnectCallback", Bridge_Events_RegisterOnClientDisconnectCallback);
@@ -138,4 +144,5 @@ DEFINE_NATIVE("Events.RegisterOnMapLoadCallback", Bridge_Events_RegisterOnMapLoa
 DEFINE_NATIVE("Events.RegisterOnMapUnloadCallback", Bridge_Events_RegisterOnMapUnloadCallback);
 DEFINE_NATIVE("Events.RegisterOnEntityTakeDamageCallback", Bridge_Events_RegisterOnEntityTakeDamageCallback);
 DEFINE_NATIVE("Events.RegisterOnPrecacheResourceCallback", Bridge_Events_RegisterOnPrecacheResourceCallback);
-DEFINE_NATIVE("Events.RegisterOnPreworldUpdateCallback", Bridge_Events_RegisterOnPreworldUpdateCallback);
+DEFINE_NATIVE("Events.RegisterOnPreworldUpdateCallback", Bridge_Events_RegisterOnPreworldUpdateCallback);
+DEFINE_NATIVE("Events.RegisterOnStartupServerCallback", Bridge_Events_RegisterOnStartupServerCallback);