quarry

os.loadAPI("inv")
os.loadAPI("t")

local x = 0
local y = 0
local z = 0
local lastx = 0
local lasty = 0
local lastz = 0
local first = true
local max = 32
local deep = 64
local facingfw = true
local avoidBlocks = {}
local COAL = "coal"
local CHAR = "char"

local OK = 0
local ERROR = 1
local LAYERCOMPLETE = 2
local OUTOFFUEL = 3
local FULLINV = 4
local BLOCKEDMOV = 5
local USRINTERRUPT = 6

local CHARCOALONLY = false
local USEMODEM = false


-- Arguments
local tArgs = {...}
for i=1,#tArgs do
    local arg = tArgs[i]
    if string.find(arg, "-") == 1 then
        for c=2,string.len(arg) do
            local ch = string.sub(arg,c,c)
            if ch == 'c' then
                CHARCOALONLY = true
            elseif ch == 'm' then
                USEMODEM = true
            else
                write("Invalid flag '")
                write(ch)
                print("'")
            end
        end
    end
end


function out(s)

    s2 = s .. " @ [" .. x .. ", " .. y .. ", " .. z .. "]"

    print(s2)
    if USEMODEM then
        rednet.broadcast(s2, "miningTurtle")
    end
end

function dropInChest()
    turtle.turnLeft()

    local success, data = turtle.inspect()
    out(data.name)
    if success then
      if string.find(data.name,"chest") then
            out("Dropping items in chest")

            for i=1, 16 do
                turtle.select(i)

                data = turtle.getItemDetail()

                if data ~= nil and
        (data.name == "minecraft:coal" or data.name == "minecraft:charcoal") == false and
                    (data.damage == nil or data.name ~= "minecraft:coal" or data.name ~= "minecraft:charcoal") then
        turtle.drop()
    end
            end
  end
    end

    turtle.turnRight()

end

function goDown()
    while true do
        if turtle.getFuelLevel() <= fuelNeededToGoBack() then
            if not refuel() then
                return OUTOFFUEL
            end
        end

        if not turtle.down() then
            turtle.up()
            z = z+1
            return
        end
        z = z-1
    end
end

function fuelNeededToGoBack()
    return -z + x + y + 2
end

function refuel()
    for i=1, 16 do
        -- Only run on Charcoal
        turtle.select(i)

        item = turtle.getItemDetail()
        if item and string.find(item.name,"coal") and turtle.refuel(1) then
            return true
        end
    end

    return false
end

function moveH()
    if inv.isInventoryFull() then
        out("Dropping thrash")
        inv.dropThrash()

        if inv.isInventoryFull() then
            out ("Stacking items")
            inv.stackItems()
        end

        if inv.isInventoryFull() then
            out("Full inventory!")
            return FULLINV
        end
    end

    if turtle.getFuelLevel() <= fuelNeededToGoBack() then
        if not refuel() then
            out("Out of fuel!")
            return OUTOFFUEL
        end
    end

    local sd, bd = turtle.inspectDown()

    if sd then

        if has_value(avoidBlocks, bd.name) then

            out("Hit bad block, remove manually, then restart")
            return BLOCKEDMOV

        end
    end


    local su, bu = turtle.inspectUp()

    if su then

        if has_value(avoidBlocks, bu.name) then

            out("Hit bad block, remove manually, then restart")
            return BLOCKEDMOV

        end
    end

    local sf, bf = turtle.inspect()

    if sf then

        if has_value(avoidBlocks, bf.name) then

            out("Hit bad block, remove manually, then restart")
            return BLOCKEDMOV

        end
    end

    if facingfw and y<max-1 then
    -- Going one way
        local dugFw = t.dig()
        if dugFw == false then
            out("Hit bedrock, can't keep going")
            return BLOCKEDMOV
        end
        t.digUp()
        t.digDown()

        if t.fw() == false then
            return BLOCKEDMOV
        end

        y = y+1

    elseif not facingfw and y>0 then
    -- Going the other way
        t.dig()
        t.digUp()
        t.digDown()

        if t.fw() == false then
            return BLOCKEDMOV
        end

        y = y-1

    else
        if x+1 >= max then
            t.digUp()
            t.digDown()
            return LAYERCOMPLETE -- Done with this Y level
        end

        -- If not done, turn around
        if facingfw then
            turtle.turnRight()
        else
            turtle.turnLeft()
        end

        t.dig()
        t.digUp()
        t.digDown()

        if t.fw() == false then
            return BLOCKEDMOV
        end

        x = x+1

        if facingfw then
            turtle.turnRight()
        else
            turtle.turnLeft()
        end

        facingfw = not facingfw
    end

    return OK
end

function digLayer()

    local errorcode = OK

    while errorcode == OK do
        if USEMODEM then
            local msg = rednet.receive(1)
            if msg ~= nil and string.find(msg, "return") ~= nil then
                return USRINTERRUPT
            end
        end
        errorcode = moveH()
    end

    if errorcode == LAYERCOMPLETE then
        return OK
    end

    return errorcode
end

function goToOrigin(up)

    if facingfw then

        turtle.turnLeft()

        t.fw(x)

        turtle.turnLeft()

        t.fw(y)

        turtle.turnRight()
        turtle.turnRight()

    else

        turtle.turnRight()

        t.fw(x)

        turtle.turnLeft()

        t.fw(y)

        turtle.turnRight()
        turtle.turnRight()

    end

    x = 0
    y = 0
    facingfw = true

    if up then
        goUp()
    end

end

function has_value(tab, val)
    for index, value in ipairs(tab) do
        if value == val then
            return true
        end
    end

    return false
end

function goUp()

    while z < 0 do

        t.up()

        z = z+1

    end

end

function returnToLast()

    t.down(lastz)

    t.fw(lasty)

    turtle.turnRight()

    t.fw(lastx)

    if lastx % 2 == 1 then
        turtle.turnRight()
    else
        turtle.turnLeft()
    end

    x = lastx
    y = lasty
    z = -lastz
    facingfw = true

end

function mainloop()

    while true do

        local errorcode = digLayer()

        if errorcode ~= OK then
            lastx = x
            lasty = y
            lastz = -z
            goToOrigin(true)
            return errorcode
        end

        goToOrigin(false)

        for i=1, 3 do
            t.digDown()
            success = t.down()

            if not success then
                lastx = x
                lasty = y
                lastz = -z
                goToOrigin(true)
                return BLOCKEDMOV
            end

            z = z-1
            out("Z: " .. z)

        end
    end
end

if USEMODEM then
    rednet.open("right")
end

out("\n\n\n-- WELCOME TO THE MINING TURTLE --\n\n")

while true do
    if first then
        goDown()
        first = false
    else
        returnToLast()
    end

    local errorcode = mainloop()
    dropInChest()

    if errorcode ~= FULLINV then
        break
    end
end

if USEMODEM then
    rednet.close("right")
end