Totoro Horizontal Miner 0.41

-- Totoro Recursive Miner 0.41 --
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 = 32
TECH_SLOTS = 8
VANILLA_CHEST = false
MAX = 1024

local moves = 0

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

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

local a = {...}
local l = tonumber(a[1]) or 10
local w = tonumber(a[2]) or 5
local comeback = a[3] or 'false'
local halfw = math.floor(w/2)

if a[1] == '?' then
  print("Usage: mine <length> [width]")
  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
    moves = moves + 1
    return true
  else
    return false
  end
end
function up()
  if robot.up() then
    loc.z = loc.z+1
    moves = moves + 1
    return true
  else
    return false
  end
end
function down()
  if robot.down() then
    loc.z = loc.z-1
    moves = moves + 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
    if gen.count() == 0 then
      for i=empty_slot, INV_SIZE do
        robot.select(i)
        if gen.insert() then break end
      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
    if not VANILLA_CHEST then
      robot.select(chest_slot)
      robot.swing()
    end
  end
end

-- =================================== M I N I N G ===================================== --
function qforward()
  while not forward() do robot.swing() end
end

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)
  -- dig one row
  for x=1, w do
    -- check up/down
    if not trash(sides.down) then go(sides.down) end
    if not trash(sides.up) then go(sides.up) end
    -- tonnel for player
    if loc.y == halfw then
      while robot.detectDown() do robot.swingDown() end
      while robot.detectUp() do robot.swingUp() end
    end
    -- move
    if x<w then
      while not forward() do robot.swing() end
    end
  end

  -- next one
  if progress%2==1 then
    turnRight()
    qforward()
    turnRight()
  else
    turnLeft()
    qforward()
    turnLeft()
  end

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

function goBack()
  while loc.y ~= 0 do
    qforward()
  end
  if loc.d == 0 then turnLeft()
  elseif loc.d == 2 then turnRight() end
  while loc.x ~= 0 do
    qforward()
  end
  turnRight()
end

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

-- statistic value
moves = 0

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

if comeback == 'true' then goBack() end

cloader.setActive(false)
print("[INFO] Chunk Loader deactivated.")
print("[STATISTIC] Moves: "..moves)