--GET ICON/ICONS FROM FLUID/ITEM
local function get_icon_or_icons(object_name)
   local prototype_list = {'item', 'fluid' }
   for _, prototype in pairs(prototype_list) do
       local object = data.raw[prototype][object_name]
       if object then
           if object.icon then
               return object.icon
           end
           if object.icons then
               return object.icons
           end
       end
   end
   --something is wrong here but we need to return something
   return '__angelsrefining__/graphics/icons/void.png'
end

--MOD ICON DEFINED AS TABLE
function mod_icon(t_icon, scale, shift)
	   local new_icon = table.deepcopy(t_icon)
	   if scale then
	       if new_icon.icon_size then
	           new_icon.scale = scale*(32/new_icon.icon_size)
	       else
	           new_icon.scale = scale
	       end
	   end
	   if shift then
	       new_icon.shift = shift
	   end
	   return new_icon
	end

--COMPARES ARGUMENT (ARG) AGAINST A TABLE (EXCEP), RETURNS FALSE IF ARG == EXCEP ELSE TRUE
function angelsmods.functions.check_exception(arg, excep)
	for _, exception in pairs(excep) do
		if exception == arg then return false end
	end
	return true
end

--ADDS A TECH TO IGNORE FOR THE TECH MOD
function angelsmods.functions.add_exception(to_add)
	if --type(to_add) == string and 
	angelsmods.industries and angelsmods.industries.tech_exceptions then
		table.insert(angelsmods.industries.tech_exceptions, to_add)
	end
end

--REMOVE PRODUCTIVITY RESTRICTION
function angelsmods.functions.allow_productivity(recipe_name)
	if data.raw.recipe[recipe_name] then
		for i, module in pairs(data.raw.module) do
			if module.limitation and module.effect.productivity then
				table.insert(module.limitation, recipe_name)
			end
		end
	end
end

--OVERRIDES SET CONDITION FOR LIST OF ITEMS
function angelsmods.functions.override_item_conditions(override)
	for i, items in pairs(override.list) do
		if data.raw.item[items] then
			if data.raw.item[items].stack_size <= override.value then
				data.raw.item[items].stack_size = override.value
			end
		end
	end
end

--MODIFY FLAGS
function angelsmods.functions.add_flag(entity, flag)
	if data.raw.item[entity] then 
	local to_add = data.raw.item[entity]
		if to_add.flags then
			table.insert(to_add.flags, flag)
		else
			to_add.flags = {flag}
		end
	end
end

--MODIFY BARRELING RECIPES
function angelsmods.functions.disable_barreling_recipes(fluid_to_disable)
	angelsmods.functions.OV.disable_recipe("fill-"..fluid_to_disable.."-barrel")
	angelsmods.functions.OV.disable_recipe("empty-"..fluid_to_disable.."-barrel")
	angelsmods.functions.OV.disable_recipe("fill-"..fluid_to_disable.."-liquid-bot")
	angelsmods.functions.OV.disable_recipe("empty-"..fluid_to_disable.."-liquid-bot")
	for nx, item in pairs(data.raw.item) do
		if item.name == fluid_to_disable.."-barrel" then
			if item.flags then
				table.insert(item.flags,"hidden")
			else
				item.flags = {"hidden"}
			end
		end
		if item.name == fluid_to_disable.."-liquid-bot" then
			if item.flags then
				table.insert(item.flags,"hidden")
			else
				item.flags = {"hidden"}
			end
		end
	end
end

function angelsmods.functions.modify_barreling_icon()
	for ix, item in pairs(data.raw.item) do
		if item.subgroup == "fill-barrel" then
			icon_name = string.sub(item.name, 1, -8)
			for kx, fluid in pairs(data.raw.fluid) do
				if fluid.name == icon_name then
					if not item.icons then 
						item.icons = {}
						item.icon= nil 
					end
                   local fluid_icon = get_icon_or_icons(fluid.name)
                   if type(fluid_icon) == 'table' then
                       for _, icon in pairs(fluid_icon) do
                           table.insert(item.icons,mod_icon(icon, 0.5, {0, 5}))
                       end
                   else
                      table.insert(item.icons,{icon = fluid_icon, scale = 0.5, shift = {0, 5}})    --shift = {4, -8}
                   end				
				end
			end
		end
	end
end

