mirror of https://github.com/status-im/go-waku.git
88 lines
2.1 KiB
Go
88 lines
2.1 KiB
Go
package rpc
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/libp2p/go-libp2p/core/protocol"
|
|
ma "github.com/multiformats/go-multiaddr"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/waku-org/go-waku/waku/v2/node"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/filter"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/legacy_filter"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/lightpush"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/relay"
|
|
"github.com/waku-org/go-waku/waku/v2/protocol/store"
|
|
)
|
|
|
|
type AdminService struct {
|
|
node *node.WakuNode
|
|
log *zap.Logger
|
|
}
|
|
|
|
type GetPeersArgs struct {
|
|
}
|
|
|
|
type PeersArgs struct {
|
|
Peers []string `json:"peers,omitempty"`
|
|
}
|
|
|
|
type PeerReply struct {
|
|
Multiaddr string `json:"multiaddr,omitempty"`
|
|
Protocol protocol.ID `json:"protocol,omitempty"`
|
|
Connected bool `json:"connected,omitempty"`
|
|
}
|
|
|
|
type PeersReply []PeerReply
|
|
|
|
func (a *AdminService) PostV1Peers(req *http.Request, args *PeersArgs, reply *SuccessReply) error {
|
|
for _, peer := range args.Peers {
|
|
addr, err := ma.NewMultiaddr(peer)
|
|
if err != nil {
|
|
a.log.Error("building multiaddr", zap.Error(err))
|
|
return err
|
|
}
|
|
|
|
err = a.node.DialPeerWithMultiAddress(req.Context(), addr)
|
|
if err != nil {
|
|
a.log.Error("dialing peers", zap.Error(err))
|
|
return err
|
|
}
|
|
}
|
|
|
|
*reply = true
|
|
return nil
|
|
}
|
|
|
|
func isWakuProtocol(protocol protocol.ID) bool {
|
|
return protocol == legacy_filter.FilterID_v20beta1 ||
|
|
protocol == filter.FilterPushID_v20beta1 ||
|
|
protocol == filter.FilterSubscribeID_v20beta1 ||
|
|
protocol == relay.WakuRelayID_v200 ||
|
|
protocol == lightpush.LightPushID_v20beta1 ||
|
|
protocol == store.StoreID_v20beta4
|
|
}
|
|
|
|
func (a *AdminService) GetV1Peers(req *http.Request, args *GetPeersArgs, reply *PeersReply) error {
|
|
peers, err := a.node.Peers()
|
|
if err != nil {
|
|
a.log.Error("getting peers", zap.Error(err))
|
|
return nil
|
|
}
|
|
for _, peer := range peers {
|
|
for _, addr := range peer.Addrs {
|
|
for _, proto := range peer.Protocols {
|
|
if !isWakuProtocol(proto) {
|
|
continue
|
|
}
|
|
*reply = append(*reply, PeerReply{
|
|
Multiaddr: addr.String(),
|
|
Protocol: proto,
|
|
Connected: peer.Connected,
|
|
})
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|