nwaku/waku/incentivization/rpc_codec.nim

import std/options
import ../common/protobuf, ../waku_core, ./rpc

# Codec for EligibilityProof

proc encode*(epRpc: EligibilityProof): ProtoBuffer =
  var pb = initProtoBuffer()
  if epRpc.proofOfPayment.isSome():
    let proofOfPayment = epRpc.proofOfPayment.get()
    pb.write3(1, proofOfPayment)
  else:
    # there is no proof
    discard
  pb

proc decode*(T: type EligibilityProof, buffer: seq[byte]): ProtobufResult[T] =
  let pb = initProtoBuffer(buffer)
  var epRpc = EligibilityProof()
  var proofOfPayment = newSeq[byte]()
  if not ?pb.getField(1, proofOfPayment):
    epRpc.proofOfPayment = none(seq[byte])
  else:
    epRpc.proofOfPayment = some(proofOfPayment)
  ok(epRpc)

# Codec for EligibilityStatus

proc encode*(esRpc: EligibilityStatus): ProtoBuffer =
  var pb = initProtoBuffer()
  pb.write3(1, esRpc.statusCode)
  if esRpc.statusDesc.isSome():
    pb.write3(2, esRpc.statusDesc.get())
  pb

proc decode*(T: type EligibilityStatus, buffer: seq[byte]): ProtobufResult[T] =
  let pb = initProtoBuffer(buffer)
  var esRpc = EligibilityStatus()
  # status code
  var code = uint32(0)
  if not ?pb.getField(1, code):
    # status code is mandatory
    return err(ProtobufError.missingRequiredField("status_code"))
  else:
    esRpc.statusCode = code
  # status description
  var description = ""
  if not ?pb.getField(2, description):
    esRpc.statusDesc = none(string)
  else:
    esRpc.statusDesc = some(description)
  ok(esRpc)