Button API

--credits: Direwolf20: for his button api that gave birth to this creation.

function search(perToFind)
    local printMode = false
    if perToFind == "print" or perToFind==nil then
            printMode=true
    end
    if printMode == true then
        term.clear()
        term.setCursorPos(1,1)
    end
    local perip = peripheral.getNames()
    for i=1, #perip do
        if printMode == true then
            print(perip[i] .. " -- " ..peripheral.getType(perip[i]))
        else
            if perToFind==peripheral.getType(perip[i]) or perToFind==perip[i] then
                return perip[i]
            end
        end
    end
    if printMode == false then
        throw("No such peripheral: " ..perToFind)
    end
end

function wsearch (perToFind)
    return peripheral.wrap(search(perToFind))
end

function throw (text)
    printError(text)
    error()
end

local mon

function getMonitor ()
    return mon
end

function findMonitor (monitor)
    if monitor == nil then
        monitor = "monitor"
    end
    mon = wsearch(monitor)
end

function setMonitor (monitor)
    mon = monitor
end

function prepareMonitor ()
    if mon == nil then
        findMonitor()
    end
    mon.setTextScale(1)
    mon.setTextColor(colors.white)
    mon.setBackgroundColor(colors.black)
    mon.clear()
end

function clearMonitor()
    mon.clear()
end

local buttonMenus = {}
local groupInfo = {}
local group

function selectGroup (name)
    group=name
    if buttonMenus[group]== nil then
        buttonMenus[group]={}
        groupInfo[group]={}
        groupInfo[group]["num"]=0
    end
end

function addCustomButton ( name, func, ymin, ymax, xmin, xmax, mode, colorOFF, colorON)
    if group == nil then
        selectGroup("A")
    end
    buttonMenus[group][name]={}
    buttonMenus[group][name]["func"] = func
    buttonMenus[group][name]["active"] = false
    buttonMenus[group][name]["locked"] = false
    buttonMenus[group][name]["xmin"] = xmin
    buttonMenus[group][name]["ymin"] = ymin
    buttonMenus[group][name]["xmax"] = xmax
    buttonMenus[group][name]["ymax"] = ymax
    buttonMenus[group][name]["mode"] = mode
    buttonMenus[group][name]["custom"] = true
    if colorOFF == nil then
        buttonMenus[group][name]["colorOFF"]=colors.red
    else
        buttonMenus[group][name]["colorOFF"]=colorOFF
    end
    if colorON == nil then
        buttonMenus[group][name]["colorON"]=colors.lime
    else
        buttonMenus[group][name]["colorON"]=colorON
    end
end

function addCustomCornerButton(name, func, mode, sizeX, sizeY, corner, colorOFF, colorON)
    local monX,monY = mon.getSize()
    local xmin,xmax,ymin,ymax
    if string.find(corner,"r")~=nil or string.find(corner,"R")~=nil then
        xmax=monX+1
        xmin=monX-sizeX+1
    else
        xmin=1
        xmax=sizeX+1
    end
    if string.find(corner,"t")~=nil or string.find(corner,"T")~=nil then
        ymax=sizeY+1
        ymin=1
    else
        ymin=monY-sizeY+1
        ymax=monY+1
    end
    addCustomButton(name,func,ymin,ymax,xmin,xmax,mode, colorOFF, colorON)
end

function addBasicButton(name, func, mode, colorOFF, colorON)
    if group == nil then
        selectGroup("A")
    end
    buttonMenus[group][name]={}
    buttonMenus[group][name]["func"] = func
    buttonMenus[group][name]["active"] = false
    buttonMenus[group][name]["mode"] = mode
    buttonMenus[group][name]["custom"] = false
    buttonMenus[group][name]["locked"] = false
    groupInfo[group]["num"]=groupInfo[group]["num"] + 1
    if colorOFF == nil then
        buttonMenus[group][name]["colorOFF"]=colors.red
    else
        buttonMenus[group][name]["colorOFF"]=colorOFF
    end
    if colorON == nil then
        buttonMenus[group][name]["colorON"]=colors.lime
    else
        buttonMenus[group][name]["colorON"]=colorON
    end
end


local heading= ""

function setHeading (text)
    heading=text
end

function printHeading()
    local head=""
    if groupInfo[group]["heading"]~= nil then
        head=groupInfo[group]["heading"]
    else
        head=heading
    end
    local w, h = mon.getSize()
    mon.setCursorPos((w-string.len(head))/2+1, 1)
    mon.write(head)
end

local button={}

function setGlobalButtonVar (width, height,Xaxis)
    button["buttonX"]= width
    button["buttonY"] = height
    button["buttonsX"] = Xaxis
end

local monitorBorder={}

function setGlobalMonitorBorder (mBT,mBR,mBD,mBL)
    monitorBorder["monitorBorderTop"] = mBT
    monitorBorder["monitorBorderRight"] = mBR
    monitorBorder["monitorBorderDown"] = mBD
    monitorBorder["monitorBorderLeft"] = mBL
