RedMew/map_gen/Diggy/Feature/CoinGathering.lua

--[[-- info
    Provides the ability to collect coins and send them to space.
]]

-- dependencies
local Event = require 'utils.event'
local Game = require 'utils.game'
local ScoreTable = require 'map_gen.Diggy.ScoreTable'
local Debug = require 'map_gen.Diggy.Debug'
local Template = require 'map_gen.Diggy.Template'
local Perlin = require 'map_gen.shared.perlin_noise'
local random = math.random
local ceil = math.ceil
local pairs = pairs
local Gui = require 'utils.gui'
local utils = require 'utils.core'

-- this
local CoinGathering = {}

-- some GUI stuff
local function redraw_table(data)
    local list = data.list
    Gui.clear(list)

    data.frame.caption = 'Scoretable'

    for name, value in pairs(ScoreTable.all()) do
        local table = list.add({type = 'table', column_count = 2})

        local key = table.add({type = 'label', name = 'Diggy.CoinGathering.Frame.List.Key', caption = name})
        key.style.minimal_width = 175

        local val = table.add({type = 'label', name = 'Diggy.CoinGathering.Frame.List.Val', caption = utils.comma_value(value)})
        val.style.minimal_width = 225
    end
end


local function toggle(event)
    local player = event.player
    local center = player.gui.left
    local frame = center['Diggy.CoinGathering.Frame']

    if (frame and event.trigger == nil) then
        Gui.destroy(frame)
        return
    elseif (frame) then
        local data = Gui.get_data(frame)
        redraw_table(data)
        return
    end

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

    local scroll_pane = frame.add({type = 'scroll-pane'})
    scroll_pane.style.maximal_height = 400

    frame.add({type = 'button', name = 'Diggy.CoinGathering.Button', caption = 'Close'})

    local data = {
        frame = frame,
        list = scroll_pane
    }

    redraw_table(data)

    Gui.set_data(frame, data)
end

local function on_player_created(event)
    Game.get_player_by_index(event.player_index).gui.top.add({
        name = 'Diggy.CoinGathering.Button',
        type = 'sprite-button',
        sprite = 'item/steel-axe',
    })
end

Gui.allow_player_to_toggle_top_element_visibility('Diggy.CoinGathering.Button')

Gui.on_click('Diggy.CoinGathering.Button', toggle)
Gui.on_custom_close('Diggy.CoinGathering.Frame', function (event)
    event.element.destroy()
end)

function CoinGathering.update_gui()
    for _, p in pairs(game.connected_players) do
        local frame = p.gui.left['Diggy.CoinGathering.Frame']

        if frame and frame.valid then
            local data = {player = p, trigger = 'update_gui'}
            toggle(data)
        end
    end
end

function CoinGathering.register(config)
    Event.add(defines.events.on_player_created, on_player_created)
    Event.on_nth_tick(61, CoinGathering.update_gui)

    ScoreTable.reset('Coins sent to space')

    local seed
    local noise_variance = config.noise_variance
    local function get_noise(surface, x, y)
        seed = seed or surface.map_gen_settings.seed + surface.index + 300
        return Perlin.noise(x * noise_variance * 0.9, y * noise_variance * 1.1, seed)
    end

    local distance_required = config.minimal_treasure_chest_distance * config.minimal_treasure_chest_distance

    Event.add(defines.events.on_rocket_launched, function (event)
        local coins = event.rocket.get_inventory(defines.inventory.rocket).get_item_count('coin')
        if coins > 0 then
            local sum = ScoreTable.add('Coins sent to space', coins)
            game.print('sent ' .. coins .. ' coins into space! The space station is now holding ' .. sum .. ' coins.')
        end
    end)

    local treasure_chest_noise_threshold = config.treasure_chest_noise_threshold
    Event.add(Template.events.on_void_removed, function (event)
        local position = event.position
        local x = position.x
        local y = position.y

        if (x * x + y * y <= distance_required) then
            return
        end

        local surface = event.surface

        if get_noise(surface, x, y) < treasure_chest_noise_threshold then
            return
        end

        local chest = surface.create_entity({name = 'steel-chest', position = position, force = game.forces.player})

        if not chest then
            return
        end

        local insert = chest.insert
        for name, prototype in pairs(config.treasure_chest_raffle) do
            if random() <= prototype.chance then
                insert({name = name, count = random(prototype.min, prototype.max)})
            end
        end
    end)

    local modifiers = config.alien_coin_modifiers
    local alien_coin_drop_chance = config.alien_coin_drop_chance

    Event.add(defines.events.on_entity_died, function (event)
        local entity = event.entity
        local force = entity.force
        if force.name ~= 'enemy' or random() > alien_coin_drop_chance then
            return
        end

        local modifier = modifiers[entity.name] or 1
        local evolution_multiplier = force.evolution_factor
        local count = random(
            ceil(2 * evolution_multiplier * modifier),
            ceil(5 * evolution_multiplier * modifier)
        )

        local coin = entity.surface.create_entity({
            name = 'item-on-ground',
            position = entity.position,
            stack = {name = 'coin', count = count}
        })

        if coin and coin.valid then
            coin.to_be_looted = true
        end
    end)

    local mining_coin_chance = config.mining_coin_chance
    local mining_coin_amount_min = config.mining_coin_amount.min
    local mining_coin_amount_max = config.mining_coin_amount.max
    Event.add(defines.events.on_pre_player_mined_item, function (event)
        local entity = event.entity
        if entity.type ~= 'simple-entity' then
            return
        end

        if random() > mining_coin_chance then
            return
        end

        local coin = entity.surface.create_entity({
            name = 'item-on-ground',
            position = entity.position,
            stack = {name = 'coin', count = random(mining_coin_amount_min, mining_coin_amount_max)}
        })

        if coin and coin.valid then
            coin.to_be_looted = true
        end
    end)

    if config.display_chest_locations then
        Event.add(defines.events.on_chunk_generated, function (event)
            local surface = event.surface
            local area = event.area

            for x = area.left_top.x, area.left_top.x + 31 do
                local sq_x = x * x
                for y = area.left_top.y, area.left_top.y + 31 do
                    if sq_x + y * y >= distance_required and get_noise(surface, x, y) >= treasure_chest_noise_threshold then
                        Debug.print_grid_value('chest', surface, {x = x, y = y}, nil, nil, true)
                    end
                end
            end
        end)
    end
end

return CoinGathering