mirror of https://github.com/waku-org/nwaku.git
92 lines
2.6 KiB
Nim
92 lines
2.6 KiB
Nim
{.push raises: [].}
|
|
|
|
import
|
|
std/strformat,
|
|
stew/byteutils,
|
|
chronicles,
|
|
json_serialization,
|
|
json_serialization/std/options,
|
|
presto/route,
|
|
presto/common
|
|
|
|
import
|
|
../../waku/node/peer_manager,
|
|
../../../waku_node,
|
|
../../waku/waku_lightpush/common,
|
|
../../handlers,
|
|
../serdes,
|
|
../responses,
|
|
../rest_serdes,
|
|
./types
|
|
|
|
export types
|
|
|
|
logScope:
|
|
topics = "waku node rest lightpush api"
|
|
|
|
const futTimeoutForPushRequestProcessing* = 5.seconds
|
|
|
|
const NoPeerNoDiscoError =
|
|
RestApiResponse.serviceUnavailable("No suitable service peer & no discovery method")
|
|
|
|
const NoPeerNoneFoundError =
|
|
RestApiResponse.serviceUnavailable("No suitable service peer & none discovered")
|
|
|
|
proc useSelfHostedLightPush(node: WakuNode): bool =
|
|
return node.wakuLightPush != nil and node.wakuLightPushClient == nil
|
|
|
|
#### Request handlers
|
|
|
|
const ROUTE_LIGHTPUSH* = "/lightpush/v1/message"
|
|
|
|
proc installLightPushRequestHandler*(
|
|
router: var RestRouter,
|
|
node: WakuNode,
|
|
discHandler: Option[DiscoveryHandler] = none(DiscoveryHandler),
|
|
) =
|
|
router.api(MethodPost, ROUTE_LIGHTPUSH) do(
|
|
contentBody: Option[ContentBody]
|
|
) -> RestApiResponse:
|
|
## Send a request to push a waku message
|
|
debug "post", ROUTE_LIGHTPUSH, contentBody
|
|
|
|
let decodedBody = decodeRequestBody[PushRequest](contentBody)
|
|
|
|
if decodedBody.isErr():
|
|
return decodedBody.error()
|
|
|
|
let req: PushRequest = decodedBody.value()
|
|
|
|
let msg = req.message.toWakuMessage().valueOr:
|
|
return RestApiResponse.badRequest("Invalid message: " & $error)
|
|
|
|
var peer = RemotePeerInfo.init($node.switch.peerInfo.peerId)
|
|
if useSelfHostedLightPush(node):
|
|
discard
|
|
else:
|
|
peer = node.peerManager.selectPeer(WakuLightPushCodec).valueOr:
|
|
let handler = discHandler.valueOr:
|
|
return NoPeerNoDiscoError
|
|
|
|
let peerOp = (await handler()).valueOr:
|
|
return RestApiResponse.internalServerError("No value in peerOp: " & $error)
|
|
|
|
peerOp.valueOr:
|
|
return NoPeerNoneFoundError
|
|
|
|
let subFut = node.lightpushPublish(req.pubsubTopic, msg, peer)
|
|
|
|
if not await subFut.withTimeout(futTimeoutForPushRequestProcessing):
|
|
error "Failed to request a message push due to timeout!"
|
|
return RestApiResponse.serviceUnavailable("Push request timed out")
|
|
|
|
if subFut.value().isErr():
|
|
if subFut.value().error == TooManyRequestsMessage:
|
|
return RestApiResponse.tooManyRequests("Request rate limmit reached")
|
|
|
|
return RestApiResponse.serviceUnavailable(
|
|
fmt("Failed to request a message push: {subFut.value().error}")
|
|
)
|
|
|
|
return RestApiResponse.ok()
|