OC-PsychOS2/lib/rpc.lua

80 lines
1.8 KiB
Lua
Raw Normal View History

2019-11-26 21:34:40 +11:00
local serial = require "serialization"
local minitel = require "minitel"
local event = require "event"
local rpc = {}
_G.rpcf = {}
rpc.port = 111
local function rpcexec(_, from, port, data)
if port == rpc.port then
local rpcrq = serial.unserialize(data)
local rpcn, rpcid = table.remove(rpcrq,1), table.remove(rpcrq,1)
if rpcf[rpcn] then
local rt = {pcall(rpcf[rpcn],table.unpack(rpcrq))}
if rt[1] == true then
table.remove(rt,1)
end
minitel.send(from,port,serial.serialize({rpcid,table.unpack(rt)}))
else
end
end
end
function rpcf.list()
local rt = {}
for k,v in pairs(rpcf) do
rt[#rt+1] = k
end
return rt
end
function rpc.call(hostname,fn,...)
if hostname == "localhost" then
return rpcf[fn](...)
end
local rv = minitel.genPacketID()
minitel.rsend(hostname,rpc.port,serial.serialize({fn,rv,...}),true)
local st = computer.uptime()
local rt = {}
repeat
local _, from, port, data = event.pull(30, "net_msg", hostname, rpc.port)
rt = serial.unserialize(data) or {}
until rt[1] == rv or computer.uptime() > st + 30
if table.remove(rt,1) == rv then
return table.unpack(rt)
end
return false
end
function rpc.proxy(hostname,filter)
filter=(filter or "").."(.+)"
local fnames = rpc.call(hostname,"list")
if not fnames then return false end
local rt = {}
for k,v in pairs(fnames) do
fv = v:match(filter)
if fv then
rt[fv] = function(...)
return rpc.call(hostname,v,...)
end
end
end
return rt
end
function rpc.register(name,fn)
local rpcrunning = false
for k,v in pairs(os.tasks()) do
if os.taskInfo(v).name == "rpc daemon" then
rpcrunning = true
end
end
if not rpcrunning then
os.spawn(function()
while true do
rpcexec(event.pull("net_msg"))
end
end,"rpc daemon")
end
rpcf[name] = fn
end
return rpc