1
0
mirror of https://github.com/Refactorio/RedMew.git synced 2024-12-14 10:13:13 +02:00
RedMew/map_gen/shared/spawn_control.lua

258 lines
6.6 KiB
Lua

local Event = require 'utils.event'
local Game = require 'utils.game'
local Utils = require 'utils.core'
local Module = {}
global.player_spawns = {} -- player_index to spawn_name
global.spawns = {} -- spawn_name to x, y, player_online_count
Module.add_spawn = function(name, x, y)
if type(name) ~= 'string' then
game.print('name must be a string')
return
end
if type(x) ~= 'number' then
game.print('x must be a number')
return
end
if type(y) ~= 'number' then
game.print('y must be a number')
return
end
global.spawns[name] = {x = x, y = y, count = 0}
end
local function get_min_count_spawn_name()
local min = 1000000
local min_spawn = nil
for name, t in pairs(global.spawns) do
local count = t.count
if min > count then
min = count
min_spawn = name
end
end
return min_spawn
end
local function player_joined_game(event)
local index = event.player_index
local spawn_name = global.player_spawns[index]
-- player already has a spawn.
if spawn_name then
local spawn = global.spawns[spawn_name]
local count = spawn.count
spawn.count = count + 1
return
end
spawn_name = get_min_count_spawn_name()
if not spawn_name then
return
end
local spawn = global.spawns[spawn_name]
global.player_spawns[index] = spawn_name
Game.get_player_by_index(index).teleport(spawn)
local count = spawn.count
spawn.count = count + 1
end
local function player_left_game(event)
local index = event.player_index
local spawn_name = global.player_spawns[index]
local spawn = global.spawns[spawn_name]
if not spawn then
return
end
local count = spawn.count
spawn.count = count - 1
end
local function player_respawned(event)
local index = event.player_index
local spawn_name = global.player_spawns[index]
local spawn = global.spawns[spawn_name]
if not spawn then
return
end
Game.get_player_by_index(index).teleport(spawn)
end
local function tp_spawn(player_name, spawn_name)
local player = Game.get_player_by_index(player_name)
if not player then
player_name = player_name or ''
game.player.print('player ' .. player_name .. ' does not exist.')
return
end
local spawn = global.spawns[spawn_name]
if not spawn then
spawn_name = spawn_name or ''
game.player.print('spawn ' .. spawn_name .. ' does not exist.')
return
end
player.teleport(spawn)
end
local function change_spawn(player_name, spawn_name)
local new_spawn = global.spawns[spawn_name]
if not new_spawn then
spawn_name = spawn_name or ''
game.player.print('spawn ' .. spawn_name .. ' does not exist.')
return
end
local player = Game.get_player_by_index(player_name)
if not player then
player_name = player_name or ''
game.player.print('player ' .. player_name .. ' does not exist.')
return
end
local index = player.index
local old_spawn_name = global.player_spawns[index]
local old_spawn = global.spawns[old_spawn_name]
if old_spawn then
local count = old_spawn.count
old_spawn.count = count - 1
end
local count = new_spawn.count
new_spawn.count = count + 1
global.player_spawns[index] = spawn_name
game.player.print(player_name .. ' spawn moved to ' .. spawn_name)
end
local function print_spawns()
for name, spawn in pairs(global.spawns) do
game.player.print(string.format('%s: (%d, %d), player count = %d', name, spawn.x, spawn.y, spawn.count))
end
end
local function print_players_for_spawn(target_spawn_name)
if not global.spawns[target_spawn_name] then
target_spawn_name = target_spawn_name or ''
game.player.print('spawn ' .. target_spawn_name .. ' does not exist.')
return
end
local str = ''
for index, spawn_name in pairs(global.player_spawns) do
if target_spawn_name == spawn_name then
local player = Game.get_player_by_index(index)
if player.connected then
str = str .. player.name .. ', '
end
end
end
if str == '' then
str = 'no players'
end
game.player.print(str)
end
local function tp_spawn_command(cmd)
if not game.player.admin then
Utils.cant_run(cmd.name)
return
end
local params = cmd.parameter
if type(params) ~= 'string' then
game.player.print('Command failed. Usage: /tpspawn <player>, <spawn_name>')
return
end
local ps = {}
for p in params:gmatch('%S+') do
table.insert(ps, p)
end
if #ps == 1 then
tp_spawn(game.player.name, ps[1])
else
tp_spawn(ps[1], ps[2])
end
end
local function change_spawn_command(cmd)
if not game.player.admin then
Utils.cant_run(cmd.name)
return
end
local params = cmd.parameter
if type(params) ~= 'string' then
game.player.print('Command failed. Usage: /changespawn <player>, <spawn_name>')
return
end
local ps = {}
for p in params:gmatch('%S+') do
table.insert(ps, p)
end
change_spawn(ps[1], ps[2])
end
local function print_spawns_command(cmd)
if not game.player.admin then
Utils.cant_run(cmd.name)
return
end
print_spawns()
end
local function print_players_for_spawn_command(cmd)
if not game.player.admin then
Utils.cant_run(cmd.name)
return
end
local params = cmd.parameter
if type(params) ~= 'string' then
game.player.print('Command failed. Usage: /playersforspawn <spawn_name>')
return
end
local ps = {}
for p in params:gmatch('%S+') do
table.insert(ps, p)
end
print_players_for_spawn(ps[1])
end
Event.add(defines.events.on_player_joined_game, player_joined_game)
Event.add(defines.events.on_player_left_game, player_left_game)
Event.add(defines.events.on_player_respawned, player_respawned)
commands.add_command('tpspawn', '<player> <spawn_name> teleports a player to the spawn point (Admins only)', tp_spawn_command) -- luacheck: ignore
commands.add_command('changespawn', '<player> <spawn_name> changes the spawn point for a player (Admins only)', change_spawn_command) -- luacheck: ignore
commands.add_command('printspawns', 'prints info on all spawn points (Admins only)', print_spawns_command) -- luacheck: ignore
commands.add_command('printplayersforspawn', '<spawn_name> prints all the connected players for a spawn (Admins only)', print_players_for_spawn_command) -- luacheck: ignore
return Module