rpc.lua

--[[

-------------------------
-- Example usage from Lua
-------------------------
local rpc = require "rpc"
local py = rpc.create()

-- Call 'bar' in connected Python client(s).
py.bar(1,2,3)

-- set up callable function 'fnord'
py.register("fnord", function(a, b)
   print("fnord called with", a, b)
end)


----------------------------
-- Example usage from Python
-- (see also hosted.py)
----------------------------
from hosted import node
lua = node.rpc()

# Connect to info-beamer/Lua and run fnord function
lua.fnord("a", "b")

# Register callable function 'bar'
@lua.call
def bar(a, b, c):
    print("called from lua", a, b, c)

]]


local function create(endpoints)
    endpoints = endpoints or {}
    local json = require "json"
    local clients = {}
    node.set_flag("close_clients", true)
    node.event("connect", function(client, prefix)
        if prefix == "rpc/python" then
            clients[client] = true
        end
    end)
    node.event("disconnect", function(client)
        clients[client] = nil
    end)
    node.event("input", function(line, client)
        if clients[client] then
            local call = json.decode(line)
            local fn = table.remove(call, 1)
            if endpoints[fn] then
                endpoints[fn](unpack(call))
            end
        end
    end)
    local function send_call(call, ...)
        local args = {...}
        table.insert(args, 1, call)
        local pkt = json.encode(args)
        local sent = false
        for client, _ in pairs(clients) do
            sent = true
            node.client_write(client, pkt) 
        end
        return sent
    end
    return setmetatable({
        register = function(name, fn)
            endpoints[name] = fn
        end,
    }, {
        __index = function(t, call)
            return function(...)
                return send_call(call, ...)
            end
        end
    })
end

return {
    create = create,
}