MCCI (StateMachine)

--[[
  Modular Chatbox Command Interpreter
    Made by LDDestroier
  pastebin get ECeJ3CpL mcci
--]]
local progname = fs.getName(shell.getRunningProgram())
local baseDir = ".mcci" --all other directories are combined with this one, you see
local commandDir = fs.combine(baseDir,"cmd") --a folder
local commandHelpDir = fs.combine(baseDir,"cmdhelp") --a folder
local chatlogDir = fs.combine(baseDir,"chatlog") --a file
local banDir = fs.combine(baseDir,"banlist")
local adminDir = fs.combine(baseDir,"adminlist")
local configDir = fs.combine(baseDir,"config")
local apiDir = fs.combine(baseDir,"statemachine")
local _

local useMoarPeriphs = (peripheral.find("chatbox_admin") or peripheral.find("chatbox") or peripheral.find("chat_box")) and true or false --either moarperipherals/computronics xor peripherals++

if not fs.exists(commandDir) then fs.makeDir(commandDir) end
if not fs.exists(commandHelpDir) then fs.makeDir(commandHelpDir) end

if not statemachine then
    if fs.exists(apiDir) then
        os.loadAPI(apiDir)
        print("Loaded StateMachine API.")
    else
        print("Downloading StateMachine API...")
        local foyle = http.get("https://pastebin.com/raw/3qe1iUQc")
        if not foyle then
            print("Could not download StateMachine API...")
            return false
        else
            local file = fs.open(apiDir,"w")
            file.write(foyle.readAll())
            file.close()
            os.loadAPI(apiDir)
            print("Loaded StateMachine API.")
        end
    end
end

local adminUserTable = { --only one who can use admin commands. default values.
    "EldidiStroyrr",
    "Notch",
}
if pairs ~= _G.pairs then
    pairs = function(tbl)
        if type(tbl) ~= "table" then
            error("expected a table here, but got "..type(tbl))
        else
            return _G.pairs(tbl)
        end
    end
end

local doRestart = false
local adminUser = function(u)
    if type(u) ~= "string" then return false end
    for a = 1, #adminUserTable do
        if adminUserTable[a]:lower() == u:lower() then
            return true
        end
    end
    return false
end
--[[
local getEvents = function(...)
    local arg = table.pack(...)
    local output
    while true do
        output = {os.pullEvent()}
        for a = 1, #arg do
            if output[1] == arg[a] then
                return table.unpack(output)
            end
        end
    end
end
--]]
--here's a more experimental getEvents...
getEvents = function(...)
    local arg, funky, output = table.pack(...), {}
    if #arg <= 1 then
        return os.pullEvent(table.unpack(arg))
    else
        for a = 1, #arg do
            funky[a] = function() output = {os.pullEvent(arg[a])} end
        end
        parallel.waitForAny(table.unpack(funky))
        return table.unpack(output)
    end
end

local mcci = {
    prefix = "(>",
    ignoreErrors = true,
    mcciLabel = "MCCI",
}

local chatbox
local usingAdminChatbox = false
local chatDelay = 0.1 --seconds. regular chatboxes in P+1 will error if you use them too fast
local chatbox = peripheral.find("chatbox_admin") or peripheral.find("chatbox") or peripheral.find("chatBox") or peripheral.find("chat_box")
local cLabel
local setLabel = function(newLabel)
    cLabel = newLabel
    if chatbox.setLabel then
        chatbox.setLabel(newLabel)
    elseif chatbox.setName then
        chatbox.setName(newLabel)
    end
end
if useMoarPeriphs then setLabel(mcci.mcciLabel) end
local SAY = chatbox.say
local WHISPER = commands.tellraw and function(user,message)
    commands.exec("tellraw "..user..' {"text":"['..cLabel..'] '..message..'"}')
