require("carbon-recipe-final-stacking")
require("carbon-recipe-final-k2")
require("carbon-recipe-final-se")
require("carbon-recipe-final-rrr")
require("low-density-structure-nanotubes")
require("carbon-recipe-modules") -- Should be last import, need in both updates and final-fixes

local util = require("data-util");

if util.me.use_rough_diamond() then
  local amt = 0
  if data.raw.recipe["vtk-deepcore-mining-ore-chunk-refining"] then
    for i, product in pairs(data.raw.recipe["vtk-deepcore-mining-ore-chunk-refining"].results) do
      if product.name == "vtk-deepcore-mining-rough-diamond-chunk" then
        amt = product.amount - 1
      end
      if product[1] == "vtk-deepcore-mining-rough-diamond-chunk" then
        amt = product[2]
      end
    end
    util.add_to_product("vtk-deepcore-mining-ore-chunk-refining",
                        "vtk-deepcore-mining-rough-diamond-chunk", 0-amt)
    util.add_to_product("vtk-deepcore-mining-ore-chunk-refining",
                        "vtk-deepcore-mining-flake-graphite-chunk", amt)
  end
end


-- Green circuits in final fixes due to K2 shenanigans 
if mods.MDbobelectronics or mods.bobelectronics then
  util.replace_ingredient("basic-electronic-components", "coal", "graphite")
  util.replace_ingredient("basic-electronic-components", "coke", "graphite")
else
  util.replace_some_ingredient("electronic-circuit", "copper-cable", 1, "graphite", 1)
  -- AAI alternate green circuit
  util.replace_some_ingredient("electronic-circuit-stone", "copper-cable", 1, "graphite", 1)
end

if data.raw.recipe["low-density-structure"] and 
  not data.raw.recipe["low-density-structure"].icon and
  not data.raw.recipe["low-density-structure"].icons
  then
  util.set_subgroup("low-density-structure", data.raw.item["low-density-structure"].subgroup)
  util.set_icons("low-density-structure", {
    { icon = "__base__/graphics/icons/low-density-structure.png", icon_size = 64, icon_mipmaps = 4},
  })
  util.set_subgroup("se-low-density-structure-beryllium", data.raw.item["low-density-structure"].subgroup)
  util.set_icons("se-low-density-structure-beryllium", {
    { icon = "__base__/graphics/icons/low-density-structure.png", icon_size = 64, icon_mipmaps = 4},
  })
  util.set_subgroup("low-density-structure-RS", data.raw.item["low-density-structure"].subgroup)
  util.set_icons("low-density-structure-RS", {
    { icon = "__base__/graphics/icons/low-density-structure.png", icon_size = 64, icon_mipmaps = 4},
  })
end


-- Only electric smelters can make Diamonds
function diamondsInElectricSmelter(entity)
  if entity.energy_source.type == "electric" then
    local isSmelter = false
    local isSmelterRS = false
    for j, category in pairs(entity.crafting_categories) do
      if category == "smelting" then
        isSmelter = true
      end
      if category == "smelting-RS" then
        isSmelterRS = true
      end
    end
    if isSmelter then
     table.insert(entity.crafting_categories,"diamond-synthesis")
    end
    if isSmelterRS then
     table.insert(entity.crafting_categories,"diamond-synthesis-RS")
    end
  end
end
for i, entity in pairs(data.raw.furnace) do
  diamondsInElectricSmelter(entity)
end
for i, entity in pairs(data.raw["assembling-machine"]) do
  diamondsInElectricSmelter(entity)
end 

-- furnace recipe rework, remove crucible ingredients one more time
for i, furnace in pairs(util.me.furnaces) do
  for j, ingredient in pairs(util.me.crucible_ingredients)  do
    util.remove_ingredient(furnace, ingredient)
  end
end

-- Vanilla burner phase tweaks -- green circuits after electronics
if not mods.Krastorio2 and not mods["aai-industry"] and not mods.bzaluminum then
  util.replace_ingredient("offshore-pump", "electronic-circuit", "copper-cable")
  util.replace_ingredient("lab", "electronic-circuit", "copper-cable")
  util.replace_ingredient("electric-mining-drill", "electronic-circuit", "copper-cable")
  util.replace_ingredient("assembling-machine-1", "electronic-circuit", "copper-plate")
  util.replace_ingredient("radar", "electronic-circuit", "copper-plate")
  util.replace_ingredient("splitter", "electronic-circuit", "copper-cable", 20)

  -- Keep repair pack raw ingredients the same:
  util.remove_ingredient("repair-pack", "electronic-circuit")
  util.add_ingredient("repair-pack", "copper-cable", 6)
  util.set_ingredient("repair-pack", "iron-gear-wheel", 3)

  util.add_effect("electronics", { type = "unlock-recipe", recipe = "electronic-circuit" })
  util.add_effect("electronics", { type = "unlock-recipe", recipe = "inserter" })
  util.add_effect("electronics", { type = "unlock-recipe", recipe = "long-handed-inserter" })
  util.remove_recipe_effect("automation", "long-handed-inserter")
  util.set_enabled("electronic-circuit", false)
  util.set_enabled("inserter", false)
  util.add_prerequisite("logistic-science-pack", "electronics")
end

if not mods["aai-industry"] then
  if not mods.Krastorio2 then
    util.set_enabled("electric-mining-drill", true)
  end
end


if mods["aai-industry"] then
  -- AAI Industry now unlocks steam much later. Move it to an earlier sensible location in the tech tree.
  util.set_prerequisite("steam-power", {"basic-fluid-handling"})
  util.set_tech_recipe("steam-power", {{"automation-science-pack", 1}})
end

if mods["vtk-deepcore-mining"] then
end


-- Must be last
util.create_list()