TreeFarm2

-- ************ TreeFarm **************
-- * 2.0 changed expected farm layout, faster
-- * 1.6 fixed bug: everthing in slot 1 can be fuel
-- * 1.5 fixed bug: crash when 1x1 tree in 2x2 farm
-- * 1.4 track support in progress (still not functional)
-- * 1.3 fixed bug in go()
-- * 1.2 fixed sapling collection at the sides, code refactor, 2x2 tree girth support
-- * 1.1 inventroy fix, intro sign, wood counter, timer
-- * 1.0 release!
-- ************************************

verNum = "v2.0"

local startTime = os.clock()

local tArgs = { ... }
local delay = 0.25

local level = 1
local nextTurn = 0
local mode
local nWood
local chopDownTree
local screenX, screenY = term.getSize()

local tTurns = {
  [0] = turtle.turnRight,
  [1] = turtle.turnLeft,
}

local forward = {
  go = turtle.forward,
  dig = turtle.dig,
  detect = turtle.detect,
  inspect = turtle.inspect,
  attack = turtle.attack,
  drop = turtle.drop,
}

local back = {
  go = turtle.back,
  inspect = function () return false end,
}

local up = {
  go = turtle.up,
  dig = turtle.digUp,
  detect = turtle.detectUp,
  inspect = turtle.inspectUp,
  attack = turtle.attackUp,
  drop = turtle.dropUp,
}

local down = {
  go = turtle.down,
  dig = turtle.digDown,
  detect = turtle.detectDown,
  inspect = turtle.inspectDown,
  attack = turtle.attackDown,
  place = turtle.placeDown,
  drop = turtle.dropDown,
}

local inventoryStorageSlots = {
  [15] = {name = "minecraft:coal", slot = 16, commonName = "Charcoalz"},
  [13] = {name = "minecraft:sapling", slot = 15, commonName = "tree thingies"},
}

local signalBlock = {
  ["minecraft:stained_hardened_clay"] = true,
}

local sapling = {
  name = "Sapling",
  ["minecraft:sapling"] = true,
  ["minecraft:golden_rail"] = true,
  ["Railcraft:track"] = true,
  ["IC2:blockRubSapling"] = true,
}

local leaves = {
  ["minecraft:leaves"] = true,
  ["IC2:blockRubLeaves"] = true,
}

local wood = {
  ["minecraft:log"] = true,
  ["minecraft:log2"] = true,
  ["IC2:blockRubWood"] = true,
}

local fuel = {
  ["minecraft:coal"] = 80,
  ["minecraft:log"] = 15,
  ["minecraft:log2"] = 15,
  ["minecraft:stick"] = 5,
  ["IC2:blockRubWood"] = 15,
}

container = {
  ["minecraft:chest"] = true,
  ["minecraft:trapped_chest"] = true,
  ["JABBA:barrel"] = true,
  ["IronChest:BlockIronChest"] = true,
  ["EnderStorage:enderChest"] = true,
  ["CarpentersBlocks:blockCarpentersSafe"] = true,
}

local function displayLogo()
  term.clear()
  term.setCursorPos(1,2)
  print("**** KaosKrise TreeFarm "..verNum.." ****")
  print()
  print("Welcom, Gud Sir! Im Gonna be ur guide on dis marvelous advendchur.")
  print()
end

local function formatSeconds(seconds)
  if seconds >= 3600 then
    return string.format("%d:%.2d:%.2d", math.floor(seconds/3600), math.floor(seconds/60%60), math.floor(seconds%60))
  else
    return string.format("%d:%.2d", math.floor(seconds/60%60), math.floor(seconds%60))
  end
end

