--wreckage yields scrap
local table_insert = table.insert

local Scrap = require 'modules.scrap_towny_ffa.scrap'

-- loot chances and amounts for scrap entities

local entity_loot_chance = {
    {name = 'advanced-circuit', chance = 5},
    --{name = "artillery-shell", chance = 1},
    {name = 'battery', chance = 20},
    {name = 'cannon-shell', chance = 2},
    --{name = "cluster-grenade", chance = 2},
    {name = 'construction-robot', chance = 1},
    {name = 'copper-cable', chance = 250},
    {name = 'copper-plate', chance = 500},
    {name = 'crude-oil-barrel', chance = 30},
    {name = 'defender-capsule', chance = 5},
    {name = 'destroyer-capsule', chance = 1},
    {name = 'distractor-capsule', chance = 2},
    {name = 'electric-engine-unit', chance = 2},
    {name = 'electronic-circuit', chance = 200},
    {name = 'empty-barrel', chance = 10},
    {name = 'engine-unit', chance = 4},
    {name = 'explosive-cannon-shell', chance = 2},
    --{name = "explosive-rocket", chance = 3},
    --{name = "explosive-uranium-cannon-shell", chance = 1},
    {name = 'explosives', chance = 5},
    {name = 'green-wire', chance = 10},
    {name = 'grenade', chance = 10},
    {name = 'heat-pipe', chance = 1},
    {name = 'heavy-oil-barrel', chance = 15},
    {name = 'iron-gear-wheel', chance = 500},
    {name = 'iron-plate', chance = 750},
    {name = 'iron-stick', chance = 50},
    {name = 'land-mine', chance = 3},
    {name = 'light-oil-barrel', chance = 15},
    {name = 'logistic-robot', chance = 1},
    {name = 'low-density-structure', chance = 1},
    {name = 'lubricant-barrel', chance = 20},
    --{name = "nuclear-fuel", chance = 1},
    {name = 'petroleum-gas-barrel', chance = 15},
    {name = 'pipe', chance = 100},
    {name = 'pipe-to-ground', chance = 10},
    {name = 'plastic-bar', chance = 5},
    {name = 'processing-unit', chance = 2},
    {name = 'red-wire', chance = 10},
    --{name = "rocket", chance = 3},
    --{name = "rocket-control-unit", chance = 1},
    --{name = "rocket-fuel", chance = 3},
    {name = 'solid-fuel', chance = 100},
    {name = 'steel-plate', chance = 150},
    {name = 'sulfuric-acid-barrel', chance = 15},
    --{name = "uranium-cannon-shell", chance = 1},
    --{name = "uranium-fuel-cell", chance = 1},
    --{name = "used-up-uranium-fuel-cell", chance = 1},
    {name = 'water-barrel', chance = 10},
    {name = 'tank', chance = 1},
    {name = 'car', chance = 1}
}

local entity_loot_amounts = {
    ['advanced-circuit'] = 2,
    --["artillery-shell"] = 0.3,
    ['battery'] = 2,
    ['cannon-shell'] = 2,
    --["cluster-grenade"] = 0.3,
    ['construction-robot'] = 0.3,
    ['copper-cable'] = 24,
    ['copper-plate'] = 16,
    ['crude-oil-barrel'] = 3,
    ['defender-capsule'] = 2,
    ['destroyer-capsule'] = 0.3,
    ['distractor-capsule'] = 0.3,
    ['electric-engine-unit'] = 2,
    ['electronic-circuit'] = 8,
    ['empty-barrel'] = 3,
    ['engine-unit'] = 2,
    ['explosive-cannon-shell'] = 2,
    --["explosive-rocket"] = 2,
    --["explosive-uranium-cannon-shell"] = 2,
    ['explosives'] = 4,
    ['green-wire'] = 8,
    ['grenade'] = 2,
    ['heat-pipe'] = 1,
    ['heavy-oil-barrel'] = 3,
    ['iron-gear-wheel'] = 8,
    ['iron-plate'] = 16,
    ['iron-stick'] = 16,
    ['land-mine'] = 1,
    ['light-oil-barrel'] = 3,
    ['logistic-robot'] = 0.3,
    ['low-density-structure'] = 0.3,
    ['lubricant-barrel'] = 3,
    --["nuclear-fuel"] = 0.1,
    ['petroleum-gas-barrel'] = 3,
    ['pipe'] = 8,
    ['pipe-to-ground'] = 1,
    ['plastic-bar'] = 4,
    ['processing-unit'] = 1,
    ['red-wire'] = 8,
    --["rocket"] = 2,
    --["rocket-control-unit"] = 0.3,
    --["rocket-fuel"] = 0.3,
    ['solid-fuel'] = 4,
    ['steel-plate'] = 4,
    ['sulfuric-acid-barrel'] = 3,
    --["uranium-cannon-shell"] = 2,
    --["uranium-fuel-cell"] = 0.3,
    --["used-up-uranium-fuel-cell"] = 1,
    ['water-barrel'] = 3,
    ['tank'] = 1,
    ['car'] = 1
}

local scrap_raffle = {}
for _, t in pairs(entity_loot_chance) do
    for _ = 1, t.chance, 1 do
        table_insert(scrap_raffle, t.name)
    end
end

local size_of_scrap_raffle = #scrap_raffle

local function on_player_mined_entity(event)
    local entity = event.entity
    if not entity.valid then
        return
    end
    local position = entity.position
    if Scrap.is_scrap(entity) == false then
        return
    end
    if entity.name == 'crash-site-chest-1' or entity.name == 'crash-site-chest-2' then
        return
    end

    -- scrap entities drop loot
    event.buffer.clear()

    local scrap = scrap_raffle[math.random(1, size_of_scrap_raffle)]

    local amount_bonus = (game.forces.enemy.evolution_factor * 2) + (game.forces.player.mining_drill_productivity_bonus * 2)
    local r1 = math.ceil(entity_loot_amounts[scrap] * (0.3 + (amount_bonus * 0.3)))
    local r2 = math.ceil(entity_loot_amounts[scrap] * (1.7 + (amount_bonus * 1.7)))
    local amount = math.random(r1, r2)

    local player = game.players[event.player_index]
    local inserted_count = player.insert({name = scrap, count = amount})

    if inserted_count ~= amount then
        local amount_to_spill = amount - inserted_count
        entity.surface.spill_item_stack(position, {name = scrap, count = amount_to_spill}, true)
    end

    entity.surface.create_entity(
        {
            name = 'flying-text',
            position = position,
            text = '+' .. amount .. ' [img=item/' .. scrap .. ']',
            color = {r = 0.98, g = 0.66, b = 0.22}
        }
    )
end

local Event = require 'utils.event'
Event.add(defines.events.on_player_mined_entity, on_player_mined_entity)