Fluid Cow Sorter

os.loadAPI("touchpoint") --Personalized and modified Touchpoint API
-- API Pastebin: 5QLmjqT1
local storage = peripheral.wrap("right")
local chestDir = "east"
local chestSize = 54
local mon = peripheral.wrap("top")

local totalCows = 0
local cowList = {}
local cowListNames = {}
mon.setTextScale(1)
local maxX, maxY = mon.getSize()
local keyboardPos = {}
local searchChars = {}
local currPage = 1
local lastPage = 1
local tp
local buttonList = {}
local emptyLabel = {
  " Empty Chest ",
  "Into Storage ",
  label = "emptyChest"
}
local refreshLabel = {
  "  Refresh  ",
  "  Screen   ",
  label = "refreshScreen"
}
local searchLabel = {
  " _ ",
  "|_|",
  "/  ",
  label = "search"
}

local pages = {}
for i=1,5,1 do
  pages[i] = touchpoint.new("top")
  pages[i]:add(emptyLabel, emptyChest, 1, maxY-1, 14, maxY, colors.blue, colors.lightBlue)
  pages[i]:add(refreshLabel, refreshScreen, maxX-10, maxY-1, maxX, maxY, colors.blue, colors.lightBlue)
  pages[i]:add(searchLabel, searchCows, 20, maxY-2, 22, maxY, colors.purple, colors.magenta)
end
pages["search"] = touchpoint.new("top")

function refreshScreen(str)
  --doesn't need to do anything else
end

function emptyChest(str)
  local cnt = 0
  for i = 1, 54, 1 do
    local num = storage.pullItem(chestDir, i)
    cnt = cnt + num
  end
  print("Moved "..cnt.." items into storage.")
  pollStorage()
  makeCowList()
end

function nextPage(str)
  lastPage = currPage
  currPage = currPage + 1
  if currPage > 5 then
    print("too many pages!")
    currPage = 5
  end
  tp = pages[currPage]
end

function prevPage(str)
  lastPage = currPage
  currPage = currPage - 1
  if currPage < 1 then
    print("Trying to go before page 1, ERROR!")
    currPage = 1
  end
  tp = pages[currPage]
end

function searchCows(str)
  print("opening Search page?")
  lastPage = currPage
  currPage = "search"
  tp = pages[currPage]
  print("currPage: "..currPage)
end

function testThis()
end

function closeSearch(str)
  currPage = lastPage
  lastPage = "search"
  tp = pages[currPage]
end


-- function drawBox(StartX, lengthX, StartY, lengthY, Text, Color, BkgColor) -- does what is says on the tin.
  -- --local x, y = term.getSize()
  -- local oldColor = mon.getTextColor()
  -- local oldBG = mon.getBackgroundColor()
  -- if Color then mon.setTextColor(Color) end
  -- if BkgColor then mon.setBackgroundColor(BkgColor) end
  -- if not Text then Text = " " end
  -- lengthX = lengthX - 1
  -- lengthY = lengthY - 1
  -- EndX = StartX + lengthX
  -- EndY = StartY + lengthY
  -- mon.setCursorPos(StartX, StartY)
  -- mon.write(string.rep(Text, lengthX))
  -- mon.setCursorPos(StartX, EndY)
  -- mon.write(string.rep(Text, lengthX))
  -- for i = StartY, EndY do
    -- mon.setCursorPos(StartX, i)
    -- mon.write(Text)
    -- mon.setCursorPos(EndX, i)
    -- mon.write(Text)
  -- end
  -- mon.setTextColor(oldColor)
  -- mon.setBackgroundColor(oldBG)
  -- return true
-- end

function makeKeyboard(leftX, topY)
  keyboardPos.x = leftX
  keyboardPos.y = topY
  tRow1 = {"1", "2", "3", "4", "5", "6", "7", "8", "9", "0", "(", ")",}
  tRow2 = {"Q", "W", "E", "R", "T", "Y", "U", "I", "O", "P", ":", "_",}
  tRow3 = {"A", "S", "D", "F", "G", "H", "J", "K", "L", "#", ";", "=",}
  tRow4 = {"/", "Z", "X", "C", "V", "B", "N", "M", "@", "~", "'", "+",}
  tRow5 = {"[", "]", "{", "}", "&", "$", "%", "-", "*", "!", "?", ".",}

  -- addBox(name, startX, Xsize, startY, Ysize, textColor, BGColor)
  pages["search"]:addBox("box1", leftX, 30, topY, 10, colors.white, colors.blue)
  pages["search"]:addBox("box2", leftX, 28, topY, 1, colors.black, colors.white)
  pages["search"]:addBox("box3", leftX, 28, topY, 5, colors.black, colors.white)
  spaceLabel = {
    "  SPACE  ",
    label = "SPACE"
  }
  pages["search"]:add(spaceLabel, pressKey, leftX+3, topY+9, leftX+11, topY+9, colors.black, colors.white)
  backspaceLabel = {
    "  <-BS-  ",
    label = "<-"
  }
  pages["search"]:add(backspaceLabel, pressKey, leftX+13, topY+9, leftX+21, topY+9, colors.black, colors.white)
  pages["search"]:add(">", nil, leftX+1, topY+1, leftX+1, topY+1, colors.gray, colors.white)

  i = leftX + 1
  for _, v in ipairs(tRow1) do
          pages["search"]:add(v, pressKey, i, topY+4, i, topY+4, colors.black, colors.white)
          i = i + 2
  end
  i = leftX + 1
  for _,v in ipairs(tRow2) do
          pages["search"]:add(v, pressKey, i, topY+5, i, topY+5, colors.black, colors.white)
          i = i + 2
  end
  i = leftX + 1
  for _,v in ipairs(tRow3) do
          pages["search"]:add(v, pressKey, i, topY+6, i, topY+6, colors.black, colors.white)
          i = i + 2
  end
  i = leftX + 1
  for _,v in ipairs(tRow4) do
          pages["search"]:add(v, pressKey, i, topY+7, i, topY+7, colors.black, colors.white)
          i = i + 2
  end
  i = leftX + 1
  for _,v in ipairs(tRow5) do
          pages["search"]:add(v, pressKey, i, topY+8, i, topY+8, colors.black, colors.white)
          i = i + 2
  end
