Computercraft mining program that detects and refuels using lava

drillmine.lua

---------------------------------------
-- drillmine.lua by valvate --
-- But why? you ask there are
-- other drill mining progarms
-- well this one detects lava
-- and uses it to refuel
-- don't worry about running out of fuel
-- he automaticaly checks his fuel level
-- and tells you if he has enough fuel
-- for another trip!

-- WARNING IF HIS CHUNK UNLOADS
-- HE WILL STOP AND NEVER COME BACK
----------------------------------------
-- Usage:
-- Simply place an empty bucket in his 16th slot
-- And run the script
-- eg drillmine 68
-- "68" is the turtles y coradanite
-- WARNING if you put something lower then his Y
-- Then he might get stuck in bedrock
----------------------------------------
-- TODO
-- Create custom move function that deals
-- with falling blocks and
-- keeps track of y posision
-- create config file and
-- create startup file so he can keep
-- going even if the chunk unloads
-- add gps support so you don't need to
-- give it its y every time
-- clean up the code its ugly af rn
--------------------------------------
-- Manage args --
local args = { ... }
if #args ~= 1 then
    print('Usage: drillmine <CurrentY>')
    return
end
-- Define varables
local depth2dig = tonumber(args[1]) - 5
local debug = false
local vals = {
    'minecraft:diamond_ore',
    'minecraft:iron_ore',
    'minecraft:coal_ore',
    'minecraft:gold_ore',
    'minecraft:redstone_ore',
    'minecraft:emerald_ore',
    'quark:biotite_ore'
}

-- Print Functions

function printDebug(message)
    if debug then
        term.setTextColor(term.isColor() and colors.gray or colors.white)
        print("[DEBUG]: "..message)
        term.setTextColor(colors.white)
        rednet.broadcast('[DEBUG]: '..message)
    end
end


function printInfo(message)
    term.setTextColor(term.isColor() and colors.blue or colors.white)
    print("[INFO]: "..message)
    rednet.broadcast('[INFO]: '..message)
    term.setTextColor(colors.white)
end

function printError(message)
    term.setTextColor(term.isColor() and colors.blue or colors.white)
    print("[ERROR]: "..message)
    rednet.broadcast('[ERROR]: '..message)
    term.setTextColor(colors.white)
end
-- End print functions

function checkfuel()
    -- Check if we have enough fuel for a run
    if turtle.getFuelLevel() <= (depth2dig * 2) + 3 then
        printError('You need '..tostring(((blocks2dig * 2) + 3) - turtle.getFuelLevel())..' More fuel')
    end
end

function process_block()
    local _,block = turtle.inspect()
    if block.name then
      printDebug('checking block '..block.name)
      for i=1,#vals do
         if block.name == vals[i] then
            printInfo('Found '..block.name)
            turtle.dig()
        elseif block.name == 'minecraft:lava' then
            printInfo('Found lava refueling...')
            turtle.select(16)
            turtle.place()
            turtle.refuel(1)
            printDebug('Fuel level is '..turtle.getFuelLevel())
        end
    end
    end
end
-- Start of program --
-- Going down loop
rednet.open('right')
printDebug('depth2dig set to: '..depth2dig)
checkfuel()
for i=1,depth2dig do
    if turtle.detectDown() then
        turtle.digDown()
        turtle.down()
    else
        turtle.down()
    end
    for i=1,3 do
        -- This should start with a check and end with a check but i'm not sure how do to it
        -- So i've used this ugly method
        process_block()
        turtle.turnRight()
    end
    -- This is ugly but faster i hope i can find a cleaner way
    process_block()
    if i%10 == 0 then
      printInfo('at depth '..i)
    end
end
-- TODO Find a way to avoid having two loops for virtually the same thing
-- Going up loop
printDebug('Moving to next shaft')
for i=1,3 do
  if turtle.detect() then
    turtle.dig()
  end
  turtle.forward()
end
printInfo('Going up')
for i=1,depth2dig do
    if turtle.detectUp() then
        turtle.digUp()
        turtle.up()
    else
        turtle.up()
    end
    for i=1,3 do
        -- This should start with a check and end with a check but i'm not sure how do to it
        -- So i've used this ugly method
        process_block()
        turtle.turnRight()
    end
    -- This is ugly but faster i hope i can find a cleaner way
    process_block()
    if i%10 == 0 then
      printInfo('At depth '..i)
    end
end
printInfo('Returned to surface!')

Leave comments with suggestions!