Allow loading mod maps only

twhl-community · Jan 14, 2024 · 8005659 · 8005659
1 parent f37b24e
commit 8005659
Showing 1 changed file with 24 additions and 3 deletions.
diff --git a/dlls/client.cpp b/dlls/client.cpp
@@ -798,7 +798,10 @@ static void LoadNextMap()
 	SERVER_COMMAND(UTIL_VarArgs("map \"%s\"\n", mapName.c_str()));
 }
 
-static void LoadAllMaps()
+// If modOnly is set to false all maps from the base game plus the mod will be
+// loaded. If set to true, only the maps present in the mod maps directory will
+// be loaded.
+static void LoadMaps(bool modOnly)
 {
 	if (!g_MapsToLoad.empty())
 	{
@@ -809,7 +812,13 @@ static void LoadAllMaps()
 
 	FileFindHandle_t handle = FILESYSTEM_INVALID_FIND_HANDLE;
 
-	const char* fileName = g_pFileSystem->FindFirst("maps/*.bsp", &handle);
+	const char* fileName = nullptr;
+	if (modOnly) {
+		auto dir = std::string(FileSystem_GetModDirectoryName()) + "/maps/*.bsp";
+		fileName = g_pFileSystem->FindFirst(dir.c_str(), &handle);
+	} else {
+		fileName = g_pFileSystem->FindFirst("maps/*.bsp", &handle);
+	}
 
 	if (fileName != nullptr)
 	{
@@ -866,11 +875,23 @@ static void LoadAllMaps()
 	}
 }
 
+static void LoadModMaps()
+{
+	LoadMaps(true);
+}
+
+static void LoadAllMaps()
+{
+	LoadMaps(false);
+}
+
 void InitMapLoadingUtils()
 {
+	g_engfuncs.pfnAddServerCommand("sv_load_mod_maps", &LoadModMaps);
 	g_engfuncs.pfnAddServerCommand("sv_load_all_maps", &LoadAllMaps);
+
 	// Escape hatch in case the command is executed in error.
-	g_engfuncs.pfnAddServerCommand("sv_stop_loading_all_maps", []()
+	g_engfuncs.pfnAddServerCommand("sv_stop_loading_maps", []()
 		{ g_MapsToLoad.clear(); });
 }