-- biters yield coins -- by mewmew

local Event = require 'utils.event'
local insert = table.insert
local math_floor = math.floor

local coin_yield = {
    ['behemoth-biter'] = 5,
    ['behemoth-spitter'] = 5,
    ['behemoth-worm-turret'] = 20,
    ['big-biter'] = 3,
    ['big-spitter'] = 3,
    ['big-worm-turret'] = 16,
    ['biter-spawner'] = 32,
    ['medium-biter'] = 2,
    ['medium-spitter'] = 2,
    ['medium-worm-turret'] = 12,
    ['small-biter'] = 1,
    ['small-spitter'] = 1,
    ['small-worm-turret'] = 8,
    ['spitter-spawner'] = 32
}

local entities_that_earn_coins = {
    ['artillery-turret'] = true,
    ['gun-turret'] = true,
    ['laser-turret'] = true,
    ['flamethrower-turret'] = true
}

--extra coins for "boss" biters from biter_health_booster.lua
local function get_coin_count(entity)
    local coin_count = coin_yield[entity.name]
    if not coin_count then
        return
    end
    if not storage.biter_health_boost_units then
        return coin_count
    end
    local unit_number = entity.unit_number
    if not unit_number then
        return coin_count
    end
    if not storage.biter_health_boost_units[unit_number] then
        return coin_count
    end
    if not storage.biter_health_boost_units[unit_number][3] then
        return coin_count
    end
    local m = 1 / storage.biter_health_boost_units[unit_number][2]
    coin_count = math_floor(coin_count * m)
    if coin_count < 1 then
        return 1
    end
    return coin_count
end

local function on_entity_died(event)
    local entity = event.entity
    if not entity.valid then
        return
    end
    if entity.force.index ~= 2 then
        return
    end

    local coin_count = get_coin_count(entity)
    if not coin_count then
        return
    end

    local players_to_reward = {}
    local reward_has_been_given = false

    if event.cause then
        if event.cause.valid then
            if event.cause.name == 'character' then
                insert(players_to_reward, event.cause)
                reward_has_been_given = true
            end
            if event.cause.type == 'car' then
                local player = event.cause.get_driver()
                local passenger = event.cause.get_passenger()
                if player then
                    insert(players_to_reward, player.player)
                end
                if passenger then
                    insert(players_to_reward, passenger.player)
                end
                reward_has_been_given = true
            end
            if event.cause.type == 'locomotive' then
                local train_passengers = event.cause.train.passengers
                if train_passengers then
                    for _, passenger in pairs(train_passengers) do
                        insert(players_to_reward, passenger)
                    end
                    reward_has_been_given = true
                end
            end
            for _, player in pairs(players_to_reward) do
                player.insert({ name = 'coin', count = coin_count })
            end
        end
        if entities_that_earn_coins[event.cause.name] then
            event.entity.surface.spill_item_stack({position = event.cause.position, stack = { name = 'coin', count = coin_count }, enable_looted = true})
            reward_has_been_given = true
        end
    end

    if reward_has_been_given == false then
        event.entity.surface.spill_item_stack({position = event.entity.position, stack = { name = 'coin', count = coin_count }, enable_looted = true})
    end
end

Event.add(defines.events.on_entity_died, on_entity_died)