end
local closeLabel = {
  "X",
  label = "CLOSE"
}
pages["search"]:add(closeLabel, closeSearch, maxX-1, 1, maxX, 1, colors.red, colors.green)
makeKeyboard(1, 1)

function pressKey(str)
  local searchString = ""
  if str == "<-" then
    table.remove(searchChars)
  else
    table.insert(searchChars, str)
  end
  for pos, searchChar in ipairs(searchChars) do
    pages["search"]:add(searchChar, nil, keyboardPos.x+2+pos, keyboardPos.y+1, keyboardPos.x+2+pos, keyboardPos.y+1, colors.black, colors.white, colors.white, colors.black)
    searchString = searchString..searchChar
  end
  searchString = string.upper(searchString)
end


local prevLabel = {
  "   ",
  " < ",
  "   ",
  label = "prevPage"
}
local nextLabel = {
  "   ",
  " > ",
  "   ",
  label = "nextPage"
}


function pollStorage()
  local tbl = storage.getAllStacks()
  totalCows = #tbl
  cowList = {}
  cowListNames = {}
  for k,v in ipairs(tbl) do
    local slot = v.all()
    if cowList[slot.display_name] then
      table.insert(cowList[slot.display_name].inSlots, k)
    else
      table.insert(cowListNames, slot.display_name)
      cowList[slot.display_name] = {}
      cowList[slot.display_name].inSlots = {}
      cowList[slot.display_name].inSlots[1] = k
    end
  end
end

function moveCows(str)
  local line = tonumber(string.match(str, "%d+"))
  for i = #cowList[cowListNames[line]].inSlots, 1, -1 do
    local num = storage.pushItem(chestDir, cowList[cowListNames[line]].inSlots[i])
    if num == 0 or num == nil then
      print("error moving "..cowListNames[line].." from slot "..cowList[cowListNames[line]].inSlots[i])
      break;
    end
  end
end

function makeCowList()
  for page,buttons in pairs(buttonList) do
    temp = pages[page]
    for k,button in pairs(buttons) do
      temp:remove(button)
    end
  end
  temp = pages[1]
  local pageCnt = 1
  buttonList = {}
  buttonList[pageCnt] = {}
  local line
  for k,v in ipairs(cowListNames) do
    if k % 16 == 0 then
      pageCnt = math.floor(k/16) + 1
      buttonList[pageCnt] = {}
      temp:remove("nextPage")
      temp:add(nextLabel, nextPage, 24, maxY-2, 26, maxY, colors.cyan, colors.lightBlue)
      temp = pages[pageCnt]
      temp:remove("prevPage")
      temp:add(prevLabel, prevPage, 16, maxY-2, 18, maxY, colors.cyan, colors.lightBlue)
    end

    if pageCnt > 1 then
      line = k - ((pageCnt - 1) * 16) + 1
    else
      line = k
    end
    local str = k..". "..v..": "..tostring(#cowList[v].inSlots).." -----------------"
    temp:add(str, nil, 1, line, maxX-7, line, colors.black, colors.black)
    table.insert(buttonList[pageCnt], str)
    temp:add("Move "..k, moveCows, maxX-6, line, maxX-1, line, colors.red, colors.green)
    table.insert(buttonList[pageCnt], "Move "..k)
    --print("Total "..k..": "..tostring(#v.inSlots))
    --12. Nutrient Distillation total: 12
  end
  temp:add("Total Cows: "..totalCows, nil, 3, line+1, 22, line+1, colors.black, colors.black, colors.gray, colors.gray)
  table.insert(buttonList[pageCnt], "Total Cows: "..totalCows)
end

pollStorage()
makeCowList()
tp = pages[1]
tp:draw()
os.startTimer(5)

while true do

  local event, p1 = tp:handleEvents(os.pullEvent())   ---button_click, name
  if event == "button_click" then
    --remove toggling and simplify button running
    if type(tp.buttonList[p1].func) == "function" then
        tp:flash(p1)
        tp.buttonList[p1].func(p1)
    end
    pollStorage()
    makeCowList()
    tp:draw()
  end
  if event == "timer" then
    os.startTimer(5)
  end
end