end or chatbox.tell
local LISTEN = function(username,checkPrefix)
    local evt
    local user, message
    while true do
        evt = {getEvents("chat_message","chatbox_command","chat")}
        user, message = evt[3 - (useMoarPeriphs and 0 or 1)], evt[4 - (useMoarPeriphs and 0 or 1)]
        if (user == (username or user)) then
            if checkPrefix then
                if message:sub(1,#tostring(checkPrefix)) == checkPrefix then
                    message = message:sub(#tostring(checkPrefix)+1)
                    break
                end
            else
                break
            end
        end
    end
    return user, message
end

local explode = function(div,str)
    if (div=='') then return false end
    local pos,arr = 0,{}
    for st,sp in function() return string.find(str,div,pos,false) end do
        table.insert(arr,string.sub(str,pos,st-1))
        pos = sp + 1
    end
    table.insert(arr,string.sub(str,pos))
    return arr
end

local chatlog = {}
local blacklist = {}
local whitelist = {}
local userActions = {}
local userActionNames = {}

local addToChatlog = function(user,message)
    if chatlog[1] then
        if chatlog[#chatlog][1] == user and chatlog[#chatlog][2] == message then
            chatlog[#chatlog] = {user,message,chatlog[#chatlog][3]+1}
        else
            chatlog[#chatlog+1] = {user,message,1}
        end
    else
        chatlog[#chatlog+1] = {user,message,1}
    end
end

local writeChatlog = function()
    prettyOutput = "("..chatlog[#chatlog][3] or "0"..") <"..chatlog[#chatlog][1].."> "..chatlog[#chatlog][2]
    local file = fs.open(chatlogDir,"a")
    file.writeLine(prettyOutput)
    file.close()
end

local saveBanlist = function()
    local file = fs.open(banDir,"w")
    for k,v in pairs(blacklist) do
        file.writeLine(k)
        file.writeLine(v)
        file.writeLine("")
    end
    file.close()
end

local loadBanlist = function()
    blacklist = {}
    local file = fs.open(banDir,"r")
    local name,reason = ""
    while name do
        name = file.readLine()
        if name then
            reason = file.readLine()
            file.readLine() --blank space
            blacklist[name] = reason
        end
    end
    file.close()
    return blacklist
end

local saveAdminlist = function()
    local file = fs.open(adminDir,"w")
    for a = 1, #adminUserTable do
        file.writeLine(adminUserTable[a])
    end
    file.close()
end

local loadAdminlist = function()
    adminUserTable = {}
    local file = fs.open(adminDir,"r")
    local line = ""
    while line do
        line = file.readLine()
        if line then
            adminUserTable[#adminUserTable+1] = line
        end
    end
    file.close()
    return adminUserTable
end

saveConfig = function()
    local file = fs.open(configDir,"w")
    file.writeLine("prefix:"..textutils.serialize(mcci.prefix))
    file.writeLine("mcciLabel:"..textutils.serialize(mcci.mcciLabel))
    file.writeLine("ignoreErrors:"..tostring(mcci.ignoreErrors))
    file.close()
end

local fixInput = function(input)
    if tonumber(input) then
        return tonumber(input)
    elseif input == "false" then
        return false
    elseif input == "true" then
        return true
    elseif textutils.unserialize(input) then
        return textutils.unserialize(input)
    else
        return tostring(input)
    end
end

loadConfig = function()
    local file = fs.open(configDir,"r")
    local line = ""
    while line do
        line = file.readLine()
        if line then
            local name = line:sub(1,line:find(":")-1)
            local value = line:sub(line:find(":")+1)
            value = fixInput(value)
            mcci[name] = value
        end
    end
    file.close()
end

local scr_x, scr_y = term.getSize()

local render = statemachine.unblock(function()
    term.setTextColor(colors.white)
    term.setBackgroundColor(colors.black)
    term.clear()
    term.setCursorPos(1,1)
    print("("..mcci.mcciLabel.."/Prefix='"..mcci.prefix.."')")
    print("(ignoreErrors="..tostring(mcci.ignoreErrors)..")")
    print(("-"):rep(scr_x))
    print("Active Users:")
    for k,v in pairs(userActions) do
        print("-"..k.." ("..userActionNames[k]..")")
    end
end)

local initCommand = function(u,cmd,...)
    local yarg = table.pack(...)
    local golly_G = setmetatable({}, {__index=(_ENV or getfenv())})
    golly_G.chatbox = chatbox
    golly_G.WHISPER = WHISPER
    golly_G.SAY = SAY
    golly_G.LISTEN = LISTEN
    golly_G._USERNAME = u
    golly_G._PREFIX = mcci.prefix
    golly_G._ADMIN = adminUser(u)
    golly_G.pairs = pairs
    userActions[u] = statemachine.unblock(function()
        local func = loadfile(fs.combine(commandDir,cmd))
        func = setfenv(func, golly_G)
        local output
        if mcci.ignoreErrors then
            output = {pcall( function() return func(table.unpack(yarg)) end )}
        else
            output = func(table.unpack(yarg))
        end
        return table.unpack(output)
    end)
    userActionNames[u] = cmd
    render()
end

local dCommands,dCommandsHelp
dCommandsHelp = {
    ["belay"] = mcci.prefix.."belay [playername] ; Belays all commands or a specific player's.",
    ["ban"] = mcci.prefix.."ban playername ; Bans a player from using this chat command interpreter.",
    ["unban"] = mcci.prefix.."unban playername ; Unbans a player who was banned from using this command interpreter.",
    ["banlist"] = mcci.prefix.."banlist ; Lists every player banned from using this command interpreter.",
    ["help"] = mcci.prefix.."help [commandname] ; Gives a more detailed description of a command.",
    ["update"] = mcci.prefix.."update ; Redownloads the MCCI program and overwrites the source if it's different.",
    ["restart"] = mcci.prefix.."restart ; Restarts the program. One would usually do this after updating.",
    ["checkadmin"] = mcci.prefix.."checkadmin [username] ; Without argument, checks if you are an admin. Also can check if another user is an admin.",
    ["admin"] = mcci.prefix.."admin username ; Sets a username as an administrator.",
    ["sudo"] = mcci.prefix.."sudo username command ; Runs a command as a specific user. Can't be used to abdicate.",
    ["abdicate"] = mcci.prefix.."abdicate ; Use with no arguments to tick yourself off the list of admins. Use with care, there's no confirming.",
}
dCommands = { --default commands
    ["belay"] = function(u,username)
        if adminUser(u) then
            if username then
                if userActions[username] then
                    userActions[username] = nil
                    WHISPER(u,username.."'s command has been belayed.")
                    WHISPER(username,"Your command has been belayed. Ready.")
                    render()
                else
                    WHISPER(u,"That user isn't executing a cancelable command.")
                end
            else
                WHISPER(u,"All commands have been belayed.")
                for k,v in pairs(userActions) do
                    WHISPER(k,"Your command has been belayed. Ready.")
                end
                userActions = {}
                render()
            end
        else
            WHISPER(k,"You can't do that!")
        end
    end,
    ["ban"] = function(u,baduser,...)
        local arg = table.pack(...)
        if adminUser(u) then
            if adminUser(baduser) then
                WHISPER(u,"Very funny. You can't ban admins.")
            elseif not baduser then
                WHISPER(u,"Do "..mcci.prefix.."ban [username]")
            else
                if blacklist[baduser] then
                    WHISPER(u,"That dude was already banned.")
                else
                    blacklist[baduser] = table.concat(arg," ") or "Unspecified reason."
                    userActions[baduser] = nil
                    userActionNames[baduser] = nil
                    render()
                    WHISPER(u,"'"..baduser.."' has been banned from making commands.")
                    WHISPER(baduser,"You are hereby BANNED from making MCCI commands!")
                end
            end
        else
            WHISPER(u,"You can't do that!")
        end
    end,
    ["banlist"] = function(u)
        WHISPER(u,"## USERS BANNED FROM THIS CHAT AI ##")
        for k,v in pairs(blacklist) do
            WHISPER(u," * "..k.." ("..v..")")
        end
    end,
    ["unban"] = function(u,baduser)
        if adminUser(u) then
            if adminUser(baduser) then
                WHISPER(u,"Admins can't be banned to begin with, you dolt.")
            elseif (not baduser) then
                WHISPER(u,"Do "..mcci.prefix.."unban [username]")
            else
                if (not blacklist[baduser]) then
                    WHISPER(u,"That dude isn't banned.")
                else
                    blacklist[baduser] = nil
                    WHISPER(u,"'"..baduser.."' has been unbanned from making commands.")
                    WHISPER(baduser,"You've been unbanned from making commands.")
                end
            end
        else
            if blacklist[gooduser] then
                WHISPER(u,"You can't do that!")
            else
                WHISPER(u,"You couldn't do that even if "..gooduser.." were banned!")
            end
        end
    end,
    ["order"] = function(u)
        WHISPER(u,"What, do I look like a pizzeria to you?")
    end,
    ["help"] = function(u,cmd)
        if cmd then
            if dCommands[cmd] then
                WHISPER(u,dCommandsHelp[cmd])
            else
                if not fs.exists(fs.combine(commandDir,cmd)) then
                    WHISPER(u,"That command doesn't exist, internally or externally.")
                else
                    if fs.exists(fs.combine(commandHelpDir,cmd)) then
                        local file = fs.open(fs.combine(commandHelpDir,cmd),"r")
                        local cont = file.readAll(file)
                        file.close()
                        cont = cont:gsub("$PREFIX",mcci.prefix):gsub("\n"," ; ")
                        WHISPER(u,cont)
                    else
                        WHISPER(u,"That external command does not have a help file.")
                    end
                end
            end
        else
            local list = {}
            local cmdlist = fs.list(commandDir)
            for k,v in pairs(dCommands) do
                list[#list+1] = k
            end
            for a = 1, #cmdlist do
                list[#list+1] = cmdlist[a]
            end
            for a = 1, #list do
                WHISPER(u,mcci.prefix..list[a])
            end
        end
    end,
    ["update"] = function(u)
        if adminUser(u) then
            WHISPER(u,"Downloading latest version...")
            local foyle = http.get("https://pastebin.com/raw/ECeJ3CpL")
            if foyle then
                local cont = foyle.readAll()
                local mccifoyle = fs.open(shell.getRunningProgram(),"r")
                local mccicont = mccifoyle.readAll()
                mccifoyle.close()
                if mccicont == cont then
                    WHISPER(u,"You were using the latest version. Aborting.")
                else
                    local file = fs.open(shell.getRunningProgram(),"w")
                    file.write(cont)
                    file.close()
                    WHISPER(u,"Update complete. Use "..mcci.prefix.."restart to run the latest version.")
                end
            end
        else
            WHISPER(u,"Yeah, try that again as an admin, doofus.")
        end
    end,
    ["restart"] = function(u)
        if adminUser(u) then
            doRestart = true
        end
    end,
    ["checkadmin"] = function(u,user)
        if not user then
            if adminUser(u) then
                WHISPER(u,"You're an admin.")
            else
                WHISPER(u,"You're decidedly not an admin.")
            end
        else
            if adminUser(user) then
                WHISPER(u,"'"..user.."' is an admin.")
            else
                WHISPER(u,"'"..user.."' is decidedly not an admin.")
            end
        end
    end,
    ["admin"] = function(u,user)
        if not user then
            if adminUser(u) then
                WHISPER(u,"Expected an argument with who you wish to make an admin, dunce.")
            else
                WHISPER(u,"Only admins can make other people admins, you twat.")
            end
        else
            if adminUser(u) then
                if adminUser(user) then
                    if user == u then
                        WHISPER(u,"Oh my god, stop it.")
                    else
                        WHISPER(u,"That person is already an admin.")
                    end
                else
                    table.insert(adminUserTable,user)
                    WHISPER(u,"'"..user.."' is now an admin.")
                end
            else
                WHISPER(u,"This is an admin command, lamebrain.")
            end
        end
    end,
    ["abdicate"] = function(u,user)
        if adminUser(u) then
            if user then
                WHISPER(u,"You can't abdicate another user from their adminship.")
            else
                for a = 1, #adminUserTable do
                    if adminUserTable[a] == u then
                        table.remove(adminUserTable,a)
                        WHISPER(u,"Lo, you have abdicated the throne of adminship.")
                        return
                    end
                end
                WHISPER(u,"That's weird. You weren't on the list, but you still did this command.")
            end
        else
            if user then
                WHISPER(u,"If you really don't like an admin, don't go complaining to me.")
            else
                WHISPER(u,"Abdicate from what? Only admins can do this!")
            end
        end
    end,
    ["sudo"] = function(u,user,cmd,...)
        if adminUser(u) then
            if cmd == "abdicate" then
                WHISPER(u,"You can't use sudo to abdicate people!")
            else
                if user then
                    if userActions[user] then
                        WHISPER(u,"That user is already in a command.")
                    else
                        if cmd then
                            initCommand(user,cmd,...)
                            WHISPER(u,"Ran '"..cmd.."' as '"..user.."'.")
                        else
                            WHISPER(u,"Sudo requires a second argument for the command.")
                        end
                    end
                else
                    WHISPER(u,"Sudo requires a first argument for the user.")
                end
            end
        else
            WHISPER(u,"Sudo is an admin's command!")
        end
    end,
}

local handleDcommands = function(u,f,...) --heh heh heh
    local arg = table.pack(...)
    if dCommands[f] then
        return true, dCommands[f](u,table.unpack(arg))
    else
        return false
    end
end

local handleNormalCommands = function(u,cmd,...)
    local yarg = table.pack(...)
    local cmdlist = fs.list(commandDir)
    for a = 1, #cmdlist do
        if cmdlist[a] == cmd then
            initCommand(u,cmd,table.unpack(yarg))
            if useMoarPeriphs then setLabel(mcci.mcciLabel..":"..userActionNames[u]) end
            return true
        end
    end
    return false
end

local fuckingDoIt = statemachine.unblock(function()
    local user, message, isDcommand, isNormalCommand
    while true do
        user, message = LISTEN()
        sleep(0) --to ensure that chatbox output comes AFTER player chat
        addToChatlog(user,message)
        writeChatlog()
        loadBanlist()
        loadAdminlist()
        loadConfig()
        if useMoarPeriphs then setLabel(mcci.mcciLabel) end
        render()
        if not blacklist[user] then
            if message:sub(1,#mcci.prefix) == mcci.prefix then
                message = message:sub(#mcci.prefix+1)
                if userActions[user] then
                    if message == "nvm" then
                        userActions[user] = nil
                        userActionNames[user] = nil
                        WHISPER(user,"Command cancelled. Ready.")
                        render()
                    else
                        if useMoarPeriphs then setLabel(mcci.mcciLabel..":"..userActionNames[user]) end
                        os.queueEvent("followup_command_"..user,message)
                    end
                else
                    isNormalCommand = handleNormalCommands(user,table.unpack(explode(" ",message)))
                    if not isNormalCommand then
                        isDcommand = handleDcommands(user,table.unpack(explode(" ",message)))
                        if doRestart then
                            return
                        else
                            if not isDcommand then
                                WHISPER(user,"No such command exists, internally or externally.")
                            end
                        end
                    end
                end
            end
            saveBanlist()
            saveAdminlist()
            saveConfig()
        end
    end
end)

if not fs.exists(adminDir) then
    saveAdminlist()
end
if not fs.exists(banDir) then
    saveBanlist()
end
if not fs.exists(configDir) then
    saveConfig()
end

loadAdminlist()
loadBanlist()
loadConfig()

render()

local rootEvent = {}
local getRootEvent = function()
    rootEvent = {os.pullEvent()}
end

--this used to be a function called doUserActions
local doBreak, vres
local grodyList
statemachine.setLoop(function()
    fuckingDoIt()
    getRootEvent()
    local vres
    grodyList = {}
    --if #rootEvent > 0 then os.queueEvent(table.unpack(rootEvent)) end
    for k,v in pairs(userActions) do
        vres = nil
        if v then vres = v() end
        if vres then
            if useMoarPeriphs then setLabel(mcci.mcciLabel) end
            WHISPER(k,"Ready.")
            grodyList[#grodyList+1] = k
        end
    end
    for a = 1, #grodyList do
        userActions[grodyList[a]] = nil
        userActionNames[grodyList[a]] = nil
    end
    render()
    return doRestart
end)
statemachine.run()

if doRestart then
    SAY("MCCI Restarting.")
    shell.run(shell.getRunningProgram())
end