Lightshot

--
--  Lighshot Screen Recorder
--  Made by GravityScore
--


--  -------- Variables

--  Lower the loop rate to decrease recording lag but to decrease compression
--  Do not set to below 50 else the file could become really big
local loopRate = 300

-- Version
local version = "1.5"

-- Terminal
local oldTerm = {}
local newTerm = {}
local w, h = term.getSize()

-- Events
local event_exitRecording = "lightshot_exitRecordingEvent"

-- Locations
local lightshotURL = "https://raw.github.com/GravityScore/Lightshot/master/lightshot.lua"
local lightshotLocation = "/" .. shell.getRunningProgram()
local recordLocation = "/.lightshot_recording"

-- Variables
local clock = 0
local nfaRecording = false
local paused = false
local handle = nil

local recordHeader = [[
--
--  Recorded by Lightshot
--

local function sp(...) return sleep(...) end
local function c(...) return term.write(...) end
local function d(...) return term.setCursorPos(...) end
local function e(...) return term.setBackgroundColor(...) end
local function f(...) return term.setTextColor(...) end
local function g(...) return term.clear(...) end
local function h(...) return term.clearLine() end
local function i(...) return term.setCursorBlink(...) end
local function j(...) return term.scroll(...) end

-- sD here...

]]


--  -------- Utilities

local function modRead(properties)
    local w, h = term.getSize()
    local defaults = {replaceChar = nil, history = nil, visibleLength = nil, textLength = nil,
        liveUpdates = nil, exitOnKey = nil}
    if not properties then properties = {} end
    for k, v in pairs(defaults) do if not properties[k] then properties[k] = v end end
    if properties.replaceChar then properties.replaceChar = properties.replaceChar:sub(1, 1) end
    if not properties.visibleLength then properties.visibleLength = w end

    local sx, sy = term.getCursorPos()
    local line = ""
    local pos = 0
    local historyPos = nil

    local function redraw(repl)
        local scroll = 0
        if properties.visibleLength and sx + pos > properties.visibleLength + 1 then
            scroll = (sx + pos) - (properties.visibleLength + 1)
        end

        term.setCursorPos(sx, sy)
        local a = repl or properties.replaceChar
        if a then term.write(string.rep(a, line:len() - scroll))
        else term.write(line:sub(scroll + 1, -1)) end
        term.setCursorPos(sx + pos - scroll, sy)
    end

    local function sendLiveUpdates(event, ...)
        if type(properties.liveUpdates) == "function" then
            local ox, oy = term.getCursorPos()
            properties.liveUpdates(line, event, ...)
            if a == true and data == nil then
                term.setCursorBlink(false)
                return line
            elseif a == true and data ~= nil then
                term.setCursorBlink(false)
                return data
            end
            term.setCursorPos(ox, oy)
        end
    end

    term.setCursorBlink(true)
    while true do
        local e, but, x, y, p4, p5 = os.pullEvent()

        if e == "char" then
            local s = false
            if properties.textLength and line:len() < properties.textLength then s = true
            elseif not properties.textLength then s = true end

            local canType = true
            if not properties.grantPrint and properties.refusePrint then
                local canTypeKeys = {}
                if type(properties.refusePrint) == "table" then
                    for _, v in pairs(properties.refusePrint) do
                        table.insert(canTypeKeys, tostring(v):sub(1, 1))
                    end
                elseif type(properties.refusePrint) == "string" then
                    for char in properties.refusePrint:gmatch(".") do
                        table.insert(canTypeKeys, char)
                    end
                end
                for _, v in pairs(canTypeKeys) do if but == v then canType = false end end
            elseif properties.grantPrint then
                canType = false
                local canTypeKeys = {}
                if type(properties.grantPrint) == "table" then
                    for _, v in pairs(properties.grantPrint) do
                        table.insert(canTypeKeys, tostring(v):sub(1, 1))
                    end
                elseif type(properties.grantPrint) == "string" then
                    for char in properties.grantPrint:gmatch(".") do
                        table.insert(canTypeKeys, char)
                    end
                end
                for _, v in pairs(canTypeKeys) do if but == v then canType = true end end
            end

            if s and canType then
                line = line:sub(1, pos) .. but .. line:sub(pos + 1, -1)
                pos = pos + 1
                redraw()
            end
        elseif e == "key" then
            if but == keys.enter then break
            elseif but == keys.left then if pos > 0 then pos = pos - 1 redraw() end
            elseif but == keys.right then if pos < line:len() then pos = pos + 1 redraw() end
            elseif (but == keys.up or but == keys.down) and properties.history then
                redraw(" ")
                if but == keys.up then
                    if historyPos == nil and #properties.history > 0 then
                        historyPos = #properties.history
                    elseif historyPos > 1 then
                        historyPos = historyPos - 1
                    end
                elseif but == keys.down then
                    if historyPos == #properties.history then historyPos = nil
                    elseif historyPos ~= nil then historyPos = historyPos + 1 end
                end

                if properties.history and historyPos then
                    line = properties.history[historyPos]
                    pos = line:len()
                else
                    line = ""
                    pos = 0
                end

                redraw()
                sendLiveUpdates("history")
            elseif but == keys.backspace and pos > 0 then
                redraw(" ")
                line = line:sub(1, pos - 1) .. line:sub(pos + 1, -1)
                pos = pos - 1
                redraw()
                sendLiveUpdates("delete")
            elseif but == keys.home then
                pos = 0
                redraw()
            elseif but == keys.delete and pos < line:len() then
                redraw(" ")
                line = line:sub(1, pos) .. line:sub(pos + 2, -1)
                redraw()
                sendLiveUpdates("delete")
            elseif but == keys["end"] then
                pos = line:len()
                redraw()
            elseif properties.exitOnKey then
                if but == properties.exitOnKey or (properties.exitOnKey == "control" and
                        (but == 29 or but == 157)) then
                    term.setCursorBlink(false)
                    return nil
                end
            end
        end

        sendLiveUpdates(e, but, x, y, p4, p5)
    end

    term.setCursorBlink(false)
    if line ~= nil then line = line:gsub("^%s*(.-)%s*$", "%1") end
    return line
