-- This is a ComputerCraft script to easily run turtle API commands from the turtle's command line.
-- Written by al@inventwithpython.com
-- Find my other scripts at https://github.com/asweigart/al-computercraft
-- or http://turtlescripts.com/profile/AlSweigart


local tArgs = {...}
if #tArgs < 1 then
  print('Usage: do <cmd> [<repeat>] [more]')
  print('  Commands are:')
  print('    f, b, up, dn, l, r - move')
  print('    d, du, dd - dig (up, down)')
  print('    i, iu, id - inspect (up, down)')
  print('    sel <num> - select inv num')
  print('    s, su, sd - suck items (up, down)')
  print('    dr, dru, drd - drop (up, down)')
  print('    p, pu, pd - place (up, down)')
  print('Example: do r f 2')
  print('(Turn right, move forward twice.)')
  return
end

local i, j, k, v

local function outOfFuel()
  if turtle.getFuelLevel() == 0 then
    print('Out of fuel!')
    return true
  else
    return false
  end
end

local function inspectAndPrintResults(direction)
  local success, inspectResults, icmd
  if direction == '' then
    success, inspectResults = turtle.inspect()
  elseif direction == 'down' then
    success, inspectResults = turtle.inspectDown()
  elseif direction == 'up' then
    success, inspectResults = turtle.inspect()
  end

  if direction == '' then
    icmd = 'inspect '
  elseif direction == 'down' then
    icmd = 'inspectDown '
  elseif direction == 'up' then
    icmd = 'inspectUp '
  end

  if not success then
    print(icmd .. ': false')
  else
    print(icmd .. ': true')
    for k, v in pairs(inspectResults) do
      print(k, '=', v)
    end
    io.write('\n')
  end
end

-- main program
for i = 1,#tArgs do
  cmd = tArgs[i] -- get the command
  if #tArgs < i + 1 then
    reps = 1 -- end of cmdline args, so set this to 1
  else
    if tonumber(tArgs[i+1]) ~= nil then -- check if next arg is numeric
      reps = tonumber(tArgs[i+1]) -- set
    else
      -- "reps" is actually the next command, so set it to 1
      reps = 1
    end
  end
  if tArgs[i] == 'f' then
    for j = 1,reps do
      success, errMsg = turtle.forward()
      print('forward: ' .. tostring(success) .. ' ' .. errMsg)
      if not success then return success, errMsg end
    end
  elseif tArgs[i] == 'b' then
    for j = 1,reps do
      success, errMsg = turtle.back()
      print('back: ' .. tostring(success) .. ' ' .. errMsg)
      if not success then return success, errMsg end

    end
  elseif tArgs[i] == 'l' then
    for j = 1,reps do
        print('left: ' .. tostring(turtle.turnLeft()))
    end
  elseif tArgs[i] == 'r' then
    for j = 1,reps do
        print('right: ' .. tostring(turtle.turnRight()))
    end
  elseif tArgs[i] == 'up' then
    for j = 1,reps do
      success, errMsg = turtle.up()
      print('up: ' .. tostring(success) .. ' ' .. errMsg)
      if not success then return success, errMsg end
    end
  elseif tArgs[i] == 'dn' then
    for j = 1,reps do
      success, errMsg = turtle.down()
      print('down: ' .. tostring(success) .. ' ' .. errMsg)
      if not success then return success, errMsg end
    end
  elseif tArgs[i] == 'd' then
    for j = 1,reps do
      success, errMsg = turtle.dig()
      print('dig: ' .. tostring(success) .. ' ' .. errMsg)
      if not success then return success, errMsg end
    end
  elseif tArgs[i] == 'du' then
    for j = 1,reps do
      success, errMsg = turtle.digUp()
      print('digUp: ' .. tostring(success) .. ' ' .. errMsg)
      if not success then return success, errMsg end
    end
  elseif tArgs[i] == 'dd' then
    for j = 1,reps do
      success, errMsg = turtle.digDown()
      print('digDown: ' .. tostring(success) .. ' ' .. errMsg)
      if not success then return success, errMsg end
    end
  elseif tArgs[i] == 'i' then
    for j = 1,reps do
      inspectAndPrintResults('')
    end
  elseif tArgs[i] == 'iu' then
    for j = 1,reps do
        inspectAndPrintResults('up')
    end
  elseif tArgs[i] == 'id' then
    for j = 1,reps do
      inspectAndPrintResults('down')
    end
  elseif tArgs[i] == 'sel' then
    -- in this case, reps is the inventory number
    print('select ' .. reps .. ': ' .. tostring(turtle.select(reps)))
  elseif tArgs[i] == 'item' then
    -- display info about the item stack
    local itemData = turtle.getItemDetail()
    if itemData ~= nil then
      for k, v in pairs(itemDetail) do
        print(k, '=', v)
      end
    else
      print('No item at slot #' .. tostring(turtle.getSelectedSlot()))
    end
  elseif tArgs[i] == 's' then
    for j = 1,reps do
        print('suck: ' .. tostring(turtle.suck()))
    end
  elseif tArgs[i] == 'su' then
    for j = 1,reps do
        print('suckUp: ' .. tostring(turtle.suckUp()))
    end
  elseif tArgs[i] == 'sd' then
    for j = 1,reps do
        print('suckDown: ' .. tostring(turtle.suckDown()))
    end
  elseif tArgs[i] == 'p' then
    for j = 1,reps do
        print('place: ' .. tostring(turtle.place()))
    end
  elseif tArgs[i] == 'pu' then
    for j = 1,reps do
        print('placeUp: ' .. tostring(turtle.placeUp()))
    end
  elseif tArgs[i] == 'pd' then
    for j = 1,reps do
        print('placeDown: ' .. tostring(turtle.placeDown()))
    end
  elseif tArgs[i] == 'dr' then
    for j = 1,reps do
        print('drop: ' .. tostring(turtle.drop()))
    end
  elseif tArgs[i] == 'dru' then
    for j = 1,reps do
        print('dropUp: ' .. tostring(turtle.dropUp()))
    end
  elseif tArgs[i] == 'drd' then
    for j = 1,reps do
        print('dropDown: ' .. tostring(turtle.dropDown()))
    end
  end
end