status-im/status-go

package pairing

import (
    "runtime"
    "sync"
    "time"

    "go.uber.org/zap"

    "github.com/status-im/status-go/common"
    "github.com/status-im/status-go/logutils"
    "github.com/status-im/status-go/server"
    "github.com/status-im/status-go/server/pairing/peers"
    "github.com/status-im/status-go/signal"
)

type PeerNotifier struct {
    logger     *zap.Logger
    stop       chan struct{}
    terminator sync.Once
}

func NewPeerNotifier() *PeerNotifier {
    logger := logutils.ZapLogger().Named("PeerNotifier")
    stop := make(chan struct{})

    return &PeerNotifier{
        logger: logger,
        stop:   stop,
    }
}

func (p *PeerNotifier) terminateIn(d time.Duration) {
    p.terminator.Do(func() {
        time.Sleep(d)
        p.stop <- struct{}{}
    })
}

func (p *PeerNotifier) handler(hello *peers.LocalPairingPeerHello) {
    signal.SendLocalPairingEvent(Event{Type: EventPeerDiscovered, Action: ActionPeerDiscovery, Data: hello})
    p.logger.Debug("received peers.LocalPairingPeerHello message", zap.Any("hello message", hello))
    p.terminateIn(5 * time.Second)
}

func (p *PeerNotifier) Search() error {
    // TODO until we can resolve Android errors when calling net.Interfaces() just noop. Sorry Android
    if common.OperatingSystemIs(common.AndroidPlatform) {
        return nil
    }

    dn, err := server.GetDeviceName()
    if err != nil {
        return err
    }

    return peers.Search(dn, runtime.GOOS, p.handler, p.stop, p.logger)
}