RedMew/map_gen/Diggy/Feature/MarketExchange.lua

--[[-- info
    Provides the ability to purchase items from the market.
]]

-- dependencies
local Event = require 'utils.event'
local Token = require 'utils.global_token'
local Task = require 'utils.Task'
local Gui = require 'utils.gui'
local Debug = require 'map_gen.Diggy.Debug'
local Template = require 'map_gen.Diggy.Template'
local Global = require 'utils.global'

-- this
local MarketExchange = {}

local config = {}

local stone_tracker = {
    first_time_market_item = nil,
    stone_sent_to_surface = 0,
    previous_stone_sent_to_surface = 0,
}
local mining_efficiency = {
    active_modifier = 0,
    research_modifier = 0,
    market_modifier = 0,
}
local inventory_slots = {
    active_modifier = 0,
    research_modifier = 0,
    market_modifier = 0,
}

Global.register({
    stone_tracker = stone_tracker,
    mining_efficiency = mining_efficiency,
    inventory_slots = inventory_slots,
}, function(tbl)
    stone_tracker = tbl.stone_tracker
    mining_efficiency = tbl.mining_efficiency
    inventory_slots = tbl.inventory_slots
end)

local on_market_timeout_finished = Token.register(function(params)
    Template.market(params.surface, params.position, params.player_force, params.currency_item, {})
end)

local function on_init()
    Task.set_timeout_in_ticks(50, on_market_timeout_finished, {
        surface = game.surfaces.nauvis,
        position = config.market_spawn_position,
        player_force = game.forces.player,
        currency_item = config.currency_item,
    })
end

local function update_mining_speed(force)
    -- remove the current buff
    local old_modifier = force.manual_mining_speed_modifier - mining_efficiency.active_modifier

    -- update the active modifier
    mining_efficiency.active_modifier = mining_efficiency.research_modifier + mining_efficiency.market_modifier

    -- add the new active modifier to the non-buffed modifier
    force.manual_mining_speed_modifier = old_modifier + mining_efficiency.active_modifier
end

local function update_inventory_slots(force)
    -- remove the current buff
    local old_modifier = force.character_inventory_slots_bonus - inventory_slots.active_modifier

    -- update the active modifier
    inventory_slots.active_modifier = inventory_slots.research_modifier + inventory_slots.market_modifier

    -- add the new active modifier to the non-buffed modifier
    force.character_inventory_slots_bonus = old_modifier + inventory_slots.active_modifier
end

local function update_market_contents(market)
    local should_update_mining_speed = false
    local should_update_inventory_slots = false

    if (nil ~= stone_tracker.first_time_market_item) then
        market.add_market_item(stone_tracker.first_time_market_item)
        stone_tracker.first_time_market_item = nil
    end

    for _, unlockable in pairs(config.unlockables) do
        local is_in_range = unlockable.stone > stone_tracker.previous_stone_sent_to_surface and unlockable.stone <= stone_tracker.stone_sent_to_surface

        -- only add the item to the market if it's between the old and new stone range
        if (is_in_range and unlockable.type == 'market') then
            market.add_market_item({
                price = {{config.currency_item, unlockable.prototype.price}},
                offer = {type = 'give-item', item = unlockable.prototype.item_name, count = 1}
            })
        elseif (is_in_range and unlockable.type == 'buff' and unlockable.prototype.name == 'mining_speed') then
            should_update_mining_speed = true
            mining_efficiency.market_modifier = mining_efficiency.market_modifier + (unlockable.prototype.value / 100)
        elseif (is_in_range and unlockable.type == 'buff' and unlockable.prototype.name == 'inventory_slot') then
            should_update_inventory_slots = true
            inventory_slots.market_modifier = inventory_slots.market_modifier + unlockable.prototype.value
        end
    end

    local force

    if (should_update_mining_speed) then
        force = force or game.forces.player
        update_mining_speed(force)
    end

    if (should_update_inventory_slots) then
        force = force or game.forces.player
        update_inventory_slots(force)
    end
end

local function on_research_finished(event)
    local force = game.forces.player
    local current_modifier = mining_efficiency.research_modifier
    local new_modifier = force.mining_drill_productivity_bonus * config.mining_speed_productivity_multiplier / 2

    if (current_modifier == new_modifier) then
        -- something else was researched
        return
    end

    mining_efficiency.research_modifier = new_modifier
    inventory_slots.research_modifier = (force.mining_drill_productivity_bonus / 2) * 100

    update_inventory_slots(force)
    update_mining_speed(force)
end

local function on_market_item_purchased(event)
    if (1 ~= event.offer_index) then
        return
    end

    stone_tracker.previous_stone_sent_to_surface = stone_tracker.stone_sent_to_surface
    stone_tracker.stone_sent_to_surface = stone_tracker.stone_sent_to_surface + (config.stone_to_surface_amount * event.count)

    update_market_contents(event.market)
end

local function on_placed_entity(event)
    if ('market' ~= event.entity.name) then
        return
    end

    update_market_contents(event.entity)
end

--[[--
    Registers all event handlers.
]]
function MarketExchange.register(cfg)
    config = cfg

    stone_tracker.first_time_market_item = {
        price = {{config.currency_item, 50}},
        offer = {type = 'nothing', effect_description = 'Send ' .. config.stone_to_surface_amount .. ' stone to the surface'}
    }

    Event.add(defines.events.on_research_finished, on_research_finished)
    Event.add(defines.events.on_market_item_purchased, on_market_item_purchased)
    Event.add(Template.events.on_placed_entity, on_placed_entity)
end

function MarketExchange.get_extra_map_info(config)
    return 'Market Exchange, trade your stone or send it to the surface'
end

function MarketExchange.on_init()
    if ('function' ~= type(on_init)) then
        error('Expected local on_init in MarketExchange to have a function assigned.')
    end

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

local function toggle(event)
    local player = event.player
    local center = player.gui.center
    local main_frame = center['Diggy.MarketExchange.Frame']

    if (main_frame) then
        Gui.destroy(main_frame)
        return
    end

    main_frame = center.add({name = 'Diggy.MarketExchange.Frame', type = 'frame', direction = 'vertical'})
end

--[[--
    @param config LuaGuiElement
]]
function MarketExchange.top_menu_button_prototype(config)
    return {
        name = 'Diggy.MarketExchange.Button',
        type = 'sprite-button',
        sprite = 'item/stone',
        toggle = toggle,
    }
end

return MarketExchange