function angelsmods.functions.modify_barreling_recipes()
	angelsmods.functions.modify_barreling_icon()
	local auto_barreling = angelsmods.trigger.enable_auto_barreling
	for _, recipe in pairs(data.raw.recipe) do
		if recipe.subgroup == "empty-barrel" or recipe.subgroup == "fill-barrel" then
			if auto_barreling then
				recipe.hidden=true
			end
			recipe.category = "barreling-pump"
		end
	end
end

--CREATE VOID RECIPES
function angelsmods.functions.make_void(fluid_name, void_category) --categories: chemical (flare-stack), water(clarifier)
	--LOCAL DEFINITIONS
	local void_hidden = angelsmods.trigger.enable_hide_void
	local recipe = {}
	
	if data.raw.fluid[fluid_name] then
		local fluid_icon = get_icon_or_icons(fluid_name)
		if type(fluid_icon) == 'table' then
		   recipe.icons = fluid_icon
		else
		   recipe.icon = fluid_icon
		end
		if data.raw.fluid[fluid_name].icon_size then
			recipe.icon_size = data.raw.fluid[fluid_name].icon_size
		else
			recipe.icon_size = 32
		end		
		if void_category == "water" then
			void_amount = 400
			void_item = "water-void"
			void_time = 5
			void_type = "fluid"
			void = 0
		end
		if void_category == "chemical" then
			void_amount = 100
			void_item = "chemical-void"
			void_time = 1
			void_type = "fluid"
			void = 0
		end
	else if data.raw.item[fluid_name] then
			fluid_icon = data.raw.item[fluid_name].icon
			if void_category == "bio" then
				void_amount = 1
				void_item = "solid-compost"
				void_time = 1
				void_type = "item"
				void = 1
			end
		end
	end
		
	recipe.type = "recipe"
	recipe.name = "angels-"..void_category.."-void-"..fluid_name
	recipe.category = "angels-"..void_category.."-void"
	recipe.enabled = "true"
	recipe.hidden = void_hidden
	recipe.energy_required = void_time
	recipe.ingredients = {{type= void_type, name= fluid_name, amount= void_amount}}
	recipe.results = {{type="item", name= void_item, amount=1, probability= void}}
	recipe.subgroup = "angels-void"
	recipe.order = "angels-"..void_category.."-void-"..fluid_name
	
	data:extend({recipe})
end

--CREATE CONVERTER RECIPES (PETROCHEM)
function angelsmods.functions.make_converter(fluid_name_other, fluid_name_angels)
	if angelsmods.trigger.enableconverter then
		if data.raw.fluid[fluid_name_angels] and data.raw.fluid[fluid_name_other] then
		
		--LOCALS
		hide_converter = angelsmods.trigger.hideconverter
		
		--ORDER COUNTER
		if not angelsmods.functions.converter_counter then
			angelsmods.functions.converter_counter = 0
		end
		angelsmods.functions.converter_counter = angelsmods.functions.converter_counter +1
			
		data:extend(
		{
			{
			type = "recipe",
			name = "converter-other-"..fluid_name_other,
			localised_name = {"recipe-name.converter-angels", {"fluid-name." .. fluid_name_other}, {"fluid-name." .. fluid_name_angels}},
			category = "angels-converter",
			subgroup = "angels-converter",
			energy_required = 0.5,
			enabled = "true",
			hidden = hide_converter,
			ingredients ={
				{type="fluid", name=fluid_name_angels, amount=50}
			},
			results=
			{
				{type="fluid", name=fluid_name_other, amount=50},
			},
			icon_size = 32,
			order = "a"..angelsmods.functions.converter_counter,
			},
			{
			type = "recipe",
			name = "converter-angels-"..fluid_name_angels,
			localised_name = {"recipe-name.converter-angels", {"fluid-name." .. fluid_name_other}, {"fluid-name." .. fluid_name_angels}},
			category = "angels-converter",
			subgroup = "angels-converter",
			energy_required = 0.5,
			enabled = "true",
			hidden = hide_converter,
			ingredients ={
				{type="fluid", name=fluid_name_other, amount=50}
			},
			results=
			{
				{type="fluid", name=fluid_name_angels, amount=50},
			},
			icon_size = 32,
			order = "b"..angelsmods.functions.converter_counter,
			},
			}
		)
			if angelsmods.trigger.hideconverter then
				angelsmods.functions.OV.hide_recipe({"converter-other-"..fluid_name_other, "converter-angels-"..fluid_name_angels})
			end
		end
	end
