local Alert = require 'utils.alert'
local Balance = require 'maps.chronosphere.balance'
local Event_functions = require 'maps.chronosphere.event_functions'
local Chrono_table = require 'maps.chronosphere.table'
local Minimap = require "maps.chronosphere.minimap"
local Ores = require "maps.chronosphere.ores"
local World_functions = require "maps.chronosphere.world_functions"

local Public = {}

function Public.on_player_changed_position(event)
	local objective = Chrono_table.get_table()
	if objective.world.id == 1 and objective.world.variant.id == 11 then --lava planet
		Event_functions.lava_planet(event)
	end
end

function Public.on_research_finished(event)
	Event_functions.research_loot(event)
	Event_functions.flamer_nerfs()
	Event_functions.mining_buffs(event)
end

function Public.on_player_mined_entity(event)
	local objective = Chrono_table.get_table()
	local entity = event.entity
	if not entity.valid then return end
	if entity.type == "tree" then
		Event_functions.tree_loot()
		if objective.world.id == 4 then --choppy planet
			Event_functions.trap(entity, false)
			Event_functions.choppy_loot(event)
		end
	end
	if entity.name == "rock-huge" or entity.name == "rock-big" or entity.name == "sand-rock-big" then
		if objective.world.id == 3 then --rocky worlds
			event.buffer.clear()
		elseif objective.world.id == 5 then --maze worlds
			--nothing
		else
			Ores.prospect_ores(entity, entity.surface, entity.position)
		end
	elseif World_functions.is_scrap(entity.name) then
		Event_functions.scrap_loot(event)
		event.buffer.clear()
	end
end

function Public.pre_player_mined_item(event)
	local objective = Chrono_table.get_table()
	if objective.world.id == 3 then --rocky worlds
		if event.entity.name == "rock-huge" or event.entity.name == "rock-big" or event.entity.name == "sand-rock-big" then
			Event_functions.trap(event.entity, false)
			event.entity.destroy()
			Event_functions.rocky_loot(event)
		end
	end
end

