local Public = require 'maps.fish_defender_v2.table'

local random = math.random
local sqrt = math.sqrt

local radius = 6
local targets = {}
local acid_splashes = {
    ['big-biter'] = 'acid-stream-worm-medium',
    ['behemoth-biter'] = 'acid-stream-worm-big',
    ['big-spitter'] = 'acid-stream-worm-medium',
    ['behemoth-spitter'] = 'acid-stream-worm-big'
}
local acid_lines = {
    ['big-spitter'] = 'acid-stream-spitter-medium',
    ['behemoth-spitter'] = 'acid-stream-spitter-big'
}
for x = radius * -1, radius, 1 do
    for y = radius * -1, radius, 1 do
        if sqrt(x ^ 2 + y ^ 2) <= radius then
            targets[#targets + 1] = {x = x, y = y}
        end
    end
end

local function acid_nova(event)
    for _ = 1, random(16, 32) do
        local i = random(1, #targets)
        event.entity.surface.create_entity(
            {
                name = acid_splashes[event.entity.name],
                position = event.entity.position,
                force = event.entity.force.name,
                source = event.entity.position,
                target = {x = event.entity.position.x + targets[i].x, y = event.entity.position.y + targets[i].y},
                max_range = radius,
                speed = 0.001
            }
        )
    end
end

function Public.died(event)
    local acid_lines_delay = Public.get('acid_lines_delay')
    if acid_splashes[event.entity.name] then
        acid_nova(event)
    end
    if acid_lines_delay[event.entity.unit_number] then
        Public.get('acid_lines_delay')[event.entity.unit_number] = nil
    end
    Public.get('boss_biters')[event.entity.unit_number] = nil
end

local function acid_line(surface, name, source, target)
    local distance = sqrt((source.x - target.x) ^ 2 + (source.y - target.y) ^ 2)

    if distance > 16 then
        return false
    end

    local modifier = {(target.x - source.x) / distance, (target.y - source.y) / distance}

    local position = {source.x, source.y}

    for i = 1, distance + 4, 1 do
        if random(1, 3) == 1 then
            surface.create_entity(
                {
                    name = name,
                    position = source,
                    force = 'enemy',
                    source = source,
                    target = position,
                    max_range = 25,
                    speed = 1
                }
            )
        end
        position = {position[1] + modifier[1], position[2] + modifier[2]}
    end

    return true
end

function Public.damaged_entity(event)
    local entity = event.entity
    local cause = event.cause

    if acid_lines[cause.name] then
        local acid_lines_delay = Public.get('acid_lines_delay')
        if not acid_lines_delay[cause.unit_number] then
            Public.set('acid_lines_delay')[cause.unit_number] = 0
        end

        if acid_lines_delay[cause.unit_number] < game.tick then
            if acid_line(cause.surface, acid_lines[cause.name], cause.position, entity.position) then
                Public.set('acid_lines_delay')[cause.unit_number] = game.tick + 180
            end
        end
    end
end

return Public