mirror of
https://github.com/etlegacy/wolfadmin.git
synced 2024-11-23 12:42:43 +00:00
107 lines
4.3 KiB
Lua
107 lines
4.3 KiB
Lua
|
|
-- WolfAdmin module for Wolfenstein: Enemy Territory servers.
|
|
-- Copyright (C) 2015-2016 Timo 'Timothy' Smit
|
|
|
|
-- This program is free software: you can redistribute it and/or modify
|
|
-- it under the terms of the GNU General Public License as published by
|
|
-- the Free Software Foundation, either version 3 of the License, or
|
|
-- at your option any later version.
|
|
|
|
-- This program is distributed in the hope that it will be useful,
|
|
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
-- GNU General Public License for more details.
|
|
|
|
-- You should have received a copy of the GNU General Public License
|
|
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
local constants = require "luascripts.wolfadmin.util.constants"
|
|
local util = require "luascripts.wolfadmin.util.util"
|
|
local events = require "luascripts.wolfadmin.util.events"
|
|
local timers = require "luascripts.wolfadmin.util.timers"
|
|
local settings = require "luascripts.wolfadmin.util.settings"
|
|
|
|
local balancer = {}
|
|
|
|
local evenerCount = 0
|
|
|
|
function balancer.balance(byAdmin, forceBalance)
|
|
local teams = {
|
|
[1] = {},
|
|
[2] = {},
|
|
[3] = {}
|
|
}
|
|
|
|
for playerId = 0, et.trap_Cvar_Get("sv_maxclients") - 1 do
|
|
if wolfa_isPlayer(playerId) then
|
|
local team = tonumber(et.gentity_get(playerId, "sess.sessionTeam"))
|
|
|
|
table.insert(teams[team], playerId)
|
|
end
|
|
end
|
|
|
|
local teamGreater = constants.TEAM_SPECTATORS
|
|
local teamSmaller = constants.TEAM_SPECTATORS
|
|
|
|
local teamsDifference = math.abs(#teams[constants.TEAM_AXIS] - #teams[constants.TEAM_ALLIES])
|
|
|
|
if #teams[constants.TEAM_AXIS] > #teams[constants.TEAM_ALLIES] then
|
|
teamGreater = constants.TEAM_AXIS
|
|
teamSmaller = constants.TEAM_ALLIES
|
|
elseif #teams[constants.TEAM_ALLIES] > #teams[constants.TEAM_AXIS] then
|
|
teamGreater = constants.TEAM_ALLIES
|
|
teamSmaller = constants.TEAM_AXIS
|
|
end
|
|
|
|
local teamGreaterName = util.getTeamName(teamGreater)
|
|
local teamSmallerName = util.getTeamName(teamSmaller)
|
|
|
|
local teamGreaterColor = util.getTeamColor(teamGreater)
|
|
local teamSmallerColor = util.getTeamColor(teamSmaller)
|
|
|
|
if settings.get("g_evenerMaxDifference") > 0 and teamsDifference >= settings.get("g_evenerMaxDifference") then
|
|
evenerCount = evenerCount + 1
|
|
|
|
if forceBalance or evenerCount >= 2 then
|
|
et.trap_SendConsoleCommand(et.EXEC_APPEND, "!shuffle;")
|
|
et.trap_SendConsoleCommand(et.EXEC_APPEND, "cpm \"^devener: ^7THE TEAMS HAVE BEEN ^qSHUFFLED^7!\";")
|
|
|
|
evenerCount = 0
|
|
else
|
|
et.trap_SendConsoleCommand(et.EXEC_APPEND, "cpm \"^devener: ^1EVEN THE TEAMS ^7OR ^1SHUFFLE\";")
|
|
end
|
|
elseif teamsDifference >= settings.get("g_evenerMinDifference") then
|
|
evenerCount = evenerCount + 1
|
|
|
|
if forceBalance or evenerCount >= 3 then
|
|
for i = 1, (teamsDifference / 2) do
|
|
local rand = math.random(#teams[teamGreater])
|
|
|
|
et.trap_SendConsoleCommand(et.EXEC_APPEND, "!put "..teams[teamGreater][rand].." "..(teamGreater == constants.TEAM_AXIS and constants.TEAM_ALLIES_SC or constants.TEAM_AXIS_SC)..";")
|
|
et.trap_SendConsoleCommand(et.EXEC_APPEND, "chat \"^devener: ^9thank you, ^7"..et.gentity_get(teams[teamGreater][rand], "pers.netname")..", ^9for helping to even the teams.\";")
|
|
|
|
teams[teamSmaller][rand] = teams[teamGreater][rand]
|
|
teams[teamGreater][rand] = nil
|
|
end
|
|
|
|
evenerCount = 0
|
|
else
|
|
et.trap_SendConsoleCommand(et.EXEC_APPEND, "chat \"^devener: ^9teams seem unfair, would someone from "..teamGreaterColor..teamGreaterName.." ^9please switch to "..teamSmallerColor..teamSmallerName.."^9?\";")
|
|
end
|
|
else
|
|
evenerCount = 0
|
|
|
|
if byAdmin then
|
|
et.trap_SendConsoleCommand(et.EXEC_APPEND, "chat \"^devener: ^9teams are even.\";")
|
|
end
|
|
end
|
|
end
|
|
|
|
function balancer.oninit()
|
|
if settings.get("g_evenerInterval") > 0 then
|
|
timers.add(balancer.balance, settings.get("g_evenerInterval") * 1000, 0, false, false)
|
|
end
|
|
end
|
|
events.handle("onGameInit", balancer.oninit)
|
|
|
|
return balancer
|