local Public = {}
local Team = require 'maps.junkyard_pvp.team'

function Public.spectate_button(player)
    if player.gui.top.spectate_button then
        return
    end
    local button = player.gui.top.add({type = 'button', name = 'spectate_button', caption = 'Spectate'})
    button.style.font = 'default-bold'
    button.style.font_color = {r = 0.0, g = 0.0, b = 0.0}
    button.style.minimal_height = 38
    button.style.minimal_width = 38
    button.style.top_padding = 2
    button.style.left_padding = 4
    button.style.right_padding = 4
    button.style.bottom_padding = 2
end

local function create_spectate_confirmation(player)
    if player.gui.center.spectate_confirmation_frame then
        return
    end
    local frame = player.gui.center.add({type = 'frame', name = 'spectate_confirmation_frame', caption = 'Are you sure you want to spectate this round?'})
    frame.style.font = 'default'
    frame.style.font_color = {r = 0.3, g = 0.65, b = 0.3}
    frame.add({type = 'button', name = 'confirm_spectate', caption = 'Spectate'})
    frame.add({type = 'button', name = 'cancel_spectate', caption = 'Cancel'})
end

function Public.rejoin_question(player)
    if player.gui.center.rejoin_question_frame then
        return
    end
    local frame = player.gui.center.add({type = 'frame', name = 'rejoin_question_frame', caption = 'Rejoin the game?'})
    frame.style.font = 'default'
    frame.style.font_color = {r = 0.3, g = 0.65, b = 0.3}
    frame.add({type = 'button', name = 'confirm_rejoin', caption = 'Rejoin'})
    frame.add({type = 'button', name = 'cancel_rejoin', caption = 'Cancel'})
end

local function on_gui_click(event)
    if not event then
        return
    end
    if not event.element then
        return
    end
    if not event.element.valid then
        return
    end
    local player = game.players[event.element.player_index]

    if event.element.name == 'confirm_rejoin' then
        player.gui.center['rejoin_question_frame'].destroy()
        Team.assign_force_to_player(player)
        Team.teleport_player_to_active_surface(player)
        Team.put_player_into_random_team(player)
        game.print(player.name .. ' has rejoined the game!')
        return
    end
    if event.element.name == 'cancel_rejoin' then
        player.gui.center['rejoin_question_frame'].destroy()
        return
    end

    if player.force.name == 'spectator' then
        return
    end
    if event.element.name == 'cancel_spectate' then
        player.gui.center['spectate_confirmation_frame'].destroy()
        return
    end
    if event.element.name == 'confirm_spectate' then
        player.gui.center['spectate_confirmation_frame'].destroy()
        Team.set_player_to_spectator(player)
        game.print(player.name .. ' has turned into a spectator ghost.')
        return
    end
    if event.element.name == 'spectate_button' then
        if player.gui.center['spectate_confirmation_frame'] then
            player.gui.center['spectate_confirmation_frame'].destroy()
        else
            create_spectate_confirmation(player)
        end
        return
    end
end

local event = require 'utils.event'
event.add(defines.events.on_gui_click, on_gui_click)

return Public