status-im/status-go

package gethbridge

import (
    "errors"

    "go.uber.org/zap"

    "github.com/status-im/status-go/waku"
    "github.com/status-im/status-go/wakuv2"

    "github.com/ethereum/go-ethereum/node"
    "github.com/ethereum/go-ethereum/p2p/enode"

    gethens "github.com/status-im/status-go/eth-node/bridge/geth/ens"
    "github.com/status-im/status-go/eth-node/types"
    enstypes "github.com/status-im/status-go/eth-node/types/ens"
)

type gethNodeWrapper struct {
    stack *node.Node
    waku1 *waku.Waku
    waku2 *wakuv2.Waku
}

func NewNodeBridge(stack *node.Node, waku1 *waku.Waku, waku2 *wakuv2.Waku) types.Node {
    return &gethNodeWrapper{stack: stack, waku1: waku1, waku2: waku2}
}

func (w *gethNodeWrapper) Poll() {
    // noop
}

func (w *gethNodeWrapper) NewENSVerifier(logger *zap.Logger) enstypes.ENSVerifier {
    return gethens.NewVerifier(logger)
}

func (w *gethNodeWrapper) SetWaku1(waku *waku.Waku) {
    w.waku1 = waku
}

func (w *gethNodeWrapper) SetWaku2(waku *wakuv2.Waku) {
    w.waku2 = waku
}

func (w *gethNodeWrapper) GetWaku(ctx interface{}) (types.Waku, error) {
    if w.waku1 == nil {
        return nil, errors.New("waku service is not available")
    }

    return NewGethWakuWrapper(w.waku1), nil
}

func (w *gethNodeWrapper) GetWakuV2(ctx interface{}) (types.Waku, error) {
    if w.waku2 == nil {
        return nil, errors.New("waku service is not available")
    }

    return NewGethWakuV2Wrapper(w.waku2), nil
}

func (w *gethNodeWrapper) AddPeer(url string) error {
    parsedNode, err := enode.ParseV4(url)
    if err != nil {
        return err
    }

    w.stack.Server().AddPeer(parsedNode)

    return nil
}

func (w *gethNodeWrapper) RemovePeer(url string) error {
    parsedNode, err := enode.ParseV4(url)
    if err != nil {
        return err
    }

    w.stack.Server().RemovePeer(parsedNode)

    return nil
}

func (w *gethNodeWrapper) PeersCount() int {
    if w.stack.Server() == nil {
        return 0
    }
    return len(w.stack.Server().Peers())
}