Touchpoint API Augment

-- This is an augmented version of the Touchpoint API by Lyqyd. If he does not want this up here. Please ask me, I will remove it. I apologize in advance
--[[
The MIT License (MIT)

Copyright (c) 2013 Lyqyd

Permission is hereby granted, free of charge, to any person obtaining a copy
of this software and associated documentation files (the "Software"), to deal
in the Software without restriction, including without limitation the rights
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
copies of the Software, and to permit persons to whom the Software is
furnished to do so, subject to the following conditions:

The above copyright notice and this permission notice shall be included in
all copies or substantial portions of the Software.

THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
THE SOFTWARE.
--]]

local function setupLabel(buttonLen, minY, maxY, name)
    local labelTable = {}
    if type(name) == "table" then
        for i = 1, #name do
            labelTable[i] = name[i]
        end
        name = name.label
    elseif type(name) == "string" then
        local buttonText = string.sub(name, 1, buttonLen - 2)
        if #buttonText < #name then
            buttonText = " "..buttonText.." "
        else
            local labelLine = string.rep(" ", math.floor((buttonLen - #buttonText) / 2))..buttonText
            buttonText = labelLine..string.rep(" ", buttonLen - #labelLine)
        end
        for i = 1, maxY - minY + 1 do
            if maxY == minY or i == math.floor((maxY - minY) / 2) + 1 then
                labelTable[i] = buttonText
            else
                labelTable[i] = string.rep(" ", buttonLen)
            end
        end
    end
    return labelTable, name
end

local Button = {
    draw = function(self)
        local old = term.redirect(self.mon)
        term.setTextColor(colors.black)
        term.setBackgroundColor(colors.orange)
        term.clear()
        for name, buttonData in pairs(self.buttonList) do
            if buttonData.active then
                term.setBackgroundColor(buttonData.activeColor)
                term.setTextColor(buttonData.activeText)
            else
                term.setBackgroundColor(buttonData.inactiveColor)
                term.setTextColor(buttonData.inactiveText)
            end
            for i = buttonData.yMin, buttonData.yMax do
                term.setCursorPos(buttonData.xMin, i)
                term.write(buttonData.label[i - buttonData.yMin + 1])
            end
        end
        if old then
            term.redirect(old)
        else
            term.restore()
        end
    end,
    add = function(self, name, func, xMin, yMin, xMax, yMax, inactiveColor, activeColor, inactiveText, activeText)
        local label, name = setupLabel(xMax - xMin + 1, yMin, yMax, name)
        if self.buttonList[name] then error("button already exists", 2) end
        local x, y = self.mon.getSize()
        if xMin < 1 or yMin < 1 or xMax > x or yMax > y then error("button out of bounds", 2) end
        self.buttonList[name] = {
            func = func,
            xMin = xMin,
            yMin = yMin,
            xMax = xMax,
            yMax = yMax,
            active = false,
            inactiveColor = inactiveColor or colors.red,
            activeColor = activeColor or colors.lime,
            inactiveText = inactiveText or colors.white,
            activeText = activeText or colors.white,
            label = label,
        }
        for i = xMin, xMax do
            for j = yMin, yMax do
                if self.clickMap[i][j] ~= nil then
                    --undo changes
                    for k = xMin, xMax do
                        for l = yMin, yMax do
                            if self.clickMap[k][l] == name then
                                self.clickMap[k][l] = nil
                            end
                        end
                    end
                    self.buttonList[name] = nil
                    error("overlapping button", 2)
                end
                self.clickMap[i][j] = name
            end
        end
    end,
    remove = function(self, name)
        if self.buttonList[name] then
            local button = self.buttonList[name]
            for i = button.xMin, button.xMax do
                for j = button.yMin, button.yMax do
                    self.clickMap[i][j] = nil
                end
            end
            self.buttonList[name] = nil
        end
    end,
    run = function(self)
        while true do
            --self:draw()
            local event = {self:handleEvents(os.pullEvent(self.side == "term" and "mouse_click" or "monitor_touch"))}
            if event[1] == "button_click" then
                self.buttonList[event[2]].func()
            end
        end
    end,
    handleEvents = function(self, ...)
        local event = {...}
        local side = event[2]
        if #event == 0 then event = {os.pullEvent()} end
        if (self.side == "term" and event[1] == "mouse_click") or (self.side ~= "term" and event[1] == "monitor_touch" and event[2] == self.side) then
            local clicked = self.clickMap[event[3]][event[4]]
            if clicked and self.buttonList[clicked] then
                return "button_click", clicked, side
            end
        end
        return unpack(event)
    end,
    toggleButton = function(self, name, noDraw)
        self.buttonList[name].active = not self.buttonList[name].active
        if not noDraw then self:draw() end
    end,
    flash = function(self, name, duration)
        self:toggleButton(name)
        sleep(tonumber(duration) or 0.15)
        self:toggleButton(name)
    end,
    rename = function(self, name, newName)
        self.buttonList[name].label, newName = setupLabel(self.buttonList[name].xMax - self.buttonList[name].xMin + 1, self.buttonList[name].yMin, self.buttonList[name].yMax, newName)
        if not self.buttonList[name] then error("no such button", 2) end
        if name ~= newName then
            self.buttonList[newName] = self.buttonList[name]
            self.buttonList[name] = nil
            for i = self.buttonList[newName].xMin, self.buttonList[newName].xMax do
                for j = self.buttonList[newName].yMin, self.buttonList[newName].yMax do
                    self.clickMap[i][j] = newName
                end
            end
        end
        self:draw()
    end,
}

function new(monSide)
    local buttonInstance = {
        side = monSide or "term",
        mon = monSide and peripheral.wrap(monSide) or term.current(),
        buttonList = {},
        clickMap = {},
    }
    local x, y = buttonInstance.mon.getSize()
    for i = 1, x do
        buttonInstance.clickMap[i] = {}
    end
    setmetatable(buttonInstance, {__index = Button})
    return buttonInstance
end