local Event = require "utils.event"
local UserGroups = require "user_groups"
local Utils = require "utils.utils"
local Game = require 'utils.game'

function allowed_to_nuke(player)
  if type(player) == "table" then
  return player.admin or UserGroups.is_regular(player.name) or ((player.online_time / 216000) > global.scenario.config.nuke_control.nuke_min_time_hours)
  elseif type(player) == "number" then
    return allowed_to_nuke(Game.get_player_by_index(player))
  end
end

local function ammo_changed(event)
  local player = Game.get_player_by_index(event.player_index)
    if allowed_to_nuke(player) then return end
  local nukes = player.remove_item({name="atomic-bomb", count=1000})
  if nukes > 0 then
    game.print(player.name .. " tried to use a nuke, but instead dropped it on his foot.")

    local character = player.character
    if character and character.valid then
      for _,p in ipairs(game.connected_players) do
        if p ~= player then
          p.add_custom_alert(character, {type = 'item', name = 'atomic-bomb'}, player.name, true)
        end
      end
    end
    player.character.health = 0
  end
end

local function on_player_deconstructed_area(event)
  local player = Game.get_player_by_index(event.player_index)
    if allowed_to_nuke(player) then return end
    player.remove_item({name="deconstruction-planner", count=1000})

    --Make them think they arent noticed
    Utils.print_except(player.name .. " tried to deconstruct something, but instead deconstructed themself.", player)
    player.print("Only regulars can mark things for deconstruction, if you want to deconstruct something you may ask an admin to promote you.")

    local character = player.character
    if character and character.valid then
      for _,p in ipairs(game.connected_players) do
        if p ~= player then
          p.add_custom_alert(character, {type = 'item', name = 'deconstruction-planner'}, player.name, true)
        end
      end
    end
    character.health = 0

    local area = event.area
    local left_top, right_bottom = area.left_top, area.right_bottom
    if left_top.x == right_bottom.x and left_top.y == right_bottom.y then
      return
    end

    local entities = player.surface.find_entities_filtered{area = area, force = player.force}
    if #entities > 1000 then
      Utils.print_admins("Warning! " .. player.name .. " just tried to deconstruct " .. tostring(#entities) .. " entities!")
    end
    for _,entity in pairs(entities) do
      if entity.valid and entity.to_be_deconstructed(Game.get_player_by_index(event.player_index).force) then
        entity.cancel_deconstruction(Game.get_player_by_index(event.player_index).force)
      end
    end
end

local function item_not_sanctioned(item)
  local name = item.name
  return (
    name:find("capsule") or
    name == "cliff-explosives" or
    name == "raw-fish" or
    name == "discharge-defense-remote"
  )
end

global.entities_allowed_to_bomb = {
  ["stone-wall"] = true,
  ["transport-belt"] = true,
  ["fast-transport-belt"] = true,
  ["express-transport-belt"] = true,
  ["construction-robot"] = true,
  ["player"] = true,
  ["gun-turret"] = true,
  ["laser-turret"] = true,
  ["flamethrower-turret"] = true,
  ["rail"] = true,
  ["rail-chain-signal"] = true,
  ["rail-signal"] = true,
  ["tile-ghost"] = true,
  ["entity-ghost"] = true,
  ["gate"] = true,
  ["electric-pole"] = true,
  ["small-electric-pole"] = true,
  ["medium-electric-pole"] = true,
  ["big-electric-pole"] = true,
  ["logistic-robot"] = true,
  ["defender"] = true,
  ["destroyer"] = true,
  ["distractor"] = true
}

local function entity_allowed_to_bomb(entity)
  return global.entities_allowed_to_bomb[entity.name]
end
global.players_warned = {}
local function on_capsule_used(event)
  local item = event.item
  local player = Game.get_player_by_index(event.player_index)

  if not player or not player.valid or 
    (global.scenario.config.nuke_control.enable_autokick and global.scenario.config.nuke_control.enable_autoban) then
    return
  end

  if item.name == 'artillery-targeting-remote' then
    player.surface.create_entity{name = 'flying-text', text = player.name, color = player.color, position = event.position}
  end

  if item_not_sanctioned(item) then return end

  if (not allowed_to_nuke(player)) then
    local area = {{event.position.x-5, event.position.y-5}, {event.position.x+5, event.position.y+5}}
    local count = 0
    local entities = player.surface.find_entities_filtered{force=player.force, area=area}
    for _,e in pairs(entities) do
      if not entity_allowed_to_bomb(e) then count = count + 1 end
    end
    if count > 8 then
      if global.players_warned[event.player_index] then
        if global.scenario.config.nuke_control.enable_autokick then  
          game.ban_player(player, string.format("Damaged %i entities with %s. This action was performed automatically. If you want to contest this ban please visit redmew.com/discord.", count, event.item.name))
        end
      else
        global.players_warned[event.player_index] = true
        if global.scenario.config.nuke_control.enable_autoban then
          game.print(player, string.format("Damaged %i entities with %s -Antigrief", count, event.item.name))
        end
      end
    end
  end
end

Event.add(defines.events.on_player_ammo_inventory_changed, ammo_changed)
Event.add(defines.events.on_player_deconstructed_area, on_player_deconstructed_area)
--Event.add(defines.events.on_player_mined_entity, on_player_mined_item)
Event.add(defines.events.on_player_used_capsule, on_capsule_used)