package net import ( "git.lumeweb.com/LumeWeb/libs5-go/encoding" "go.uber.org/zap" "net/url" ) //go:generate mockgen -source=peer.go -destination=../mocks/net/peer.go -package=net var ( _ Peer = (*BasePeer)(nil) ) // EventCallback type for the callback function type EventCallback func(event []byte) error // CloseCallback type for the OnClose callback type CloseCallback func() // ErrorCallback type for the onError callback type ErrorCallback func(args ...interface{}) // ListenerOptions struct for options type ListenerOptions struct { OnClose *CloseCallback OnError *ErrorCallback Logger *zap.Logger } type Peer interface { SendMessage(message []byte) error RenderLocationURI() string ListenForMessages(callback EventCallback, options ListenerOptions) End() error SetId(id *encoding.NodeId) Id() *encoding.NodeId SetChallenge(challenge []byte) Challenge() []byte SetSocket(socket interface{}) Socket() interface{} SetConnected(isConnected bool) IsConnected() bool SetConnectionURIs(uris []*url.URL) ConnectionURIs() []*url.URL } type BasePeer struct { connectionURIs []*url.URL isConnected bool challenge []byte socket interface{} id *encoding.NodeId } func (b *BasePeer) IsConnected() bool { return b.isConnected } func (b *BasePeer) SetConnected(isConnected bool) { b.isConnected = isConnected } func (b *BasePeer) SendMessage(message []byte) error { panic("must implement in child class") } func (b *BasePeer) RenderLocationURI() string { panic("must implement in child class") } func (b *BasePeer) ListenForMessages(callback EventCallback, options ListenerOptions) { panic("must implement in child class") } func (b *BasePeer) End() error { panic("must implement in child class") } func (b *BasePeer) Challenge() []byte { return b.challenge } func (b *BasePeer) SetChallenge(challenge []byte) { b.challenge = challenge } func (b *BasePeer) Socket() interface{} { return b.socket } func (b *BasePeer) SetSocket(socket interface{}) { b.socket = socket } func (b *BasePeer) Id() *encoding.NodeId { return b.id } func (b *BasePeer) SetId(id *encoding.NodeId) { b.id = id } func (b *BasePeer) SetConnectionURIs(uris []*url.URL) { b.connectionURIs = uris } func (b *BasePeer) ConnectionURIs() []*url.URL { return b.connectionURIs }