local Public = {}

local BiterRaffle = require "functions.biter_raffle"
local LootRaffle = require "functions.loot_raffle"
local Get_noise = require "utils.get_noise"

local table_shuffle_table = table.shuffle_table
local table_insert = table.insert
local table_remove = table.remove
local math_random = math.random
local math_abs = math.abs
local math_floor = math.floor

function Public.get_dungeon_evolution_factor(surface_index)
	local e = global.dungeons.depth[surface_index] * 0.0005
	if global.dungeons.tiered then
		e = math.min(e, surface_index * 0.05 - 0.05)
	end
	return e
end

local function blacklist(surface_index, special)
	local evolution_factor = Public.get_dungeon_evolution_factor(surface_index)
	local blacklist = {}
	--general unused items on dungeons
	blacklist["cliff-explosives"] = true
	--items that would trivialize stuff if dropped too early
	if global.dungeons.item_blacklist and not special then
		if evolution_factor < 0.9 then
			blacklist["discharge-defense-equipment"] = true
			blacklist["power-armor-mk2"] = true
			blacklist["fusion-reactor-equipment"] = true
			blacklist["rocket-silo"] = true
			blacklist["discharge-defense-remote"] = true
		end
		if evolution_factor < 0.7 then
			blacklist["energy-shield-mk2-equipment"] = true
			blacklist["personal-laser-defense-equipment"] = true
			blacklist["personal-roboport-mk2-equipment"] = true
			blacklist["battery-mk2-equipment"] = true
			blacklist["nuclear-reactor"] = true
			blacklist["artillery-turret"] = true
			blacklist["artillery-wagon"] = true
			blacklist["power-armor"] = true
		end
		if evolution_factor < 0.4 then
			blacklist["steam-turbine"] = true
			blacklist["heat-exchanger"] = true
			blacklist["heat-pipe"] = true
			blacklist["express-loader"] = true
			blacklist["modular-armor"] = true
			blacklist["solar-panel-equipment"] = true
			blacklist["energy-shield-equipment"] = true
			blacklist["battery-equipment"] = true
			blacklist["solar-panel-equipment"] = true
		end
	end
	return blacklist
end

function Public.roll_spawner_name()
	if math_random(1, 3) == 1 then
		return "spitter-spawner"
	end
	return "biter-spawner"
end

function Public.roll_worm_name(surface_index)
	return BiterRaffle.roll("worm", Public.get_dungeon_evolution_factor(surface_index))
end

function Public.get_crude_oil_amount(surface_index)
	local amount = math_random(200000, 400000) + Public.get_dungeon_evolution_factor(surface_index) * 500000
	if global.dungeons.tiered then amount = amount / 4 end
	return amount
end

function Public.get_common_resource_amount(surface_index)
	local amount = math_random(350, 700) + Public.get_dungeon_evolution_factor(surface_index) * 16000
	if global.dungeons.tiered then amount = amount / 4 end
	return amount
end

function Public.common_loot_crate(surface, position, special)
	local item_stacks = LootRaffle.roll(Public.get_dungeon_evolution_factor(surface.index) * 4000 + math_random(8, 16), 16, blacklist(surface.index, special))
	local container = surface.create_entity({name = "wooden-chest", position = position, force = "neutral"})
	for _, item_stack in pairs(item_stacks) do
		container.insert(item_stack)
	end
	container.minable = false
end

function Public.uncommon_loot_crate(surface, position, special)
	local item_stacks = LootRaffle.roll(Public.get_dungeon_evolution_factor(surface.index) * 8000 + math_random(32, 64), 16, blacklist(surface.index, special))
	local container = surface.create_entity({name = "iron-chest", position = position, force = "neutral"})
	for _, item_stack in pairs(item_stacks) do
		container.insert(item_stack)
	end
	container.minable = false
end

function Public.rare_loot_crate(surface, position, special)
	local item_stacks = LootRaffle.roll(Public.get_dungeon_evolution_factor(surface.index) * 16000 + math_random(128, 256), 32, blacklist(surface.index, special))
	local container = surface.create_entity({name = "steel-chest", position = position, force = "neutral"})
	for _, item_stack in pairs(item_stacks) do
		container.insert(item_stack)
	end
	container.minable = false
end

function Public.epic_loot_crate(surface, position, special)
	local item_stacks = LootRaffle.roll(Public.get_dungeon_evolution_factor(surface.index) * 32000 + math_random(512, 1024), 48, blacklist(surface.index, special))
	local container = surface.create_entity({name = "steel-chest", position = position, force = "neutral"})
	for _, item_stack in pairs(item_stacks) do
		container.insert(item_stack)
	end
	container.minable = false
