Bio_Industries_2/Bio_Industries_2/libs/item-functions.lua

local BioInd = require('common')('Bio_Industries_2')


if not thxbob.lib.item then
  thxbob.lib.item = {}
end

function thxbob.lib.item.get_type(name) --returns actual item type
  local item_type = nil
  if type(name) == "string" then
    for type_name, _ in pairs(defines.prototypes["item"]) do
      if data.raw[type_name] and data.raw[type_name][name] then
        item_type = type_name
      end
    end
    if data.raw.fluid and data.raw.fluid[name] then
      item_type = "fluid"
    end
  else
    log("Item name is not a string")
  end
  return item_type
end

function thxbob.lib.item.get_basic_type(name) --returns fluid for fluid, item for all other types.
  local item_type = thxbob.lib.item.get_type(name)
  if not (item_type == "fluid" or item_type == nil) then
    item_type = "item"
  end
  return item_type
end

function thxbob.lib.item.get_basic_type_simple(name) --assumes type is item, even if the item doesn't exist
  local item_type = "item"
  if data.raw.fluid[name] then
    item_type = "fluid"
  end
  return item_type
end

function thxbob.lib.item.ingredient_simple(inputs) --doesn't care if the item actually exists or not, returns if a valid ingredient structure can be determined.
  local item = {}

  if type(inputs) == "table" then
    if inputs.name and type(inputs.name) == "string" then
      item.name = inputs.name
    else
      log(debug.traceback())
      log("Unable to determine an ingredient name")
      return nil
    end

    if inputs.amount and type(inputs.amount) == "number" then
      item.amount = inputs.amount
    else
      log(debug.traceback())
      log("Unable to determine an ingredient amount")
      return nil
    end

    if inputs.type then
      item.type = inputs.type
    else
      log(debug.traceback())
      log("Unable to determine an ingredient type")
      return nil
    end

    if item.type == "item" then
      if type(item.amount) ~= "number" or item.amount < 1 then
        item.amount = 1
      else
        item.amount = math.floor(item.amount)
      end
    end
    if item.type == "fluid" then
      item.temperature = inputs.temperature
      item.minimum_temperature = inputs.minimum_temperature
      item.maximum_temperature = inputs.maximum_temperature
      item.fluidbox_index = inputs.fluidbox_index
      item.fluidbox_multiplier = inputs.fluidbox_multiplier
    end

    item.ignored_by_stats = inputs.ignored_by_stats

    return item
  else
    log(debug.traceback())
    thxbob.lib.error.ingredient(item)
    return nil
  end
end

function thxbob.lib.item.ingredient(inputs) --returns a valid ingredient only if the item exists.
  local item = thxbob.lib.item.ingredient_simple(inputs)
  if item then
    return item
  else
    if inputs and inputs.name then
      log(inputs.name)
    end
    log(debug.traceback())
    thxbob.lib.error.ingredient(inputs)
    return nil
  end
end

--Same as ingredient, but has support for amount_min, amount_max and probability
function thxbob.lib.item.result_simple(inputs)
  local item = {}

  if type(inputs) == "table" then
    if inputs.name and type(inputs.name) == "string" then
      item.name = inputs.name
    else
      log(debug.traceback())
      log("Unable to determine a result name")
      return nil
    end

    if inputs.amount and type(inputs.amount) == "number" then
      item.amount = inputs.amount
    elseif inputs.amount_min and inputs.amount_max then
      item.amount_min = inputs.amount_min
      item.amount_max = inputs.amount_max
    else
      log(debug.traceback())
      log("Unable to determine a result amount")
      return nil
    end

    if inputs.probability then
      item.probability = inputs.probability
    end

    if inputs.type then
      item.type = inputs.type
    else
      item.type = thxbob.lib.item.get_basic_type_simple(item.name)
    end

    if item.type == "item" then
      if item.amount then
        if type(item.amount) ~= "number" or item.amount < 1 then
          item.amount = 1
        else
          item.amount = math.floor(item.amount)
        end
      end
      if item.amount_min then
        if type(item.amount_min) ~= "number" or item.amount_min < 1 then
          item.amount_min = 0
        else
          item.amount_min = math.floor(item.amount_min)
        end
      end
      if item.amount_max then
        if type(item.amount_max) ~= "number" or item.amount_max < 1 then
          item.amount_max = 1
        else
          item.amount_max = math.ceil(item.amount_max)
        end
      end
    end
    if item.type == "fluid" then
      item.fluidbox_index = inputs.fluidbox_index
      item.temperature = inputs.temperature
    else
      item.extra_count_fraction = inputs.extra_count_fraction
      item.percent_spoiled = inputs.percent_spoiled
    end

    item.ignored_by_stats = inputs.ignored_by_stats
    item.ignored_by_productivity = inputs.ignored_by_productivity
    item.show_details_in_recipe_tooltip = inputs.show_details_in_recipe_tooltip
  end

  if
    type(item.name) == "string"
    and (type(item.amount) == "number" or (type(item.amount_min) == "number" and type(item.amount_max) == "number"))
    and (item.probability == nil or type(item.probability) == "number")
    and (item.type == "item" or item.type == "fluid")
  then
    return item
  else
    log(debug.traceback())
    thxbob.lib.error.result(item)
    return nil
  end
