wood

--Var
local robot = require("robot")
local gps = require("component").navigation
local inv = require("component").inventory_controller
local comp = require("computer")
local s = require("sides")
local tr = require("component").tractor_beam

local Xc,Xn,Xt,Yn,Yc,Yt,Zn,Zc,Zt
Xn,Yn,Zn = gps.getPosition()
nSide = gps.getFacing()
--Function

function forward()
local n = 0
while true do
if robot.forward() then
while tr.suck() do end return true
else
robot.swing()
n = n+1
end
if n == 20 then print("no run forward, program stop!") os.exit() end
end
end

function up()
local n = 0
while true do
if robot.up() then return true else
robot.swingUp()
n = n+1
end
if n == 20 then print("no run up, program stop!") os.exit() end
end
end

function down()
local n = 0
while true do
if robot.down() then return true else
robot.swingDown()
n = n+1
end
if n == 20 then print("no run down, program stop!") os.exit() end
end
end


function sapling()
robot.select(2)
up()
forward()
robot.placeDown()
forward()
robot.placeDown()
robot.turnLeft()
forward()
robot.placeDown()
robot.turnLeft()
forward()
robot.placeDown()
forward()
robot.turnLeft()
forward()
robot.turnLeft()
down()
end

function wood()
local n=0
robot.swing()
forward()
while true do
  if robot.detectUp() then
  n=n+1
  robot.swing()
  robot.swingUp()
  up()
  else break end
end
robot.turnLeft()
robot.swing()
forward()
robot.turnRight()
for i=1,n do
robot.swing()
robot.swingDown()
down()
end
robot.swing()
robot.turnRight()
forward()
robot.turnRight()
forward()
robot.turnAround()
end

function new_tools()
   if robot.durability() == nil then return false end
   if robot.durability()<150 then
   robot.select(1)
   inv.equip()
       if robot.durability()<150 then
       return false
       else
       return true
    end
    else
    return true
    end
end

function detect_wood()
 if robot.detect() then
  up()
   if robot.detect() then
    down()
    return "wood"
   else
    down()
    return "sapling"
   end
 else
  return "clear"
 end
end

function run_a()
robot.turnLeft()
for i=1,10 do
forward()
end
robot.turnRight()
end

function run_b()
robot.turnRight()
forward()
robot.turnRight()
for i=1,10 do
forward()
end
robot.turnRight()
forward()
robot.turnRight()
end

function run_c()
robot.turnRight()
for i=1,10 do
forward()
end
robot.turnLeft()
end

function run_d()
robot.turnLeft()
forward()
forward()
robot.turnRight()
for i=1,10 do
forward()
end
robot.turnRight()
forward()
forward()
robot.turnLeft()
end

function bksp(area)
if area == "chest" then
  robot.turnLeft()
  forward()
  forward()
  robot.turnRight()
  Xc,Yc,Zc = gps.getPosition()
  Xt,Yt,Zt = gps.getPosition()
    if (nSide == 2) or (nSide == 3) then
      while Xn~=Xt do
      forward()
      Xt,Yt,Zt = gps.getPosition()  end
    robot.turnRight()
    else
      while Zn~=Zt do
      forward()
      Xt,Yt,Zt = gps.getPosition() end
    robot.turnRight()
    end
     if (nSide == 2) or (nSide == 3) then
      while Zn~=Zt do
      forward()
      Xt,Yt,Zt = gps.getPosition() end
    else
      while Xn~=Xt do
      forward()
      Xt,Yt,Zt = gps.getPosition()   end
    end
elseif area == "wood" then
  Xt,Yt,Zt = gps.getPosition()
  robot.turnAround()
    if (nSide == 2) or (nSide == 3) then
      while Zc~=Zt do
      forward()
      Xt,Yt,Zt = gps.getPosition() end
   robot.turnLeft()
 else
      while Xc~=Xt do
      forward()
      Xt,Yt,Zt = gps.getPosition() end
   robot.turnLeft()
    end
    if (nSide == 2) or (nSide == 3) then
      while Xc~=Xt do
      forward()
      Xt,Yt,Zt = gps.getPosition() end
 else
      while Zc~=Zt do
      forward()
      Xt,Yt,Zt = gps.getPosition() end
    end
  robot.turnLeft()
  forward()
  forward()
  robot.turnLeft()
end
end

function chest(tools)
    for i=3,16 do
  robot.select(i)
  if (robot.count(i) ~= 0) and (not robot.drop(64)) then robot.turnAround() print("chest full, stop!") os.exit() end
  end
  robot.turnRight()
  robot.select(2)
   for i=1,5 do
   if robot.count(2) ~= 64 then robot.suck(robot.space(2)) else break end
  end
  robot.turnLeft()
  if not count_sapl() then return "sapling" end
if not tools then
  robot.select(1)
  robot.drop()
  inv.equip()
  robot.drop()
  if not robot.suckDown(1) then return "tools" end
  inv.equip()
  if not robot.suckDown(1) then return "tools" end
  end
  return true
end

function count_sapl()
if robot.count(2) < 5 then return false else return true end
end

function full()
if robot.count(13) == 0  then return true else return false end
end

function energy()
if comp.energy() < 1800 then os.sleep(180) end
end

-- MultiFunction
function woodAll()
  while true do
       if count_sapl() and full() and new_tools() then
       detect = detect_wood()
           if detect == "clear" then sapling() break
           elseif detect == "wood" then wood() sapling() break
           elseif detect == "sapling" then break end
   else
    bksp("chest")
 _exit = chest(new_tools())
   if "tools" == _exit then robot.turnAround() print("not tools, stop!") os.exit()
    elseif "sapling" ==_exit then robot.turnAround() print("not sapling, stop!") os.exit()
   end
 bksp("wood")
  end
 end
end

--Programs:

robot.turnRight()
while true do
  for i=1,3 do
   energy()
   woodAll()
   os.sleep(10)
   energy()
   run_a()
   end
energy()
woodAll()
os.sleep(10)
energy()
run_b()
for i=1,3 do
   energy()
   woodAll()
   os.sleep(10)
   energy()
   run_c()
  end
energy()
woodAll()
os.sleep(10)
energy()
run_d()
end