feat(API/Engine): WorkshopId

Author: skuzzis

managed/src/SwiftlyS2.Core/Modules/Engine/EngineService.cs

@@ -24,6 +24,8 @@ public EngineService( CommandTrackerManager commandTrackedManager )
 
     public string Map => GlobalVars.MapName.Value;
 
+    public string WorkshopId => NativeEngineHelpers.GetWorkshopId();
+
     public int MaxPlayers => GlobalVars.MaxClients;
 
     public float CurrentTime => GlobalVars.CurrentTime;

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

@@ -169,4 +169,18 @@ public unsafe static string GetGameDirectoryPath() {
       return retString;
     }
   }
+
+  private unsafe static delegate* unmanaged<byte*, int> _GetWorkshopId;
+
+  public unsafe static string GetWorkshopId() {
+    var ret = _GetWorkshopId(null);
+    var pool = ArrayPool<byte>.Shared;
+    var retBuffer = pool.Rent(ret + 1);
+    fixed (byte* retBufferPtr = retBuffer) {
+      ret = _GetWorkshopId(retBufferPtr);
+      var retString = Encoding.UTF8.GetString(retBufferPtr, ret);
+      pool.Return(retBuffer);
+      return retString;
+    }
+  }
 }

managed/src/SwiftlyS2.Shared/Modules/Engine/IEngineService.cs

@@ -16,6 +16,11 @@ public interface IEngineService
     [Obsolete("Use GlobalVars.MapName instead.")]
     public string Map { get; }
 
+    /// <summary>
+    /// Gets the Workshop ID of the current map.
+    /// </summary>
+    public string WorkshopId { get; }
+
     /// <summary>
     /// Gets a reference to the global variables structure.
     /// </summary>

natives/engine/helpers.native

@@ -11,4 +11,5 @@ string GetNativeVersion = void
 string GetMenuSettings = void
 ptr GetGlobalVars = void
 string GetCSGODirectoryPath = void
-string GetGameDirectoryPath = void
+string GetGameDirectoryPath = void
+string GetWorkshopId = void

src/scripting/engine/enginehelpers.cpp

@@ -27,6 +27,7 @@
 #include <public/filesystem.h>
 #include <public/steam/isteamgameserver.h>
 #include <public/tier0/platform.h>
+#include <public/iserver.h>
 #include <igamesystemfactory.h>
 
 #include <core/bridge/metamod.h>
@@ -53,7 +54,7 @@ bool Bridge_EngineHelpers_IsMapValid(const char* map_name)
         engine->IsMapValid(map_name) ||
         Files::ExistsPath(fmt::format("{}steamapps/workshop/content/730/{}/{}.vpk", s_sWorkingDir.Get(), map_name, map_name)) ||
         Files::ExistsPath(fmt::format("{}steamapps/workshop/content/730/{}/{}_dir.vpk", s_sWorkingDir.Get(), map_name, map_name))
-    );
+        );
 }
 
 void Bridge_EngineHelpers_ExecuteCommand(const char* command)
@@ -182,6 +183,21 @@ int Bridge_EngineHelpers_GetIP(char* out)
     return s.size();
 }
 
+int Bridge_EngineHelpers_GetWorkshopId(char* out)
+{
+    auto networkServerService = g_ifaceService.FetchInterface<INetworkServerService>(NETWORKSERVERSERVICE_INTERFACE_VERSION);
+    if (!networkServerService) return 0;
+
+    auto server = networkServerService->GetIGameServer();
+    if (!server) return 0;
+
+    std::string addonId = server->GetAddonName();
+
+    if (out != nullptr) strcpy(out, addonId.c_str());
+
+    return addonId.size();
+}
+
 DEFINE_NATIVE("EngineHelpers.GetIP", Bridge_EngineHelpers_GetIP);
 DEFINE_NATIVE("EngineHelpers.IsMapValid", Bridge_EngineHelpers_IsMapValid);
 DEFINE_NATIVE("EngineHelpers.ExecuteCommand", Bridge_EngineHelpers_ExecuteCommand);
@@ -193,4 +209,5 @@ DEFINE_NATIVE("EngineHelpers.GetNativeVersion", Bridge_EngineHelpers_GetNativeVe
 DEFINE_NATIVE("EngineHelpers.GetMenuSettings", Bridge_EngineHelpers_GetMenuSettings);
 DEFINE_NATIVE("EngineHelpers.GetGlobalVars", Bridge_EngineHelpers_GetGlobalVars);
 DEFINE_NATIVE("EngineHelpers.GetCSGODirectoryPath", Bridge_EngineHelpers_GetCSGODirectoryPath);
-DEFINE_NATIVE("EngineHelpers.GetGameDirectoryPath", Bridge_EngineHelpers_GetGameDirectoryPath);
+DEFINE_NATIVE("EngineHelpers.GetGameDirectoryPath", Bridge_EngineHelpers_GetGameDirectoryPath);
+DEFINE_NATIVE("EngineHelpers.GetWorkshopId", Bridge_EngineHelpers_GetWorkshopId);