local BUILDINGS_ver = '7.3.6'
local QC_Mod = true



require ("util")
require ("stdlib/event/event")
require ("control_bio_cannon")
require ("control_artifact_collector")
require ("control_acs")
require ("control_terraforming")
require ("control_evolution")


if remote.interfaces.EvoGUI then
	require ("libs/EvoGUI")
end


--- Settup Settings
if not global.NE_Buildings then global.NE_Buildings = {} end
if not global.NE_Buildings.Settings then global.NE_Buildings.Settings = {} end
global.NE_Buildings.Settings.Battle_Marker = settings.startup["NE_Battle_Marker"].value



---------------------------------------------				 
local function On_Init()
 writeDebug("NE Buildings Initialize")
 
 	--- Artifact Collector
	global.world = {}
	global.world.itemCollectorLookup = {}
	global.world.itemCollectorEvents = {}
	

---- Evolution_MOD
	if global.Evolution_MOD == nil then
      global.Evolution_MOD = {}
	end
	
	---- Alien Control Initialization ----	
	if not global.beacons then
		global.beacons = {}
	end
	if not global.minds then
		global.minds = {}
	end
	if not global.hiveminds then
		global.hiveminds = {} -- Bases / Spawners
	end

	--- Alien_Control_Station Difficulty settings	
	if global.NE_Buildings.Settings.Conversion_Difficulty == 1 then
		global.minds.difficulty = 3 -- Easy difficulty
	elseif global.NE_Buildings.Settings.Conversion_Difficulty == 2 then
		global.minds.difficulty = 5 -- Normal 
	else 
		global.minds.difficulty = 10 -- Hard
	end

	---- Terraforming Initialization ----	
	if not global.numTerraformingStations then
          global.numTerraformingStations = 0
    elseif global.numTerraformingStations < 0 then
          global.numTerraformingStations = 0
	end
	
	if not global.factormultiplier then
		global.factormultiplier = 0
    elseif global.factormultiplier < 0 then
       global.factormultiplier = 0
	end	

	if not global.Total_TerraformingStations_Evo_Deduction then
        global.Total_TerraformingStations_Evo_Deduction = 0
    elseif global.Total_TerraformingStations_Evo_Deduction < 0 then
        global.Total_TerraformingStations_Evo_Deduction = 0
	end

	
	if global.deduction_constant == nil or global.deduction_constant == 0 then
		global.deduction_constant = 0.00025 -------- DEDUCTION CONSTANT
	end
	
	---- Living Wall Init
	if global.Living_Walls_Table == nil then
          global.Living_Walls_Table = {}
	end

	
	-- enable researched recipes
	for i, force in pairs(game.forces) do
		force.reset_technologies()
		force.reset_recipes()
	end
	
	if global.Terraforming_Station_Table == nil then
			global.Terraforming_Station_Table = {}
	end

		
	--- Settup Settings
	if not global.NE_Buildings then global.NE_Buildings = {} end
	if not global.NE_Buildings.Settings then global.NE_Buildings.Settings = {} end

	global.NE_Buildings.Settings.Conversion_Difficulty = settings.startup["NE_Conversion_Difficulty"].value
	global.NE_Buildings.Settings.Battle_Marker = settings.startup["NE_Battle_Marker"].value
	global.NE_Buildings.Settings.Search_Distance = settings.startup["NE_Conversion_Search_Distance"].value
	global.NE_Buildings.Settings.Artifact_Collector_Radius = settings.startup["NE_Artifact_Collector_Radius"].value
	global.NE_Buildings.Settings.Artifact_Item_Count = settings.startup["NE_Artifact_Item_Count"].value

	
end
		 

---------------------------------------------				 
local function On_Load()



end