end
setGlobalMonitorBorder(0,0,0,0)

local space={}
function setGlobalButtonBorder (sX,sY)
    space["spaceX"] = sX
    space["spaceY"] = sY
end
------------------------------------------------------------------------------------------------------------------
function setGroupMonitorBorder (mBT,mBR,mBD,mBL)
    if groupInfo[group]["MB"]==nil then
        groupInfo[group]["MB"]={}
    end
    groupInfo[group]["MB"]["monitorBorderTop"] = mBT
    groupInfo[group]["MB"]["monitorBorderRight"] = mBR
    groupInfo[group]["MB"]["monitorBorderDown"] = mBD
    groupInfo[group]["MB"]["monitorBorderLeft"] = mBL
end

function setGroupButtonVar (width, height,Xaxis)
    if groupInfo[group]["BV"]==nil then
        groupInfo[group]["BV"]={}
    end
    groupInfo[group]["BV"]["buttonX"]= width
    groupInfo[group]["BV"]["buttonY"]= height
    groupInfo[group]["BV"]["buttonsX"] = Xaxis
end
function setGroupButtonBorder (sX,sY)
    if groupInfo[group]["BB"]==nil then
        groupInfo[group]["BB"]={}
    end
    groupInfo[group]["BB"]["spaceX"] = sX
    groupInfo[group]["BB"]["spaceY"] = sY
end
function setGroupHeading (text)
    groupInfo[group]["heading"]=text
end
----------------------------------------------------------------------------------------------------------

function calculateButtons()
    local monX,monY = mon.getSize()

    local cButton , cVar , cMonitor

    local buttonsY
    local buttonColumn, buttonRow
    for name, groupData in pairs(buttonMenus) do

        if groupInfo[name]["BB"] ~= nil then
            cButton=groupInfo[name]["BB"]
        else
            cButton=space
        end

        if groupInfo[name]["MB"] ~= nil then
            cMonitor=groupInfo[name]["MB"]
        else
            cMonitor=monitorBorder
        end

        if groupInfo[name]["BV"] ~= nil then
            cVar=groupInfo[name]["BV"]
        else
            cVar=button
        end

        if cButton == nil then
            throw("You forgot to call 'setBasicButtonVar'")
        end


        if heading ~= "" then
            if cMonitor["monitorBorderTop"] < 2 then
                cMonitor["monitorBorderTop"]= 2
            end
        end
        if cButton["spaceX"] == nil  then
            cButton["spaceX"]=math.floor  ((monX - (cVar["buttonX"] * cVar["buttonsX"]) ) / (cVar["buttonsX"]+ 1 ) )

        end
        buttonsY=math.ceil(groupInfo[name]["num"] / cVar["buttonsX"])

        if cButton["spaceY"] == nil then
            cButton["spaceY"]=math.floor(( monY  - (cVar["buttonY"] * buttonsY) ) / (buttonsY + 1 ) )
        end

        buttonColumn=-1
        buttonRow=0
        for bName, bData in pairs(groupData) do
            if bData["custom"] == false then
                buttonColumn=buttonColumn+1
                if buttonColumn==cVar["buttonsX"] then
                    buttonColumn=0
                    buttonRow=buttonRow+1
                end

                local areaNeededX = (cVar["buttonX"] * cVar["buttonsX"]) + (cButton["spaceX"] * (cVar["buttonsX"] - 1))
                local areaNeededY = (cVar["buttonY"] * buttonsY) + (cButton["spaceY"] * (buttonsY - 1))
                local arealeftX = monX - (areaNeededX + cMonitor["monitorBorderLeft"] + cMonitor["monitorBorderRight"])
                local arealeftY = monY - (areaNeededY + cMonitor["monitorBorderTop"] + cMonitor["monitorBorderDown"])
                local shiftX = math.floor(arealeftX/2)
                local shiftY = math.ceil(arealeftY/2)

                bData["xmin"] = (cMonitor["monitorBorderLeft"]+shiftX+((cButton["spaceX"] + cVar["buttonX"])  * buttonColumn))  +1
                bData["xmax"]= (cMonitor["monitorBorderLeft"]+shiftX+((cButton["spaceX"] + cVar["buttonX"])  * buttonColumn)  + cVar["buttonX"] )+1
                bData["ymin"]=  (cMonitor["monitorBorderTop"]+shiftY+((cButton["spaceY"] + cVar["buttonY"])  * buttonRow) ) +1
                bData["ymax"]= (cMonitor["monitorBorderTop"]+shiftY+((cButton["spaceY"] + cVar["buttonY"])  * buttonRow) + cVar["buttonY"]) +1

                if bData["xmax"]-1>(monX-cMonitor["monitorBorderRight"]) or bData["ymax"]-1>(monY-cMonitor["monitorBorderDown"]) then
                    throw("Button outside border ")
                end
                if bData["xmin"]<=cMonitor["monitorBorderLeft"] or bData["ymin"]<=cMonitor["monitorBorderTop"] then
                    throw("Button outside border ")
                end
            end
        end
    end
