mirror of https://github.com/waku-org/nwaku.git
90 lines
3.6 KiB
Nim
90 lines
3.6 KiB
Nim
{.push raises: [Defect].}
|
|
|
|
import
|
|
std/[options, json],
|
|
eth/keys,
|
|
../../../v1/node/rpc/hexstrings,
|
|
../../protocol/waku_store,
|
|
../../protocol/waku_message,
|
|
../../utils/time,
|
|
../../utils/pagination,
|
|
../waku_payload,
|
|
./jsonrpc_types
|
|
|
|
export hexstrings
|
|
|
|
## Json marshalling
|
|
|
|
proc `%`*(value: WakuMessage): JsonNode =
|
|
## This ensures that seq[byte] fields are marshalled to hex-format JStrings
|
|
## (as defined in `hexstrings.nim`) rather than the default JArray[JInt]
|
|
let jObj = newJObject()
|
|
for k, v in value.fieldPairs:
|
|
jObj[k] = %v
|
|
return jObj
|
|
|
|
## Conversion tools
|
|
## Since the Waku v2 JSON-RPC API has its own defined types,
|
|
## we need to convert between these and the types for the Nim API
|
|
|
|
proc toPagingInfo*(pagingOptions: StorePagingOptions): PagingInfo =
|
|
PagingInfo(pageSize: pagingOptions.pageSize,
|
|
cursor: if pagingOptions.cursor.isSome: pagingOptions.cursor.get else: Index(),
|
|
direction: if pagingOptions.forward: PagingDirection.FORWARD else: PagingDirection.BACKWARD)
|
|
|
|
proc toPagingOptions*(pagingInfo: PagingInfo): StorePagingOptions =
|
|
StorePagingOptions(pageSize: pagingInfo.pageSize,
|
|
cursor: some(pagingInfo.cursor),
|
|
forward: if pagingInfo.direction == PagingDirection.FORWARD: true else: false)
|
|
|
|
proc toStoreResponse*(historyResponse: HistoryResponse): StoreResponse =
|
|
StoreResponse(messages: historyResponse.messages,
|
|
pagingOptions: if historyResponse.pagingInfo != PagingInfo(): some(historyResponse.pagingInfo.toPagingOptions()) else: none(StorePagingOptions))
|
|
|
|
proc toWakuMessage*(relayMessage: WakuRelayMessage, version: uint32): WakuMessage =
|
|
const defaultCT = ContentTopic("/waku/2/default-content/proto")
|
|
var t: Timestamp
|
|
if relayMessage.timestamp.isSome:
|
|
t = relayMessage.timestamp.get
|
|
else:
|
|
# incoming WakuRelayMessages with no timestamp will get 0 timestamp
|
|
t = Timestamp(0)
|
|
WakuMessage(payload: relayMessage.payload,
|
|
contentTopic: if relayMessage.contentTopic.isSome: relayMessage.contentTopic.get else: defaultCT,
|
|
version: version,
|
|
timestamp: t)
|
|
|
|
proc toWakuMessage*(relayMessage: WakuRelayMessage, version: uint32, rng: ref HmacDrbgContext, symkey: Option[SymKey], pubKey: Option[keys.PublicKey]): WakuMessage =
|
|
# @TODO global definition for default content topic
|
|
const defaultCT = ContentTopic("/waku/2/default-content/proto")
|
|
|
|
let payload = Payload(payload: relayMessage.payload,
|
|
dst: pubKey,
|
|
symkey: symkey)
|
|
|
|
var t: Timestamp
|
|
if relayMessage.timestamp.isSome:
|
|
t = relayMessage.timestamp.get
|
|
else:
|
|
# incoming WakuRelayMessages with no timestamp will get 0 timestamp
|
|
t = Timestamp(0)
|
|
|
|
WakuMessage(payload: payload.encode(version, rng[]).get(),
|
|
contentTopic: if relayMessage.contentTopic.isSome: relayMessage.contentTopic.get else: defaultCT,
|
|
version: version,
|
|
timestamp: t)
|
|
|
|
proc toWakuRelayMessage*(message: WakuMessage, symkey: Option[SymKey], privateKey: Option[keys.PrivateKey]): WakuRelayMessage =
|
|
# @TODO global definition for default content topic
|
|
|
|
let
|
|
keyInfo = if symkey.isSome(): KeyInfo(kind: Symmetric, symKey: symkey.get())
|
|
elif privateKey.isSome(): KeyInfo(kind: Asymmetric, privKey: privateKey.get())
|
|
else: KeyInfo(kind: KeyKind.None)
|
|
decoded = decodePayload(message, keyInfo)
|
|
|
|
WakuRelayMessage(payload: decoded.get().payload,
|
|
contentTopic: some(message.contentTopic),
|
|
timestamp: some(message.timestamp))
|
|
|