nwaku/waku/node/v2/rpc/wakurpc.nim

53 lines
1.9 KiB
Nim
Raw Normal View History

import
json_rpc/rpcserver, tables, options,
eth/[common, rlp, keys, p2p],
2020-04-29 05:02:42 +00:00
../../../protocol/v2/waku_protocol,
nimcrypto/[sysrand, hmac, sha2, pbkdf2],
2020-05-22 06:12:05 +00:00
../../v1/rpc/[rpc_types, hexstrings, key_storage],
../waku_types,
libp2p/protocols/pubsub/pubsub
from stew/byteutils import hexToSeqByte, hexToByteArray
# Instead of using rlpx waku_protocol here, lets do mock waku2_protocol
# This should wrap GossipSub, not use EthereumNode here
2020-04-29 05:19:48 +00:00
# In Waku0/1 we use node.protocolState(Waku) a lot to get information
# Also keys to get priate key, etc
# Where is the equivalent in Waku/2?
# TODO: Extend to get access to protocol state and keys
#proc setupWakuRPC*(node: EthereumNode, keys: KeyStorage, rpcsrv: RpcServer) =
proc setupWakuRPC*(wakuProto: WakuProto, rpcsrv: RpcServer) =
# Seems easy enough, lets try to get this first
rpcsrv.rpc("waku_version") do() -> string:
2020-05-19 03:49:25 +00:00
## Returns string of the current Waku protocol version.
result = WakuSubCodec
# TODO: Implement symkey etc logic
2020-05-22 07:28:51 +00:00
rpcsrv.rpc("waku_publish") do(topic: string, message: string) -> bool:
let data = cast[seq[byte]](message)
# Assumes someone subscribing on this topic
#let wakuSub = wakuProto.switch.pubsub
let wakuSub = cast[WakuSub](wakuProto.switch.pubSub.get())
# XXX also future return type
2020-05-22 07:28:51 +00:00
discard wakuSub.publish(topic, data)
return true
#if not result:
# raise newException(ValueError, "Message could not be posted")
# TODO: Handler / Identifier logic
rpcsrv.rpc("waku_subscribe") do(topic: string) -> bool:
let wakuSub = cast[WakuSub](wakuProto.switch.pubSub.get())
# XXX: Hacky in-line handler
proc handler(topic: string, data: seq[byte]) {.async, gcsafe.} =
info "Hit subscribe handler", topic=topic, data=data
discard wakuSub.subscribe(topic, handler)
return true
#if not result:
# raise newException(ValueError, "Message could not be posted")