RedMew/map_gen/Diggy/Feature/MiningEfficiency.lua

--[[-- info
    Provides the ability to increase mining efficiency while preserving the original value.
]]

-- dependencies
local Event = require 'utils.event'

-- this
local MiningEfficiency = {}

global.MiningEfficiency = {
    default_mining_speed = 0,
    active_modifier = 0,
    current_modifier = 0,
    research_modifier = 0,
}

local function update_mining_speed(player_force)
    -- recalculate current modifier
    global.MiningEfficiency.current_modifier =
        global.MiningEfficiency.default_mining_speed + global.MiningEfficiency.research_modifier

    -- remove the current buff
    player_force.manual_mining_speed_modifier =
        player_force.manual_mining_speed_modifier - global.MiningEfficiency.active_modifier

    -- update the active modifier
    global.MiningEfficiency.active_modifier = global.MiningEfficiency.current_modifier

    -- add the new active modifier
    player_force.manual_mining_speed_modifier =
        player_force.manual_mining_speed_modifier + global.MiningEfficiency.active_modifier
end

--[[--
    Registers all event handlers.
]]
function MiningEfficiency.register(config)
    global.MiningEfficiency.default_mining_speed = config.default_mining_speed

    Event.add(
        defines.events.on_research_finished,
        function(event)
            local player_force = game.forces.player

            global.MiningEfficiency.research_modifier =
                player_force.mining_drill_productivity_bonus * config.mining_speed_productivity_multiplier / 2

            update_mining_speed(player_force)
        end
    )
end

function MiningEfficiency.get_extra_map_info(config)
    return [[Mining Efficiency, increase your mining efficiency by researching mining productivity!
Efficiency increase per level: ]] .. config.mining_speed_productivity_multiplier .. '%'
end

function MiningEfficiency.on_init()
    update_mining_speed(game.forces.player)
end

return MiningEfficiency