status-im/status-go

package mailserver

import (
    "errors"
    "time"
)

const (
    maxMessagesRequestPayloadLimit = 1000
)

// MessagesRequestPayload is a payload sent to the Mail Server.
type MessagesRequestPayload struct {
    // Lower is a lower bound of time range for which messages are requested.
    Lower uint32
    // Upper is a lower bound of time range for which messages are requested.
    Upper uint32
    // Bloom is a bloom filter to filter envelopes.
    Bloom []byte
    // Topics is a list of topics to filter envelopes.
    Topics [][]byte
    // Limit is the max number of envelopes to return.
    Limit uint32
    // Cursor is used for pagination of the results.
    Cursor []byte
    // Batch set to true indicates that the client supports batched response.
    Batch bool
}

func (r *MessagesRequestPayload) SetDefaults() {
    if r.Limit == 0 {
        r.Limit = maxQueryLimit
    }

    if r.Upper == 0 {
        r.Upper = uint32(time.Now().Unix() + whisperTTLSafeThreshold)
    }
}

func (r MessagesRequestPayload) Validate() error {
    if r.Upper < r.Lower {
        return errors.New("query range is invalid: lower > upper")
    }
    if len(r.Bloom) == 0 && len(r.Topics) == 0 {
        return errors.New("bloom filter and topics is empty")
    }
    if r.Limit > maxMessagesRequestPayloadLimit {
        return errors.New("limit exceeds the maximum allowed value")
    }
    return nil
}