end

local function centerPrint(text, ny)
    if type(text) == "table" then for _, v in pairs(text) do centerPrint(v) end
    else
        local x, y = term.getCursorPos()
        local w, h = term.getSize()
        term.setCursorPos(w/2 - text:len()/2 + (#text % 2 == 0 and 1 or 0), ny or y)
        print(text)
    end
end


--  -------- Updating

local function download(url, path)
    for i = 1, 3 do
        local response = http.get(url)
        if response then
            local data = response.readAll()
            response.close()
            if path then
                local f = io.open(path, "w")
                f:write(data)
                f:close()
            end
            return true
        end
    end

    return false
end

local function updateClient()
    local updateLocation = "/.lightshot-update"
    fs.delete(updateLocation)

    download(lightshotURL, updateLocation)
    local a = io.open(updateLocation, "r")
    local b = io.open(lightshotLocation, "r")
    local new = a:read("*a")
    local cur = b:read("*a")
    a:close()
    b:close()

    if cur ~= new then
        fs.delete(lightshotLocation)
        fs.move(updateLocation, lightshotLocation)
        return true
    else
        fs.delete(updateLocation)
        return false
    end
end


--  -------- Compression

local sD = {{}}
local cD = {}

local function proccessFunction(data)
    if data:len() < 8 then
        if data:sub(-1,-1) == "]" then return "[[" .. data .. "] .. \"]\""
        else return "[[" .. data .. "]]" end
    end
    if cD[v] then return cD[v] end
    for k, v in pairs(sD[#sD]) do
        if v == data then
            cD[v] = ("sD[" .. #sD .. "][" .. k .. "]")
            return("sD[" .. #sD .. "][" .. k .. "]")
        end
    end
    table.insert(sD[#sD], data)
    local returnData = ("sD[".. #sD .. "][" .. #sD[#sD] .. "]")
    if #sD[#sD] > loopRate then sD[#sD + 1] = {} end
    return returnData
end


--  -------- Terminal Override

local function add(...)
    if not handle then
        if fs.exists(recordLocation) then
            handle = io.open(recordLocation, "a")
        else
            handle = io.open(recordLocation, "w")
        end
    end

    for _, v in pairs({...}) do
        handle:write(v)
    end
end

local bg, tc, blnk = -1, -1, nil
for k, v in pairs(term.native) do oldTerm[k] = v end

newTerm.write = function(...)
    local text = ""
    for k, v in pairs({...}) do text = text .. tostring(v) end

    local a = ""
    if not paused and os.clock() - clock > 0 then
        a = "sp(" .. os.clock() - clock .. ") "
    end
    text = proccessFunction(text)
    local b = "c(" .. text .. ")\n"

    add(a .. b)
    clock = os.clock()

    if not nfaRecording then return oldTerm.write(...) end
end

newTerm.setCursorPos = function(x, y)
    local a = ""
    if not paused and os.clock() - clock > 0 then
        a = "sp(" .. os.clock() - clock .. ") "
    end
    add(a .. "d(" .. tostring(x) .. ", " .. tostring(y) .. ")\n")
    clock = os.clock()

    return oldTerm.setCursorPos(x, y)
end

newTerm.getCursorPos = function(...) return oldTerm.getCursorPos(...) end

newTerm.setBackgroundColor = function(col)
    if bg ~= col then
        local a = ""
        if not paused and not nfaRecording and os.clock() - clock > 0 then
            a = "sp(" .. os.clock() - clock .. ") "
        end

        add(a .. "e(" .. tostring(col) .. ")\n")
        clock = os.clock()
        bg = col
    end

    return oldTerm.setBackgroundColor(col)
end

newTerm.setTextColor = function(col)
    if tc ~= col then
        local a = ""
        if not paused and not nfaRecording and os.clock() - clock > 0 then
            a = "sp(" .. os.clock() - clock .. ") "
        end

        add(a .. "f(" .. tostring(col) .. ")\n")
        clock = os.clock()
        tc = col
    end

    return oldTerm.setTextColor(col)
end

newTerm.setBackgroundColour = function(col) return term.setBackgroundColor(col) end
newTerm.setTextColour = function(col) return term.setTextColor(col) end

newTerm.clear = function(...)
    local a = ""
    if not paused and os.clock() - clock > 0 then
        a = "sp(" .. os.clock() - clock .. ") "
    end
    add(a .. "g()\n")
    clock = os.clock()

    return oldTerm.clear(...)
end

newTerm.clearLine = function(...)
    local a = ""
    if not paused and os.clock() - clock > 0 then
        a = "sp(" .. os.clock() - clock .. ") "
    end
    add(a .. "h()\n")
    clock = os.clock()

    return oldTerm.clearLine(...)
end

newTerm.setCursorBlink = function(flag)
    if flag ~= blnk then
        local a = ""
        if not paused and os.clock() - clock > 0 then
            a = "sp(" .. os.clock() - clock .. ") "
        end
        add(a .. "i(" .. tostring(flag) .. ")\n")
        clock = os.clock()
        blnk = flag
    end

    return oldTerm.setCursorBlink(flag)
end

newTerm.scroll = function(n)
    local a = ""
    if not paused and os.clock() - clock > 0 then
        a = "sp(" .. os.clock() - clock .. ") "
    end
    add(a .. "j(" .. tostring(n) .. ")\n")
    clock = os.clock()

    return oldTerm.scroll(n)
end

newTerm.getSize = function(...) return oldTerm.getSize(...) end
newTerm.redirect = function(...) return oldTerm.redirect(...) end
newTerm.restore = function(...) return oldTerm.restore(...) end
newTerm.isColor = function(...) return oldTerm.isColor and oldTerm.isColor(...) end
newTerm.isColour = function(...) return term.isColor(...) end


--  -------- Recording

local function record(location)
    while true do
        local e, key = os.pullEventRaw()
        if (e == "key" and key == 59) or e == event_exitRecording or e == "terminate" then
            local a = ""
            if os.clock() - clock > 0 then a = "sp(" .. os.clock() - clock .. ") " end
            add(a)

            add("\n\nterm.setCursorBlink(false)\n")
            add("if term.isColor() then term.setTextColor(colors.yellow)\n")
            add("else term.setTextColor(colors.white) end\n")
            add("term.setBackgroundColor(colors.black)\n")
            add("term.clear()\n")
            add("term.setCursorPos(1, 1)\n")
            add("print(\"End of Recording!\")\n")

            local sd = "local sD = " .. textutils.serialize(sD) .. "\n"

            term.restore()
            term.setCursorBlink(false)

            handle:close()
            handle = nil
            local f = io.open(recordLocation, "r")
            local ncont = f:read("*a")
            f:close()

            ncont = ncont:gsub("%-%- sD here...\n", sd)
            local f = io.open(location, "w")
            f:write(ncont)
            f:close()
            fs.delete(recordLocation)

            break
        elseif e == "key" and key == 61 then
            paused = not paused
            if paused then
                add("sp(2)\n")
            end
        end
    end
end


--  -------- Movie

local function loadNfa(path)
    local ret = {}
    if fs.exists(path) and not fs.isDir(path) then
        local f = io.open(path, "r")
        local l = f:read("*l")
        local curFrame = ""
        while l do
            if l ~= "" and l ~= "~" then curFrame = curFrame .. l .. "\n"
            elseif l == "~" then
                table.insert(ret, curFrame)
                curFrame = ""
            end
            l = f:read("*l")
        end
        f:close()
    end

    return ret
end

local function movie(location, duration)
    nfaRecording = true
    local frames = loadNfa(location)
    if frames ~= {} then
        for i, v in ipairs(frames) do
            term.setTextColor(colors.white)
            term.setBackgroundColor(colors.black)
            term.clear()

            local tempImageLocation = "/.lightshot-temp-image"
            local f = io.open(tempImageLocation, "w")
            f:write(v)
            f:close()
            local a = paintutils.loadImage(tempImageLocation)
            paintutils.drawImage(a, 1, 1)
            fs.delete(tempImageLocation)
            add("sp(" .. duration .. ")\n")
        end

        add("sp(" .. duration .. ")\n")
    end

    add("\n\nterm.setCursorBlink(false)\n")
    add("if term.isColor() then term.setTextColor(colors.yellow)\n")
    add("else term.setTextColor(colors.white) end\n")
    add("term.setBackgroundColor(colors.black)\n")
    add("term.clear()\n")
    add("term.setCursorPos(1, 1)\n")
    add("print(\"The End! :D\")\n")

    nfaRecording = false
    term.restore()
    term.setCursorBlink(false)
    if term.isColor() then term.setTextColor(colors.yellow)
    else term.setTextColor(colors.white) end
    term.setBackgroundColor(colors.black)
    term.clear()
    term.setCursorPos(1, 1)
    print("Movie Recorded Successfully!")

    handle:close()
    handle = nil
    local f = io.open(recordLocation, "r")
    local ncont = f:read("*a")
    f:close()

    local sd = "local sD = " .. textutils.serialize(sD) .. "\n"
    ncont = ncont:gsub("%-%- sD here...\n", sd)
    local f = io.open(location, "w")
    f:write(ncont)
    f:close()
    fs.delete(recordLocation)
end


--  -------- Main

local theme = {}
if term.isColor and term.isColor() then
    theme = {
        ["prompt"] = "cyan",
        ["promptHighlight"] = "lightBlue",
        ["textColor"] = "white",
    }
else
    theme = {
        ["prompt"] = "black",
        ["promptHighlight"] = "black",
        ["textColor"] = "white",
    }
end

local function drawButton(v, sel)
    if sel then term.setBackgroundColor(v.highlight or colors[theme.promptHighlight])
    else term.setBackgroundColor(v.bg or colors[theme.prompt]) end
    term.setTextColor(v.tc or colors[theme.textColor])
    for i = -1, 1 do
        term.setCursorPos(v[2], v[3] + i)
        term.write(string.rep(" ", v[1]:len() + 4))
    end

    term.setCursorPos(v[2], v[3])
    if sel then
        term.setBackgroundColor(v.highlight or colors[theme.promptHighlight])
        term.write(" > ")
    else term.write(" - ") end
    term.write(v[1] .. " ")
end

local function prompt(list, dir)
    local function draw(sel)
        for i, v in ipairs(list) do
            if i == sel then term.setBackgroundColor(v.highlight or colors[theme.promptHighlight])
            else term.setBackgroundColor(v.bg or colors[theme.prompt]) end
            term.setTextColor(v.tc or colors[theme.textColor])
            for i = -1, 1 do
                term.setCursorPos(v[2], v[3] + i)
                term.write(string.rep(" ", v[1]:len() + 4))
            end

            term.setCursorPos(v[2], v[3])
            if i == sel then
                term.setBackgroundColor(v.highlight or colors[theme.promptHighlight])
                term.write(" > ")
            else term.write(" - ") end
            term.write(v[1] .. " ")
        end
    end

    local sel = 1
    draw(sel)

    while true do
        local e, but, x, y = os.pullEvent()
        if e == "key" and but == 28 then return list[sel][1]
        elseif e == "key" and but == 200 and sel > 1 then
            sel = sel - 1
            draw(sel)
        elseif e == "key" and but == 208 and ((err == true and sel < #list - 1) or (sel < #list)) then
            sel = sel + 1
            draw(sel)
        elseif e == "key" and but == 203 and sel > 2 then
            sel = sel - 2
            draw(sel)
        elseif e == "key" and but == 205 and sel < 3 then
            sel = sel + 2
            draw(sel)
        elseif e == "mouse_click" then
            for i, v in ipairs(list) do
                if x >= v[2] - 1 and x <= v[2] + v[1]:len() + 3 and y >= v[3] - 1 and y <= v[3] + 1 then
                    return list[i][1]
                end
            end
        end
    end
end

local function menu()
    term.setBackgroundColor(colors.gray)
    term.setTextColor(colors.white)
    term.clear()
    term.setCursorPos(1, 1)

    term.setBackgroundColor(colors.lightGray)
    for i = 2, 4 do term.setCursorPos(1, i) term.clearLine() end
    term.setCursorPos(3, 3)
    term.write("Lightshot " .. version)

    local opt = prompt({{"Record a Video", w/2 - 16, 9}, {"Record an Animation", w/2 - 21, 13},
        {"Update Lightshot", w/2 + 4, 9}, {"Exit", w/2 + 4, 13, bg = (term.isColor and term.isColor())
        and colors.red or colors.black, highlight = (term.isColor and term.isColor()) and colors.pink
        or colors.black}}, "vertical")
    if opt == "Record a Video" or opt == "Record an Animation" then
        term.setBackgroundColor(colors.gray)
        for i = 7, 14 do term.setCursorPos(1, i) term.clearLine() end
        term.setBackgroundColor(colors.cyan)
        for i = 8, 11 do term.setCursorPos(7, i) term.write(string.rep(" ", w - 14)) end
        drawButton({"Control to Return to Menu", 7, 14}, true)
        term.setBackgroundColor(colors.cyan)
        term.setCursorPos(8, 9)
        term.write("Where to save the recording?")
        term.setCursorPos(9, 10)
        term.write("/")
        local loc = modRead({visibleLength = w - 14, exitOnKey = "control"})
        if not loc or loc == "" then return "menu" end

        term.setCursorPos(8, 9)
        term.write(string.rep(" ", 28))
        term.setCursorPos(8, 9)
        loc = "/" .. loc
        if fs.isReadOnly(loc) then term.write("File is Read Only!") sleep(1.6) return "menu"
        elseif fs.isDir(loc) then term.write("Location is a directory!") sleep(1.6) return "menu"
        elseif opt == "Record a Video" then
            if fs.exists(loc) then fs.delete(loc) end
            term.setTextColor(colors.white)
            term.setBackgroundColor(colors.black)
            term.clear()
            term.setCursorPos(2, 5)
            term.write("Press F1 to end the recording")
            term.setCursorPos(2, 6)
            term.write("Press F3 to pause or unpause the recording")

            if term.isColor and term.isColor() then term.setTextColor(colors.yellow) end
            for i = 1, 3 do
                term.setCursorPos(2, 3)
                term.clearLine()
                term.write("Recording in " .. 4 - i .. "...")
                sleep(1)
            end

            return "record", loc
        elseif opt == "Record an Animation" then
            if loc:sub(-4, -1) ~= ".nfa" then
                term.write("File Not An nPaintPro Animation!")
                sleep(1.6)
                return "menu"
            end

            term.setCursorPos(8, 9)
            term.write("Duration between frames (seconds):")
            term.setCursorPos(9, 10)
            term.write(string.rep(" ", w - 16))
            term.setCursorPos(9, 10)
            local dur = modRead({visibleLength = w - 14, exitOnKey = "control"})
            if not dur or dur == "" then return "menu" end

            term.setCursorPos(8, 9)
            term.write(string.rep(" ", 38))
            term.setCursorPos(8, 9)
            dur = tonumber(dur)
            if not dur then term.write("Duration must be an integer!") sleep(1.6) return "menu"
            else return "tomovie", loc, dur end
        end
    elseif opt == "Update Lightshot" then
        term.setTextColor(colors.white)
        term.setBackgroundColor(colors.lightGray)
        term.setCursorPos(3, 3)
        term.clearLine()
        term.write("Checking for Updates...")
        term.setCursorPos(3, 3)
        if updateClient() then
            term.clearLine()
            term.write("Updated!")
            sleep(1.6)
            opt = "Exit"
        else
            term.clearLine()
            term.write("No Updates Found!")
            sleep(1.6)
            return "menu"
        end
    end if opt == "Exit" then return nil end
end

local function main()
    local action, location, duration = "menu", nil, nil
    while action == "menu" do action, location, duration = menu() end

    if action and location then
        clock = os.clock()
        add(recordHeader)
        term.redirect(newTerm)

        term.setTextColor(colors.white)
        term.setBackgroundColor(colors.black)
        term.clear()
        term.setCursorPos(1, 1)
        if action == "record" then
            parallel.waitForAny(function()
                record(location)
            end, function()
                term.setBackgroundColor(colors.black)
                term.setTextColor(colors.white)
                term.clear()
                term.setCursorPos(1, 1)

--[[
                -- Run startup
                if fs.exists("/startup") and not fs.isDir("/startup") then shell.run("/startup") end
                term.setBackgroundColor(colors.black)
                term.setTextColor(colors.white)
                term.clear()
                term.setCursorPos(1, 1)
]]--

                -- Run shell
                shell.run("/rom/programs/shell")
                os.queueEvent(event_exitRecording)
            end)

            if term.isColor() then term.setTextColor(colors.yellow)
            else term.setTextColor(colors.white) end
            term.setBackgroundColor(colors.black)
            term.clear()
            term.setCursorPos(2, 3)
            print("Recording Saved!")
            sleep(1.1)
        elseif action == "tomovie" then
            movie(location, duration)
        end
    else
        return "exit"
    end
end

-- Run
local oldDir = shell.dir()
while main() ~= "exit" do end
shell.setDir(oldDir)

-- Exit Message
term.setBackgroundColor(colors.black)
term.setTextColor(colors.white)
term.clear()
term.setCursorPos(1, 1)
centerPrint("Thanks for Using Lightshot " .. version .. "!")
centerPrint("Made by GravityScore and 1lann")