end

function printButton (name, bData)
    local color
    local on = bData["active"]
    if on == true then color = bData["colorON"] else color = bData["colorOFF"] end
   mon.setBackgroundColor(color)
   if bData["ymin"] == nil then
    throw ("You forgot to call calculateButtons")
    end
   local yspot = math.floor((bData["ymin"] + bData["ymax"]) /2)
   local xspot = math.floor((bData["xmax"] - bData["xmin"] - string.len(name)) /2) +0

   for j = bData["ymin"], (bData["ymax"] - 1) do
      mon.setCursorPos(bData["xmin"], j)
      if j == yspot then
        if xspot<0 then
            mon.setCursorPos((bData["xmin"]+xspot), j)
            mon.write(name)
        else
             for k=0, bData["xmax"] - bData["xmin"] - string.len(name)  do
                if k == xspot then
                   mon.write(name)
                else
                   mon.write(" ")
                end
             end
            end
      else
         for i = bData["xmin"], bData["xmax"]-1 do
            mon.write(" ")
         end
      end
   end
   mon.setBackgroundColor(colors.black)
end

function refresh (name)
    printButton(name,buttonMenus[group][name])
end

local flashTime=0.2

function setGlobalFlashTime(tim)
    flashTime=tim
end

function toggle(name,button)
    if button["locked"]~=true then
        button["active"] = not button["active"]
        refresh(name)
        button["func"]()
    end
end

function flash(name,button)
    if button["locked"]~=true then
        local tgroup=group
        if button["active"]==false then
            toggle(name,button)
        end

        if tgroup==group then
            sleep(flashTime)
            button["active"] = not button["active"]
            refresh(name)
        else
            button["active"] = not button["active"]
        end
    end
end

function flashInvert(name,button)
    if button["locked"]~=true then
        local tgroup=group
        if tgroup==group then
            sleep(flashTime)
            button["active"] = not button["active"]
            refresh(name)
        else
            button["active"] = not button["active"]
        end
        if button["active"]==false then
            toggle(name,button)
        end
    end
end

function no(name,button)
    if button["locked"]~=true then
        button["func"]()
    end
end


function empty() end

function redrawScreen()
    clearMonitor()
    printHeading()
    for name,data in pairs(buttonMenus[group]) do
        printButton(name, data)
    end
end

function setButtonLock(state, name, gr)
    if gr==nil then
        gr=group
    end
    if buttonMenus[gr][name]==nil then
        throw("button does not exist")
    end
    buttonMenus[gr][name]["locked"]=state
end
function changeButtonLock( name, gr)
    if gr==nil then
        gr=group
    end
    if buttonMenus[gr][name]==nil then
        throw("button does not exist")
    end
    buttonMenus[gr][name]["locked"]=not buttonMenus[gr][name]["locked"]
end
function getButtonLock(name,gr)
    if gr==nil then
        gr=group
    end
    if buttonMenus[gr][name]==nil then
        throw("button does not exist")
    end
    return buttonMenus[gr][name]["locked"]
end

function useButton(name,gr)
    if gr==nil then
        gr=group
    end
    if buttonMenus[gr][name]==nil then
        throw("button does not exist")
    end
    if buttonMenus[gr][name]["locked"]~=true then
        buttonMenus[gr][name]["mode"](name,buttonMenus[gr][name])
    end
end

function setButtonState(state,name,gr)
    if gr==nil then
        gr=group
    end
    if buttonMenus[gr][name]==nil then
        throw("button does not exist")
    end
    if buttonMenus[gr][name]["active"]~= state then
        toggle (name,buttonMenus[gr][name])
    end
end

function getButtonState(name,gr)
    if gr==nil then
        gr=group
    end
    if buttonMenus[gr][name]==nil then
        throw("button does not exist")
    end
    return buttonMenus[gr][name]["active"]
end

function getButton(name,gr)
    if gr==nil then
        gr=group
    end
    if buttonMenus[gr][name]==nil then
        throw("button does not exist")
    end
    return buttonMenus[gr][name]
end

function getGroup(gr)
    if gr==nil then
        gr=group
    end
    return buttonMenus[gr]
end

function checkxy(x, y)
   for name, data in pairs(buttonMenus[group]) do

          if y>=data["ymin"] and  y < data["ymax"] then
             if x>=data["xmin"] and x< data["xmax"] then
                 print ("")print ("")
                    data["mode"](name,data)
                    return true
             end
          end

    end
   return false
end

function getClick()
    local event,side,x,y = os.pullEvent("monitor_touch")
    checkxy(x,y)
end

function printButtonData()
    local monX,monY = mon.getSize()
    print(monX)
    print (monY)
    for name, data in pairs(buttonMenus) do

        for names, dataup in pairs(data) do
            print (name.."  "..names .." xmin:"..dataup["xmin"].." xmax:"..dataup["xmax"].." ymin:"..dataup["ymin"].." ymax:"..dataup["ymax"])
        end

    end

end