Tree Farm v4.0 by rhyser9

local treeType = 0
local resupplyOn = 0
local sapling = turtle.getItemCount(1) - 1
local bonemeal = turtle.getItemCount(2) - 1
local resupplySlot = 3
local resupplyCount = 0
local resupplyMax = 11

--************
--*Print Logo*
--************
function logo()
  shell.run('clear')
  print "***************************************"
  print "*         Variable Tree Farms         *"
  print "*             Version 4.0             *"
  print "*         Created By: rhyser9         *"
  print "***************************************"
  term.setCursorPos(1,6)
end

--*****************
--*Terminate Error*
--*****************
function terminate()
  sleep(1)
  print "Terminating Program..."
  sleep(3)
  print "Program Terminated"
  error()
end

--*******************
--*Configure Program*
--*******************
function configure()
  logo()
  print "What tree type would you like to farm?"
  write("(Oak, Birch, Spruce, Jungle, or Redwood): ")
  input = read()
  if input == "Oak" or input == "oak" then
    treeType = 0
    print "Oak Trees Selected"
  elseif input == "Birch" or input == "birch" then
    treeType = 0
    print "Birch Trees Selected"
  elseif input == "Spruce" or input == "spruce" then
    treeType = 0
    print "Spruce Trees Selected"
  elseif input == "Jungle" or input == "jungle" then
    treeType = 0
    print "Jungle Trees Selected"
  elseif input == "Redwood" or input == "RedWood" or input == "redwood" then
    treeType = 1
    resupplySlot = 4
    resupplyMax = 10
    print "Redwood Trees Selected"
  else
    print "Couldn't understand tree selection"
    treeSelection = 0
    print "Current Mode: 1x1 Vanilla Trees"
  end
  sleep(1)

  logo()
  print "Would you like to enable the Resupply function?"
  write("(Y/N): ")
  input = read()
  if input == "Y" or input == "y" then
    resupplyOn = 1
    print "Resupply Mode Enabled"
  elseif input == "N" or input == "n" then
    resupplyOn = 0
    print "Resupply Mode Not Enabled"
  else
    print "Couldn't Understand Input"
    resupplyOn = 0
    print "Resupply Mode not Enabled"
  end
  sleep(1)
end

--******************
--*Gather Materials*
--******************
function materials()
  logo()
  print "Please place Saplings in Slot #1."
    turtle.select(1)
    while turtle.getItemCount(1) == 0 do
      sleep(1)
    end

  logo()
  print "Please place bonemeal in Slot #2."
    turtle.select(2)
    while turtle.getItemCount(2) == 0 do
      sleep(1)
    end

  if treeType == 1 then
    logo()
    print "Please place Dirt in Slot #3."
      turtle.select(3)
      while turtle.getItemCount(3) == 0 do
        sleep(1)
      end
  end

  logo()
  print "Please place a Log in Slot #15."
    turtle.select(15)
    while turtle.getItemCount(15) == 0 do
      sleep(1)
    end

  logo()
  print "Please place fuel in Slot #16."
    turtle.select(16)
    while turtle.getItemCount(16) == 0 do
      sleep(1)
    end
end

--*******************
--*Calculate Runtime*
--*******************
function runtime()
  if treeType == 1 then
    sapling = math.floor((turtle.getItemCount(1) - 1) / 4)
  end

  logo()
  print "Calculating Runtime..."
  sleep(2)
  if sapling <= bonemeal then
    if sapling == 0 then
      print "Not Enough Items to Run"
      sleep(1)
      print "Terminating Program"
      sleep(3)
      print "Program Terminated"
      error()
    elseif sapling == 1 then
      print "The program should run 1 time."
    else
      write("The program should run "..sapling.." times.\n")
    end
  else
    if bonemeal == 0 then
      print "Not Enough Items to Run"
      sleep(1)
      print "Terminating Program"
      sleep(3)
      print "Program Terminated"
      error()
    elseif bonemeal == 1 then
      print "The program should run 1 time."
    else
      write("The program should run "..bonemeal.." times.\n")
    end
  end

  for i=0,4 do
    write("\nThe program will initiate in "..5 - i.." seconds")
    sleep(1)
  end
  shell.run('clear')
end

--********
--*Refuel*
--********
function refuel()
  while turtle.getFuelLevel() < 50 do
    print "  Low Fuel!"
    print "  Refueling..."
    turtle.select(16)
    turtle.refuel(2)
    sleep(2)
    if turtle.refuel() == false then
      print "  Refueling Failed!"
      terminate()
    else
      print "  Turtle Refueled"
    end
  end
