mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-01-05 23:43:07 +00:00
83 lines
2.6 KiB
Nim
83 lines
2.6 KiB
Nim
{.push raises: [].}
|
|
|
|
import std/options, results, chronos, libp2p/peerid
|
|
import ../waku_core, ./rpc, ../waku_relay/protocol
|
|
|
|
from ../waku_core/codecs import WakuLightPushCodec
|
|
export WakuLightPushCodec
|
|
|
|
type LightpushStatusCode* = enum
|
|
SUCCESS = uint32(200)
|
|
BAD_REQUEST = uint32(400)
|
|
PAYLOAD_TOO_LARGE = uint32(413)
|
|
INVALID_MESSAGE_ERROR = uint32(420)
|
|
UNSUPPORTED_PUBSUB_TOPIC = uint32(421)
|
|
TOO_MANY_REQUESTS = uint32(429)
|
|
INTERNAL_SERVER_ERROR = uint32(500)
|
|
SERVICE_NOT_AVAILABLE = uint32(503)
|
|
OUT_OF_RLN_PROOF = uint32(504)
|
|
NO_PEERS_TO_RELAY = uint32(505)
|
|
|
|
type ErrorStatus* = tuple[code: LightpushStatusCode, desc: Option[string]]
|
|
type WakuLightPushResult* = Result[uint32, ErrorStatus]
|
|
|
|
type PushMessageHandler* = proc(
|
|
peer: PeerId, pubsubTopic: PubsubTopic, message: WakuMessage
|
|
): Future[WakuLightPushResult] {.async.}
|
|
|
|
const TooManyRequestsMessage* = "Request rejected due to too many requests"
|
|
|
|
func isSuccess*(response: LightPushResponse): bool =
|
|
return response.statusCode == LightpushStatusCode.SUCCESS.uint32
|
|
|
|
func toPushResult*(response: LightPushResponse): WakuLightPushResult =
|
|
if isSuccess(response):
|
|
return ok(response.relayPeerCount.get(0))
|
|
else:
|
|
return err((response.statusCode.LightpushStatusCode, response.statusDesc))
|
|
|
|
func lightpushSuccessResult*(relayPeerCount: uint32): WakuLightPushResult =
|
|
return ok(relayPeerCount)
|
|
|
|
func lightpushResultInternalError*(msg: string): WakuLightPushResult =
|
|
return err((LightpushStatusCode.INTERNAL_SERVER_ERROR, some(msg)))
|
|
|
|
func lighpushErrorResult*(
|
|
statusCode: LightpushStatusCode, desc: Option[string]
|
|
): WakuLightPushResult =
|
|
return err((statusCode, desc))
|
|
|
|
func lighpushErrorResult*(
|
|
statusCode: LightpushStatusCode, desc: string
|
|
): WakuLightPushResult =
|
|
return err((statusCode, some(desc)))
|
|
|
|
func mapPubishingErrorToPushResult*(
|
|
publishOutcome: PublishOutcome
|
|
): WakuLightPushResult =
|
|
case publishOutcome
|
|
of NoTopicSpecified:
|
|
return err(
|
|
(LightpushStatusCode.INVALID_MESSAGE_ERROR, some("Empty topic, skipping publish"))
|
|
)
|
|
of DuplicateMessage:
|
|
return err(
|
|
(LightpushStatusCode.INVALID_MESSAGE_ERROR, some("Dropping already-seen message"))
|
|
)
|
|
of NoPeersToPublish:
|
|
return err(
|
|
(
|
|
LightpushStatusCode.NO_PEERS_TO_RELAY,
|
|
some("No peers for topic, skipping publish"),
|
|
)
|
|
)
|
|
of CannotGenerateMessageId:
|
|
return err(
|
|
(
|
|
LightpushStatusCode.INTERNAL_SERVER_ERROR,
|
|
some("Error generating message id, skipping publish"),
|
|
)
|
|
)
|
|
else:
|
|
return err((LightpushStatusCode.INTERNAL_SERVER_ERROR, none[string]()))
|