85 lines
2.3 KiB
Lua
85 lines
2.3 KiB
Lua
|
local URL_REFILL_MACHINE = "http://gitea.fse-media.group/Henry_Du/FSE-Ticket.sys/src/commit/d6aa03d3a78f8b6fbd2754bf3f4199392c6ef4c1/refillmachine.lua"
|
||
|
|
|
||
|
|
local function writeFile(path, content, binary)
|
||
|
|
local mode = binary and "wb" or "w"
|
||
|
|
local f = fs.open(path, mode)
|
||
|
|
if not f then return false end
|
||
|
|
f.write(content)
|
||
|
|
f.close()
|
||
|
|
return true
|
||
|
|
end
|
||
|
|
|
||
|
|
local function atomicWrite(path, content, binary)
|
||
|
|
local tmp = path .. ".new"
|
||
|
|
if fs.exists(tmp) then fs.delete(tmp) end
|
||
|
|
if not writeFile(tmp, content, binary) then return false end
|
||
|
|
if fs.exists(path) then fs.delete(path) end
|
||
|
|
fs.move(tmp, path)
|
||
|
|
return true
|
||
|
|
end
|
||
|
|
|
||
|
|
local function httpGet(url)
|
||
|
|
if not http then return false, "HTTP API disabled" end
|
||
|
|
local okReq, err = pcall(function()
|
||
|
|
http.request({ url = url, method = "GET" })
|
||
|
|
end)
|
||
|
|
if not okReq then return false, tostring(err) end
|
||
|
|
|
||
|
|
while true do
|
||
|
|
local ev, p1, p2, p3 = os.pullEvent()
|
||
|
|
if ev == "http_success" and p1 == url then
|
||
|
|
local res = p2
|
||
|
|
if type(res) == "table" and type(res.readAll) == "function" then
|
||
|
|
local data = res.readAll()
|
||
|
|
res.close()
|
||
|
|
return true, data
|
||
|
|
end
|
||
|
|
return false, "invalid http response"
|
||
|
|
end
|
||
|
|
if ev == "http_failure" and p1 == url then
|
||
|
|
local failErr = p2
|
||
|
|
local res = p3
|
||
|
|
if type(p2) == "table" and type(p2.readAll) == "function" then
|
||
|
|
res = p2
|
||
|
|
failErr = p3
|
||
|
|
end
|
||
|
|
if type(res) == "table" and type(res.readAll) == "function" then
|
||
|
|
local data = res.readAll()
|
||
|
|
res.close()
|
||
|
|
return false, data
|
||
|
|
end
|
||
|
|
return false, tostring(failErr or "http_failure")
|
||
|
|
end
|
||
|
|
end
|
||
|
|
end
|
||
|
|
|
||
|
|
term.clear()
|
||
|
|
term.setCursorPos(1, 1)
|
||
|
|
print("Refill Machine Updater")
|
||
|
|
print("")
|
||
|
|
print("Downloading refill machine program...")
|
||
|
|
|
||
|
|
local ok, code = httpGet(URL_REFILL_MACHINE)
|
||
|
|
if not ok or type(code) ~= "string" or #code == 0 then
|
||
|
|
print("Download failed: " .. tostring(code or ""))
|
||
|
|
return
|
||
|
|
end
|
||
|
|
|
||
|
|
if not atomicWrite("startup.lua", code, false) then
|
||
|
|
print("Write failed: startup.lua")
|
||
|
|
return
|
||
|
|
end
|
||
|
|
if not atomicWrite("startup", code, false) then
|
||
|
|
print("Write failed: startup")
|
||
|
|
return
|
||
|
|
end
|
||
|
|
if not atomicWrite("refillmachine.lua", code, false) then
|
||
|
|
print("Write failed: refillmachine.lua")
|
||
|
|
return
|
||
|
|
end
|
||
|
|
|
||
|
|
print("")
|
||
|
|
print("Done.")
|
||
|
|
print("refillmachine.lua has been updated and installed as startup.")
|
||
|
|
print("Reboot the computer to apply the update.")
|