netbird/client/internal/peer/notifier.go

package peer

import (
	"sync"
)

const (
	stateDisconnected = iota
	stateConnected
	stateConnecting
)

type notifier struct {
	serverStateLock    sync.Mutex
	listenersLock      sync.Mutex
	listeners          map[Listener]struct{}
	currentServerState bool
	currentClientState bool
	lastNotification   int
}

func newNotifier() *notifier {
	return &notifier{
		listeners: make(map[Listener]struct{}),
	}
}

func (n *notifier) addListener(listener Listener) {
	n.listenersLock.Lock()
	defer n.listenersLock.Unlock()

	n.serverStateLock.Lock()
	go n.notifyListener(listener, n.lastNotification)
	n.serverStateLock.Unlock()
	n.listeners[listener] = struct{}{}
}

func (n *notifier) removeListener(listener Listener) {
	n.listenersLock.Lock()
	defer n.listenersLock.Unlock()
	delete(n.listeners, listener)
}

func (n *notifier) updateServerStates(mgmState bool, signalState bool) {
	n.serverStateLock.Lock()
	defer n.serverStateLock.Unlock()

	var newState bool
	if mgmState && signalState {
		newState = true
	} else {
		newState = false
	}

	if !n.isServerStateChanged(newState) {
		return
	}

	n.currentServerState = newState
	n.lastNotification = n.calculateState(newState, n.currentClientState)

	go n.notifyAll(n.lastNotification)
}

func (n *notifier) clientStart() {
	n.serverStateLock.Lock()
	defer n.serverStateLock.Unlock()
	n.currentClientState = true
	n.lastNotification = n.calculateState(n.currentServerState, true)
	go n.notifyAll(n.lastNotification)
}

func (n *notifier) clientStop() {
	n.serverStateLock.Lock()
	defer n.serverStateLock.Unlock()
	n.currentClientState = false
	n.lastNotification = n.calculateState(n.currentServerState, false)
	go n.notifyAll(n.lastNotification)
}

func (n *notifier) isServerStateChanged(newState bool) bool {
	return n.currentServerState != newState
}

func (n *notifier) notifyAll(state int) {
	n.listenersLock.Lock()
	defer n.listenersLock.Unlock()

	for l := range n.listeners {
		n.notifyListener(l, state)
	}
}

func (n *notifier) notifyListener(l Listener, state int) {
	switch state {
	case stateDisconnected:
		l.OnDisconnected()
	case stateConnected:
		l.OnConnected()
	case stateConnecting:
		l.OnConnecting()
	}
}

func (n *notifier) calculateState(serverState bool, clientState bool) int {
	if serverState && clientState {
		return stateConnected
	}

	if !clientState {
		return stateDisconnected
	}

	return stateConnecting
}

func (n *notifier) peerListChanged(numOfPeers int) {
	n.listenersLock.Lock()
	defer n.listenersLock.Unlock()

	for l := range n.listeners {
		l.OnPeersListChanged(numOfPeers)
	}
}