mirror of https://github.com/status-im/go-waku.git
62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package protocol
|
|
|
|
import (
|
|
"crypto/sha256"
|
|
|
|
wpb "github.com/waku-org/go-waku/waku/v2/protocol/pb"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/store/pb"
|
|
)
|
|
|
|
// Envelope contains information about the pubsub topic of a WakuMessage
|
|
// and a hash used to identify a message based on the bytes of a WakuMessage
|
|
// protobuffer
|
|
type Envelope struct {
|
|
msg *wpb.WakuMessage
|
|
size int
|
|
hash []byte
|
|
index *pb.Index
|
|
}
|
|
|
|
// NewEnvelope creates a new Envelope that contains a WakuMessage
|
|
// It's used as a way to know to which Pubsub topic belongs a WakuMessage
|
|
// as well as generating a hash based on the bytes that compose the message
|
|
func NewEnvelope(msg *wpb.WakuMessage, receiverTime int64, pubSubTopic string) *Envelope {
|
|
messageHash, dataLen, _ := msg.Hash()
|
|
hash := sha256.Sum256(append([]byte(msg.ContentTopic), msg.Payload...))
|
|
return &Envelope{
|
|
msg: msg,
|
|
size: dataLen,
|
|
hash: messageHash,
|
|
index: &pb.Index{
|
|
Digest: hash[:],
|
|
ReceiverTime: receiverTime,
|
|
SenderTime: msg.Timestamp,
|
|
PubsubTopic: pubSubTopic,
|
|
},
|
|
}
|
|
}
|
|
|
|
// Message returns the WakuMessage associated to an Envelope
|
|
func (e *Envelope) Message() *wpb.WakuMessage {
|
|
return e.msg
|
|
}
|
|
|
|
// PubsubTopic returns the topic on which a WakuMessage was received
|
|
func (e *Envelope) PubsubTopic() string {
|
|
return e.index.PubsubTopic
|
|
}
|
|
|
|
// Hash returns a 32 byte hash calculated from the WakuMessage bytes
|
|
func (e *Envelope) Hash() []byte {
|
|
return e.hash
|
|
}
|
|
|
|
// Size returns the byte size of the WakuMessage
|
|
func (e *Envelope) Size() int {
|
|
return e.size
|
|
}
|
|
|
|
func (env *Envelope) Index() *pb.Index {
|
|
return env.index
|
|
}
|