-- pastebin.com/xg7p10Vt
-- pastebin.com/hrXuWsAS

local robot = require("robot")
local computer = require("computer")
local sides = require("sides")
local component = require("component")
local ser = require("serialization")

local c = component.crafting
local g = component.generator
local inventory = component.inventory_controller

local left = robot.turnLeft
local right = robot.turnRight
local TURN = robot.turnAround
local d = robot.detect
local dUp = robot.detectUp
local dDown = robot.detectDown

local side = sides.front

local args = {...} -- ... gets the program's arguments somehow
local rowsToDig = args[1] -- getting the first argument in the argument table


local nRail_names = {}
nRail_names[1] = "minecraft:rail"
nRail_names[2] = "TConstruct:rail.wood"

local Fuel_names = {}
Fuel_names[1] = "minecraft:coal"
Fuel_names[2] = "minecraft:planks"
Fuel_names[3] = "minecraft:log"
Fuel_names[4] = "minecraft:blaze_rod"
Fuel_names[5] = "minecraft:lava_bucket"


print(robot.detect())


-- Functions for movements

local function goUp()
  while dUp() do
    robot.swingUp()
    os.sleep(2)
  end
  robot.up()
  os.sleep(0.8)
end

local function goDown()
  while dDown() do
    robot.swingDown()
    os.sleep(2)
  end
  robot.down()
  os.sleep(0.8)
end

local function go()
  while d() do    --- while robot.detect == true do
    robot.swing()
    os.sleep(2)
  end
  robot.forward()
  os.sleep(0.8)
end

-- Extra Functions to break blocks without moving, clear sand blocks or anormalies

local function rSand()
  while d() do    --- while robot.detect == true do
    robot.swing()
    os.sleep(2)
  end
  robot.turnAround()
  while d() do
    robot.swing()
  os.sleep(2)
end
end


-- Place Blocks with Dev/ Null

local function Bf()
  if not d() then
    robot.select(13)
    robot.place()
  end
end

local function Bd()
  if not dDown() then
    robot.select(13)
    robot.placeDown()
  end
end

local function Bu()
  if not dUp() then
    robot.select(13)
    robot.placeUp()
  end
end


-- Movement script

local function Movement()
Bd() ; go()
Bd() ; left() ; go() ; Bf() ; Bd() ; TURN() ; go() ; go() ; Bd() ; Bf() ; goUp()
Bf() ; TURN() ; go() ; go() ; Bf() ; goUp()
Bf() ; Bu() ; TURN() ; go() ; Bu() ; go() ; Bf() ; Bu()
TURN() ; go() ; goDown() ; goDown() ; rSand() ; left()
end





local function getRail(stack_info)
  if robot.count(4) <=2 then
right() ; robot.select(5) ; robot.drop()
robot.select(14) ; robot.place() ; robot.select(4)

if stack_info == nil then
    return false
  end
  for index, known_rail_name in ipairs(nRail_names) do
    if stack_info.name == known_rail_name then
      return true
    end
  end
  return false
end

for slot=1,27 do
  local stack_info = inventory.getStackInSlot(side, slot)

  --- inspect the stack_info when it is not nil
  if getRail(stack_info) then
    print(stack_info.name, "found in slot", slot)
    inventory.suckFromSlot(sides.front, slot)
    break
  end
end
robot.select(5) ; inventory.dropIntoSlot(side, 1) ; robot.swing() ; robot.transferTo(14) ; left()
end



local function refuel(stack_info)
  if stack_info == nil then
    return false
  end
  for index, known_fuel_name in ipairs(Fuel_names) do
    if stack_info.name == known_fuel_name then
      return true
    end
  end
  return false
end

if computer.energy() <=200 then
right() ; robot.select(14) ; robot.place() ; robot.select(16) ; robot.drop() ; robot.select(15)
for slot=1,27 do
  local stack_info = inventory.getStackInSlot(side, slot)

  --- inspect the stack_info when it is not nil
  if refuel(stack_info) then
    print(stack_info.name, "found in slot", slot)
    inventory.suckFromSlot(sides.front, slot)
    break
  end
end
robot.select(14) ; robot.swing()
robot.select(8) ; robot.place() ; robot.select(16) ; g.remove() ; inventory.dropIntoSlot(side, 1) ; robot.select(15) ; g.insert() ; os.sleep(20)
robot.select(8) ; robot.swing()
left()
end

local function digit()
  for i = 1, rowsToDig do

refuel(stack_info)
getRail(stack_info)
TURN() ; robot.select(4) ; robot.place() ; TURN()
Movement()
end
end

digit()