Item Collector

-- ************ ItemCollector **************
-- * 1.0 release!
-- ******************************************

verNum = "v1.0"

local startTime = os.clock()

local tArgs = { ... }
local delay = 0.5

local level = 1
local nextTurn = 0
local itemsCollected = 0
local screenX, screenY = term.getSize()

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

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

local up = {
  go = turtle.up,
  detect = turtle.detectUp,
  inspect = turtle.inspectUp,
  suck = turtle.suckUp,
  drop = turtle.dropUp,
}

local down = {
  go = turtle.down,
  detect = turtle.detectDown,
  inspect = turtle.inspectDown,
  suck = turtle.suckDown,
  drop = turtle.dropDown,
}

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

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:stick"] = 5,
}

local 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 Itamz collect0r "..verNum.." ****")
  print()
  print("Iz tiem 2 clean up dis mes!!")
  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 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 getInventoryCount()
  local inventoryCount = 0
  for i=1,16 do
    inventoryCount = inventoryCount + turtle.getItemCount(i)
  end
  return inventoryCount
end

local function suck(direction)
  direction = direction or forward
  local pickedUp = getInventoryCount()
  if direction.suck() then
    pickedUp = getInventoryCount() - pickedUp
  else
    return false
  end
  itemsCollected = itemsCollected + pickedUp
  local x, y = term.getCursorPos()
  term.setCursorPos(1,screenY)
  write("I has found "..itemsCollected.." itamz!")
  term.setCursorPos(x, y)
  return true
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 goDirection(direction)
  suck(direction)
  if not refuel() then
    return false
  end
  while displayTime() and not direction.go() do
    sleep(delay)
  end
  return true
end

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

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

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

local function patrol()
  while true do
    repeat
      suck()
      refuel()
      displayTime()
    until not turtle.forward()
    local block, meta = inspectBlock()
    if block then
      return block, meta
    else
      sleep(delay)
    end
  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()
--    local item = isItem(16)
--  if "minecraft:coal" ~= item then
--    turtle.select(16)
--    success, slot =  nextFreeSlot()
--    if success then
--      turtle.transferTo(slot)
--    else
--    turtle.dropDown()
--    end
--  end
--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 and item == "minecraft:coal" then
        turtle.transferTo(16)
        if turtle.getItemCount(i) > 0 then
          dropCount = dropCount + turtle.getItemCount()
          direction.drop()
        end
      else
        dropCount = dropCount + turtle.getItemCount()
        direction.drop()
      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 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)
      prnLines("I needz "..turtle.getItemSpace().." "..inventoryStorageSlots[meta].commonName)
      turtle.suckDown(turtle.getItemSpace())
      turtle.select(1)
    end
  else
    return false
  end
end

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




-- ************MAIN************
turtle.select(1)

displayLogo()

while true do
  local block, meta = patrol()
--  prnLines(block,meta)
  if block then
    if leaves[block] then
      if not turtle.dig() then
        return false
      end
    elseif signalBlock[block] then
      if meta == 15 then
        refill(meta)
      elseif meta == 0 then
        unload(down)
      end
      turn()
    else
      turn()
    end
  else
    return false
  end
end


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