Totoro Recursive Miner

-- Totoro Recursive Miner 0.2 --
local robot = require('robot')
local event = require('event')
local sides = require('sides')
local computer = require('computer')
local com = require('component')
local gen = com.generator
local cloader = com.chunkloader

INV_SIZE = 24
TECH_SLOTS = 8
MAX = 1024

local trash_slots = TECH_SLOTS - 1
local chest_slot = TECH_SLOTS
local empty_slot = TECH_SLOTS + 1

local stepside = {}
stepside[0] = sides.up
stepside[1] = sides.down

local loc = {x=0, y=0, z=0, d=0}
local lode = {}

local a = {...}
local l = tonumber(a[1]) or 10
local h = tonumber(a[2]) or 4

if a[1] == '?' then
  print("Usage: mine <length> [height]")
  return
end

-- ============================= N A V I G A T I O N ====================================== --
function forward()
  if robot.forward() then
    if loc.d == 0 then loc.y = loc.y+1
    elseif loc.d == 1 then loc.x = loc.x+1
    elseif loc.d == 2 then loc.y = loc.y-1
    else loc.x = loc.x-1 end
    return true
  else
    return false
  end
end
function up()
  if robot.up() then
    loc.z = loc.z+1
    return true
  else
    return false
  end
end
function down()
  if robot.down() then
    loc.z = loc.z-1
    return true
  else
    return false
  end
end
function turnRight()
  loc.d = (loc.d+1)%4
  robot.turnRight()
end
function turnAround()
  loc.d = (loc.d+2)%4
  robot.turnAround()
end
function turnLeft()
  loc.d = (loc.d+3)%4
  robot.turnLeft()
end

-- ============================= F U E L  C O N T R O L ================================ --
function check_fuel()
  if (computer.maxEnergy() - computer.energy()) > 1000 then
    for i=empty_slot, INV_SIZE do
      robot.select(i)
      if gen.insert() then break end
    end
  end
end

-- ============================= I T E M  C O N T R O L ================================ --
function check_loot()
  for i=empty_slot, INV_SIZE do
    if robot.count(i) == 0 then return false end
  end
  return true
end

function unload()
  -- place ender chest
  robot.select(chest_slot)
  robot.swing()
  if robot.place() then
    -- put items
    for i=empty_slot, INV_SIZE do
      robot.select(i)
      while robot.drop() do end
    end

    -- grab ender chest
    robot.select(chest_slot)
    robot.swing()
  end
end

-- =================================== M I N I N G ===================================== --
function trash(side)
  for i=1, trash_slots do
    robot.select(i)
    if side == sides.up then
      a,t = robot.detectUp()
      if t == 'air' or t=='liquid' then return true end
      if robot.compareUp() then return true end
    elseif side == sides.front then
      a,t = robot.detect()
      if t == 'air' or t=='liquid' then return true end
      if robot.compare() then return true end
    else
      a,t = robot.detectDown()
      if t == 'air' or t=='liquid' then return true end
      if robot.compareDown() then return true end
    end
  end
  return false
end

function mine(side)
  local direct = loc.d
  local backdir = (direct+2)%4

  if side == sides.up then
    c = 0
    while not up() do
      robot.swingUp()
      c = c + 1
      if c>30 then return end
    end
  elseif side == sides.front then
    c = 0
    while not forward() do
      robot.swing()
      c = c + 1
      if c>30 then return end
    end
  elseif side == sides.down then
    c = 0
    while not down() do
      robot.swingDown()
      c = c + 1
      if c>30 then return end
    end
  end

  lode[loc.x*MAX*MAX + loc.y*MAX + loc.z] = true

  -- check further direction
  if lode[loc.x*MAX*MAX + loc.y*MAX + (loc.z+1)] == nil then
    if not trash(sides.up) then mine(sides.up)
      else lode[loc.x*MAX*MAX + loc.y*MAX + (loc.z+1)] = false end
  end
  if lode[loc.x*MAX*MAX + loc.y*MAX + (loc.z-1)] == nil then
    if not trash(sides.down) then mine(sides.down)
    else lode[loc.x*MAX*MAX + loc.y*MAX + (loc.z-1)] = false end
  end

  for i=loc.d, loc.d+3 do
    local a = i%4
    local x = loc.x
    local y = loc.y
    if a == 0 then y = y + 1
    elseif a == 1 then x = x + 1
    elseif a == 2 then y = y - 1
    else x = x - 1 end
    if lode[x*MAX*MAX + y*MAX + loc.z] == nil then
      while loc.d < a do turnRight() end
      while loc.d > a do turnLeft() end
      if not trash(sides.front) then mine(sides.front)
      else lode[x*MAX*MAX + y*MAX + loc.z] = false end
    end
  end

  -- come back
  if side == sides.up then
    down()
  elseif side == sides.front then
    while loc.d < backdir do turnRight() end
    while loc.d > backdir do turnLeft() end
    while not forward() do robot.swing() end
  elseif side == sides.down then
    up()
  end
end

function go(side)
  direct = loc.d
  lode = {}
  mine(side)
  while loc.d < direct do turnRight() end
  while loc.d > direct do turnLeft() end
end

function step(progress)
  -- check top/down
  if not trash(stepside[progress%2]) then go(stepside[progress%2]) end
  turnLeft()

  -- dig one column
  for y=1, h do
    -- check sides
    if not trash(sides.front) then go(sides.front) end
    turnAround()
    if not trash(sides.front) then go(sides.front) end

    -- move up or down
    if y<h then
      if progress%2 == 1 then
        while not up() do robot.swingUp() end
      else
        while not down() do robot.swingDown() end
      end
    end
  end

  -- check down/top
  if not trash(stepside[(progress+1)%2]) then go(stepside[(progress+1)%2]) end

  -- turn forward
  if h%2 == 0 then
    turnRight()
  else
    turnLeft()
  end

  -- move forward
  if progress<l then
    c = 0
    while not forward() do
      robot.swing()
      c = c + 1
      if c>30 then return end
    end
  end

  -- fuel checking
  check_fuel()
  -- loot checking
  if check_loot() then unload() end
end

-- ===================================== M A I N ======================================= --
cloader.setActive(true)
print("[INFO] Chunk Loader is active.")
print("[INFO] Press any key to break mining...")

for i=1, l do
  step(i)
end

while down() do end

cloader.setActive(false)
print("[INFO] Chunk Loader deactivated.")