local Recipe = require("__cf-lib__/data/Recipe")
local Technology = require("__cf-lib__/data/Technology")

-- Entities

table.merge(data.raw["locomotive"]["mini-locomotive"], {
    max_power = "230kW",
    max_speed = 0.35,
    friction_force = 0.20,
    braking_force = 1,
    energy_source = {
        fuel_inventory_size = 2
    }
})

table.merge(data.raw["cargo-wagon"]["mini-cargo-wagon"], {
    max_speed = 0.35,
    friction_force = 0.1,
    braking_force = 0.5
})

table.merge(data.raw["fluid-wagon"]["mini-fluid-wagon"], {
    max_speed = 0.35,
    friction_force = 0.1,
    braking_force = 0.5,
    capacity = 25000
})


-- Items

table.assign(data.raw.item["mini-locomotive"], {
    stack_size = 10,
    order = "c[rolling-stock]-a[tier1]-a[locomotive]",
    subgroup = "train-transport"
})
table.assign(data.raw.item["mini-cargo-wagon"], {
    stack_size = 10,
    order = "c[rolling-stock]-a[tier1]-b[cargo-wagon]",
    subgroup = "train-transport"
})
table.assign(data.raw.item["mini-fluid-wagon"], {
    stack_size = 10,
    order = "c[rolling-stock]-a[tier1]-c[fluid-wagon]",
    subgroup = "train-transport"
})


-- Recipes

Recipe:new("mini-locomotive")
    :replaceIngredient("engine-unit", "iron-gear-wheel", 10)


-- Technologies

Technology:new("railway")
    :setPrerequisites({ "logistics", "steel-processing" })
    :removeIngredient("logistic-science-pack")
    :removeRecipe("locomotive")
    :removeRecipe("cargo-wagon")
    :addRecipe("mini-locomotive")
    :addRecipe("mini-cargo-wagon")
    :assign({
        icon = "__Mini_Trains__/data/icons/tech256.png",
        icon_size = 256
    })

Technology:new("automated-rail-transportation")
    :removeIngredient("logistic-science-pack")
    :merge({
        unit = {
            count = 75
        }
    })

Technology:new("fluid-wagon")
    :removeRecipe("fluid-wagon")
    :addRecipe("mini-fluid-wagon")
    :assign({
        icon = "__Mini_Trains__/data/icons/fluid256.png",
        icon_size = 256
    })

data.raw.technology["mini-trains"] = nil