pbconv.go

package util

import (
	"errors"

	"github.com/John-LittleBearLabs/go-libp2p/core/peer"
	pbv1 "github.com/John-LittleBearLabs/go-libp2p/p2p/protocol/circuitv1/pb"
	pbv2 "github.com/John-LittleBearLabs/go-libp2p/p2p/protocol/circuitv2/pb"

	ma "github.com/multiformats/go-multiaddr"
)

func PeerToPeerInfoV1(p *pbv1.CircuitRelay_Peer) (peer.AddrInfo, error) {
	if p == nil {
		return peer.AddrInfo{}, errors.New("nil peer")
	}

	id, err := peer.IDFromBytes(p.Id)
	if err != nil {
		return peer.AddrInfo{}, err
	}

	var addrs []ma.Multiaddr
	if len(p.Addrs) > 0 {
		addrs = make([]ma.Multiaddr, 0, len(p.Addrs))
	}

	for _, addrBytes := range p.Addrs {
		a, err := ma.NewMultiaddrBytes(addrBytes)
		if err == nil {
			addrs = append(addrs, a)
		}
	}

	return peer.AddrInfo{ID: id, Addrs: addrs}, nil
}

func PeerInfoToPeerV1(pi peer.AddrInfo) *pbv1.CircuitRelay_Peer {
	var addrs [][]byte
	if len(pi.Addrs) > 0 {
		addrs = make([][]byte, 0, len(pi.Addrs))
	}

	for _, addr := range pi.Addrs {
		addrs = append(addrs, addr.Bytes())
	}

	p := new(pbv1.CircuitRelay_Peer)
	p.Id = []byte(pi.ID)
	p.Addrs = addrs

	return p
}

func PeerToPeerInfoV2(p *pbv2.Peer) (peer.AddrInfo, error) {
	if p == nil {
		return peer.AddrInfo{}, errors.New("nil peer")
	}

	id, err := peer.IDFromBytes(p.Id)
	if err != nil {
		return peer.AddrInfo{}, err
	}

	var addrs []ma.Multiaddr
	if len(p.Addrs) > 0 {
		addrs = make([]ma.Multiaddr, 0, len(p.Addrs))
	}

	for _, addrBytes := range p.Addrs {
		a, err := ma.NewMultiaddrBytes(addrBytes)
		if err == nil {
			addrs = append(addrs, a)
		}
	}

	return peer.AddrInfo{ID: id, Addrs: addrs}, nil
}

func PeerInfoToPeerV2(pi peer.AddrInfo) *pbv2.Peer {
	var addrs [][]byte

	if len(pi.Addrs) > 0 {
		addrs = make([][]byte, 0, len(pi.Addrs))
	}

	for _, addr := range pi.Addrs {
		addrs = append(addrs, addr.Bytes())
	}

	p := new(pbv2.Peer)
	p.Id = []byte(pi.ID)
	p.Addrs = addrs

	return p
}