--[[ * Buttons API by LBPHacker, 2013
     * You can use it whenever and whereever you want,
       but give credit! (at least something like 'LBP')
]]

--edited by Shazbot

local container = {}
local termw, termh = term.getSize()
local bgColorDefault = colors.black
local textColorDefault = colors.white

local forcedEvent
local getClickEvent = function()
    local clickEvent
    if forcedEvent then
        clickEvent = forcedEvent
    else
        --[[ * Below is a pretty way to determine wheter
               the terminal is redirected to a monitor or not.
             * As you might know, .setTextScale can be found
               in monitors only.
        ]]
        if term.setTextScale then
            clickEvent = "monitor_touch"
        else
            clickEvent = "mouse_click"
        end
    end
    return clickEvent
end

register = function(xorg, yorg, width, height, color, text, func)
    local newButtonID = false
    for i = 1, #container do if not container[i].alive and not newButtonID then newButtonID = i end end
    if not newButtonID then
        newButtonID = #container + 1
        container[newButtonID] = {}
    end
    container[newButtonID].alive = true
    container[newButtonID].xorg = xorg
    container[newButtonID].yorg = yorg
    container[newButtonID].width = width
    container[newButtonID].height = height
    container[newButtonID].color = color
    container[newButtonID].text = text
    container[newButtonID].func = func
    container[newButtonID].enabled = true
    container[newButtonID].visible = true
    return newButtonID
end

exists = function(buttonID)
    if container[buttonID] then
        return container[buttonID].alive
    else
        return false
    end
end

unregister = function(buttonID)
    if not exists(buttonID) then error("Invalid identifier: " .. tostring(buttonID), 2) end
    container[buttonID].alive = false
end

event = function(eventArray)
    if eventArray[1] == getClickEvent() then
        local button, mousex, mousey = eventArray[2], eventArray[3], eventArray[4]
        for i = 1, #container do
            if container[i].alive then
                if container[i].enabled and (button == 1 or getClickEvent() == "monitor_touch") and mousex > container[i].xorg - 1 and mousey > container[i].yorg - 1 and mousex < container[i].xorg + container[i].width and mousey < container[i].yorg + container[i].height then container[i].func() end
            end
        end
    end
end

draw = function()
    term.setBackgroundColor(32768)
    term.clear()
    for i = 1, #container do
        if container[i].alive and container[i].visible then
            for j = container[i].yorg, container[i].yorg + container[i].height - 1 do
                term.setCursorPos(container[i].xorg, j)
                term.setTextColor(2 ^ (container[i].color % 16))
                term.setBackgroundColor(2 ^ math.floor(container[i].color / 16))
                term.write(string.rep(" ", container[i].width))
            end
            term.setCursorPos(container[i].xorg + math.floor((container[i].width - #container[i].text) / 2), container[i].yorg + math.floor((container[i].height - 1) / 2))
            term.write(container[i].text)
        end
    end
    term.setTextColor(textColorDefault)
    term.setBackgroundColor(bgColorDefault)
end

enable = function(buttonID, bEnabled)
    if not exists(buttonID) then error("Invalid identifier: " .. tostring(buttonID), 2) end
    container[buttonID].enabled = bEnabled
end

show = function(buttonID, bVisible)
    if not exists(buttonID) then error("Invalid identifier: " .. tostring(buttonID), 2) end
    container[buttonID].visible = bVisible
end

setPosition = function(buttonID, xorg, yorg, width, height)
    if not exists(buttonID) then error("Invalid identifier: " .. tostring(buttonID), 2) end
    container[buttonID].xorg = xorg
    container[buttonID].yorg = yorg
    container[buttonID].width = width
    container[buttonID].height = height
end

setText = function(buttonID, sText)
    if not exists(buttonID) then error("Invalid identifier: " .. tostring(buttonID), 2) end
    container[buttonID].text = sText
end

setColor = function(buttonID, color)
    if not exists(buttonID) then error("Invalid identifier: " .. tostring(buttonID), 2) end
    container[buttonID].color = color
end

setForcedEvent = function(fEvent)
    forcedEvent = fEvent
end

setBgColorDefault = function(color)
    bgColorDefault = color
end

setTextColorDefault = function(color)
    textColorDefault = color
end