Rednet Turtle Forester [CC]

args = { ... }
thisType = 0
workerID = 0
helperID = 0
serverID = 0
x, y, d = 0, 0, 0
sapling = false
protocol = "TurtleFProto"
timer = 0
refuel = false
delay = 600
working = false

function line(y, backcolor, text)
        term.setCursorPos(1, y)
        term.setBackgroundColor(backcolor)
        term.clearLine()
        term.write(text)
end

function findModem()
        sides = {"front","back","left","right","top","bottom"}
        for i=1,6 do
                if peripheral.getType(sides[i]) == "modem" then
                        rednet.open(sides[i])
                        return
                end
        end
        print("Need wireless modem!")
        error(r)
end

function getName()
    if thisType == 1 then
        workerID = os.getComputerID()
        return "worker"
    elseif thisType == 2 then
        serverID = os.getComputerID()
        return "server"
    elseif thisType == 3 then
        helperID = os.getComputerID()
        return "helper"
    end
end

function findFriends()
    rednet.broadcast("zdarova_ya_"..getName(), protocol)
    while true do
            local event, sender, msg, prot = os.pullEvent("rednet_message")
            if prot == protocol then
            if string.find(msg, "zdarova") then
                rednet.send(sender, "privet_ya_"..getName(), protocol)
            end
            if string.find(msg, "ya_server") and thisType ~= 2 then
                serverID = sender
                print("Server ID: "..serverID)
            elseif string.find(msg, "ya_worker") and thisType ~= 1 then
                workerID = sender
                print("Worker ID: "..workerID)
            elseif string.find(msg, "ya_helper") and thisType ~= 3 then
                helperID = sender
                print("Helper ID: "..helperID)
            end
            if serverID ~= 0 and workerID ~= 0 and helperID ~= 0 then
                if thisType == 2 then
                    print("Look at the monitor! Program has been started.")
                        term.redirect(monitor)
                        monitor.setTextScale(1)
                    line(1, colors.black, "  RTF")
                        line(2, colors.green, "FUEL")
                        line(3, colors.green, "SAPLING")
                        line(4, colors.gray, "START")
                        line(5, colors.gray, "EXIT")
                end
                return
            end
        end
    end
end

function checkArgs()
    if args[1] == nil then
        print("Need args!")
        error(r)
    else
        thisType = tonumber(args[1])
    end
    if args[4] == nil and thisType == 1 then
        print("Need args!")
        error(r)
    elseif thisType == 1 then
        x, y, d = tonumber(args[2]), tonumber(args[3]), tonumber(args[4])
        if args[5] ~= nil then
            protocol = args[5]
        end
    elseif thisType == 2 and args[2] ~= nil then
        delay = tonumber(args[2])
        if args[3] ~= nil then
            protocol = args[3]
        end
    elseif thisType == 3 and args[2] ~= nil then
        protocol = args[2]
    end
end

function move(w)
    turtle.suck()
    if turtle.getFuelLevel() == 0 then
        turtle.refuel(1)
    end
    if w == "forward" then
        while true do
            local res = turtle.forward()
            if res then
                break
            else
                turtle.dig()
                turtle.select(16)
                turtle.equipLeft()
                turtle.attack()
                turtle.equipLeft()
                turtle.select(1)
                findModem()
                sleep(1)
            end
        end
    elseif w == "up" then
        turtle.up()
    elseif w == "down" then
        turtle.down()
    end
end

function findMonitor()
        local e = peripheral.getNames()
        for i=1,#e do
                if peripheral.getType(e[i]) == "monitor" then
                        local m = peripheral.wrap(e[i])
                        if m.isColor() then
                monitor = m
                                return
                        end
                end
        end
        if monitor == nil then
                print("Need advanced monitor!")
                error(r)
        end
end

function dropItems()
    checkSapling()
    for i=1,15 do
        if turtle.getItemCount(i) > 0 then
            local d = turtle.getItemDetail(i)
            turtle.select(i)
            if d.name == "minecraft:log" or d.name == "minecraft:apple" then
                turtle.dropDown()
            elseif d.name ~= "minecraft:coal" and d.name ~= "minecraft:sapling" then
                turtle.dropUp()
            end
        end
    end
    turtle.select(1)
end

function checkSapling()
    for i=1,15 do
        if turtle.getItemCount(i) > 0 then
            local d = turtle.getItemDetail(i)
            if d.name == "minecraft:sapling" then
                turtle.select(i)
                if turtle.getItemCount(2) < 64 then
                    turtle.transferTo(2, turtle.getItemSpace(2))
                end
                if turtle.getItemCount(i) > 0 then
                    return
                end
            end
        end
    end
end