end

--MODIFY LOCALIZATION STRINGS
function angelsmods.functions.add_localization(res_name, ore1, ore2, ore3, ore4, ore5, ore6)
	if not ore1 then ore1 = "missing" end
	if not ore2 then ore2 = "missing" end
	if not ore3 then ore3 = "missing" end
	if not ore4 then ore4 = "missing" end
	if not ore5 then ore5 = "missing" end
	if not ore6 then ore6 = "missing" end
	if data.raw["resource"][res_name] then
		data.raw["resource"][res_name].localised_description = {"item-description."..res_name, {"item-description.loc-"..ore1}, {"item-description.loc-dot"}, {"item-description.loc-"..ore2}, {"item-description.loc-dot"}, {"item-description.loc-"..ore3}, {"item-description.loc-dot"}, {"item-description.loc-"..ore4}, {"item-description.loc-dot"}, {"item-description.loc-"..ore5}, {"item-description.loc-and"}, {"item-description.loc-"..ore6}, "", ""}
	end
	if data.raw["resource"]["infinite-"..res_name] then
		data.raw["resource"]["infinite-"..res_name].localised_description = {"item-description."..res_name, {"item-description.loc-"..ore1}, {"item-description.loc-dot"}, {"item-description.loc-"..ore2}, {"item-description.loc-dot"}, {"item-description.loc-"..ore3}, {"item-description.loc-dot"}, {"item-description.loc-"..ore4}, {"item-description.loc-dot"}, {"item-description.loc-"..ore5}, {"item-description.loc-and"}, {"item-description.loc-"..ore6}, "", ""}
	end
	if data.raw["item"][res_name] then
		data.raw["item"][res_name].localised_description = {"item-description."..res_name, {"item-description.loc-"..ore1}, {"item-description.loc-dot"}, {"item-description.loc-"..ore2}, {"item-description.loc-dot"}, {"item-description.loc-"..ore3}, {"item-description.loc-dot"}, {"item-description.loc-"..ore4}, {"item-description.loc-dot"}, {"item-description.loc-"..ore5}, {"item-description.loc-and"}, {"item-description.loc-"..ore6}, "", ""}
	end
	if data.raw["item"][res_name.."-crushed"] then
		data.raw["item"][res_name.."-crushed"].localised_description = {"item-description."..res_name.."-crushed", {"item-description.loc-"..ore1}, {"item-description.loc-dot"}, {"item-description.loc-"..ore2}, {"item-description.loc-and"}, {"item-description.loc-slag"}}
	end
	if data.raw["item"][res_name.."-chunk"] then
		data.raw["item"][res_name.."-chunk"].localised_description = {"item-description."..res_name.."-chunk", {"item-description.loc-"..ore1}, {"item-description.loc-dot"}, {"item-description.loc-"..ore2}, {"item-description.loc-dot"}, {"item-description.loc-"..ore3}, {"item-description.loc-dot"}, {"item-description.loc-"..ore4}, {"item-description.loc-and"}, {"item-description.loc-slag"}, "", ""}
	end
	if data.raw["item"][res_name.."-crystal"] then
		data.raw["item"][res_name.."-crystal"].localised_description = {"item-description."..res_name.."-crystal", {"item-description.loc-"..ore1}, {"item-description.loc-dot"}, {"item-description.loc-"..ore2}, {"item-description.loc-dot"}, {"item-description.loc-"..ore3}, {"item-description.loc-dot"}, {"item-description.loc-"..ore4}, {"item-description.loc-dot"}, {"item-description.loc-"..ore5}, {"item-description.loc-and"}, {"item-description.loc-slag"}, "", ""}
	end
	if data.raw["item"][res_name.."-pure"] then
		data.raw["item"][res_name.."-pure"].localised_description = {"item-description."..res_name.."-pure", {"item-description.loc-"..ore1}, {"item-description.loc-dot"}, {"item-description.loc-"..ore2}, {"item-description.loc-dot"}, {"item-description.loc-"..ore3}, {"item-description.loc-dot"}, {"item-description.loc-"..ore4}, {"item-description.loc-dot"}, {"item-description.loc-"..ore5}, {"item-description.loc-and"}, {"item-description.loc-"..ore6}, "", ""}
	end
end