From f9c800008044d529bbffbb24e87b3ba49cdcd719 Mon Sep 17 00:00:00 2001
From: XeonSquared <redtailedairways@gmail.com>
Date: Tue, 26 Nov 2019 21:34:40 +1100
Subject: [PATCH] added an mtrpc library

---
 lib/rpc.lua | 79 +++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 79 insertions(+)
 create mode 100644 lib/rpc.lua

diff --git a/lib/rpc.lua b/lib/rpc.lua
new file mode 100644
index 0000000..fd9286f
--- /dev/null
+++ b/lib/rpc.lua
@@ -0,0 +1,79 @@
+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