chore: get workshop map id (#73)

Author: lukepolo

src/FiveStack.Entities/Map.cs

@@ -3,6 +3,5 @@ namespace FiveStack.Entities;
 public class Map
 {
     public string name { get; set; } = "";
-    public string label { get; set; } = "";
     public string workshop_map_id { get; set; } = "";
 }

src/FiveStack.Services/MatchManager.cs

@@ -1,3 +1,4 @@
+using System.Runtime.InteropServices;
 using CounterStrikeSharp.API;
 using CounterStrikeSharp.API.Core;
 using CounterStrikeSharp.API.Modules.Cvars;
@@ -10,6 +11,9 @@
 
 namespace FiveStack;
 
+[UnmanagedFunctionPointer(CallingConvention.ThisCall)]
+delegate IntPtr GetAddonNameDelegate(IntPtr self);
+
 public class MatchManager
 {
     private MatchData? _matchData;
@@ -28,6 +32,7 @@ public class MatchManager
     // public CoachSystem _coachSystem;
     public KnifeSystem knifeSystem;
     public CaptainSystem captainSystem;
+    public INetworkServerService _networkServerService;
 
     public MatchManager(
         ILogger<MatchManager> logger,
@@ -38,7 +43,8 @@ public MatchManager(
         KnifeSystem KnifeSystem,
         ReadySystem ReadySystem,
         CaptainSystem CaptainSystem,
-        EnvironmentService environmentService
+        EnvironmentService environmentService,
+        INetworkServerService networkServerService
     )
     {
         _logger = logger;
@@ -50,6 +56,7 @@ EnvironmentService environmentService
         captainSystem = CaptainSystem;
         _backUpManagement = backUpManagement;
         _environmentService = environmentService;
+        _networkServerService = networkServerService;
     }
 
     public void Init(MatchData match)
@@ -241,10 +248,22 @@ public void SetupMatch(MatchData match)
             return;
         }
 
-        _logger.LogInformation($"Game State {_currentMap.status} on {_currentMap.map.label} {_currentMap.map.name}  / {Server.MapName}");
+        _logger.LogInformation(
+            $"Game State {_currentMap.status} on ({_currentMap.map.name}) / {Server.MapName}"
+        );
 
-        // attempt to reduce the number of bad map naming issues
-        if (!new[] { _currentMap.map.name, _currentMap.map.label.ToLower() }.Contains(Server.MapName.ToLower()))
+        if (_currentMap.map.workshop_map_id is not null)
+        {
+            string currentWorkshopID = GetWorkshopID();
+            _logger.LogInformation($"Checking Workshop Map {_currentMap.map.workshop_map_id} / {currentWorkshopID}");
+
+            if (_currentMap.map.workshop_map_id != currentWorkshopID)
+            {
+                ChangeMap(_currentMap.map);
+                return;
+            }
+        }
+        else if (!Server.MapName.ToLower().Contains(_currentMap.map.name.ToLower()))
         {
             ChangeMap(_currentMap.map);
             return;
@@ -270,6 +289,16 @@ public void SetupMatch(MatchData match)
         }
     }
 
+    private string GetWorkshopID()
+    {
+        IntPtr networkGameServer = _networkServerService.GetIGameServerHandle();
+        IntPtr vtablePtr = Marshal.ReadIntPtr(networkGameServer);
+        IntPtr functionPtr = Marshal.ReadIntPtr(vtablePtr + (25 * IntPtr.Size));
+        var getAddonName = Marshal.GetDelegateForFunctionPointer<GetAddonNameDelegate>(functionPtr);
+        IntPtr result = getAddonName(networkGameServer);
+        return Marshal.PtrToStringAnsi(result)!.Split(',')[0];
+    }
+
     public void SetupTeamNames()
     {
         MatchMap? _currentMap = GetCurrentMap();

src/FiveStack.Services/NetworkServices.cs

@@ -0,0 +1,21 @@
+using System.Runtime.InteropServices;
+using CounterStrikeSharp.API;
+using CounterStrikeSharp.API.Core;
+using CounterStrikeSharp.API.Modules.Memory;
+
+public class INetworkServerService : NativeObject
+{
+    private readonly VirtualFunctionWithReturn<nint, nint> GetIGameServerFunc;
+
+    public INetworkServerService()
+        : base(NativeAPI.GetValveInterface(0, "NetworkServerService_001"))
+    {
+        int offset = RuntimeInformation.IsOSPlatform(OSPlatform.Windows) ? 23 : 24;
+        this.GetIGameServerFunc = new VirtualFunctionWithReturn<nint, nint>(this.Handle, offset);
+    }
+
+    public nint GetIGameServerHandle()
+    {
+        return this.GetIGameServerFunc.Invoke(this.Handle);
+    }
+}

src/FiveStackServiceCollection.cs

@@ -15,6 +15,7 @@ public void ConfigureServices(IServiceCollection serviceCollection)
         serviceCollection.AddSingleton<GameBackUpRounds>();
         serviceCollection.AddSingleton<SurrenderSystem>();
         serviceCollection.AddSingleton<EnvironmentService>();
+        serviceCollection.AddSingleton<INetworkServerService>();
 
         serviceCollection.AddTransient<MatchManager>();
         serviceCollection.AddTransient<VoteSystem>();