end

function thxbob.lib.item.result(inputs) --returns a valid result only if the item exists.
  local item = thxbob.lib.item.result_simple(inputs)
  if item then
    return item
  else
    if inputs and inputs.name then
      log(inputs.name)
    end
    log(debug.traceback())
    thxbob.lib.error.result(inputs)
    return nil
  end
end

function thxbob.lib.item.combine(item1_in, item2_in)
  local item = {}
  local item1 = thxbob.lib.item.result(item1_in)
  local item2 = thxbob.lib.item.result(item2_in)

  if item1 and item2 then
    item.name = item1.name
    item.type = item1.type

    if item1.amount and item2.amount then
      item.amount = item1.amount + item2.amount
    elseif item1.amount_min and item1.amount_max and item2.amount_min and item2.amount_max then
      item.amount_min = item1.amount_min + item2.amount_min
      item.amount_max = item1.amount_max + item2.amount_max
    elseif item1.amount_min and item1.amount_max and item2.amount then
      item.amount_min = item1.amount_min + item2.amount
      item.amount_max = item1.amount_max + item2.amount
    elseif item1.amount and item2.amount_min and item2.amount_max then
      item.amount_min = item1.amount + item2.amount_min
      item.amount_max = item1.amount + item2.amount_max
    end

    if item1.probability and item2.probability then
      item.probability = (item1.probability + item2.probability) / 2
    elseif item1.probability then
      item.probability = (item1.probability + 1) / 2
    elseif item2.probability then
      item.probability = (item2.probability + 1) / 2
    end

    if item1.ignored_by_productivity and item2.ignored_by_productivity then
      item.ignored_by_productivity = item1.ignored_by_productivity + item2.ignored_by_productivity
    elseif item1.ignored_by_productivity then
      item.ignored_by_productivity = item1.ignored_by_productivity
    elseif item2.ignored_by_productivity then
      item.ignored_by_productivity = item2.ignored_by_productivity
    end

    if item1.ignored_by_stats and item2.ignored_by_stats then
      item.ignored_by_stats = item1.ignored_by_stats + item2.ignored_by_stats
    elseif item1.ignored_by_stats then
      item.ignored_by_stats = item1.ignored_by_stats
    elseif item2.ignored_by_stats then
      item.ignored_by_stats = item2.ignored_by_stats
    end

    item.fluidbox_index = item1.fluidbox_index or item2.fluidbox_index

    return item
  else
    return nil
  end
end

function thxbob.lib.item.add(list, item_in) --increments amount if exists
  local item = thxbob.lib.item.result(item_in)
  if type(list) == "table" and item then
    local addit = true
    for i, object in pairs(list) do
      if object.name == item.name then
        addit = false
        list[i] = thxbob.lib.item.combine(object, item)
      end
    end
    if addit then
      table.insert(list, item)
    end
  end
end

function thxbob.lib.item.add_new(list, item_in) --ignores if exists
  local item = thxbob.lib.item.result(item_in)
  if type(list) == "table" and item then
    local addit = true
    for i, object in pairs(list) do
      local basic_object = thxbob.lib.item.result(object)
      if basic_object and item.name == basic_object.name then
        addit = false
      end
    end
    if addit then
      table.insert(list, item)
    end
  end
end

function thxbob.lib.item.remove(list, item)
  if type(list) == "table" and type(item) == "string" then
    for i, object in ipairs(list) do
      if object.name == item then
        table.remove(list, i)
      end
    end
  else
    log(debug.traceback())
    thxbob.lib.error.item(item)
  end
end

function thxbob.lib.item.set(list, item_in)
  local item = thxbob.lib.item.result(item_in)
  if type(list) == "table" and item then
    local addit = true
    for i, object in pairs(list) do
      if object.name == item.name then
        list[i] = item
        addit = false
      end
    end
    if addit then
      table.insert(list, item)
    end
  end
end

function thxbob.lib.item.hide(item_name)
  if type(item_name) == "string" then
    local item = data.raw.item[item_name] or data.raw.fluid[item_name]
    if item then
      item.hidden = true
    end
  else
    log(debug.traceback())
    thxbob.lib.error.item(item_name)
  end
end

function thxbob.lib.item.hide_entity(type_name, entity_name)
  if type(type_name) == "string" and type(entity_name) == "string" then
    local entities = data.raw[type_name]
    if entities then
      local entity = entities[entity_name]
      if entity then
        entity.hidden = true
      end
    end
  else
    log(debug.traceback())
    thxbob.lib.error.item(entity_name)
  end
end

function thxbob.lib.item.set_subgroup(item_name, subgroup)
  if type(item_name) == "string" and type(subgroup) == "string" then
    local item = data.raw.item[item_name]
    if item then
      item.subgroup = subgroup
    else
      item = data.raw.fluid[item_name]
      if item then
        item.subgroup = subgroup
      end
    end
  else
    log(debug.traceback())
    thxbob.lib.error.item(item_name)
  end
end