autoexec/plugin/autoexec.sp

#pragma semicolon 1
#include <sourcemod>
#include <sdktools>
#include <tf2>

public Plugin:myinfo = {
	name = "autoexec",
	author = "Icewind",
	description = "Automatically execute the right config for a map",
	version = "0.4",
	url = "https://spire.tf"
};

new StringMap:mapPrefixMap;
new StringMap:mapOverwriteMap;
new StringMap:mapTypeMap;
new StringMap:gameModeMap;
new StringMap:configOverwriteMap;
bool inAutoExec = false;
bool warned[4] = {false, false, false, false}; // size 4 so we can use team indexes

new Handle:CvarLeague = INVALID_HANDLE;
new Handle:CvarMode = INVALID_HANDLE;
new Handle:CvarAutoset = INVALID_HANDLE;
new Handle:CvarWhitelist = INVALID_HANDLE;

public OnPluginStart() {
	mapPrefixMap = new StringMap();
	mapOverwriteMap = new StringMap();
	mapTypeMap = new StringMap();
	gameModeMap = new StringMap();
	configOverwriteMap = new StringMap();

	CvarLeague = CreateConVar("sm_autoexec_league", "ugc", "league to execute the configs for (ugc or etf2l)", FCVAR_PROTECTED);
	CvarMode = CreateConVar("sm_autoexec_mode", "9v9", "game mode to execute the config for (9v9, 6v6 or 4v4)", FCVAR_PROTECTED);
	CvarAutoset = CreateConVar("sm_autoexec_autoset", "true", "try to set league and mode when a config is manually loaded (true or false)", FCVAR_PROTECTED);

	CvarWhitelist = FindConVar("mp_tournament_whitelist");

	RegServerCmd("sm_autoexec", AutoExec, "Execute the config for the current map and select league and game mode");
	RegServerCmd("sm_getexec", GetExec, "Get the name of the config for a specific map");
	RegServerCmd("exec", HandleExecAction);

	mapTypeMap.SetString("5cp", "standard");

	mapPrefixMap.SetString("pl_", "stopwatch");
	mapPrefixMap.SetString("cp_", "5cp");
	mapPrefixMap.SetString("koth_", "koth");
	mapPrefixMap.SetString("ctf_", "ctf");

	configOverwriteMap.SetString("ultiduo_", "etf2l_ultiduo");
	configOverwriteMap.SetString("bball_", "etf2l_bball");
	configOverwriteMap.SetString("ctf_bball", "etf2l_bball");

	mapOverwriteMap.SetString("cp_steel", "stopwatch");
	mapOverwriteMap.SetString("cp_gravelpit", "stopwatch");
	mapOverwriteMap.SetString("cp_hadal", "stopwatch");
	mapOverwriteMap.SetString("cp_alloy", "stopwatch");

	gameModeMap.SetString("9v9", "hl");
	gameModeMap.SetString("6v6", "6v");

	HookEvent("tournament_stateupdate", OnReadyUp, EventHookMode_Post);
}

public OnMapStart() {
    ExecMapCfg();

	CreateTimer(10.0, ExecMapCfg);
}


public ExecMapCfg() {
    decl String:config[128];
	decl String:map[128];
	GetCurrentMap(map, sizeof(map));
	GetConfig(map, config);

	ExecCFG(config);
}

public Action:HandleExecAction(args) {
	new String:cfg[128];
	GetCmdArg(1, cfg, sizeof(cfg));
	decl String:autoset[8];
	GetConVarString(CvarAutoset, autoset, sizeof(autoset));

	warned[2] = false;
	warned[3] = false;

	if (strncmp("true", autoset, sizeof(autoset)) != 0 || inAutoExec) {
		return Plugin_Continue;
	}

	if (StrContains(cfg, "etf2l_") == 0) {
		PrintToChatAll("Setting league to etf2l");
		SetConVarString(CvarLeague, "etf2l");
	}

	if (StrContains(cfg, "ugc_") == 0) {
		PrintToChatAll("Setting league to ugc");
		SetConVarString(CvarLeague, "ugc");
	}

	if ((StrContains(cfg, "9v9_") > 0) || (StrContains(cfg, "hl_") > 0)) {
		PrintToChatAll("Setting game mode to 9v9");
		SetConVarString(CvarMode, "9v9");
	}

	if (StrContains(cfg, "6v6_") > 0) {
		PrintToChatAll("Setting game mode to 6v6");
		SetConVarString(CvarMode, "6v6");
	}

	if (StrContains(cfg, "6v_") > 0) {
		PrintToChatAll("Setting game mode to 6v6");
		SetConVarString(CvarMode, "6v6");
	}

	if (StrContains(cfg, "4v4_") > 0) {
		PrintToChatAll("Setting game mode to 4v4");
		SetConVarString(CvarMode, "4v4");
	}

	return Plugin_Continue;
}

public Action:GetExec(args) {
	new String:map[128];
	GetCmdArg(1, map, sizeof(map));
	decl String:config[128];
	GetConfig(map, config);

	PrintToChatAll("Config: %s", config);
	return Plugin_Handled;
}

public Action:AutoExec(args) {
	decl String:config[128];
	decl String:map[128];
	GetCurrentMap(map, sizeof(map));
	GetConfig(map, config);

	ExecCFG(config);
	return Plugin_Handled;
}

