InfOS/Programs/Assembly Line/recipeServer.lua

Component = require("component")
Event = require("event")
Term = require("term")
AU = require("util")
local uc = require("unicode")
local S = require("serialization")
local fluidMap, dictionary = require("dictionary")
local network = Component.modem
local mainChannel = 100
-- mainChannel = Main channel for bi-directional communication

local knownAssemblyLines = 0
local assemblyStatus = {}
local run = true
local function getIDs()
    local knownIDs = io.open("IDs", "r")
    if knownIDs == nil then
        knownAssemblyLines = 0
    else
        for line in io.lines("IDs") do
            knownAssemblyLines = knownAssemblyLines + 1
            network.open(mainChannel + knownAssemblyLines)
            assemblyStatus[mainChannel + knownAssemblyLines] = false
        end
    end
end
local function contains(t, value)
    for k, v in pairs(t) do
        if v == value then
            return true
        end
    end
    return false
end
local function getFree()
    for i = mainChannel, mainChannel + knownAssemblyLines, 1 do
        if assemblyStatus[i] == false then
            return i
        end
    end
    return nil
end

local function processMessage(type, localAddress, remoteAddress, port, distance, eventType, value2, value3)
    if eventType == "requestID" then
        knownAssemblyLines = knownAssemblyLines + 1
        local newID = mainChannel + knownAssemblyLines
        assemblyStatus[newID] = false
        network.broadcast(mainChannel, "sendID", newID)
        local knownIDs = io.open("IDs", "a")
        knownIDs:write(newID .. "\n")
        knownIDs:close()
    elseif eventType == "complete" then
        assemblyStatus[port] = false
    end
end

local processKey = {}

local function quit()
    Term.write("Quitting...")
    Event.ignore("modem_message", processMessage)
    Event.ignore("key_up", processKey)
    run = false
end

local function processKey(event, address, key, code, player)
    local value = uc.char(key)
    if value == "." then
        quit()
    elseif value == "a" then
        AU.getRecipes(RL)
    end
end

local function startAssembly(assemblyport, recipe)
    assemblyStatus[assemblyport] = recipe.label
    network.broadcast(assemblyport, "startAssembly", S.serialize(recipe))
end
local function spairs(t, order)
    local keys = {}
    for k in pairs(t) do
        keys[#keys + 1] = k
    end
    if order then
        table.sort(
            keys,
            function(a, b)
                return order(t, a, b)
            end
        )
    else
        table.sort(keys)
    end
    local i = 0
    return function()
        i = i + 1
        if keys[i] then
            return keys[i], t[keys[i]]
        end
    end
end
local function matchRecipe(recipes)
    local items = Component.me_interface.getItemsInNetwork()
    local foundItems = {}
    if #items > 0 then
        for i = 1, #items, 1 do
            foundItems[items[i].label] = items[i].size
        end
    end
    for outputLabel, recipe in spairs(
        recipes,
        function(t, a, b)
            return t[b].inputs < t[a].inputs
        end
    ) do
        local found = 0
        local craftable = 1000
        for i = 1, recipe.inputs, 1 do
            local label, requiredAmount = recipe["input" .. i].name, recipe["input" .. i].amount
            if dictionary[label] ~= nil then
                label = dictionary[label]
            end
            if foundItems[label] == nil then
                break
            else
                local existingAmount = foundItems[label]
                if existingAmount >= requiredAmount then
                    found = found + 1
                    craftable = math.min(math.floor(existingAmount / requiredAmount), craftable)
                end
            end
        end
        for i = 1, recipe.fluids, 1 do
            local label, requiredAmount
            if fluidMap[recipe["fluid" .. i].name] ~= nil then
                label, requiredAmount =
                    fluidMap[recipe["fluid" .. i].name].name,
                    recipe["fluid" .. i].amount / fluidMap[recipe["fluid" .. i].name].size
            else
                break
            end
            if foundItems[label] == nil then
                break
            else
                local existingAmount = foundItems[label]
                if existingAmount >= requiredAmount then
                    found = found + 1
                    craftable = math.min(math.floor(existingAmount / requiredAmount), craftable)
                end
            end
        end
        if found == recipe.inputs + recipe.fluids then
            return recipe, craftable
        end
    end
    return nil
end
RL = {}
local function scheduleTasks()
    local recipe, craftable = matchRecipe(RL)
    if recipe ~= nil then
        if craftable <= 8 then
            if not contains(assemblyStatus, recipe.label) then
                local taskid = getFree()
                if taskid ~= nil then
                    Term.write("Started assembly of " .. recipe.label .. " with AL #" .. taskid .. "\n")
                    startAssembly(taskid, recipe)
                else
                    Term.write("No free assembly lines.\n")
                end
            end
            return true
        else
            while craftable > 0 do
                local taskid = getFree()
                if taskid ~= nil then
                    startAssembly(taskid, recipe)
                    Term.write("Started assembly of " .. recipe.label .. " with AL #" .. taskid .. "\n")
                else
                    Term.write("No free assembly lines.\n")
                end
                craftable = craftable - 8
            end
        end
        return true
    else
        return false
    end
end

--"gt.metaitem.01.32606.name"
local function initializeServer()
    network.open(mainChannel)
    getIDs()
    AU.getRecipes(RL)
end

initializeServer()
Event.listen("modem_message", processMessage)
Event.listen("key_up", processKey)

while run do
    scheduleTasks()
    os.sleep(2)
end