package protocol

import (
	wpb "github.com/waku-org/go-waku/waku/v2/protocol/pb"
	"github.com/waku-org/go-waku/waku/v2/protocol/store/pb"
	"github.com/waku-org/go-waku/waku/v2/utils"
)

// 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 := utils.SHA256(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
}