--luacheck: ignore
local Public = {}
local math_random = math.random

local starting_items = { ['iron-plate'] = 32, ['iron-gear-wheel'] = 16, ['stone'] = 20, ['pistol'] = 1, ['firearm-magazine'] = 16 }

function Public.reset_forces()
    for _, force_name in pairs({ 'west', 'east' }) do
        storage.map_forces[force_name].unit_health_boost = 1
        storage.map_forces[force_name].unit_count = 0
        storage.map_forces[force_name].units = {}
        storage.map_forces[force_name].player_count = 0
        local force = game.forces[force_name]
        force.reset()
        force.share_chart = true
        force.technologies['artillery'].enabled = false
        force.technologies['artillery-shell-range-1'].enabled = false
        force.technologies['artillery-shell-speed-1'].enabled = false
        force.technologies['land-mine'].enabled = false
    end
    game.forces.west.set_friend('spectator', true)
    game.forces.east.set_friend('spectator', true)
    game.forces.west.set_spawn_position({ -210, 0 }, game.surfaces.nauvis)
    game.forces.east.set_spawn_position({ 210, 0 }, game.surfaces.nauvis)
end

function Public.create_forces()
    game.create_force('west')
    game.create_force('east')
    game.create_force('spectator')
    game.forces.spectator.set_friend('west', true)
    game.forces.spectator.set_friend('east', true)
    for _, force_name in pairs({ 'west', 'east' }) do
        storage.map_forces[force_name].max_unit_count = 1280
    end
    local surface = game.surfaces.nauvis
    game.forces.spectator.set_spawn_position({ 0, -128 }, surface)
end

function Public.assign_random_force_to_active_players()
    local player_indexes = {}
    for _, player in pairs(game.connected_players) do
        if player.force.name ~= 'spectator' then
            player_indexes[#player_indexes + 1] = player.index
        end
    end
    if #player_indexes > 1 then
        table.shuffle_table(player_indexes)
    end
    local a = math_random(0, 1)
    for key, player_index in pairs(player_indexes) do
        if key % 2 == a then
            game.players[player_index].force = game.forces.west
        else
            game.players[player_index].force = game.forces.east
        end
    end
end

function Public.assign_force_to_player(player)
    player.spectator = false
    if math_random(1, 2) == 1 then
        if #game.forces.east.connected_players > #game.forces.west.connected_players then
            player.force = game.forces.west
        else
            player.force = game.forces.east
        end
    else
        if #game.forces.east.connected_players < #game.forces.west.connected_players then
            player.force = game.forces.east
        else
            player.force = game.forces.west
        end
    end
end

function Public.teleport_player_to_spawn(player)
    local surface = game.surfaces.nauvis
    local position
    if player.force.name == 'spectator' then
        position = player.force.get_spawn_position(surface)
        position = { x = (position.x - 160) + math_random(0, 320), y = (position.y - 16) + math_random(0, 32) }
    else
        position = surface.find_non_colliding_position('character', player.force.get_spawn_position(surface), 48, 1)
        if not position then
            position = player.force.get_spawn_position(surface)
        end
    end
    player.teleport(position, surface)
end

function Public.add_player_to_team(player)
    if player.character then
        if player.character.valid then
            player.character.destroy()
        end
    end
    player.character = nil
    player.set_controller({ type = defines.controllers.god })
    player.create_character()
    for item, amount in pairs(starting_items) do
        player.insert({ name = item, count = amount })
    end
    storage.map_forces[player.force.name].player_count = storage.map_forces[player.force.name].player_count + 1
end

function Public.set_player_to_spectator(player)
    if player.character and player.character.valid then
        player.character.die()
    end
    player.force = game.forces.spectator
    player.character = nil
    player.spectator = true
    player.set_controller({ type = defines.controllers.spectator })
end

function Public.spawn_players(hatchery)
    if game.tick % 90 ~= 0 then
        return
    end
    -- game.print('spawning characters', {150, 150, 150})
    for _, player in pairs(game.connected_players) do
        if player.force.name ~= 'spectator' then
            Public.assign_force_to_player(player)
            Public.add_player_to_team(player)
            Public.teleport_player_to_spawn(player)
        end
    end
    hatchery.gamestate = 'rejoin_question'
    print(hatchery.gamestate)
end

function Public.init(hatchery)
    game.reset_time_played()
    Public.reset_forces()

    for _, player in pairs(game.forces.spectator.players) do
        Public.teleport_player_to_spawn(player)
    end

    for _, player in pairs(game.forces.west.players) do
        if player.connected then
            if player.character and player.character.valid then
                player.character.destroy()
            end
            player.character = nil
            player.spectator = true
            player.set_controller({ type = defines.controllers.spectator })
            player.force = game.forces.player
            Public.teleport_player_to_spawn(player)
        else
            player.force = game.forces.player
        end
    end

    for _, player in pairs(game.forces.east.players) do
        if player.connected then
            if player.character and player.character.valid then
                player.character.destroy()
            end
            player.character = nil
            player.spectator = true
            player.set_controller({ type = defines.controllers.spectator })
            player.force = game.forces.player
            Public.teleport_player_to_spawn(player)
        else
            player.force = game.forces.player
        end
    end
    hatchery.gamestate = 'reset_nauvis'
    print(hatchery.gamestate)
end

return Public
