mirror of
https://github.com/logos-messaging/logos-messaging-nim.git
synced 2026-05-11 04:49:26 +00:00
65 lines
2.1 KiB
Nim
65 lines
2.1 KiB
Nim
import tables, std/options, chronicles
|
|
import ../waku_lightpush/[rpc, common]
|
|
import libp2p/peerid
|
|
|
|
const BadLightPushErrorCodes* = [
|
|
LightPushErrorCode.INTERNAL_SERVER_ERROR,
|
|
LightPushErrorCode.SERVICE_NOT_AVAILABLE,
|
|
LightPushErrorCode.OUT_OF_RLN_PROOF,
|
|
LightPushErrorCode.NO_PEERS_TO_RELAY
|
|
]
|
|
|
|
type
|
|
ResponseQuality* = enum
|
|
BadResponse
|
|
NeutralResponse
|
|
GoodResponse
|
|
|
|
# Encode reputation indicator as Option[bool]:
|
|
# some(true) => GoodRep
|
|
# some(false) => BadRep
|
|
# none(bool) => unknown / not set
|
|
ReputationManager* = ref object
|
|
reputationOf*: Table[PeerId, Option[bool]]
|
|
|
|
proc init*(T: type ReputationManager): ReputationManager =
|
|
return ReputationManager(reputationOf: initTable[PeerId, Option[bool]]())
|
|
|
|
proc setReputation*(
|
|
manager: var ReputationManager, peer: PeerId, repValue: Option[bool]
|
|
) =
|
|
manager.reputationOf[peer] = repValue
|
|
|
|
proc getReputation*(manager: ReputationManager, peer: PeerId): Option[bool] =
|
|
if peer in manager.reputationOf:
|
|
result = manager.reputationOf[peer]
|
|
else:
|
|
result = none(bool)
|
|
|
|
### Lightpush-specific functionality ###
|
|
|
|
# Evaluate the quality of a LightPushResponse by checking its status code
|
|
proc evaluateResponse*(response: LightPushResponse): ResponseQuality =
|
|
if response.isSuccess():
|
|
return GoodResponse
|
|
elif response.statusCode in BadLightPushErrorCodes:
|
|
return BadResponse
|
|
else:
|
|
return NeutralResponse
|
|
|
|
# Update reputation of the peer based on LightPushResponse quality
|
|
proc updateReputationFromResponse*(
|
|
manager: var ReputationManager, peer: PeerId, response: LightPushResponse
|
|
) =
|
|
let respQuality = evaluateResponse(response)
|
|
case respQuality
|
|
of BadResponse:
|
|
debug "Assign bad reputation for peer", peer = peer
|
|
manager.setReputation(peer, some(false)) # false => BadRep
|
|
of GoodResponse:
|
|
debug "Assign good reputation for peer", peer = peer
|
|
manager.setReputation(peer, some(true)) # true => GoodRep
|
|
of NeutralResponse:
|
|
debug "Neutral response - reputation unchanged for peer", peer = peer
|
|
# Don't change reputation for neutral responses
|