rtmc/rtmc.lua

local __BUNDLER_FILES = {}
local __VERSION = "1.0.0d"
local __DEFAULT_IMPORT = require
local require = function(path)
    if __BUNDLER_FILES[path] then
        return __BUNDLER_FILES[path]()
    elseif __BUNDLER_FILES[path .. ".lua"] then
        return __BUNDLER_FILES[path .. ".lua"]()
    else
        return __DEFAULT_IMPORT(path)
    end
end
rawset(__BUNDLER_FILES, "main", function ()
    local updater = require("updater");
    local log = require("log");
    local notifier = require("notifier")
    local json     = require("json")
    local MODULE_NAME = "keypadOS";
    local MODULE_URL = "https://git.mcorangehq.xyz/xomf/keypadOS/raw/branch/keypad/keypadOS.lua";
    if not __VERSION then
        __VERSION = "DEV";
    end
    local function updaterLoop(upd)
        while true do
            sleep(1)
            upd:checkAndUpdateAll();
        end
    end
    local function _start()
        log.info("Starting bootloader (" .. __VERSION .. ")");
        local upd = updater.new();
        upd:addEntry(MODULE_NAME .. ".lua", "keypad", MODULE_URL);
        local lib_exports = {
            log = log,
            updater = upd,
            notifier = notifier,
            json = json
        };
        parallel.waitForAny(
            function()
                updaterLoop(upd)
            end,
            function ()
                require(MODULE_NAME).main(lib_exports);
            end
        );
    end
    local mod = {};
    mod.main = _start;
    return mod;
end)
rawset(__BUNDLER_FILES, "log", function ()
    local log = {};
    log.__index = log;
    function log.error(...)
        print("[ERROR] " .. ...);
    end
    function log.warn(...)
        print("[WARN] " .. ...);
    end
    function log.info(...)
        print("[INFO] " .. ...);
    end
    function log.debug(...)
        print("[DEBUG] " .. ...);
    end
    return log;
end)
rawset(__BUNDLER_FILES, "json", function ()
    local json = { _version = "0.1.2" }
    local encode
    local escape_char_map = {
      [ "\\" ] = "\\",
      [ "\"" ] = "\"",
      [ "\b" ] = "b",
      [ "\f" ] = "f",
      [ "\n" ] = "n",
      [ "\r" ] = "r",
      [ "\t" ] = "t",
    }
    local escape_char_map_inv = { [ "/" ] = "/" }
    for k, v in pairs(escape_char_map) do
      escape_char_map_inv[v] = k
    end
    local function escape_char(c)
      return "\\" .. (escape_char_map[c] or string.format("u%04x", c:byte()))
    end
    local function encode_nil(val)
      return "null"
    end
    local function encode_table(val, stack)
      local res = {}
      stack = stack or {}
      if stack[val] then error("circular reference") end
      stack[val] = true
      if rawget(val, 1) ~= nil or next(val) == nil then
        local n = 0
        for k in pairs(val) do
          if type(k) ~= "number" then
            error("invalid table: mixed or invalid key types")
          end
          n = n + 1
        end
        if n ~= #val then
          error("invalid table: sparse array")
        end
        for i, v in ipairs(val) do
          table.insert(res, encode(v, stack))
        end
        stack[val] = nil
        return "[" .. table.concat(res, ",") .. "]"
      else
        for k, v in pairs(val) do
          if type(k) ~= "string" then
            error("invalid table: mixed or invalid key types")
          end
          table.insert(res, encode(k, stack) .. ":" .. encode(v, stack))
        end
        stack[val] = nil
        return "{" .. table.concat(res, ",") .. "}"
      end
    end
    local function encode_string(val)
      return '"' .. val:gsub('[%z\1-\31\\"]', escape_char) .. '"'
    end
    local function encode_number(val)
      if val ~= val or val <= -math.huge or val >= math.huge then
        error("unexpected number value '" .. tostring(val) .. "'")
      end
      return string.format("%.14g", val)
    end
    local type_func_map = {
      [ "nil"     ] = encode_nil,
      [ "table"   ] = encode_table,
      [ "string"  ] = encode_string,
      [ "number"  ] = encode_number,
      [ "boolean" ] = tostring,
    }
    encode = function(val, stack)
      local t = type(val)
      local f = type_func_map[t]
      if f then
        return f(val, stack)
      end
      error("unexpected type '" .. t .. "'")
    end
    function json.encode(val)
      return ( encode(val) )
    end
    local parse
    local function create_set(...)
      local res = {}
      for i = 1, select("#", ...) do
        res[ select(i, ...) ] = true
      end
      return res
    end
    local space_chars   = create_set(" ", "\t", "\r", "\n")
    local delim_chars   = create_set(" ", "\t", "\r", "\n", "]", "}", ",")
    local escape_chars  = create_set("\\", "/", '"', "b", "f", "n", "r", "t", "u")
    local literals      = create_set("true", "false", "null")
    local literal_map = {
      [ "true"  ] = true,
      [ "false" ] = false,
      [ "null"  ] = nil,
    }
    local function next_char(str, idx, set, negate)
      for i = idx, #str do
        if set[str:sub(i, i)] ~= negate then
          return i
        end
      end
      return #str + 1
    end
    local function decode_error(str, idx, msg)
      local line_count = 1
      local col_count = 1
      for i = 1, idx - 1 do
        col_count = col_count + 1
        if str:sub(i, i) == "\n" then
          line_count = line_count + 1
          col_count = 1
        end
      end
      error( string.format("%s at line %d col %d", msg, line_count, col_count) )
    end
    local function codepoint_to_utf8(n)
      local f = math.floor
      if n <= 0x7f then
        return string.char(n)
      elseif n <= 0x7ff then
        return string.char(f(n / 64) + 192, n % 64 + 128)
      elseif n <= 0xffff then
        return string.char(f(n / 4096) + 224, f(n % 4096 / 64) + 128, n % 64 + 128)
      elseif n <= 0x10ffff then
        return string.char(f(n / 262144) + 240, f(n % 262144 / 4096) + 128,
                           f(n % 4096 / 64) + 128, n % 64 + 128)
      end
      error( string.format("invalid unicode codepoint '%x'", n) )
    end
    local function parse_unicode_escape(s)
      local n1 = tonumber( s:sub(1, 4),  16 )
      local n2 = tonumber( s:sub(7, 10), 16 )
      if n2 then
        return codepoint_to_utf8((n1 - 0xd800) * 0x400 + (n2 - 0xdc00) + 0x10000)
      else
        return codepoint_to_utf8(n1)
      end
    end
    local function parse_string(str, i)
      local res = ""
      local j = i + 1
      local k = j
      while j <= #str do
        local x = str:byte(j)
        if x < 32 then
          decode_error(str, j, "control character in string")
        elseif x == 92 then -- `\`: Escape
          res = res .. str:sub(k, j - 1)
          j = j + 1
          local c = str:sub(j, j)
          if c == "u" then
            local hex = str:match("^[dD][89aAbB]%x%x\\u%x%x%x%x", j + 1)
                     or str:match("^%x%x%x%x", j + 1)
                     or decode_error(str, j - 1, "invalid unicode escape in string")
            res = res .. parse_unicode_escape(hex)
            j = j + #hex
          else
            if not escape_chars[c] then
              decode_error(str, j - 1, "invalid escape char '" .. c .. "' in string")
            end
            res = res .. escape_char_map_inv[c]
          end
          k = j + 1
        elseif x == 34 then -- `"`: End of string
          res = res .. str:sub(k, j - 1)
          return res, j + 1
        end
        j = j + 1
      end
      decode_error(str, i, "expected closing quote for string")
    end
    local function parse_number(str, i)
      local x = next_char(str, i, delim_chars)
      local s = str:sub(i, x - 1)
      local n = tonumber(s)
      if not n then
        decode_error(str, i, "invalid number '" .. s .. "'")
      end
      return n, x
    end
    local function parse_literal(str, i)
      local x = next_char(str, i, delim_chars)
      local word = str:sub(i, x - 1)
      if not literals[word] then
        decode_error(str, i, "invalid literal '" .. word .. "'")
      end
      return literal_map[word], x
    end
    local function parse_array(str, i)
      local res = {}
      local n = 1
      i = i + 1
      while 1 do
        local x
        i = next_char(str, i, space_chars, true)
        if str:sub(i, i) == "]" then
          i = i + 1
          break
        end
        x, i = parse(str, i)
        res[n] = x
        n = n + 1
        i = next_char(str, i, space_chars, true)
        local chr = str:sub(i, i)
        i = i + 1
        if chr == "]" then break end
        if chr ~= "," then decode_error(str, i, "expected ']' or ','") end
      end
      return res, i
    end
    local function parse_object(str, i)
      local res = {}
      i = i + 1
      while 1 do
        local key, val
        i = next_char(str, i, space_chars, true)
        if str:sub(i, i) == "}" then
          i = i + 1
          break
        end
        if str:sub(i, i) ~= '"' then
          decode_error(str, i, "expected string for key")
        end
        key, i = parse(str, i)
        i = next_char(str, i, space_chars, true)
        if str:sub(i, i) ~= ":" then
          decode_error(str, i, "expected ':' after key")
        end
        i = next_char(str, i + 1, space_chars, true)
        val, i = parse(str, i)
        res[key] = val
        i = next_char(str, i, space_chars, true)
        local chr = str:sub(i, i)
        i = i + 1
        if chr == "}" then break end
        if chr ~= "," then decode_error(str, i, "expected '}' or ','") end
      end
      return res, i
    end
    local char_func_map = {
      [ '"' ] = parse_string,
      [ "0" ] = parse_number,
      [ "1" ] = parse_number,
      [ "2" ] = parse_number,
      [ "3" ] = parse_number,
      [ "4" ] = parse_number,
      [ "5" ] = parse_number,
      [ "6" ] = parse_number,
      [ "7" ] = parse_number,
      [ "8" ] = parse_number,
      [ "9" ] = parse_number,
      [ "-" ] = parse_number,
      [ "t" ] = parse_literal,
      [ "f" ] = parse_literal,
      [ "n" ] = parse_literal,
      [ "[" ] = parse_array,
      [ "{" ] = parse_object,
    }
    parse = function(str, idx)
      local chr = str:sub(idx, idx)
      local f = char_func_map[chr]
      if f then
        return f(str, idx)
      end
      decode_error(str, idx, "unexpected character '" .. chr .. "'")
    end
    function json.decode(str)
      if type(str) ~= "string" then
        error("expected argument of type string, got " .. type(str))
      end
      local res, idx = parse(str, next_char(str, 1, space_chars, true))
      idx = next_char(str, idx, space_chars, true)
      if idx <= #str then
        decode_error(str, idx, "trailing garbage")
      end
      return res
    end
    return json
end)
rawset(__BUNDLER_FILES, "updater", function ()
    local log = require("log");
    local json = require("json");
    local notifier = require("notifier");
    local _;
    local updater  = {};
    updater.__index = updater;
    function updater.new()
    	local c = setmetatable({}, updater);
        c.last_check = os.time();
        c.threshold = ((20*60)/24/60/60) * 5;
        c.api_url = "https://git.mcorangehq.xyz/api/v1/repos/xomf/keypadOS";
        c.curr_commit_hash = "";
        c.updated_files = {};
        c:addEntry("startup.lua", "main", "https://git.mcorangehq.xyz/xomf/keypadOS/raw/branch/main/rtmc.lua");
    	return c
    end
    function updater:addEntry(path, branch, url)
        self.updated_files[path] = {
            branch = branch,
            url = url
        };
        self:update(path, url);
    end
    function updater:checkAndUpdateAll()
        local updated = false;
        for path, entry in pairs(self.updated_files) do
            if self:check(entry.branch) then
                self:update(path, entry.url);
                updated = true;
            end
        end
        if updated then
            log.info("Rebooting!");
            sleep(3);
            os.reboot();
        end
    end
    function updater:check(branch)
        local curr = os.time();
        if math.abs(curr - self.last_check) < self.threshold then
            if self.curr_commit_hash ~= "" then
                return false;
            end
        end
        self.last_check = curr;
        local req, rerr = http.get(self.api_url .. "/commits?sha="..branch);
        if not req then
            log.error("Updater:check: Could not send request: " .. rerr);
            return false;
        end
        local body, berr = req.readAll();
        if not body then
            log.error("Updater:check: Could not get body of request: " .. berr);
            return false;
        end
        local data = json.decode(body);
        if self.curr_commit_hash == "" then
            log.debug("No commit hash found, setting");
            self.curr_commit_hash = data[1].sha;
        elseif data[1].sha ~= self.curr_commit_hash then
            log.debug("Commit hash doesnt match, probbably an update");
            return true;
        end
        if data[1].hash == self.curr_commit_hash then
            log.debug("Commit hash matches");
            log.debug(data[1].sha .. " (old) => (new) " .. self.curr_commit_hash);
        end
        return false;
    end
    function updater:update(path, url)
        local req, rerr = http.get(url .. "?x=" .. tostring(math.random(0,69420)));
        if not req then
            log.error("Updater:update: Could not send request for update: " .. rerr);
            return;
        end
        local body, berr = req.readAll();
        if not body then
            log.error("Updater:update: Could not get file for update: " .. berr);
            return;
        end
        if fs.exists(path .. ".bak") then
            fs.delete(path .. ".bak");
        end
        if fs.exists(path) then
            fs.copy(path, path..".bak");
            fs.delete(path);
        end
        local fd = fs.open(path, "w");
        fd.write(body);
        fd.close();
        local notif = "Computer #" .. tostring( os.getComputerID() ) .. " updating";
        notifier.notify("5", notif);
        log.debug("New update written to disk (" .. path .. ")");
    end
    return updater;
end)
rawset(__BUNDLER_FILES, "notifier", function ()
    local notifier = {};
    notifier.__index = notifier;
    function notifier.notify(priority, body)
        local headers = {
            ["Priority"] = priority;
        };
        http.post("https://ntfy.sh/keypadOS_alerts", body, headers);
    end
    return notifier;
end)
if pcall(debug.getlocal, 4, 1) then
    local exports = {};
    for k, v in pairs(__BUNDLER_FILES) do
        if k ~= "main" then
            exports[k] = v();
        end
    end
    return exports;
else
    require("main").main()
end