7f513a45bf
I had to type in the updated code manually, since for some reason, the version here uses tabs instead of two spaces. Personally I don't like the way that these external libraries are included here - maybe we could do something like urlfs but only download them once instead of every boot - but hopefully we won't need to worry about that when there is an alternative krist library (ETA next century)
144 lines
2.9 KiB
Lua
144 lines
2.9 KiB
Lua
local jua = nil
|
|
local idPatt = "#R%d+"
|
|
|
|
if not ((socket and socket.websocket) or http.websocketAsync) then
|
|
error("You do not have CC:Tweaked/CCTweaks installed or you are not on the latest version.")
|
|
end
|
|
|
|
local newws = socket and socket.websocket or http.websocketAsync
|
|
local async
|
|
if socket and socket.websocket then
|
|
async = false
|
|
else
|
|
async = true
|
|
end
|
|
|
|
callbackRegistry = {}
|
|
wsRegistry = {}
|
|
|
|
local function gfind(str, patt)
|
|
local t = {}
|
|
for found in str:gmatch(patt) do
|
|
table.insert(t, found)
|
|
end
|
|
|
|
if #t > 0 then
|
|
return t
|
|
else
|
|
return nil
|
|
end
|
|
end
|
|
|
|
local function findID(url)
|
|
local found = gfind(url, idPatt)
|
|
if not found then return nil end
|
|
return tonumber(found[#found]:sub(found[#found]:find("%d+")))
|
|
end
|
|
|
|
local function newID()
|
|
return #callbackRegistry + 1
|
|
end
|
|
|
|
local function trimID(url)
|
|
local found = gfind(url, idPatt)
|
|
local s, e = url:find(found[#found])
|
|
return url:sub(1, s-1)
|
|
end
|
|
|
|
function open(callback, url, headers)
|
|
local id
|
|
if async then
|
|
id = newID()
|
|
end
|
|
local newUrl
|
|
if async then
|
|
newUrl = url .. "#R" .. id
|
|
newws(newUrl, headers)
|
|
else
|
|
if headers then
|
|
error("Websocket headers not supported under CCTweaks")
|
|
end
|
|
local ws = newws(url)
|
|
ws.send = ws.write
|
|
id = ws.id()
|
|
wsRegistry[id] = ws
|
|
end
|
|
callbackRegistry[id] = callback
|
|
return id
|
|
end
|
|
|
|
function init(jua)
|
|
jua = jua
|
|
if async then
|
|
jua.on("websocket_success", function(event, url, handle)
|
|
local id = findID(url)
|
|
if id and callbackRegistry[id].success then
|
|
callbackRegistry[id].success(id, handle)
|
|
end
|
|
end)
|
|
|
|
jua.on("websocket_failure", function(event, url)
|
|
local id = findID(url)
|
|
if id then
|
|
if callbackRegistry[id].failure then
|
|
callbackRegistry[id].failure(id)
|
|
end
|
|
table.remove(callbackRegistry, id)
|
|
end
|
|
end)
|
|
|
|
jua.on("websocket_message", function(event, url, data)
|
|
local id = findID(url)
|
|
if id and callbackRegistry[id].message then
|
|
callbackRegistry[id].message(id, data)
|
|
end
|
|
end)
|
|
|
|
jua.on("websocket_closed", function(event, url)
|
|
local id = findID(url)
|
|
if id then
|
|
if callbackRegistry[id].closed then
|
|
callbackRegistry[id].closed(id)
|
|
end
|
|
table.remove(callbackRegistry, id)
|
|
end
|
|
end)
|
|
else
|
|
jua.on("socket_connect", function(event, id)
|
|
if id and callbackRegistry[id].success then
|
|
callbackRegistry[id].success(id, wsRegistry[id])
|
|
end
|
|
end)
|
|
|
|
jua.on("socket_error", function(event, id, msg)
|
|
if id then
|
|
if callbackRegistry[id].failure then
|
|
callbackRegistry[id].failure(id)
|
|
end
|
|
table.remove(callbackRegistry, id)
|
|
end
|
|
end)
|
|
|
|
jua.on("socket_message", function(event, id)
|
|
if id and callbackRegistry[id].message then
|
|
local data = wsRegistry[id].read()
|
|
callbackRegistry[id].message(id, data)
|
|
end
|
|
end)
|
|
|
|
jua.on("socket_closed", function(event, id)
|
|
if id then
|
|
if callbackRegistry[id].closed then
|
|
callbackRegistry[id].closed(id)
|
|
end
|
|
table.remove(callbackRegistry, id)
|
|
end
|
|
end)
|
|
end
|
|
end
|
|
|
|
return {
|
|
open = open,
|
|
init = init
|
|
}
|