--luacheck: ignore
local event = require 'utils.event'
local math_random = math.random
local math_floor = math.floor
global.biter_command = {}
global.biter_command.active_unit_groups = {}
global.biter_command.enabled = true
global.biter_command.whitelist = {}
global.biter_command.admin_mode = true --if only admins can see and use the panel

global.biter_command.teleporting = false --if teleporting is allowed for non-admins
global.biter_command.buildings = true ---if player can trigger building nests and worms

local worm_raffle = {
    'small-worm-turret',
    'small-worm-turret',
    'medium-worm-turret',
    'small-worm-turret',
    'medium-worm-turret',
    'medium-worm-turret',
    'big-worm-turret',
    'medium-worm-turret',
    'big-worm-turret',
    'big-worm-turret',
    'behemoth-worm-turret',
    'big-worm-turret',
    'behemoth-worm-turret',
    'behemoth-worm-turret',
    'behemoth-worm-turret',
    'big-worm-turret',
    'behemoth-worm-turret'
}

local function shuffle(tbl)
    local size = #tbl
    for i = size, 1, -1 do
        local rand = math_random(size)
        tbl[i], tbl[rand] = tbl[rand], tbl[i]
    end
    return tbl
end

local function is_closer(pos1, pos2, pos)
    return ((pos1.x - pos.x) ^ 2 + (pos1.y - pos.y) ^ 2) < ((pos2.x - pos.x) ^ 2 + (pos2.y - pos.y) ^ 2)
end

local function shuffle_distance(tbl, position)
    local size = #tbl
    for i = size, 1, -1 do
        local rand = math_random(size)
        if is_closer(tbl[i].position, tbl[rand].position, position) and i > rand then
            tbl[i], tbl[rand] = tbl[rand], tbl[i]
        end
    end
    return tbl
end

local function get_evo(force)
    local evo = math_floor(game.forces['enemy'].evolution_factor * 20)
    local nests = math_random(1 + evo, 2 + evo * 2)
end

local function place_nest_near_unit_group(group)
    if not global.biter_command.buildings then
        return false
    end
    if not group.members then
        return false
    end
    if #group.members < 5 then
        return false
    end
    local units = group.members
    shuffle(units)
    for i = 1, 5, 1 do
        if not units[i].valid then
            return false
        end
    end
    local name = 'biter-spawner'
    if math_random(1, 3) == 1 then
        name = 'spitter-spawner'
    end
    local position = group.surface.find_non_colliding_position(name, group.position, 16, 1)
    if not position then
        return false
    end
    group.surface.create_entity({name = name, position = position, force = group.force})
    group.surface.create_entity({name = 'blood-explosion-huge', position = position})
    for i = 1, 5, 1 do
        units[i].surface.create_entity({name = 'blood-explosion-huge', position = units[i].position})
        units[i].destroy()
    end
    return true
end

local function build_worm(group)
    if not global.biter_command.buildings then
        return false
    end
    if not group.members then
        return false
    end
    if #group.members < 5 then
        return false
    end
    local units = group.members
    shuffle(units)
    for i = 1, 5, 1 do
        if not units[i].valid then
            return false
        end
    end
    local position = group.surface.find_non_colliding_position('assembling-machine-1', group.position, 8, 1)
    local worm = worm_raffle[math_random(1 + math_floor(group.force.evolution_factor * 8), math_floor(1 + group.force.evolution_factor * 16))]
    if not position then
        return false
    end
    group.surface.create_entity({name = worm, position = position, force = group.force})
    group.surface.create_entity({name = 'blood-explosion-huge', position = position})
    for i = 1, 5, 1 do
        units[i].surface.create_entity({name = 'blood-explosion-huge', position = units[i].position})
        units[i].destroy()
    end
    return true
end

