control.lua

require("objects.global")
require("objects.balancer")
require("objects.part")
require("objects.belt")
require("helper.message-handler")
require("test")

-- on new savegame and on adding mod to existing save
script.on_init(function()
    -- set defaults and initialize values in global table
    global.next_balancer_unit_number = 1
    global.next_lane_unit_number = 1
    global.next_belt_check = nil
    global.balancer = {}
    global.parts = {}
    global.belts = {}
    global.lanes = {}
    global.events = {}
end)

script.on_load(reregister_on_tick)

-- If some mod is changed, check if boblogistics got added and do stuff :)
script.on_configuration_changed(
    function(e)
        ---@type ModConfigurationChangedData
        local boblogistics_changes = e.mod_changes["boblogistics"]

        if boblogistics_changes and boblogistics_changes.old_version == nil and boblogistics_changes.new_version and settings.startup["bobmods-logistics-beltoverhaul"].value == true then
            -- on boblogistics got added!
            for _, force in pairs(game.forces) do
                local technologies = force.technologies
                local recipes = force.recipes

                technologies["belt-balancer-0"].researched = technologies["belt-balancer-1"].researched
                recipes["belt-balancer-basic-belt"].enabled = technologies["belt-balancer-1"].researched
            end
        end
    end
)


-- Custom command to print out some statistics
commands.add_command("belt-balancer-statistics", "", function(e)
    local balancer_amount = table_size(global.balancer)
    local balancer_part_amount = table_size(global.parts)
    local balancer_input_belt_amount = 0
    local balancer_output_belt_amount = 0
    local balancer_input_lane_amount = 0
    local balancer_output_lane_amount = 0

    for _, balancer in pairs(global.balancer) do
        for _, v in pairs(balancer.parts) do
            balancer_input_belt_amount = balancer_input_belt_amount + table_size(global.parts[v].input_belts)
            balancer_output_belt_amount = balancer_output_belt_amount + table_size(global.parts[v].output_belts)
        end
        balancer_input_lane_amount = balancer_input_lane_amount + table_size(balancer.input_lanes)
        balancer_output_lane_amount = balancer_output_lane_amount + table_size(balancer.output_lanes)
    end

    local output = "balancers: " .. balancer_amount ..
        "\nbalancer-parts: " .. balancer_part_amount ..
        "\nbalancer_input_belts: " .. balancer_input_belt_amount ..
        "\nbalancer_output_belts: " .. balancer_output_belt_amount ..
        "\nbalancer_input_lanes: " .. balancer_input_lane_amount ..
        "\nbalancer_output_lanes: " .. balancer_output_lane_amount
    game.get_player(e.player_index).print(output)
    print(output)
end)

-- only add this command, if `debug` is available and creative-mod is activated
-- `debug` can be activated, when calling factorio with `--instrument-mod belt-balancer`
if debug and script.active_mods["creative-mod"] then
    commands.add_command("belt-balancer-test", "", function(e)
        test_mod(e.player_index)
    end)

    commands.add_command("belt-balancer-test-logic", "", function(e)
        test_mod_logic(e.player_index)
    end)

    commands.add_command("belt-balancer-print", "", function(e)
        print("Balancer:")
        print(serpent.block(global.balancer))
        print("Parts:")
        print(serpent.block(global.parts))
        print("Belts:")
        print(serpent.block(global.belts))
        print("Lanes:")
        print(serpent.block(global.lanes))
    end)
end

