client

local modem = nil
local serverName = ""
local port = 0
local serverPort = 0
local selected = 1

local w, h = term.getSize()
local pos = 0
local color = colors.white
local infoText = ""
local index = 1
local cmds = {}
local originID = ""
local currentID = ""
local isText = false
local toOpen = ""
local cancel = false
local name = ""
local listenerID = -1

local menu = require("menu")
menu.loadMenus()

local function ask(targetPort, request, timeout)
    timeout = timeout or 1
    modem.transmit(targetPort, port, request)

    local timerID = os.startTimer(timeout)

    local message = nil
    local status = "Timeout"

    local loop = true

    while loop do

        local eventData = {os.pullEvent()}
        local event = eventData[1]

        if event == "modem_message" then
            local a, side, senderChannel, replyChannel, replyMessage, distance = unpack(eventData)

            if replyChannel == targetPort then
                message = replyMessage
                status = "Success"
                break
            end

        elseif event == "timer" then
            if eventData[2] == timerID then
                os.cancelTimer(timerID)
                return nil, "Timeout"
            end
        else
            message = event
            status = "Failed"
        end

    end

    os.cancelTimer(timerID)
    return message, status
end

local function split(inputstr, sep)
    if sep == nil then
        sep = "%s"
    end
    local t={}
    for str in string.gmatch(inputstr, "([^"..sep.."]+)") do
        table.insert(t, str)
    end
    return t
end

function string.starts(String,Start)
   return string.sub(tostring(String), 1, string.len(Start))== tostring(Start)
end

local function isEmpty(s)
    return s == nil or s == ''
end

local function hasColor(targetPort, c, flag)
    local r, status = ask(targetPort, "check "..targetPort.." "..c)

    if status == "Success" then
        local f = r == "true"
        if flag then f = not f end
        if f then
            return "[On]"
        else
            return "[Off]"
        end
    else
        return "[FAILED]"
    end
end