function Public.on_pre_player_left_game(event)
	local objective = Chrono_table.get_table()
	local playertable = Chrono_table.get_player_table()
	local player = game.players[event.player_index]
	if player.controller_type == defines.controllers.editor then player.toggle_map_editor() end
	if player.character then
		playertable.offline_players[#playertable.offline_players + 1] = {index = event.player_index, tick = game.tick}
	end
end

function Public.on_player_joined_game(event)
	local objective = Chrono_table.get_table()
	local playertable = Chrono_table.get_player_table()
	local player = game.players[event.player_index]
	if not playertable.flame_boots[event.player_index] then
		playertable.flame_boots[event.player_index] = {}
	end
	playertable.flame_boots[event.player_index] = {fuel = 1}
	if not playertable.flame_boots[event.player_index].steps then playertable.flame_boots[event.player_index].steps = {} end

	local surface = game.surfaces[objective.active_surface_index]

	if player.online_time == 0 then
		player.teleport(surface.find_non_colliding_position("character", game.forces.player.get_spawn_position(surface), 32, 0.5), surface)
		for item, amount in pairs(Balance.starting_items) do
			player.insert({name = item, count = amount})
		end
	end

	if player.surface.index ~= objective.active_surface_index and player.surface.name ~= "cargo_wagon" then
		player.character = nil
		player.set_controller({type=defines.controllers.god})
		player.create_character()
		player.teleport(surface.find_non_colliding_position("character", game.forces.player.get_spawn_position(surface), 32, 0.5), surface)
		for item, amount in pairs(Balance.starting_items) do
			player.insert({name = item, count = amount})
		end
	end

	local tile = surface.get_tile(player.position)
	if tile.valid then
		if tile.name == "out-of-map" then
			player.teleport(surface.find_non_colliding_position("character", game.forces.player.get_spawn_position(surface), 32, 0.5), surface)
		end
	end
	Minimap.update_surface(player)
end

function Public.on_entity_died(event)
	local objective = Chrono_table.get_table()
	if event.entity.type == "tree" and objective.world.id == 4 then --choppy planet
		if event.cause then
			if event.cause.valid then
				if event.cause.force.name ~= "enemy" then
					Event_functions.trap(event.entity, false)
				end
			end
		end
		-- if not event.entity.valid then return end
		-- for _, entity in pairs (event.entity.surface.find_entities_filtered({area = {{event.entity.position.x - 4, event.entity.position.y - 4},{event.entity.position.x + 4, event.entity.position.y + 4}}, name = "fire-flame-on-tree"})) do
		-- 	if entity.valid then entity.destroy() end
		-- end
		--return
	end
	local entity = event.entity
	if not entity.valid then return end
	if entity.type == "unit" and entity.force == "enemy" then
		objective.active_biters[entity.unit_number] = nil

	end
	if entity.type == "rocket-silo" and entity.force.name == "enemy" then
		Event_functions.danger_silo(entity)
	end
	if entity.force.name == "scrapyard" and entity.name == "gun-turret" then
		if (objective.world.id == 2 and objective.world.variant.id == 2) or objective.world.id == 5 then --danger + hedge maze
			Event_functions.trap(entity, true)
		end
	end
	if entity.force.name == "enemy" then
		if entity.type == "unit-spawner" then
			Event_functions.spawner_loot(entity.surface, entity.position)
			if objective.world.id == 8 then
				Ores.prospect_ores(entity, entity.surface, entity.position)
			end
			if event.cause and (event.cause.name == "artillery-turret" or event.cause.name == "artillery-wagon") then
				Event_functions.nuclear_artillery(entity, event.cause)
			end
		else
			Event_functions.biter_loot(event)
			if objective.world.id == 8 then
				Event_functions.swamp_loot(event)
			end
		end
	end
	if entity.force.index == 3 then
		event.loot.clear()
		if objective.world.id == 2 and objective.world.variant.id == 3 and entity.type == "container" then --RUR robot spawns
			Event_functions.trap(entity, true)
		end
		if event.cause then
			if event.cause.valid then
				if event.cause.force.index == 2 then
					Event_functions.shred_simple_entities(entity)
				end
			end
		end
	end
end

local function protect_entity(event)
	local objective = Chrono_table.get_table()
	if event.entity.force.index ~= 1 then return end --Player Force
	if Event_functions.isprotected(event.entity) then
		if event.cause then
			if event.cause == objective.comfylatron or event.entity == objective.comfylatron then
				return
			end
			if event.cause.force.index == 2 or event.cause.force.name == "scrapyard" then
					Event_functions.set_objective_health(event.final_damage_amount)
			end
		elseif objective.world.id == 2 and objective.world.variant.id == 2 then
			Event_functions.set_objective_health(event.final_damage_amount)
		end
		if not event.entity.valid then return end
		event.entity.health = event.entity.health + event.final_damage_amount
	end
	if event.entity.name == "character" then
		if objective.upgrades[25] > 0 and event.damage_type.name == "poison" then
			event.entity.health = event.entity.health + event.final_damage_amount * (0.25 * objective.upgrades[25])
		end
	end
end

function Public.on_entity_damaged(event)
	if not event.entity.valid then return end
	protect_entity(event)
	if not event.entity.valid then return end
	if not event.entity.health then return end
	Event_functions.biters_chew_rocks_faster(event)
	if event.entity.force.name == "enemy" then
		Event_functions.biter_immunities(event)
	end
end

function Public.on_built_entity(event)
	local entity = event.created_entity
	if not entity or not entity.valid then return end
	local objective = Chrono_table.get_table()
	if entity.type == "entity-ghost" then
		entity.time_to_live = game.forces.player.ghost_time_to_live
	end
	if entity.name == "spidertron" then
		if objective.world.id ~= 7 or entity.surface.name == "cargo_wagon" then
			entity.destroy()
			local player = game.players[event.player_index]
			Alert.alert_player_warning(player, 8, {"chronosphere.spidertron_not_allowed"})
			player.insert({name = "spidertron", count = 1})
		end
	end
end

function Public.on_pre_player_died(event)
	local objective = Chrono_table.get_table()
	local player = game.players[event.player_index]
	local surface = player.surface
	local poisons = surface.count_entities_filtered{position = player.position, radius = 10, name = "poison-cloud"}
	if poisons > 0 then
		objective.poison_mastery_unlocked = objective.poison_mastery_unlocked + 1
	end
	if objective.poison_mastery_unlocked == 10 then
		game.print({"chronosphere.message_poison_mastery_unlock"}, {r=0.98, g=0.66, b=0.22})
	end
end

return Public