mirror of https://github.com/waku-org/nwaku.git
89 lines
2.2 KiB
Nim
89 lines
2.2 KiB
Nim
when (NimMajor, NimMinor) < (1, 4):
|
|
{.push raises: [Defect].}
|
|
else:
|
|
{.push raises: [].}
|
|
|
|
import std/options, results, stew/byteutils, stew/arrayops, nimcrypto/sha2
|
|
import ../waku_core, ../common/paging
|
|
|
|
## Waku message digest
|
|
|
|
type MessageDigest* = MDigest[256]
|
|
|
|
proc fromBytes*(T: type MessageDigest, src: seq[byte]): T =
|
|
var data: array[32, byte]
|
|
|
|
let byteCount = copyFrom[byte](data, src)
|
|
|
|
assert byteCount == 32
|
|
|
|
return MessageDigest(data: data)
|
|
|
|
proc computeDigest*(msg: WakuMessage): MessageDigest =
|
|
var ctx: sha256
|
|
ctx.init()
|
|
defer:
|
|
ctx.clear()
|
|
|
|
ctx.update(msg.contentTopic.toBytes())
|
|
ctx.update(msg.payload)
|
|
|
|
# Computes the hash
|
|
return ctx.finish()
|
|
|
|
## Public API types
|
|
|
|
type
|
|
#TODO Once Store v2 is removed, the cursor becomes the hash of the last message
|
|
ArchiveCursor* = object
|
|
digest*: MessageDigest
|
|
storeTime*: Timestamp
|
|
senderTime*: Timestamp
|
|
pubsubTopic*: PubsubTopic
|
|
hash*: WakuMessageHash
|
|
|
|
ArchiveQuery* = object
|
|
includeData*: bool # indicate if messages should be returned in addition to hashes.
|
|
pubsubTopic*: Option[PubsubTopic]
|
|
contentTopics*: seq[ContentTopic]
|
|
cursor*: Option[ArchiveCursor]
|
|
startTime*: Option[Timestamp]
|
|
endTime*: Option[Timestamp]
|
|
hashes*: seq[WakuMessageHash]
|
|
pageSize*: uint
|
|
direction*: PagingDirection
|
|
requestId*: string
|
|
|
|
ArchiveResponse* = object
|
|
hashes*: seq[WakuMessageHash]
|
|
messages*: seq[WakuMessage]
|
|
topics*: seq[PubsubTopic]
|
|
cursor*: Option[ArchiveCursor]
|
|
|
|
ArchiveErrorKind* {.pure.} = enum
|
|
UNKNOWN = uint32(0)
|
|
DRIVER_ERROR = uint32(1)
|
|
INVALID_QUERY = uint32(2)
|
|
|
|
ArchiveError* = object
|
|
case kind*: ArchiveErrorKind
|
|
of DRIVER_ERROR, INVALID_QUERY:
|
|
# TODO: Add an enum to be able to distinguish between error causes
|
|
cause*: string
|
|
else:
|
|
discard
|
|
|
|
ArchiveResult* = Result[ArchiveResponse, ArchiveError]
|
|
|
|
proc `$`*(err: ArchiveError): string =
|
|
case err.kind
|
|
of ArchiveErrorKind.DRIVER_ERROR:
|
|
"DIRVER_ERROR: " & err.cause
|
|
of ArchiveErrorKind.INVALID_QUERY:
|
|
"INVALID_QUERY: " & err.cause
|
|
of ArchiveErrorKind.UNKNOWN:
|
|
"UNKNOWN"
|
|
|
|
proc invalidQuery*(T: type ArchiveError, cause: string): T =
|
|
ArchiveError(kind: ArchiveErrorKind.INVALID_QUERY, cause: cause)
|