local function info(text, isCenter)
    if text == nil then
        term.setCursorPos(1, y+2)
    elseif isCenter then
        x, y = term.getCursorPos()
        x = (w - #text) / 2
        term.setCursorPos(x, y)
        write(text)
        term.setCursorPos(1, y+1)
    else
        print(text)
    end
end

local function longInfo(text, isCenter)
    line = ""
    for k,s in pairs(split(text, " ")) do
        length = string.len(line) + string.len(s) + 1
        if w >= length then
            a = line .. " " .. s
            line = a
        else
            info(line, isCenter)
            line = " "..s
        end
    end
    info(line, isCenter)
end

local function hr()
    x, y = term.getCursorPos()
    for i = 1, w, 1 do
        write("-")
    end
    term.setCursorPos(1, y+1)
end

local function cmdLine()
    flag = true
    count = 0
    while not cancel do
        term.setCursorPos(1, h)
        term.setBackgroundColor(colors.black)
        term.setTextColor(color)
        term.clearLine()
        textLine = "> "..infoText
        term.write(textLine)
        term.setCursorPos(#textLine + 1, h)
        if flag then
            term.setBackgroundColor(color)
            term.write(" ")
            term.setBackgroundColor(colors.black)
            flag = false
        else
            term.setBackgroundColor(colors.black)
            term.write(" ")
            term.setBackgroundColor(color)
            flag = true
        end
        sleep(0.5)
        count = count + 1
        if count >= 10 then
            infoText = ""
            count = 0
        end
    end
end

local function options(index)
    term.setCursorPos(x,pos)
    local ind = index
    if cmds == nil then
        return
    end
    for a = pos+1, h-1, 1 do
        cmd = cmds[ind]
        if cmd == nil then
        else
            if ind == (selected + index - 1) then
                term.setTextColor(colors.black)
                term.setBackgroundColor(color)
            else
                term.setTextColor(color)
                term.setBackgroundColor(colors.black)
            end
            term.clearLine()
            if string.starts(cmd[1], "toggle") then
                local splitCMD = split(cmd[1], " ")
                local targetPort = tonumber(splitCMD[2])
                local numb = tonumber(splitCMD[3])
                info("> "..cmd[2].." "..hasColor(targetPort, numb, cmd[3]))
            elseif string.starts(cmd[1], "trigger") then
                info("> "..cmd[2])
            elseif string.starts(cmd[1], "open") then
                info("> "..cmd[2])
            elseif string.starts(cmd[1], "read") then
                info("> "..cmd[2])
            end
        end
        ind = ind + 1
    end
end

local function buttons()
    while true do
        local event, key, isHeld = os.pullEvent("key")
        if not isHeld then
            if isText then
                if not isEmpty(originID) then
                    toOpen = originID
                    isText = false
                    currentID = ""
                    originID = ""
                    return
                end
            end
            if key == keys.up then
                if selected > 1 then
                    selected = selected - 1
                else
                    if index > 1 then
                        index = index - 1
                    end
                end
            elseif key == keys.down then
                if selected < table.getn(cmds) then
                    if selected + pos < h - 1  then
                        selected = selected + 1
                    else
                        if index < (h - pos) then
                            index = index + 1
                        end
                    end
                end
            elseif key == keys.enter then
                local command = cmds[(selected + index - 1)]
                local splitCMD = split(command[1], " ")

                local cmdString = splitCMD[1]

                if type(cmdString) ~= "string" then
                    print("Invalid command string: "..cmdString)
                    return
                end

                if cmdString == "toggle" then
                    local targetPort = tonumber(splitCMD[2])
                    local numb = tonumber(splitCMD[3])
                    local f, status = ask(targetPort, "check "..numb) -- false when off, true when on
                    local flag = f == "true"

                    if status == "Success" then
                        if command[3] then -- check the invert boolean to check if the flag should be inverted
                            flag = not flag
                        end

                        if flag then
                            infoText = "Disabling "..command[2]
                        else
                            infoText = "Enabling "..command[2]
                        end
                        ask(targetPort, "toggle "..targetPort.." "..numb)
                    end
                elseif cmdString == "trigger" then
                    local targetPort = tonumber(splitCMD[2])
                    numb = tonumber(splitCMD[3])
                    infoText = "Activating "..command[2]
                    ask(targetPort, "pulse "..targetPort.." "..numb)
                elseif cmdString == "open" then
                    id = splitCMD[2]
                    toOpen = id
                    isText = false
                    return
                elseif cmdString == "read" then
                    id = splitCMD[2]
                    toOpen = id
                    isText = true
                    return
                end
            elseif key == keys.backspace then
                if not isEmpty(originID) then
                    toOpen = originID
                    currentID = ""
                    originID = ""
                    return
                end
            end
            options(index)
        end
    end
end

function renderText(text)
    isText = true
    if isEmpty(currentID) then
        currentID = text.id
    else
        originID = currentID
        currentID = text.id
    end
    term.setTextColor(color)
    term.setBackgroundColor(colors.black)
    term.clear()
    term.setCursorPos(1,1)
    info("ROBCO INDUSTRIES UNIFIED OPERATING SYSTEM", true)
    info("COPYRIGHT 2019-2020 ROBCO INDUSTRIES", true)
    info("-Server 1-", true)
    info(nil, false)
    info(serverName, false)
    info("======================", false)
    info(text.title, false)
    hr()
    x,y = term.getCursorPos()
    pos = (h - y)
    selected = 1
    cmds = {}
    longInfo(text.txt, false)
    parallel.waitForAny(buttons, cmdLine)
    if not isEmpty(toOpen) then
        if isText then
            renderText(menu.getText(toOpen))
        else
            renderMenu(menu.getMenu(toOpen))
        end
    end
end

function renderMenu(m)
    isText = false
    if isEmpty(currentID) then
        currentID = m.id
    else
        originID = currentID
        currentID = m.id
    end
    term.setTextColor(color)
    term.setBackgroundColor(colors.black)
    term.clear()
    term.setCursorPos(1,1)
    info("ROBCO INDUSTRIES UNIFIED OPERATING SYSTEM", true)
    info("COPYRIGHT 2019-2020 ROBCO INDUSTRIES", true)
    info("-Server "..port.."-", true)
    info(nil, false)
    info(serverName, false)
    info("======================", false)
    info(m.title, false)
    hr()
    x,y = term.getCursorPos()
    pos = (h - y)
    selected = 1
    cmds = m.cmds
    options(index)
    parallel.waitForAny(buttons, cmdLine)
    if not isEmpty(toOpen) then
        if isText then
            renderText(menu.getText(toOpen))
        else
            renderMenu(menu.getMenu(toOpen))
        end
    end
end

function init(n, p, sp, side, c)
    name = n
    port = p
    serverPort = sp
    modem = peripheral.wrap(side)
    color = c

    if modem then
        modem.open(port)

        local reply, status = ask(serverPort, "name:"..name)
        if status == "Success" then
            serverName = reply
        else
            serverName = status
        end
    else
        print("Error: Modem was nil: " .. side .. ":" .. port)
        return
    end


    term.setTextColor(color)
    term.setBackgroundColor(colors.black)
    term.clear()
    term.setCursorPos(1,1)
    renderMenu(menu.getMenu("start"))
end

return {
    init = init
}