package gethbridge

import (
	"github.com/status-im/status-go/eth-node/types"
	"github.com/status-im/status-go/waku"
	"github.com/status-im/status-go/whisper/v6"
)

type whisperEnvelope struct {
	env *whisper.Envelope
}

// NewWhisperEnvelope returns an object that wraps Geth's Whisper Envelope in a types interface.
func NewWhisperEnvelope(e *whisper.Envelope) types.Envelope {
	return &whisperEnvelope{env: e}
}

func UnwrapWhisperEnvelope(e types.Envelope) (*whisper.Envelope, bool) {
	if env, ok := e.(*whisperEnvelope); ok {
		return env.env, true
	}
	return nil, false
}

func MustUnwrapWhisperEnvelope(e types.Envelope) *whisper.Envelope {
	return e.(*whisperEnvelope).env
}

func (w *whisperEnvelope) Hash() types.Hash {
	return types.Hash(w.env.Hash())
}

func (w *whisperEnvelope) Bloom() []byte {
	return w.env.Bloom()
}

func (w *whisperEnvelope) PoW() float64 {
	return w.env.PoW()
}

func (w *whisperEnvelope) Expiry() uint32 {
	return w.env.Expiry
}

func (w *whisperEnvelope) TTL() uint32 {
	return w.env.TTL
}

func (w *whisperEnvelope) Topic() types.TopicType {
	return types.TopicType(w.env.Topic)
}

func (w *whisperEnvelope) Size() int {
	return len(w.env.Data)
}

type wakuEnvelope struct {
	env *waku.Envelope
}

// NewWakuEnvelope returns an object that wraps Geth's Waku Envelope in a types interface.
func NewWakuEnvelope(e *waku.Envelope) types.Envelope {
	return &wakuEnvelope{env: e}
}

func UnwrapWakuEnvelope(e types.Envelope) (*waku.Envelope, bool) {
	if env, ok := e.(*wakuEnvelope); ok {
		return env.env, true
	}
	return nil, false
}

func MustUnwrapWakuEnvelope(e types.Envelope) *waku.Envelope {
	return e.(*wakuEnvelope).env
}

func (w *wakuEnvelope) Hash() types.Hash {
	return types.Hash(w.env.Hash())
}

func (w *wakuEnvelope) Bloom() []byte {
	return w.env.Bloom()
}

func (w *wakuEnvelope) PoW() float64 {
	return w.env.PoW()
}

func (w *wakuEnvelope) Expiry() uint32 {
	return w.env.Expiry
}

func (w *wakuEnvelope) TTL() uint32 {
	return w.env.TTL
}

func (w *wakuEnvelope) Topic() types.TopicType {
	return types.TopicType(w.env.Topic)
}

func (w *wakuEnvelope) Size() int {
	return len(w.env.Data)
}