---check if this entity was built with fast-replace
---if so, remove the fast-replaced entity from tracking.
---@param new_entity LuaEntity
function process_fast_replace(new_entity)
    local belts = {}

    for unit_number, belt in pairs(global.belts) do
        -- only run, when entities overlapping and are on the same surface
        if new_entity.surface == belt.surface
            and  belt.position.x >= new_entity.position.x - 0.5 and belt.position.x <= new_entity.position.x + 0.5
            and belt.position.y >= new_entity.position.y - 0.5 and belt.position.y <= new_entity.position.y + 0.5 then

            belts[unit_number] = belt
        end
    end

    for unit_number, belt in pairs(belts) do
        -- remove fast-replaced invalid entity
        if belt.type == "splitter" then
            belt_functions.remove_splitter(belt.entity, belt.direction, unit_number, belt.surface, belt.position)
        else
            belt_functions.remove_belt(belt.entity, belt.direction, unit_number, belt.surface, belt.position)
        end
    end
end

function built_entity(e)
    ---@type LuaEntity
    local entity

    if e.entity then
        entity = e.entity
    elseif e.created_entity then
        entity = e.created_entity
    else
        entity = e.destination -- this is the name of the entity from on_entity_cloned
    end

    if entity.name == "balancer-part" then
        part_functions.built(entity)
    end

    if entity.type == "transport-belt" then
        process_fast_replace(entity)
        belt_functions.built_belt(entity)
    end

    if entity.type == "underground-belt" then
        process_fast_replace(entity)
        belt_functions.built_belt(entity)
    end

    if entity.type == "splitter" then
        process_fast_replace(entity)
        belt_functions.built_splitter(entity)
    end
end

script.on_event(
    {
        defines.events.on_built_entity,
        defines.events.on_robot_built_entity,
        defines.events.script_raised_built,
        defines.events.script_raised_revive,
        defines.events.on_entity_cloned -- fix compatability with region cloner
    },
    built_entity
)

function remove_entity(e)
    ---@type LuaEntity
    local entity

    if e.entity then
        entity = e.entity
    else
        entity = e.created_entity
    end

    if entity.name == "balancer-part" then
        part_functions.remove(entity, e.buffer)
    end

    if entity.type == "transport-belt" then
        belt_functions.remove_belt(entity)
    end

    if entity.type == "underground-belt" then
        belt_functions.remove_belt(entity)
    end

    if entity.type == "splitter" then
        belt_functions.remove_splitter(entity)
    end
end

script.on_event(
    {
        defines.events.on_entity_died,
        defines.events.on_player_mined_entity,
        defines.events.on_robot_mined_entity,
        defines.events.script_raised_destroy
    },
    remove_entity
)

script.on_event({ defines.events.on_player_rotated_entity },
    function(e)
        if e.entity.type == "transport-belt" then
            belt_functions.remove_belt(e.entity, e.previous_direction)
            belt_functions.built_belt(e.entity)
        end

        if e.entity.type == "underground-belt" then
            belt_functions.remove_belt(e.entity, e.previous_direction)
            belt_functions.built_belt(e.entity)

            -- Neighbour is only the other end
            local neighbour = e.entity.neighbours
            if neighbour and neighbour.valid then
                -- make neighbour also have previous_direction
                local previous_direction = (neighbour.direction + 4) % 8
                belt_functions.remove_belt(neighbour, previous_direction)
                belt_functions.built_belt(neighbour)
            end
        end

        if e.entity.type == "splitter" then
            belt_functions.remove_splitter(e.entity, e.previous_direction)
            belt_functions.built_splitter(e.entity)
        end
    end
)

script.on_event(defines.events.on_tick, function()
    local unit_number = global.next_belt_check
    local belt = global.belts[unit_number]

    -- check if belt direction got changed
    if belt and belt.entity.valid and belt.direction ~= belt.entity.direction then
        -- remove belt and readd it
        if belt.type == "splitter" then
            belt_functions.remove_splitter(belt.entity, belt.direction, unit_number, belt.entity.surface, belt.position)
            belt_functions.built_splitter(belt.entity)
        else
            belt_functions.remove_belt(belt.entity, belt.direction, unit_number, belt.entity.surface, belt.position)
            belt_functions.built_belt(belt.entity)
        end
    end

    global.next_belt_check, _ = next(global.belts, unit_number)
end)