Rework LuCI build system
[project/luci.git] / libs / luci-lib-rpcc / luasrc / rpcc.lua
diff --git a/libs/luci-lib-rpcc/luasrc/rpcc.lua b/libs/luci-lib-rpcc/luasrc/rpcc.lua
new file mode 100644 (file)
index 0000000..ef94d3b
--- /dev/null
@@ -0,0 +1,84 @@
+--[[
+LuCIRPCc
+(c) 2009 Steven Barth <steven@midlink.org>
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+        http://www.apache.org/licenses/LICENSE-2.0
+
+$Id$
+]]--
+
+local util = require "luci.util"
+local json = require "luci.json"
+local ltn12 = require "luci.ltn12"
+local nixio = require "nixio", require "nixio.util"
+
+local tostring, assert, setmetatable = tostring, assert, setmetatable
+local error = error
+
+--- LuCI RPC Client.
+-- @cstyle instance
+module "luci.rpcc"
+
+RQLIMIT = 32 * nixio.const.buffersize
+
+--- Create a new JSON-RPC stream client.
+-- @class function
+-- @param fd File descriptor
+-- @param v1 Use protocol version 1.0
+-- @return RPC Client
+Client = util.class()
+
+function Client.__init__(self, fd, v1)
+       self.fd = fd
+       self.uniqueid = tostring(self):match("0x([a-f0-9]+)")
+       self.msgid = 1
+       self.v1 = v1
+end
+
+--- Request an RP call and get the response.
+-- @param method Remote method
+-- @param params Parameters
+-- @param notification Notification only?
+-- @return response 
+function Client.request(self, method, params, notification)
+       local oldchunk = self.decoder and self.decoder.chunk
+       self.decoder = json.ActiveDecoder(self.fd:blocksource(nil, RQLIMIT))
+       self.decoder.chunk = oldchunk
+       
+       local reqid = self.msgid .. self.uniqueid
+       local reqdata = json.Encoder({
+               id = (not notification) and (self.msgid .. self.uniqueid) or nil,
+               jsonrpc = (not self.v1) and "2.0" or nil,
+               method = method,
+               params = params
+       })
+       ltn12.pump.all(reqdata:source(), self.fd:sink())
+       if not notification then
+               self.msgid = self.msgid + 1
+               local response = self.decoder:get()
+               assert(response.id == reqid, "Invalid response id")
+               if response.error then
+                       error(response.error.message or response.error)
+               end
+               return response.result
+       end
+end
+
+--- Create a transparent RPC proxy.
+-- @param prefix Method prefix
+-- @return RPC Proxy object
+function Client.proxy(self, prefix)
+       prefix = prefix or ""
+       return setmetatable({}, {
+               __call = function(proxy, ...)
+                       return self:request(prefix, {...})
+               end,
+               __index = function(proxy, name)
+                       return self:proxy(prefix .. name .. ".")
+               end
+       })
+end
\ No newline at end of file