---------------------------------------------				 
local function On_Config_Change()

 	--- Artifact Collector
	if global.world == nil then
		global.world = {}
		global.world.itemCollectorLookup = {}
		global.world.itemCollectorEvents = {}
	end
	
	
	-- enable researched recipes
	for i, force in pairs(game.forces) do
		for _, tech in pairs(force.technologies) do
			if tech.researched then
				for _, effect in pairs(tech.effects) do
					if effect.type == "unlock-recipe" then          
						force.recipes[effect.recipe].enabled = true
					end
				end
			end
		end
	end
	
	if global.Terraforming_Station_Table == nil then
			global.Terraforming_Station_Table = {}
	end
	

	if game.active_mods["EndgameCombat"] or game.active_mods['EndgameCombat'] or game.active_mods[EndgameCombat] then	
		
		for i, player in pairs(game.players) do
			player.print(tostring("You have Endgame Combat Installed, not compatible with Natural Evolution Buildings"))
		end
	end
	

	--- Settup Settings	
	if not global.NE_Buildings then global.NE_Buildings = {} end
	if not global.NE_Buildings.Settings then global.NE_Buildings.Settings = {} end

	global.NE_Buildings.Settings.Conversion_Difficulty = settings.startup["NE_Conversion_Difficulty"].value
	global.NE_Buildings.Settings.Battle_Marker = settings.startup["NE_Battle_Marker"].value
	global.NE_Buildings.Settings.Search_Distance = settings.startup["NE_Conversion_Search_Distance"].value
	global.NE_Buildings.Settings.Artifact_Collector_Radius = settings.startup["NE_Artifact_Collector_Radius"].value
	global.NE_Buildings.Settings.Artifact_Item_Count = settings.startup["NE_Artifact_Item_Count"].value

	
end


---------------------------------------------
local function On_Built(event)