function work()
    for i=1,x do
        for j=1,y do
            if sapling then
                for k=1,d-1 do
                    move("forward")
                end
                sapling = false
            else
                for k=1,d+1 do
                    move("forward")
                end
            end
            local a, b = turtle.inspect()
            while b.name == "minecraft:log" do
                turtle.dig()
                if turtle.inspectUp() then
                    turtle.digUp()
                end
                move("up")
                a, b = turtle.inspect()
            end
            while not turtle.inspectDown() do
                                move("down")
                        end
            if not turtle.inspect() then
                turtle.select(2)
                turtle.place()
                turtle.select(1)
            end
            local a, b = turtle.inspect()
            if b.name == "minecraft:sapling" then
                turtle.turnLeft()
                move("forward")
                turtle.turnRight()
                move("forward")
                move("forward")
                turtle.turnRight()
                move("forward")
                turtle.turnLeft()
                sapling = true
            end
        end
        if sapling then
            for k=1,d+1 do
                move("forward")
            end
            sapling = false
        else
            for k=1,d+3 do
                move("forward")
            end
        end
        if i == x then
            if math.fmod(x,2) == 0 then
                turtle.turnRight()
                for k=1,(d+1)*(x-1) do
                    move("forward")
                end
                turtle.turnRight()
                dropItems()
                rednet.send(serverID, "finish", protocol)
            else
                turtle.turnRight()
                move("forward")
                turtle.turnRight()
                for l=1,(y+1)*(d+1)+2 do
                    move("forward")
                end
                turtle.turnRight()
                for l=1,(d+1)*(x-1)+1 do
                    move("forward")
                end
                turtle.turnRight()
                dropItems()
                rednet.send(serverID, "finish", protocol)
            end
        else
            if math.fmod(i,2) == 1 then
                turtle.turnRight()
            else
                turtle.turnLeft()
            end
            for k=1,d+1 do
                move("forward")
            end
            if math.fmod(i,2) == 1 then
                turtle.turnRight()
            else
                turtle.turnLeft()
            end
        end
    end
end

function workerEvents(event, sender, msg, prot)
    if event == "rednet_message" then
        if msg == "start" then
            if turtle.getFuelLevel() < 800 and turtle.getItemCount(1) == 0 then
                rednet.send(serverID, "needbackup", protocol)
            elseif turtle.getItemCount(2) < x*y then
                rednet.send(serverID, "needbackup", protocol)
            else
                work()
            end
        elseif msg == "catch" then
            if turtle.getItemCount(1) < 10 then
                rednet.send(helperID, "coal", protocol)
                return
            end
            if turtle.getItemCount(2) < x*y then
                rednet.send(helperID, "sapling", protocol)
                return
            end
            rednet.send(helperID, "ok", protocol)
        elseif msg == "disconnect" then
            print("Server has disconnected.")
            error(r)
        end
    end
end

function getCount()
    local a, b = 0, 0
    for i=1,16 do
        if turtle.getItemCount(i) > 0 then
            local k = turtle.getItemDetail(i)
            if k.name == "minecraft:sapling" then
                a = a + turtle.getItemCount(i)
            elseif k.name == "minecraft:coal" then
                b = b + turtle.getItemCount(i)
            end
        end
    end
    return a, b
end

function helperEvents(event, sender, msg, prot)
    if event == "rednet_message" then
        if msg == "refuel" and sender == serverID then
            local a, b = getCount()
            if a < 64 or b < 64 then
                rednet.send(serverID, "needbackup", protocol)
            else
                refuel = true
                rednet.send(workerID, "catch", protocol)
            end
        elseif msg == "coal" or msg == "sapling" then
            if sender == workerID and refuel then
                for i=1,16 do
                    local k = turtle.getItemDetail(i)
                    if turtle.getItemCount(i) > 0 and k.name == "minecraft:"..msg then
                        turtle.select(i)
                        turtle.drop(1)
                        rednet.send(workerID, "catch", protocol)
                        return
                    end
            end
            end
        elseif msg == "ok" and sender == workerID then
            rednet.send(serverID, "ok", protocol)
            refuel = false
        elseif msg == "disconnect" then
            print("Server has disconnected.")
            error(r)
        end
    end
end

function sendStart()
    line(1, colors.yellow, "  RTF")
        line(2, colors.green, "FUEL")
        line(3, colors.green, "SAPLING")
        line(4, colors.blue, "STOP")
        rednet.send(workerID, "start", protocol)
end

function serverEvents(event, arg1, arg2, arg3)
    if event == "monitor_touch" then
            if arg3 == 4 then
            if working then
                working = false
                os.cancelTimer(timer)
                line(1, colors.black, "  RTF")
                line(4, colors.blue, "START")
            elseif not refuel then
                working = true
                sendStart()
            end
        elseif arg3 == 5 then
            rednet.send(workerID, "disconnect", protocol)
            rednet.send(helperID, "disconnect", protocol)
            term.setBackgroundColor(colors.black)
            term.clear()
            term.setCursorPos(1,1)
            error(r)
                end
    elseif event == "rednet_message" and arg3 == protocol then
            if arg2 == "needbackup" and arg1 == workerID then
            working = false
            rednet.send(helperID, "refuel", protocol)
            line(1, colors.black, "  RTF")
                line(2, colors.blue, "FUEL")
                line(3, colors.blue, "SAPLING")
                line(4, colors.gray, "START")
            refuel = true
        elseif arg2 == "needbackup" and arg1 == helperID then
            line(4, colors.gray, "START")
            line(2, colors.red, "REFUEL")
            line(3, colors.red, "SAPLING")
            line(1, colors.black, "  RTF")
            refuel = false
        elseif arg2 == "ok" and arg1 == helperID then
            refuel = false
            working = true
            sendStart()
        elseif arg2 == "finish" then
            if working then
                timer = os.startTimer(delay)
            else
                line(1, colors.black, "  RTF")
                line(4, colors.gray, "START")
            end
        end
    elseif event == "timer" and arg1 == timer then
        sendStart()
        os.cancelTimer(timer)
    end
end

function start()
    term.setCursorPos(1,1)
        term.clear()
    checkArgs()
    print("I am "..getName())
    findModem()
    if thisType == 2 then
        findMonitor()
    end
    print("Searching...")
    findFriends()
    while true do
        local event, sender, msg, prot = os.pullEvent()
        if thisType == 1 then
            workerEvents(event, sender, msg, prot)
        elseif thisType == 2 then
            serverEvents(event, sender, msg, prot)
        elseif thisType == 3 then
            helperEvents(event, sender, msg, prot)
        end
    end
end

start() --Start application