mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-01-03 22:43:09 +00:00
3133aaaf71
* chore: use distinct type for Light push status codes * Make naming more explicit * test: use new light push error code in tests * fix missed line * fix thing
61 lines
2.0 KiB
Nim
61 lines
2.0 KiB
Nim
{.push raises: [].}
|
|
|
|
import results
|
|
|
|
import
|
|
../waku_core,
|
|
../waku_relay,
|
|
./common,
|
|
../waku_rln_relay,
|
|
../waku_rln_relay/protocol_types
|
|
|
|
import std/times, libp2p/peerid, stew/byteutils
|
|
|
|
proc checkAndGenerateRLNProof*(
|
|
rlnPeer: Option[WakuRLNRelay], message: WakuMessage
|
|
): Result[WakuMessage, string] =
|
|
# check if the message already has RLN proof
|
|
if message.proof.len > 0:
|
|
return ok(message)
|
|
|
|
if rlnPeer.isNone():
|
|
notice "Publishing message without RLN proof"
|
|
return ok(message)
|
|
# generate and append RLN proof
|
|
let
|
|
time = getTime().toUnix()
|
|
senderEpochTime = float64(time)
|
|
var msgWithProof = message
|
|
rlnPeer.get().appendRLNProof(msgWithProof, senderEpochTime).isOkOr:
|
|
return err(error)
|
|
return ok(msgWithProof)
|
|
|
|
proc getNilPushHandler*(): PushMessageHandler =
|
|
return proc(
|
|
peer: PeerId, pubsubTopic: string, message: WakuMessage
|
|
): Future[WakuLightPushResult] {.async.} =
|
|
return lightpushResultInternalError("no waku relay found")
|
|
|
|
proc getRelayPushHandler*(
|
|
wakuRelay: WakuRelay, rlnPeer: Option[WakuRLNRelay] = none[WakuRLNRelay]()
|
|
): PushMessageHandler =
|
|
return proc(
|
|
peer: PeerId, pubsubTopic: string, message: WakuMessage
|
|
): Future[WakuLightPushResult] {.async.} =
|
|
# append RLN proof
|
|
let msgWithProof = checkAndGenerateRLNProof(rlnPeer, message).valueOr:
|
|
return lighpushErrorResult(LightPushErrorCode.OUT_OF_RLN_PROOF, error)
|
|
|
|
(await wakuRelay.validateMessage(pubSubTopic, msgWithProof)).isOkOr:
|
|
return lighpushErrorResult(LightPushErrorCode.INVALID_MESSAGE, $error)
|
|
|
|
let publishedResult = await wakuRelay.publish(pubsubTopic, msgWithProof)
|
|
|
|
if publishedResult.isErr():
|
|
let msgHash = computeMessageHash(pubsubTopic, message).to0xHex()
|
|
notice "Lightpush request has not been published to any peers",
|
|
msg_hash = msgHash, reason = $publishedResult.error
|
|
return mapPubishingErrorToPushResult(publishedResult.error)
|
|
|
|
return lightpushSuccessResult(publishedResult.get().uint32)
|