local entity = event.created_entity
local surface = entity.surface
local position = entity.position	
local force = entity.force


	
	--- Artifact Collector	
	
    if (entity.name == "Artifact-collector-area") then --or (entity.name == "item-collector-base-rampant") then

	local x = entity.position.x
	local y = entity.position.y
	local chest

	if (entity.name == "Artifact-collector-area") then
	    entity.destroy()
	    chest = surface.create_entity({name = "Artifact-collector",
					   position = position,
					   force = force})
		position_r = {position.x + 1, position.y}
	    entity = surface.create_entity({name="Artifact-collector_r",
					    position = position_r,
					    force = force})

	else	    
	    local ghosts = surface.find_entities_filtered({name="entity-ghost",
							   area={{x=x-1,
								  y=y-1},
							       {x=x+1,
								y=y+1}},
							   limit=1})
	    if (#ghosts > 0) then
		local ghost = ghosts[1]
		if (ghost.ghost_name == "Artifact-collector") then
		    local conflicts
		    conflicts, chest = ghost.revive()
		end
	    else
		chest = surface.create_entity({name = "Artifact-collector",
					       position = position,
					       force = force})
	    end
	end
	if chest and chest.valid then
	    local pair = { chest, entity }
	    global.world.itemCollectorLookup[chest.unit_number] = pair
	    global.world.itemCollectorLookup[entity.unit_number] = pair
	    entity.backer_name = ""
	end
    end	

	
	--- Living Wall built
	if entity.valid and entity.name == "ne-living-wall" then
		if global.Living_Walls_Table == nil then
          global.Living_Walls_Table = {}
		end
		writeDebug("Living Wall has been built")				

		local Created_L_Wall = event.created_entity
		
		table.insert(global.Living_Walls_Table, Created_L_Wall)
		
	end

	
	--- Alien Control Station has been built	
	if entity.valid and entity.name == "AlienControlStation_Area" then
	local newAlienControlStation
	local surface = event.created_entity.surface
	local force = event.created_entity.force
	
	writeDebug("ACS has been built")			
	writeDebug("The ACS Difficulty is: " .. global.minds.difficulty)  					
			
		newAlienControlStation = surface.create_entity({name = "AlienControlStation", position = event.created_entity.position, force = force})
		event.created_entity.destroy()

		table.insert(global.beacons, newAlienControlStation)
	end	


	
   --- Terraforming Station has been built
	if entity.valid and entity.name == "TerraformingStation_New" then
	
		if global.numTerraformingStations < 0 then
			global.numTerraformingStations = 0
		end

		global.numTerraformingStations = global.numTerraformingStations + 1
      
		global.factormultiplier = GetFactorPerTerraformingStation(global.numTerraformingStations)
		writeDebug("The the number of Terraforming Stations: " .. global.numTerraformingStations)
	  
		position_c = {position.x - 1.7, position.y + 1.7}
		T_Station_Container = surface.create_entity({name = "TerraformingStation_c", position = position_c, direction = event.created_entity.direction, force = force})
	
		T_Station_Container.minable = false
		T_Station_Container.set_request_slot({name = "Alien-Stimulant", count = "10"}, 1)		
	
		global.Terraforming_Station_Table[entity.unit_number] = {radar=entity, inventory=T_Station_Container}
	    writeDebug("The Radar # is: "..entity.unit_number)
		writeDebug("The Container # is: "..T_Station_Container.unit_number)
	  
	end   

	
		
	--- Bio Cannon (Hive Buster) has been built
	if entity.valid and entity.name == "Bio_Cannon_Area" then
	
	local New_Bio_Cannon
	local New_Bio_CannonR
	
	writeDebug("Bio Cannon has been built")				

		New_Bio_Cannon  = surface.create_entity({name = "Bio_Cannon", position = position, direction = event.created_entity.direction, force = force})
		New_Bio_CannonR = surface.create_entity({name = "Bio_Cannon_r", position = position, direction = event.created_entity.direction, force = force})

		New_Bio_Cannon.health = event.created_entity.health
		
		event.created_entity.destroy()

		
		New_Bio_CannonR.operable = false
		New_Bio_CannonR.destructible = false
		New_Bio_CannonR.minable = false
		
		if global.Bio_Cannon_Table == nil then
			global.Bio_Cannon_Table = {}
			Event.register(defines.events.on_tick, function(event) end)
		end

		table.insert(global.Bio_Cannon_Table, {New_Bio_Cannon,New_Bio_CannonR,0})
		
	end


	
end


---------------------------------------------
local function On_Remove(event)

	local entity = event.entity	

	
	  --- fully heal the items that need a refresh in the alien hatchery anyway to avoid having multiple item stacks for damaged items.
	if event.name ~= defines.events.on_entity_died then
		if (
				entity.name == "small-worm-turret-player" or
				entity.name == "medium-worm-turret-player" or
				entity.name == "big-worm-turret-player" or
				entity.name == "Natural_Evolution_Biter-Spawner" or
				entity.name == "Natural_Evolution_Spitter-Spawner" or
				entity.name == "ne-living-wall" 
			) then
			entity.health = 100000 -- Note: Just needs to be higher or identical to maxhealth
		end
	end
	
	--Artifact collector
    if (entity.name == "Artifact-collector") then -- or (entity.name == "item-collector-base-rampant") then
	local pair = global.world.itemCollectorLookup[entity.unit_number]
	if pair then
	    local chest = pair[1]
	    local dish = pair[2]

	    if chest and chest.valid then
		global.world.itemCollectorLookup[chest.unit_number] = nil
		if destroyed and (entity == chest) then
		    chest.die()
		elseif (entity ~= chest) then
		    chest.destroy()
		end
	    end
	    if dish and dish.valid then
		global.world.itemCollectorLookup[dish.unit_number] = nil
		if destroyed and (entity ~= dish) then
		    dish.die()
		elseif (entity ~= dish) then
		    dish.destroy()
		end
	    end
	end
    end
	

   --- Alien Control Station has been removed
	if entity.valid and entity.name == "AlienControlStation" then
		ACS_Remove()
	end

		
		--- Terraforming Station has been removed
	if entity.valid and entity.name == "TerraformingStation_New"   then
		
		if global.Terraforming_Station_Table[entity.unit_number] == nil then 
		--Legacy from 7.1.5 Should remove during next update.
			T_Count()

		else 
		
			global.Terraforming_Station_Table[entity.unit_number].inventory.destroy()
			global.Terraforming_Station_Table[entity.unit_number] = nil
			T_Count()	

		end
		
	end
	

end


---------------------------------------------
local function On_Death(event)

	local entity = event.entity	
	local surface = entity.surface
	local pos = entity.position	

	


  	--Artifact collector
    if (entity.name == "Artifact-collector") then -- or (entity.name == "item-collector-base-rampant") then
	local pair = global.world.itemCollectorLookup[entity.unit_number]
	if pair then
	    local chest = pair[1]
	    local dish = pair[2]

	    if chest and chest.valid then
		global.world.itemCollectorLookup[chest.unit_number] = nil
		if destroyed and (entity == chest) then
		    chest.die()
		elseif (entity ~= chest) then
		    chest.destroy()
		end
	    end
	    if dish and dish.valid then
		global.world.itemCollectorLookup[dish.unit_number] = nil
		if destroyed and (entity ~= dish) then
		    dish.die()
		elseif (entity ~= dish) then
		    dish.destroy()
		end
	    end
	end
    end
	
	--- Terraforming Station has been removed
	if entity.valid and entity.name == "TerraformingStation_New"   then
		
		if global.Terraforming_Station_Table[entity.unit_number] == nil then 
		--Legacy from 7.1.5 Should remove during next update.
			T_Count()

		else 
		
			global.Terraforming_Station_Table[entity.unit_number].inventory.destroy()
			global.Terraforming_Station_Table[entity.unit_number] = nil
			T_Count()	

		end
		
	end
	

   --- Alien Control Station has been removed
	if entity.valid and entity.name == "AlienControlStation" then
		ACS_Remove()
	end
	

 	--------- Spawner killed
	if entity.valid and (entity.type == "unit-spawner") and (entity.force == game.forces.enemy) and global.NE_Buildings.Settings.Battle_Marker then

			writeDebug("Enemy Spawner Killed")

			local force = event.force


			Battle_Marker = surface.create_entity({name = "battle_marker", position = pos, force = force})
			Battle_Marker.destructible = false		

	end

	
	--- Conversion Turret	
	if event.force ~= nil and entity.force.name == "enemy" and  entity.type	 == "unit" and event.cause and event.cause.type == "ammo-turret" then 
	
		local name = entity.name
		local inventory = event.cause.get_inventory(defines.inventory.turret_ammo)

		if inventory then

			local inventoryContent = inventory.get_contents()		
			local AmmoType
			local Ammo = 0
				
			if inventoryContent ~= nil then
				for n,a in pairs(inventoryContent) do
					AmmoType=n
					Ammo=a
				end
			end
				
			--writeDebug("Ammo Type: " .. AmmoType)
			--writeDebug("Ammo Count: " .. Ammo)
			if AmmoType == "basic-dart-magazine_c"  or AmmoType == "enhanced-dart-magazine_c"  or AmmoType == "firearm-magazine_c"  or AmmoType == "copper-bullet-magazine_c"  or AmmoType == "piercing-rounds-magazine_c"  or AmmoType == "uranium-rounds-magazine_c"  or AmmoType == "Biological-bullet-magazine_c" then
				Convert = surface.create_entity({name = name, position = pos, force = event.cause.force.name})
				Convert.health = entity.prototype.max_health / 4
			end
		end	

	end

--NEEDS TO BE FIXED

	--- Conversion Player Ammo
	if event.force ~= nil and entity.force.name == "enemy" and  entity.type	 == "unit" and event.cause and event.cause.type == "player" then 
	
	
		local name = entity.name
		local inventory = event.cause.get_inventory(defines.inventory.player_ammo)

		if inventory then

		local current_gun = inventory[event.cause.selected_gun_index]
        local current_ammo = event.cause.get_inventory(defines.inventory.player_gun)[event.cause.selected_gun_index]    	

       --if current_ammo.valid_for_read   and current_gun.valid_for_read then
		 
		 
		
			writeDebug("Ammo Type: " .. current_ammo)
			writeDebug("Ammo Count: " .. current_gun)
			
			local inventoryContent = inventory.get_contents()		
			local AmmoType
			local Ammo = 0
				
			if inventoryContent ~= nil then
				for n,a in pairs(inventoryContent) do
					AmmoType=n
					Ammo=a
				end
			end
				
			--writeDebug("Ammo Type: " .. AmmoType)
			--writeDebug("Ammo Count: " .. Ammo)
			if AmmoType == "basic-dart-magazine_c"  or AmmoType == "enhanced-dart-magazine_c"  or AmmoType == "firearm-magazine_c"  or AmmoType == "copper-bullet-magazine_c"  or AmmoType == "piercing-rounds-magazine_c"  or AmmoType == "uranium-rounds-magazine_c"  or AmmoType == "Biological-bullet-magazine_c" then
				Convert = surface.create_entity({name = name, position = pos, force = event.cause.force.name})
				Convert.health = entity.prototype.max_health / 4
			end
			
		end	

	end
	
	
end


----------------Radars Scanning Function, used by Terraforming Sataion and Thumper -----------------------------
script.on_event(defines.events.on_sector_scanned, function(event)
	
	---- Each time a Terraforming Station scans a sector, reduce the evolution factor ----	
	if event.radar.name == "TerraformingStation_New" then
		writeDebug("The Unit # is: "..event.radar.unit_number)
		local num = (event.radar.unit_number)
		Reduce_Evolution(global.Terraforming_Station_Table[num])
		
	end

	
	--- Each time a Thumper "Scans", it will attract biters in the area
	if event.radar.name == "Thumper" then
		event.radar.surface.set_multi_command{command = {type=defines.command.attack, target=event.radar, distraction=defines.distraction.by_enemy},unit_count = 10, unit_search_distance = 500}
		writeDebug("Thumper Scanned, units should attack")   
    end   
	

    if (event.radar.name == "Artifact-collector_r") then
		local count = #global.world.itemCollectorEvents
		if (count <= 20) then	    
			global.world.itemCollectorEvents[count+1] = event.radar.unit_number
		end
    end
	
end)

--------------------------------------------
--- Living Wall Stuff
Event.register(defines.events.on_tick, function(event)	


	if game.tick % 60 == 0 and global.Living_Walls_Table ~= nil then

		for k,Living_Wall in pairs(global.Living_Walls_Table) do
			if Living_Wall.valid then
				local LW_Health = Living_Wall.health
				local Regen = 0.5
				if LW_Health < 600 then
					New_LW_Health = LW_Health + Regen
					Living_Wall.health = New_LW_Health 
				end
			else
				table.remove(global.Living_Walls_Table, k)
			end
		end
	end
	

end)



---------------------------------------------
script.on_event(defines.events.on_research_finished, function(event)

	local research = event.research.name
	if research == "Alien_Training" then
		for _, player in pairs(event.research.force.players) do
			player.insert{name="attractor-off",count=1}
		end
	end
  
    if research == "TerraformingStation-2" then
        global.deduction_constant = global.deduction_constant + (global.deduction_constant / 4)
    end      

    if research == "TerraformingStation-3" then
        global.deduction_constant = global.deduction_constant + (global.deduction_constant / 4)
    end    	
  
end)



----------------------------------------
script.on_configuration_changed(On_Config_Change)
script.on_load(On_Load)
script.on_init(On_Init)


local build_events = {defines.events.on_built_entity, defines.events.on_robot_built_entity}
script.on_event(build_events, On_Built)

local pre_remove_events = {defines.events.on_preplayer_mined_item, defines.events.on_robot_pre_mined}
script.on_event(pre_remove_events, On_Remove)

local death_events = {defines.events.on_entity_died}
script.on_event(death_events, On_Death)





-------------------- For Testing --------------
if QC_Mod == true then  

	script.on_event(defines.events.on_player_created, function(event)
	local iteminsert = game.players[event.player_index].insert
	iteminsert{name="TerraformingStation_New", count=5}
	iteminsert{name="Artifact-collector-area", count=5}
	iteminsert{name="solar-panel", count=50}
	iteminsert{name="medium-electric-pole", count=5}
	iteminsert{name="Alien-Stimulant", count=200}
	end)

end

---------------------------------------------
--- DeBug Messages 
function writeDebug(message)
	if QC_Mod == true then  
		for i, player in pairs(game.players) do
			player.print(tostring(message))
		end
	end
end