public PrefixSearch(StringMap:map, String:query[128], String:result[128]) {
	new StringMapSnapshot:keys = map.Snapshot();
	decl String:key[16];
	new length = keys.Length;
	bool found = false;
	for (new i = 0; i < length; i++) {
		keys.GetKey(i, key, sizeof(key));
		if (strncmp(key, query, strlen(key)) == 0) {
			found = true;
			map.GetString(key, result, sizeof(result));
			break;
		}
	}

	CloseHandle(keys);
	return found;
}

public GetConfig(String:map[128], String:config[128]) {
	decl String:mapType[128];
	decl String:league[8];
	decl String:gamemode[8];

	if (PrefixSearch(configOverwriteMap, map, config)) {
		return;
	}

	GetLeague(league);
	GetLeagueMapType(map, mapType, league);
	GetGameMode(gamemode, league);

	Format(config, sizeof(config), "%s_%s_%s", league, gamemode, mapType);
}

public GetLeague(String:league[8]) {
	GetConVarString(CvarLeague, league, sizeof(league));
}

public GetGameMode(String:leagueGamemode[8], String:league[8]) {
	decl String:gamemode[8];
	GetConVarString(CvarMode, gamemode, sizeof(gamemode));

	if (strncmp("ugc", league, strlen(league)) == 0) {
		if (gameModeMap.GetString(gamemode, leagueGamemode, sizeof(leagueGamemode))) {
			return;
		}
	}
	leagueGamemode = gamemode;
}

public GetLeagueMapType(String:map[128], String:leagueMapType[128], String:league[8]) {
	decl String:mapType[128];
	GetMapType(map, mapType);

	if (strncmp("ugc", league, strlen(league)) == 0) {
		if (mapTypeMap.GetString(mapType, leagueMapType, sizeof(leagueMapType))) {
			return;
		}
	}
	leagueMapType = mapType;
}

public GetMapType(String:map[128], String:mapType[128]) {
	if (PrefixSearch(mapOverwriteMap, map, mapType)) {
		return;
	}

	if (!PrefixSearch(mapPrefixMap, map, mapType)) {
		mapType = "5cp"; //fallback
	}
	return;
}

public ExecCFG(String:cfg[128]) {
	decl String:command[256];
	Format(command, sizeof(command), "exec %s", cfg);
	// dont trigger autoset
	inAutoExec = true;
	ServerCommand(command, sizeof(command));
	CreateTimer(1.0, clearInAutoExec); // give the config time to load before clearing
}

public Action clearInAutoExec(Handle timer) {
	inAutoExec = false;
	return Plugin_Continue;
}

public void OnReadyUp(Event event, const String:name[], bool:dontBroadcast) {
    bool redReady = GameRules_GetProp("m_bTeamReady", 1, .element=2) != 0;
    bool blueReady = GameRules_GetProp("m_bTeamReady", 1, .element=3) != 0;

    CheckWhitelist();

    if (redReady) {
        CheckPlayerCount(2);
    }
    if (blueReady) {
        CheckPlayerCount(3);
    }
}

public void CheckPlayerCount(int team) {
    int playerCount = GetTeamClientCount(team);

    decl String:gamemode[8];
    GetConVarString(CvarMode, gamemode, sizeof(gamemode));

    if (!warned[team]) {
        if (strcmp(gamemode, "9v9") == 0 && playerCount != 9) {
            warned[team] = true;
            PrintToChatAll("Warning, config is set to 9v9 but you readied up with %d players", playerCount);
            GameRules_SetProp("m_bTeamReady", 0, _ , team, true);
        } else if (strcmp(gamemode, "6v6") == 0 && playerCount != 6) {
            warned[team] = true;
            GameRules_SetProp("m_bTeamReady", 0, _ , team, true);
            PrintToChatAll("Warning, config is set to 6v6 but you readied up with %d players", playerCount);
        } else {
            warned[team] = true;
        }
    }
}

public void CheckWhitelist() {
    decl String:whitelist[128];
    GetConVarString(CvarWhitelist, whitelist, sizeof(whitelist));

    decl String:expected[128];
    GetExpectedWhitelist(expected);

    if (strncmp(expected, whitelist, strlen(whitelist)) == 0) {
        PrintToChatAll("Warning, incorrect whitelist loaded for game mode: %s", whitelist);
    }
}

public GetExpectedWhitelist(String:whitelist[128]) {
	decl String:league[8];
	decl String:gamemode[8];

	whitelist = "";
	if (strncmp("ugc", league, strlen(league)) == 0) {
	    if (strncmp("9v9", gamemode, strlen(gamemode)) == 0) {
			whitelist = "cfg/item_whitelist_ugc_HL.txt";
		} else if (strncmp("6v6", gamemode, strlen(gamemode)) == 0) { {
		    whitelist = "cfg/item_whitelist_ugc_6v6.txt";
		}
	} else if (strncmp("etf2l", league, strlen(league)) == 0) {
    	if (strncmp("9v9", gamemode, strlen(gamemode)) == 0) {
            whitelist = "cfg/etf2l_whitelist_9v9.txt";
        } else if (strncmp("6v6", gamemode, strlen(gamemode)) == 0) { {
            whitelist = "cfg/etf2l_whitelist_6v6.txt";
        }
	}
}