end

function Public.crash_site_chest(surface, position, special)
	local item_stacks = LootRaffle.roll(Public.get_dungeon_evolution_factor(surface.index) * 12000 + math_random(160, 320), 48, blacklist(surface.index, special))
	local container = surface.create_entity({name = "crash-site-chest-" .. math_random(1, 2), position = position, force = "neutral"})
	for _, item_stack in pairs(item_stacks) do
		container.insert(item_stack)
	end
end

function Public.market(surface, position)
	local offers = {
		{price = {{'pistol', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(1,4)}},
		{price = {{'submachine-gun', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(5,20)}},
		{price = {{'shotgun', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(5,18)}},
		{price = {{'combat-shotgun', 1}}, offer = {type = 'give-item', item = "steel-plate", count = math_random(3,10)}},
		{price = {{'rocket-launcher', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(4,10)}},
		{price = {{'flamethrower', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(5,18)}},
		{price = {{'light-armor', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(5,20)}},
		{price = {{'heavy-armor', 1}}, offer = {type = 'give-item', item = "steel-plate", count = math_random(5,20)}},
		{price = {{'modular-armor', 1}}, offer = {type = 'give-item', item = "advanced-circuit", count = math_random(10,20)}},
		{price = {{'night-vision-equipment', 1}}, offer = {type = 'give-item', item = "steel-plate", count = math_random(1,4)}},
		{price = {{'solar-panel-equipment', 1}}, offer = {type = 'give-item', item = "copper-plate", count = math_random(10,25)}},
		{price = {{'red-wire', 100}}, offer = {type = 'give-item', item = "copper-cable", count = math_random(50,100)}},
		{price = {{'green-wire', 100}}, offer = {type = 'give-item', item = "copper-cable", count = math_random(50,100)}},
		{price = {{'empty-barrel', 10}}, offer = {type = 'give-item', item = "steel-plate", count = math_random(4,8)}},
		{price = {{'arithmetic-combinator', 10}}, offer = {type = 'give-item', item = "electronic-circuit", count = math_random(10,25)}},
		{price = {{'decider-combinator', 10}}, offer = {type = 'give-item', item = "electronic-circuit", count = math_random(10,25)}},
		{price = {{'constant-combinator', 10}}, offer = {type = 'give-item', item = "electronic-circuit", count = math_random(3,12)}},
		{price = {{'power-switch', 10}}, offer = {type = 'give-item', item = "electronic-circuit", count = math_random(3,12)}},
		{price = {{'programmable-speaker', 10}}, offer = {type = 'give-item', item = "electronic-circuit", count = math_random(8,30)}},
		{price = {{'belt-immunity-equipment', 1}}, offer = {type = 'give-item', item = "advanced-circuit", count = math_random(1,3)}},
		{price = {{'discharge-defense-remote', 1}}, offer = {type = 'give-item', item = "electronic-circuit", count = 1}},
		{price = {{'rail-signal', 10}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(20,40)}},
		{price = {{'rail-chain-signal', 10}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(20,40)}},
		{price = {{'train-stop', 10}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(50,100)}},
		{price = {{'locomotive', 1}}, offer = {type = 'give-item', item = "steel-plate", count = math_random(20,40)}},
		{price = {{'cargo-wagon', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(20,40)}},
		{price = {{'fluid-wagon', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(20,40)}},
		{price = {{'car', 1}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(10,20)}},
		{price = {{'radar', 10}}, offer = {type = 'give-item', item = "iron-plate", count = math_random(10,20)}},
		{price = {{'cannon-shell', 10}}, offer = {type = 'give-item', item = "steel-plate", count = math_random(5,10)}},
		{price = {{'uranium-cannon-shell', 10}}, offer = {type = 'give-item', item = "uranium-238", count = math_random(5,10)}},

	}
	table.shuffle_table(offers)
	local market = surface.create_entity({name = "market", position = position, force = "neutral"})
	market.destructible = false
	market.minable = false
	for i = 1, math.random(6, 10), 1 do
		market.add_market_item(offers[i])
	end
end

function Public.add_room_loot_crates(surface, room)
	if not room.room_border_tiles[1] then return end
	for key, tile in pairs(room.room_tiles) do
		if math_random(1, 384) == 1 then
			Public.common_loot_crate(surface, tile.position)
		else
			if math_random(1, 1024) == 1 then
				Public.uncommon_loot_crate(surface, tile.position)
			else
				if math_random(1, 4096) == 1 then
					Public.rare_loot_crate(surface, tile.position)
				else
					if math_random(1, 16384) == 1 then
						Public.epic_loot_crate(surface, tile.position)
					end
				end
			end
		end
	end
end

function Public.set_spawner_tier(spawner, surface_index)
	local tier = math_floor(Public.get_dungeon_evolution_factor(surface_index) * 8 - math_random(0, 8)) + 1
	if tier < 1 then tier = 1 end
	global.dungeons.spawner_tier[spawner.unit_number] = tier
	--[[
	rendering.draw_text{
		text = "-Tier " .. tier .. "-",
		surface = spawner.surface,
		target = spawner,
		target_offset = {0, -2.65},
		color = {25, 0, 100, 255},
		scale = 1.25,
		font = "default-game",
		alignment = "center",
		scale_with_zoom = false
	}
	]]
end

function Public.spawn_random_biter(surface, position)
	local name = BiterRaffle.roll("mixed", Public.get_dungeon_evolution_factor(surface.index))
	local non_colliding_position = surface.find_non_colliding_position(name, position, 16, 1)
	local unit
	if non_colliding_position then
		unit = surface.create_entity({name = name, position = non_colliding_position, force = global.enemy_forces[surface.index]})
	else
		unit = surface.create_entity({name = name, position = position, force = global.enemy_forces[surface.index]})
	end
	unit.ai_settings.allow_try_return_to_spawner = false
	unit.ai_settings.allow_destroy_when_commands_fail = false
end

function Public.place_border_rock(surface, position)
	local vectors = {{0, -1}, {0, 1}, {1, 0}, {-1, 0}}
	table_shuffle_table(vectors)
	local key = false
	for k, v in pairs(vectors) do
		local tile = surface.get_tile({position.x + v[1], position.y + v[2]})
		if tile.name == "out-of-map" then
			key = k
			break
		end
	end
	local pos = {x = position.x + 0.5, y = position.y + 0.5}
	if key then
		pos = {pos.x + vectors[key][1] * 0.45, pos.y + vectors[key][2] * 0.45}
	end
	surface.create_entity({name = "rock-big", position = pos})
end

function Public.mining_events(entity)
	if math_random(1, 16) == 1 then Public.spawn_random_biter(entity.surface, entity.position) return end
	if math_random(1, 24) == 1 then Public.common_loot_crate(entity.surface, entity.position) return end
	if math_random(1, 128) == 1 then Public.uncommon_loot_crate(entity.surface, entity.position) return end
	if math_random(1, 512) == 1 then Public.rare_loot_crate(entity.surface, entity.position) return end
	if math_random(1, 1024) == 1 then Public.epic_loot_crate(entity.surface, entity.position) return end
end

function Public.draw_spawn(surface)
	local spawn_size = global.dungeons.spawn_size

	for _, e in pairs(surface.find_entities({{spawn_size * -1, spawn_size * -1}, {spawn_size, spawn_size}})) do
		e.destroy()
	end

	local tiles = {}
	local i = 1
	for x = spawn_size * -1, spawn_size, 1 do
		for y = spawn_size * -1, spawn_size, 1 do
			local position = {x = x, y = y}
			if math_abs(position.x) < 2 or math_abs(position.y) < 2 then
				tiles[i] = {name = "dirt-7", position = position}
				i = i + 1
				tiles[i] = {name = "stone-path", position = position}
				i = i + 1
			else
				tiles[i] = {name = "dirt-7", position = position}
				i = i + 1
			end
		end
	end
	surface.set_tiles(tiles, true)

	local tiles = {}
	local i = 1
	for x = -2, 2, 1 do
		for y = -2, 2, 1 do
			local position = {x = x, y = y}
			if math_abs(position.x) > 1 or math_abs(position.y) > 1 then
				tiles[i] = {name = "black-refined-concrete", position = position}
				i = i + 1
			else
				tiles[i] = {name = "purple-refined-concrete", position = position}
				i = i + 1
			end
		end
	end
	surface.set_tiles(tiles, true)

	local tiles = {}
	local i = 1
	for x = spawn_size * -1, spawn_size, 1 do
		for y = spawn_size * -1, spawn_size, 1 do
			local position = {x = x, y = y}
			local r = math.sqrt(position.x ^ 2 + position.y ^ 2)
			if r < 2 then
				tiles[i] = {name = "purple-refined-concrete", position = position}
				--tiles[i] = {name = "water-mud", position = position}
				i = i + 1
			else
				if r < 2.5 then
					tiles[i] = {name = "black-refined-concrete", position = position}
					--tiles[i] = {name = "water-shallow", position = position}
					i = i + 1
				else
					if r < 4.5 then
						tiles[i] = {name = "dirt-7", position = position}
						i = i + 1
						tiles[i] = {name = "concrete", position = position}
						i = i + 1
					end
				end
			end
		end
	end
	surface.set_tiles(tiles, true)

	local decoratives = {"brown-hairy-grass", "brown-asterisk", "brown-fluff", "brown-fluff-dry", "brown-asterisk", "brown-fluff", "brown-fluff-dry"}
	local a = spawn_size * -1 + 1
	local b = spawn_size - 1
	for _, decorative_name in pairs(decoratives) do
		local seed = game.surfaces[surface.index].map_gen_settings.seed + math_random(1, 1000000)
		for x = a, b, 1 do
			for y = a, b, 1 do
				local position = {x = x + 0.5, y = y + 0.5}
				if surface.get_tile(position).name == "dirt-7" or math_random(1, 5) == 1 then
					local noise = Get_noise("decoratives", position, seed)
					if math_abs(noise) > 0.37 then
						surface.create_decoratives{check_collision = false, decoratives = {{name = decorative_name, position = position, amount = math.floor(math.abs(noise * 3)) + 1}}}
					end
				end
			end
		end
	end

	local entities = {}
	local i = 1
	for x = spawn_size * -1 - 16, spawn_size + 16, 1 do
		for y = spawn_size * -1 - 16, spawn_size + 16, 1 do
			local position = {x = x, y = y}
			if position.x <= spawn_size and position.y <= spawn_size and position.x >= spawn_size * -1 and position.y >= spawn_size * -1 then
				if position.x == spawn_size then
					entities[i] = {name = "rock-big", position = {position.x + 0.95, position.y}}
					i = i + 1
				end
				if position.y == spawn_size then
					entities[i] = {name = "rock-big", position = {position.x, position.y + 0.95}}
					i = i + 1
				end
				if position.x == spawn_size * -1 or position.y == spawn_size * -1 then
					entities[i] = {name = "rock-big", position = position}
					i = i + 1
				end
			end
		end
	end

	for k, e in pairs(entities) do
		if k % 3 > 0 then surface.create_entity(e) end
	end

	if global.dungeons.tiered then
		if surface.index > 2 then
			table.insert(global.dungeons.transport_surfaces, surface.index)
			global.dungeons.transport_chests_inputs[surface.index] = {}
			for i = 1, 2, 1 do
				local chest = surface.create_entity({name = "blue-chest", position = {-12 + i * 8, -4}, force = "player"})
				global.dungeons.transport_chests_inputs[surface.index][i] = chest
				chest.destructible = false
				chest.minable = false
			end
			global.dungeons.transport_poles_outputs[surface.index] = {}
			for i = 1, 2, 1 do
				local pole = surface.create_entity({name = "constant-combinator", position = {-15 + i * 10, -5}, force = "player"})
				global.dungeons.transport_poles_outputs[surface.index][i] = pole
				pole.destructible = false
				pole.minable = false
			end
		end
		global.dungeons.transport_chests_outputs[surface.index] = {}
		for i = 1, 2, 1 do
			local chest = surface.create_entity({name = "red-chest", position = {-12 + i * 8, 4}, force = "player"})
			global.dungeons.transport_chests_outputs[surface.index][i] = chest
			chest.destructible = false
			chest.minable = false
		end
		global.dungeons.transport_poles_inputs[surface.index] = {}
		for i = 1, 2, 1 do
			local pole = surface.create_entity({name = "medium-electric-pole", position = {-15 + i * 10, 5}, force = "player"})
			global.dungeons.transport_poles_inputs[surface.index][i] = pole
			pole.destructible = false
			pole.minable = false
		end
	end

	local trees = { "dead-grey-trunk", "dead-tree-desert", "dry-hairy-tree", "dry-tree", "tree-04"}
	local size_of_trees = #trees
	local r = 4
	for x = spawn_size * -1, spawn_size, 1 do
		for y = spawn_size * -1, spawn_size, 1 do
			local position = {x = x + 0.5, y = y + 0.5}
			if position.x > 5 and position.y > 5 and math_random(1, r) == 1 then
				surface.create_entity({name = trees[math_random(1, size_of_trees)], position = position})
			end
			if position.x <= -4 and position.y <= -4 and math_random(1, r) == 1 then
				surface.create_entity({name = trees[math_random(1, size_of_trees)], position = position})
			end
			if position.x > 5 and position.y <= -4 and math_random(1, r) == 1 then
				surface.create_entity({name = trees[math_random(1, size_of_trees)], position = position})
			end
			if position.x <= -4 and position.y > 5 and math_random(1, r) == 1 then
				surface.create_entity({name = trees[math_random(1, size_of_trees)], position = position})
			end
		end
	end
	surface.set_tiles(tiles, true)
end

return Public