--[[-- Commands Module - Bonus
    - Adds a command that allows players to have increased stats
    @commands Bonus
]]

local Commands = require 'expcore.commands' --- @dep expcore.commands
local Roles = require 'expcore.roles' --- @dep expcore.roles
local Event = require 'utils.event' --- @dep utils.event
local Game = require 'utils.game' --- @dep utils.game
local Store = require 'expcore.store' --- @dep expcore.store
local config = require 'config.bonuses' --- @dep config.bonuses
require 'config.expcore-commands.parse_general'

-- Store bonus percentages keyed by player name
local bonus_store = Store.register(function(player)
    return player.name
end)

-- Apply a bonus amount to a player
local function apply_bonus(player,amount)
    if not amount then return end
    for bonus,min_max in pairs(config) do
        local increase = min_max[2]*amount
        player[bonus] = min_max[1]+increase
    end
end

--- Changes the amount of bonus you receive
-- @command bonus
-- @tparam number amount range 0-50 the percent increase for your bonus
Commands.new_command('bonus','Changes the amount of bonus you receive')
:add_param('amount','integer-range',0,50)
:register(function(player,amount)
    local percent = amount/100
    Store.set(bonus_store,player,percent)
    Commands.print{'expcom-bonus.set',amount}
    Commands.print({'expcom-bonus.wip'},'orange')
end)

-- When store is updated apply new bonus to the player
Store.watch(bonus_store,function(value,category)
    local player = Game.get_player_from_any(category)
    apply_bonus(player,value)
end)

-- When a player respawns re-apply bonus
Event.add(defines.events.on_player_respawned,function(event)
    local player = Game.get_player_by_index(event.player_index)
    local value = Store.get(bonus_store,player)
    apply_bonus(player,value)
end)

-- When a player dies allow them to have instant respawn
Event.add(defines.events.on_player_died,function(event)
    local player = Game.get_player_by_index(event.player_index)
    if Roles.player_has_flag(player,'instance-respawn') then
        player.ticks_to_respawn = 120
    end
end)

-- Remove bonus if a player no longer has access to the command
local function role_update(event)
    local player = Game.get_player_by_index(event.player_index)
    if not Roles.player_allowed(player,'command/bonus') then
        Store.clear(bonus_store,player)
    end
end

Event.add(Roles.events.on_role_assigned,role_update)
Event.add(Roles.events.on_role_unassigned,role_update)

return bonus_store