Implemented signature parsing to create client rpc procs based on params

This commit is contained in:
coffeepots 2018-05-22 20:48:12 +01:00
parent ff93b18296
commit 3878ee0ad0

View File

@ -1,4 +1,5 @@
import asyncnet, asyncdispatch, tables, json, oids, ethcalls, macros
import ../ ethtypes, stint, ../ jsonconverters
type
RpcClient* = ref object
@ -88,17 +89,113 @@ proc connect*(self: RpcClient, address: string, port: Port) {.async.} =
self.port = port
asyncCheck processData(self)
macro generateCalls: untyped =
## Generate templates for client calls so that:
## client.call("web3_clientVersion", params)
## can be written as:
## client.web3_clientVersion(params)
result = newStmtList()
for callName in ETHEREUM_RPC_CALLS:
let nameLit = ident(callName)
result.add(quote do:
template `nameLit`*(client: RpcClient, params: JsonNode): Future[Response] = client.call(`callName`, params) # TODO: Back to template
import jsonmarshal
proc createRpcFromSig*(rpcDecl: NimNode): NimNode =
let procNameState = rpcDecl[0]
var
parameters = rpcDecl.findChild(it.kind == nnkFormalParams).copy
path: NimNode
pathStr: string
# get proc signature's name. This becomes the path we send to the server
if procNameState.kind == nnkPostFix:
path = rpcDecl[0][1]
else:
path = rpcDecl[0]
pathStr = $path
if parameters.isNil or parameters.kind == nnkEmpty:
parameters = newNimNode(nnkFormalParams)
# if no return parameters specified (parameters[0])
if parameters.len == 0: parameters.add(newEmptyNode())
# insert rpc client as first parameter
let
clientParam =
nnkIdentDefs.newTree(
ident"client",
ident"RpcClient",
newEmptyNode()
)
parameters.insert(1, clientParam)
# For each input parameter we need to
# take the Nim type and translate to json with `%`.
# For return types, we need to take the json and
# convert it to the Nim type.
var
callBody = newStmtList()
returnType: NimNode
let jsonParamIdent = genSym(nskVar, "jsonParam")
callBody.add(quote do:
var `jsonParamIdent` = newJArray()
)
if parameters.len > 2:
# skip return type and the inserted rpc client parameter
# add the rest to json node via `%`
for i in 2 ..< parameters.len:
let curParam = parameters[i][0]
if curParam.kind != nnkEmpty:
callBody.add(quote do:
`jsonParamIdent`.add(%`curParam`)
)
if parameters[0].kind != nnkEmpty:
returnType = parameters[0]
else:
returnType = ident"JsonNode"
# convert return type to Future
parameters[0] = nnkBracketExpr.newTree(ident"Future", returnType)
# client call to server using json params
var updatedParams = newSeq[NimNode]()
# convert parameter tree to seq
for p in parameters: updatedParams.add(p)
# create new proc
result = newProc(path, updatedParams, callBody)
# convert this proc to async
result.addPragma ident"async"
# export this proc
result[0] = nnkPostFix.newTree(ident"*", ident(pathStr))
# add rpc call to proc body
var callResult = genSym(nskVar, "res")
callBody.add(quote do:
let res = await client.call(`pathStr`, `jsonParamIdent`)
if res.error: raise newException(ValueError, $res.result)
var `callResult` = res.result
)
let
procRes = ident"result"
# now we need to extract the response and build it into the expected type
if returnType != ident"JsonNode":
let setup = setupParamFromJson(procRes, returnType, callResult)
callBody.add(quote do: `setup`)
else:
callBody.add(quote do:
`procRes` = `callResult`
)
when defined(nimDumpRpcs):
echo pathStr, ":\n", result.repr
from os import getCurrentDir, DirSep
from strutils import rsplit
macro processRpcSigs(): untyped =
result = newStmtList()
const
codePath = currentSourcePath.rsplit(DirSep, 1)[0] & DirSep & "ethcallsigs.nim"
code = staticRead(codePath)
let parsedCode = parseStmt(code)
var line = 0
while line < parsedCode.len:
if parsedCode[line].kind == nnkProcDef: break
line += 1
for curLine in line ..< parsedCode.len:
var procDef = createRpcFromSig(parsedCode[curLine])
result.add(procDef)
# generate all client ethereum rpc calls
generateCalls()
processRpcSigs()