local function displayTime()
  local x, y = term.getCursorPos()
  local time = os.clock()
  local runTime = time - startTime
  local string = formatSeconds(runTime)
  term.setCursorPos(screenX -(#string) ,screenY)
  write(string)
  term.setCursorPos(x, y)
  return true
end

local function countWood()
  nWood = nWood +1
  local x, y = term.getCursorPos()
  term.setCursorPos(1,screenY)
  write("I has chopd "..nWood.." wood itamz")
  term.setCursorPos(x, y)
end

local function prnLines(...)
  local x, y = term.getCursorPos()
  for i,v in ipairs(arg) do
    term.clearLine()
    print(v)
  end
  term.setCursorPos(x,y)
end

local function isItem(slot)
  slot = slot or turtle.getSelectedSlot()
  local data = turtle.getItemDetail(slot)

  if data then
    return data.name
  else
    return false
  end
end

local function selectItem(item)
  for i=1,16 do
    if item[isItem(i)] then
      turtle.select(i)
      return true
    end
  end
  return false
end

local function inspectBlock(direction)
  if direction == nil then direction = forward end
  local success, data = direction.inspect()
  if success then
    return data.name, data.metadata
  else
    return false
  end
end

local function refuel()
  local fuelLevel = turtle.getFuelLevel()
  if fuelLevel == "unlimited" or fuelLevel > 0 then
    return true
  end

  local function tryRefuel()
    local bestFuel
    local bestFuelVal = 0
    for i=1,16 do
      local itemInSlot = isItem(i)
      if fuel[itemInSlot] and fuel[itemInSlot] > bestFuelVal then
        bestFuel = i
        bestFuelVal = fuel[itemInSlot]
      end
    end
    if bestFuel then
      turtle.select(bestFuel)
      if turtle.refuel(1) then
        turtle.select(1)
        return true
      end
    end
    return false
  end

  if not tryRefuel() then
    displayLogo()
    prnLines( "Add more fuel to continue.","" )
    while not tryRefuel() do
      displayTime()
      sleep(1)
    end
    prnLines( "Resuming...","" )
  end
  return true
end

local function tryDig(direction, block)
  block = block or inspectBlock(direction)
  turtle.select(1)
  if direction.dig() then
    if wood[block] then
      countWood()
    end
    return true
  end
  return false
end

local function goDirection(direction, block)
  block = block or inspectBlock(direction)
  if not refuel() then
    return false
  end

  while displayTime() and not direction.go() do
    if block and not tryDig(direction, block) then
      return false
    elseif direction.attack() then
      prnLines("smack!","Out of my wayz!")
    else
      sleep(delay)
    end
  end

  return true
end

--[[
Syntax:   go( string detectedBlockname)
Returns:  boolean whether the turtle succeeded.
--]]
local function go(block)
  return goDirection(forward, block)
end

local function goBack(block)
  return goDirection(back, block)
end

local function goSuck(block)
  turtle.suckDown()
  turtle.suck()
  return goDirection(forward, block)
end

local function goUp(block)
  level = level + 1
  return goDirection(up, block)
end

local function goDown(block)
  level = level -1
  return goDirection(down, block)
end

local function plant()
  if selectItem(sapling) and down.place() then
    return true
  else
    return false
  end
end

-- Tree Girth: 1x1:
local function chopDownTree_1x1()
  local block = inspectBlock(up)
  while wood[block] do
    goUp(block)
    block = inspectBlock(up)
  end
  while level > 1 do
    goDown()
  end
  tryDig(down)
end

-- Tree Girth: 2x2:
local function chopDownTree_2x2()
  local topBlockFound = false
  local turnCount

  -- helper function - turn until find wood log and return num of turns
  local function findWood()
    local turnCount = 0
    for i=1,4 do
      if wood[inspectBlock()] then
        return turnCount
      end
      turtle.turnRight()
      turnCount = i
    end
    return turnCount
  end

  -- helper function - finding pos at top and choping the lonely top block
  local function findPos()
    local block = inspectBlock()
    if not wood[block] then -- top block found
      topBlockFound = true
      go()
      goDown()
    else -- top block still there
      tryDig(forward, block)
      go()
    end

    block = inspectBlock(up)
    if wood[block] then -- top block found
      tryDig(up, block)
      topBlockFound = true
    end

    turnCount = findWood()
    if turnCount % 4 == 0 then
      --  local event, key = os.pullEvent( "key" )  -- DEBUG BREAK POINT
      goBack()
      return false
    else
      go()
      tTurns[math.floor(turnCount / 2)]()
    end

    if not topBlockFound then
      block = inspectBlock(up)
      if wood[block] then -- top block found
        tryDig(up, block)
        topBlockFound = true
      else
        goUp()
      end
    end
    tryDig(forward)
    return true
  end

  -- check if it's a little tree
  if not wood[inspectBlock()] then
    prnLines("Awwww, it's a little one!","")
    return chopDownTree_1x1()
  end

  -- going up the tree:
  local blockUp = inspectBlock(up)
  while wood[blockUp] do
    tryDig(forward)
    goUp(blockUp)
    blockUp = inspectBlock(up)
  end

  local bigTree = false
  if findPos() then
    bigTree = true
  end

  -- going back down
  while level > 1 do
    goDown()
    tryDig(forward)
  end

  -- remove the stump and replant
  if bigTree then
    for i=1,3 do
      tryDig(down)

      if not plant() then
        prnLines("Ohhh noezz ","I needz teh Saplings!")
        while not plant() do
          displayTime()
          sleep(1)
        end
      end

      plant()
      if i >= 2 and i <= 3 then
        tTurns[math.floor(turnCount / 2)]()
      end
      goSuck()
    end
  end

  tryDig(down)
end


local function unload(direction)
  turtle.select(1)
  prnLines("I has findz chest.", "Will try 2 unload mah stuff...")

  local dropCount = 0

  local item = isItem(16)
  if "minecraft:coal" ~= item then
    turtle.select(16)
    dropCount = dropCount + turtle.getItemCount()
    direction.drop()
  end

  for i=1,15 do
    if turtle.getItemCount(i) > 0 then
      turtle.select(i)
      item = isItem(i)
      if item then
        if item == "minecraft:coal" then
          turtle.transferTo(16)
          if turtle.getItemCount(i) > 0 then
            dropCount = dropCount + turtle.getItemCount()
            direction.drop()
          end
        elseif sapling[item] then
          --          if not turtle.transferTo(15) then
          turtle.transferTo(15)
          if turtle.getItemCount(i) > 0 then
            dropCount = dropCount + turtle.getItemCount()
            direction.drop()
          end
          --          end
        else
          dropCount = dropCount + turtle.getItemCount()
          direction.drop()
        end
      end
    end
  end

  turtle.select(1)
  if dropCount > 0 then
    prnLines("I has putz "..dropCount.." itamz in teh chest. Yay!")
  else
    prnLines("I has nothin 2 giv :(","maybe nxt tiem :)")
  end
end

local function nextFreeSlot()
  for i=1,16 do
    if turtle.getItemCount(i) == 0 then
      return true, i
    end
  end
  return false
end

local function refill(meta)
  if inventoryStorageSlots[meta] then
    local name = inventoryStorageSlots[meta].name
    local slot = inventoryStorageSlots[meta].slot

    while turtle.getItemCount(slot) > 0 and name ~= isItem(slot) do
      local success, freeSlot =  nextFreeSlot()
      if success then
        turtle.select(slot)
        turtle.transferTo(freeSlot)
      else
        prnLines("I has too much stuffz!","")
        sleep(delay)
      end
    end

    if turtle.getItemSpace(slot) > 0 then
      turtle.select(slot)
      goDown()
      prnLines("I needz "..turtle.getItemSpace().." "..inventoryStorageSlots[meta].commonName)
      turtle.suck(turtle.getItemSpace())
      goUp()
      turtle.select(1)
    end
  else
    return false
  end
end

local function turn()
  tTurns[nextTurn]()
  if forward.detect() then
    tTurns[nextTurn]()
  else
    go()
  end
  tTurns[nextTurn]()
  nextTurn = 1 - nextTurn
end

local function patrol()
  while true do
    repeat
      refuel()
      displayTime()
    until not forward.go()
    local block, meta = inspectBlock()
    if block then
      return block, meta
    elseif forward.attack() then
      print("smack!")
    else
      sleep(delay)
    end
  end
end


-- ************MAIN************
chopDownTree = chopDownTree_2x2

nWood = 0
turtle.select(1)

displayLogo()

while true do
  local block, meta = patrol()
  if block then
    if wood[block] then
      go()
      chopDownTree()
      plant()
      goSuck()
    elseif leaves[block] then
      goSuck()
    elseif container[block] then
      unload(forward)
      turn()
    elseif signalBlock[block] then
      refill(meta)
      turn()
    else
      turn()
    end
  else
    return false
  end
end


--  local event, key = os.pullEvent( "key" )  -- DEBUG BREAK POINT