end

--**********************
--*Plant & Grow Sapling*
--**********************
function plant()
  print "  Planting Tree..."
  turtle.select(1)
  if treeType == 1 then
    --First Sapling
    turtle.up()
    turtle.forward()
    turtle.placeDown()
    --Second Sapling
    turtle.forward()
    turtle.placeDown()
    --Third Sapling
    turtle.turnRight()
    turtle.forward()
    turtle.placeDown()
    --Fourth Sapling
    turtle.turnRight()
    turtle.forward()
    turtle.placeDown()
    --Return to Start
    turtle.turnRight()
    turtle.forward()
    turtle.turnLeft()
    turtle.forward()
    turtle.turnRight()
    turtle.turnRight()
    turtle.down()
  else
    turtle.place()
  end
  --Bonemeal
  turtle.select(2)
  turtle.place()
  --Verify Process
  turtle.select(15)
  if turtle.compare() then
    print "  Planting Process Successful"
  else
    print "  Planting Process Failed"
    print "  Possible Causes:"
    print "  * Saplings and Logs don't match"
    print "  * Planting Area Blocked"
    terminate()
  end
end

--**************
--*Harvest Tree*
--**************
function harvest()
  print "  Harvesting Tree..."
  turtle.select(15)
  if treeType == 1 then
    --Chop First Side
    turtle.dig()
    turtle.forward()
    while turtle.compareUp() == true do
      turtle.dig()
      turtle.digUp()
      turtle.up()
    end
    turtle.dig()
    --Chop Second Side
    turtle.select(3)
    turtle.turnRight()
    turtle.dig()
    turtle.forward()
    turtle.turnLeft()
    while turtle.compareDown() == false do
      turtle.dig()
      turtle.digDown()
      turtle.down()
    end
    turtle.dig()
    --Return to Start
    turtle.turnLeft()
    turtle.forward()
    turtle.turnLeft()
    turtle.forward()
    turtle.turnRight()
    turtle.turnRight()
  else
    while turtle.compare() == true do
      turtle.dig()
      turtle.digUp()
      turtle.up()
    end
    while turtle.detectDown() == false do
      turtle.down()
    end
  end
  print "  Harvest Completed"
end

--**********
--*Resupply*
--**********
function resupply()
  --Saplings
  turtle.select(1)
  turtle.suckDown()
  while resupplyCount ~= resupplyMax do
    if turtle.compareTo(resupplySlot + resupplyCount) == true then
      turtle.select(resupplySlot + resupplyCount)
      turtle.dropDown()
      turtle.select(1)
    end
    resupplyCount = resupplyCount + 1
  end
  resupplyCount = 0
  --Bonemeal
  turtle.select(2)
  turtle.turnLeft()
  turtle.suck()
  while resupplyCount ~= resupplyMax do
    if turtle.compareTo(resupplySlot + resupplyCount) == true then
      turtle.select(resupplySlot + resupplyCount)
      turtle.drop()
      turtle.select(2)
    end
    resupplyCount = resupplyCount + 1
  end
  resupplyCount = 0
  --Fuel
  turtle.select(16)
  turtle.turnRight()
  turtle.turnRight()
  turtle.suck()
  turtle.select(16)
  while resupplyCount ~= resupplyMax do
    if turtle.compareTo(resupplySlot + resupplyCount) == true then
      turtle.select(resupplySlot + resupplyCount)
      turtle.drop()
      turtle.select(16)
    end
    resupplyCount = resupplyCount + 1
  end
  resupplyCount = 0
  --Drop Off Extra
  turtle.turnLeft()
  while resupplyCount ~= resupplyMax do
    turtle.select(resupplySlot + resupplyCount)
    turtle.drop()
    resupplyCount = resupplyCount + 1
  end
  turtle.select(15)
  turtle.drop(turtle.getItemCount(15) - 1)
  turtle.turnLeft()
  turtle.turnLeft()
end

--***********
--*Main Loop*
--***********
configure()
materials()
runtime()

logo()
while turtle.getItemCount(1) > 1 and turtle.getItemCount(2) > 1 do
  logo()
  print "Farming Cycle Initiated"
  refuel()
  plant()
  harvest()
  if resupplyOn == 1 then
    turtle.turnRight()
    turtle.turnRight()
    resupply()
  end
  print "Farming Cycle Completed"
  if treeType == 1 then
    print "Waiting for Leaf Despawn..."
    sleep(480)
  else
    sleep(3)
  end
end

print "Program Completed"