local function flying_text(message, action, position, player)
    local texts = {
        {'roger', 'acknowledged', 'aye aye', 'confirmed', 'will do'},
        {'negative', 'no', 'not really', 'we are not your critters', 'go away'},
        {'fooood', 'nom nom', 'we hunger', 'killllll'},
        {'WTF', 'we wanted ACTION', 'why you hate us', 'we were good soldiers', 'go to hell'}
    }
    colors = {{r = 0, g = 220, b = 0}, {r = 220, g = 0, b = 0}, {r = 0, g = 100, b = 220}, {r = 200, g = 200, b = 0}, {r = 255, g = 255, b = 255}}
    if message then
        player.create_local_flying_text {text = message, position = position, color = colors[5]}
    else
        player.create_local_flying_text {text = texts[action][math_random(1, #texts[action])], position = position, color = colors[action]}
    end
end

-----------commands-----------

local function move_to(position, distraction)
    local command = {
        type = defines.command.go_to_location,
        destination = position,
        distraction = distraction,
        pathfind_flags = {allow_destroy_friendly_entities = true}
    }
    return command
end

-- local function attackmaincommand(target)
--   local wave_defense_table = WD.get_table()
--   if not wave_defense_table.target then return end
--   if not wave_defense_table.target.valid then return end
--   local command = {
-- 		type = defines.command.attack,
-- 		target = target,
-- 		distraction = defines.distraction.by_enemy,
-- 	}
--   return command
-- end

local function attackareacommand(position)
    local command = {
        type = defines.command.attack_area,
        destination = position,
        radius = 25,
        distraction = defines.distraction.by_enemy
    }
    return command
end

local function attackobstaclescommand(surface, position)
    local commands = {}
    local obstacles = surface.find_entities_filtered {position = position, radius = 20, type = {'simple-entity', 'tree'}, limit = 100}
    if obstacles then
        shuffle(obstacles)
        shuffle_distance(obstacles, position)
        for i = 1, #obstacles, 1 do
            if obstacles[i].valid then
                commands[#commands + 1] = {
                    type = defines.command.attack,
                    target = obstacles[i],
                    distraction = defines.distraction.by_enemy
                }
            end
        end
    end
    return commands
end

local function get_coords(group, source_player)
    local position
    if source_player.gui.screen['biter_panel'] then
        local x = tonumber(source_player.gui.screen['biter_panel']['coords']['coord_x'].text)
        local y = tonumber(source_player.gui.screen['biter_panel']['coords']['coord_y'].text)
        if x == nil or x == 'nil' then
            x = group.position.x
            source_player.gui.screen['biter_panel']['coords']['coord_x'].text = group.position.x
        end
        if y == nil or y == 'nil' then
            y = group.position.y
            source_player.gui.screen['biter_panel']['coords']['coord_y'].text = group.position.y
        end
        position = {x = x, y = y}
    end
    return position
end

-------button functions-------------

local function pan(group, source_player)
    source_player.open_map(group.position, 0.5)
end

local function teleport(group, source_player)
    if source_player.admin or global.biter_command.teleporting then
        source_player.teleport(group.position, group.surface)
    else
        flying_text('Teleporting is disabled', nil, source_player.position, source_player)
    end
end

local function disband(group, source_player)
    flying_text(nil, 4, group.position, source_player)
    group.destroy()
end

local function movetome(group, source_player)
    group.set_command(move_to(source_player.position, defines.distraction.none))
    flying_text(nil, 1, group.position, source_player)
end

local function movetoposition(group, source_player)
    local position = get_coords(group, source_player)
    if position then
        group.set_command(move_to(position, defines.distraction.none))
        flying_text(nil, 1, group.position, source_player)
    else
        flying_text(nil, 2, group.position, source_player)
    end
end

local function patroltome(group, source_player)
    group.set_command(move_to(source_player.position, defines.distraction.by_enemy))
    flying_text(nil, 1, group.position, source_player)
end

local function patroltoposition(group, source_player)
    local position = get_coords(group, source_player)
    if position then
        group.set_command(move_to(position, defines.distraction.by_enemy))
        flying_text(nil, 1, group.position, source_player)
    else
        flying_text(nil, 2, group.position, source_player)
    end
end

local function settle(group, source_player)
    local success = place_nest_near_unit_group(group)
    if success then
        flying_text(nil, 1, group.position, source_player)
    else
        flying_text(nil, 2, group.position, source_player)
        source_player.print('Settling new nest failed. Check if group has enough members(5+) and there is empty space (or nests are disabled).')
    end
end

local function siege(group, source_player)
    local success = build_worm(group)
    if success then
        flying_text(nil, 1, group.position, source_player)
    else
        flying_text(nil, 2, group.position, source_player)
        source_player.print('Making worm failed. Check if group has enough members(5+) and there is empty space (or worms are disabled).')
    end
end

local function report(group, source_player)
    local status = group.state
    local states = {'gathering', 'moving', 'attacking distraction', 'attacking target', 'finished', 'pathfinding', 'wander in group'}
    flying_text(states[status + 1], nil, group.position, source_player)
end

local function attackenemiesaround(group, source_player)
    flying_text(nil, 3, group.position, source_player)
    group.set_command(attackareacommand(group.position))
end

local function attackobstaclesaround(group, source_player)
    local commands = attackobstaclescommand(group.surface, group.position)
    if #commands > 1 then
        group.set_command(
            {
                type = defines.command.compound,
                structure_type = defines.compound_command.return_last,
                commands = commands
            }
        )
        flying_text(nil, 3, group.position, source_player)
    else
        source_player.print('No obstacles found around unit group.')
        flying_text(nil, 2, group.position, source_player)
    end
end

local function attackenemiesaroundme(group, source_player)
    group.set_command(attackareacommand(source_player.position))
    flying_text(nil, 3, group.position, source_player)
end

local function attackobstaclesaroundme(group, source_player)
    local commands = attackobstaclescommand(source_player.surface, source_player.position)
    if #commands > 1 then
        group.set_command(
            {
                type = defines.command.compound,
                structure_type = defines.compound_command.return_last,
                commands = commands
            }
        )
        flying_text(nil, 3, group.position, source_player)
    else
        source_player.print('No obstacles found around player.')
        flying_text(nil, 2, group.position, source_player)
    end
end

local function addunitsaroundme(group, source_player)
    local units = source_player.surface.find_entities_filtered {position = source_player.position, radius = 50, type = 'unit', force = group.force}
    for i = 1, #units, 1 do
        group.add_member(units[i])
    end
end

local function addunits(group, source_player)
    local units = source_player.surface.find_entities_filtered {position = group.position, radius = 50, type = 'unit', force = group.force}
    for i = 1, #units, 1 do
        group.add_member(units[i])
    end
end

local function forcemove(group, source_player)
    group.start_moving()
    flying_text(nil, 1, group.position, source_player)
end

local function creategroup(source_player)
    source_player.surface.create_unit_group {position = source_player.position, force = source_player.force}
    flying_text('Unit group created', nil, source_player.position, source_player)
end
----------------------direction panel-----------------
local function set_directions(changedx, changedy, source_player)
    if source_player.gui.screen['biter_panel'] then
        local x = tonumber(source_player.gui.screen['biter_panel']['coords']['coord_x'].text)
        local y = tonumber(source_player.gui.screen['biter_panel']['coords']['coord_y'].text)
        if x == nil or x == 'nil' then
            x = 0
        end
        if y == nil or y == 'nil' then
            y = 0
        end
        x = x + changedx
        y = y + changedy
        source_player.gui.screen['biter_panel']['coords']['coord_x'].text = x
        source_player.gui.screen['biter_panel']['coords']['coord_y'].text = y
    end
end

local function nw(source_player)
    set_directions(-25, -25, source_player)
end

local function n(source_player)
    set_directions(0, -25, source_player)
end

local function ne(source_player)
    set_directions(25, -25, source_player)
end

local function w(source_player)
    set_directions(-25, 0, source_player)
end

local function e(source_player)
    set_directions(25, 0, source_player)
end

local function sw(source_player)
    set_directions(-25, 25, source_player)
end

local function s(source_player)
    set_directions(0, 25, source_player)
end

local function se(source_player)
    set_directions(25, 25, source_player)
end

local function center(group, source_player)
    if source_player.gui.screen['biter_panel'] then
        source_player.gui.screen['biter_panel']['coords']['coord_x'].text = group.position.x
        source_player.gui.screen['biter_panel']['coords']['coord_y'].text = group.position.y
    end
end

----------------------------gui-----------------------

local function top_button(player)
    if player.gui.top['biter_commands'] then
        if global.biter_command.enabled or global.biter_command.whitelist[player.name] == true then
            player.gui.top['biter_commands'].visible = true
            return
        else
            --player.gui.top["biter_commands"].destroy()
            player.gui.top['biter_commands'].visible = false
            return
        end
    end
    if player.admin or not global.biter_command.admin_mode then
        if global.biter_command.enabled or global.biter_command.whitelist[player.name] == true then
            local button = player.gui.top.add({type = 'sprite-button', name = 'biter_commands', sprite = 'entity/medium-spitter'})
            button.style.minimal_height = 38
            button.style.minimal_width = 38
            button.style.padding = -2
        end
    end
end

local function show_info(player)
    if player.gui.screen['biter_comm_info'] then
        player.gui.screen['biter_comm_info'].destroy()
        return
    end
    local frame = player.gui.screen.add {type = 'frame', name = 'biter_comm_info', caption = 'Biter Commander needs halp', direction = 'vertical'}
    frame.location = {x = 350, y = 45}
    frame.style.minimal_height = 300
    frame.style.maximal_height = 300
    frame.style.minimal_width = 330
    frame.style.maximal_width = 630
    frame.add({type = 'label', caption = 'Create new group first, then add biters to it.'})
    frame.add({type = 'label', caption = 'You can use directionpad to navigate them, or do it in person.'})
    frame.add({type = 'label', caption = "If you input invalid coordinates, they get rewritten to current group's position."})
    frame.add({type = 'label', caption = 'You can operate only biters and create groups of your own force.'})
    frame.add({type = 'label', caption = "If group is stuck at gathering state, use 'force move' button."})
    frame.add({type = 'label', caption = 'Empty groups get autodeleted by game after a while.'})
    frame.add({type = 'button', name = 'close_info', caption = 'Close'})
end

local function build_groups(player)
    local groups = {}
    for _, g in pairs(global.biter_command.active_unit_groups) do
        if g.group.valid then
            if player.admin and global.biter_command.admin_mode then
                table.insert(groups, tostring(g.id))
            else
                if player.force == g.group.force then
                    table.insert(groups, tostring(g.id))
                end
            end
        else
            g = nil
        end
    end
    table.insert(groups, 'Select Group')
    return groups
end

local function biter_panel(player)
    if player.gui.screen['biter_panel'] then
        player.gui.screen['biter_panel'].destroy()
        return
    end

    local frame = player.gui.screen.add {type = 'frame', caption = 'Biter Commander', name = 'biter_panel', direction = 'vertical'}
    frame.location = {x = 5, y = 45}
    frame.style.minimal_height = 680
    frame.style.maximal_height = 680
    frame.style.minimal_width = 330
    frame.style.maximal_width = 330

    local groups = build_groups(player)
    local selected_index = #groups
    if global.panel_group_index then
        if global.panel_group_index[player.name] then
            if groups[global.panel_group_index[player.name]] then
                selected_index = global.paneld_group_index[player.name]
            end
        end
    end
    local t0 = frame.add({type = 'table', name = 'top', column_count = 3})
    local drop_down = t0.add({type = 'drop-down', name = 'group_select', items = groups, selected_index = selected_index})
    drop_down.style.minimal_width = 150
    drop_down.style.right_padding = 12
    drop_down.style.left_padding = 12
    t0.add({type = 'sprite-button', name = 'info', sprite = 'virtual-signal/signal-info'})
    t0.add({type = 'sprite-button', name = 'close_biters', sprite = 'virtual-signal/signal-X'})

    local l1 = frame.add({type = 'label', caption = 'Camera'})
    local t1 = frame.add({type = 'table', name = 'camera', column_count = 2})
    local l2 = frame.add({type = 'label', caption = 'Movement'})
    local t2 = frame.add({type = 'table', name = 'movement', column_count = 2})
    local l3 = frame.add({type = 'label', caption = 'Build'})
    local t3 = frame.add({type = 'table', name = 'build', column_count = 2})
    local l4 = frame.add({type = 'label', caption = 'Attack'})
    local t4 = frame.add({type = 'table', name = 'attack', column_count = 2})
    local l5 = frame.add({type = 'label', caption = 'Group Management'})
    local t5 = frame.add({type = 'table', name = 'management', column_count = 2})
    local line = frame.add {type = 'line'}
    line.style.top_margin = 8
    line.style.bottom_margin = 8
    local t6 = frame.add({type = 'table', name = 'directions', column_count = 3})
    local buttons = {
        t1.add({type = 'button', caption = 'Pan to group', name = 'pan', tooltip = 'Moves camera to group position.'}),
        t1.add({type = 'button', caption = 'TP to group', name = 'teleport', tooltip = 'Teleports to group.'}),
        t2.add({type = 'button', caption = 'Move to me', name = 'movetome', tooltip = 'Gives group order to move to your position.'}),
        t2.add({type = 'button', caption = 'Move to position', name = 'movetoposition', tooltip = 'Sends group to position with coordinates entered below.'}),
        t2.add({type = 'button', caption = 'Patrol to me ', name = 'patroltome', tooltip = 'Gives group order to move to your position and engage any enemy during movement.'}),
        t2.add(
            {
                type = 'button',
                caption = 'Patrol to position',
                name = 'patroltoposition',
                tooltip = 'Sends group to position with coordinates entered below and engage any enemy during movement.'
            }
        ),
        t3.add({type = 'button', caption = 'Settle nest', name = 'settle', tooltip = 'Group creates base. Costs 5 units.'}),
        t3.add({type = 'button', caption = 'Build worm', name = 'siege', tooltip = 'Group builds worm turret. Costs 5 units.'}),
        t4.add({type = 'button', caption = 'Attack area', name = 'attackenemiesaround', tooltip = 'Group attacks enemy things around self.'}),
        t4.add({type = 'button', caption = 'Attack obstacles', name = 'attackobstaclesaround', tooltip = 'Group attacks obstacles around self.'}),
        t4.add({type = 'button', caption = 'Attack my area', name = 'attackenemiesaroundme', tooltip = 'Group attacks enemy things around your position.'}),
        t4.add({type = 'button', caption = 'Attack my obstacles', name = 'attackobstaclesaroundme', tooltip = 'Group attacks obstacles around your position.'}),
        t5.add({type = 'button', caption = 'Report', name = 'report', tooltip = 'Reports group status.'}),
        t5.add({type = 'button', caption = 'Force Move', name = 'forcemove', tooltip = 'Makes group to start moving even if gathering is not done (unstuck).'}),
        t5.add({type = 'button', caption = 'Add units around me', name = 'addunitsaroundme', tooltip = 'Adds units around you to selected unit group.'}),
        t5.add({type = 'button', caption = 'Add units', name = 'addunits', tooltip = 'Adds units around group to it.'}),
        t5.add({type = 'button', caption = 'Create group', name = 'creategroup', tooltip = 'Creates new group on player position'}),
        t5.add({type = 'button', caption = 'Disband group', name = 'disband', tooltip = 'Disbands group.'})
    }
    local buttons2 = {
        t6.add({type = 'button', caption = '25 NW', name = 'nw', tooltip = 'Changes remote position'}),
        t6.add({type = 'button', caption = '25 N', name = 'n', tooltip = 'Changes remote position'}),
        t6.add({type = 'button', caption = '25 NE', name = 'ne', tooltip = 'Changes remote position'}),
        t6.add({type = 'button', caption = '25 W', name = 'w', tooltip = 'Changes remote position'}),
        t6.add({type = 'button', caption = 'Center', name = 'center', tooltip = 'Centers remote position to group'}),
        t6.add({type = 'button', caption = '25 E', name = 'e', tooltip = 'Changes remote position'}),
        t6.add({type = 'button', caption = '25 SW', name = 'sw', tooltip = 'Changes remote position'}),
        t6.add({type = 'button', caption = '25 S', name = 's', tooltip = 'Changes remote position'}),
        t6.add({type = 'button', caption = '25 SE', name = 'se', tooltip = 'Changes remote position'})
    }
    for _, button in pairs(buttons) do
        button.style.font = 'default-bold'
        button.style.font_color = {r = 0.99, g = 0.99, b = 0.99}
        button.style.minimal_width = 150
    end
    for _, button in pairs(buttons2) do
        button.style.font = 'default-bold'
        button.style.font_color = {r = 0.99, g = 0.99, b = 0.99}
        button.style.minimal_width = 70
    end
    local t7 = frame.add({type = 'table', name = 'coords', column_count = 2})
    t7.add({type = 'label', caption = 'X: '})
    t7.add({type = 'textfield', name = 'coord_x'})
    t7.add({type = 'label', caption = 'Y: '})
    t7.add({type = 'textfield', name = 'coord_y'})
end

local comm_functions = {
    ['pan'] = pan,
    ['teleport'] = teleport,
    ['disband'] = disband,
    ['movetome'] = movetome,
    ['movetoposition'] = movetoposition,
    ['patroltome'] = patroltome,
    ['patroltoposition'] = patroltoposition,
    ['settle'] = settle,
    ['siege'] = siege,
    ['report'] = report,
    ['attackenemiesaround'] = attackenemiesaround,
    ['attackobstaclesaround'] = attackobstaclesaround,
    ['attackenemiesaroundme'] = attackenemiesaroundme,
    ['attackobstaclesaroundme'] = attackobstaclesaroundme,
    ['addunits'] = addunits,
    ['addunitsaroundme'] = addunitsaroundme,
    ['forcemove'] = forcemove,
    ['center'] = center
}

local comm_global_functions = {
    ['creategroup'] = creategroup,
    ['nw'] = nw,
    ['n'] = n,
    ['ne'] = ne,
    ['w'] = w,
    ['e'] = e,
    ['sw'] = sw,
    ['s'] = s,
    ['se'] = se
}

local function refresh_groups(player)
    local groups = build_groups(player)
    player.gui.screen['biter_panel']['top']['group_select'].items = groups
end

local function on_gui_click(event)
    if not event then
        return
    end
    if not event.element then
        return
    end
    if not event.element.valid then
        return
    end
    local player = game.players[event.element.player_index]
    if event.element.name == 'biter_commands' then --top button press
        if global.biter_command.enabled or global.biter_command.whitelist[player.name] == true then
            biter_panel(player)
            return
        else
            top_button(player)
            player.print('Biter commander module is disabled.')
            return
        end
    else
        if global.biter_command.enabled or global.biter_command.whitelist[player.name] == true then
            top_button(player)
        end
    end
    if event.element.type ~= 'button' and event.element.type ~= 'sprite-button' then
        return
    end
    --if event.frame.name ~= "biter_panel" then return end
    local name = event.element.name
    if name == 'close_biters' then
        biter_panel(player)
        return
    end
    if name == 'info' then
        show_info(player)
        return
    end
    if name == 'close_info' then
        show_info(player)
        return
    end
    if comm_functions[name] then
        local target_group_id = event.element.parent.parent['top']['group_select'].items[event.element.parent.parent['top']['group_select'].selected_index]
        if not target_group_id then
            return
        end
        if target_group_id == 'Select Group' then
            player.print('No target group selected.', {r = 0.88, g = 0.88, b = 0.88})
            return
        end
        -- local index = index(tonumber(target_group_id))
        -- if not index then
        --   player.print("Selected group is no longer valid.", {r=0.88, g=0.88, b=0.88})
        --   return
        -- end
        local group = global.biter_command.active_unit_groups[tonumber(target_group_id)]
        if group and group.group.valid then
            comm_functions[name](group.group, player)
        else
            refresh_groups(player)
        end
        return
    end

    if comm_global_functions[name] then
        comm_global_functions[name](player)
        return
    end
end

local function refresh_panel()
    for _, player in pairs(game.connected_players) do
        if player.gui.screen['biter_panel'] then
            refresh_groups(player)
        end
    end
end

local function on_player_joined_game(event)
    top_button(game.players[event.player_index])
end

local function on_unit_group_created(event)
    if event and event.group and event.group.valid then
        global.biter_command.active_unit_groups[event.group.group_number] = {id = event.group.group_number, group = event.group}
        refresh_panel()
    end
end

local function on_unit_removed_from_group(event)
    if event and event.group and event.group.valid then
        if #event.group.members == 1 then
            global.biter_command.active_unit_groups[event.group.group_number] = nil
            refresh_panel()
        end
    end
end

event.add(defines.events.on_unit_removed_from_group, on_unit_removed_from_group)
event.add(defines.events.on_unit_group_created, on_unit_group_created)
event.add(defines.events.on_player_joined_game, on_player_joined_game)
event.add(defines.events.on_gui_click, on_gui_click)