Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

feat(cmd/rpc): add commands for the p2p module #2599

33 changes: 9 additions & 24 deletions cmd/celestia/p2p.go
Original file line number Diff line number Diff line change
Expand Up @@ -54,11 +54,7 @@ var infoCmd = &cobra.Command{
info, err := client.P2P.Info(cmd.Context())

formatter := func(data interface{}) interface{} {
peerAdd, ok := data.(peer.AddrInfo)
if !ok {
return peerAdd
}

peerAdd := data.(peer.AddrInfo)
ma := make([]string, len(info.Addrs))
for i := range peerAdd.Addrs {
ma[i] = peerAdd.Addrs[i].String()
Expand Down Expand Up @@ -92,10 +88,7 @@ var peersCmd = &cobra.Command{
}

formatter := func(data interface{}) interface{} {
conPeers, ok := data.([]string)
if !ok {
return conPeers
}
conPeers := data.([]string)
return struct {
Peers []string `json:"peers"`
}{
Expand Down Expand Up @@ -124,11 +117,7 @@ var peerInfoCmd = &cobra.Command{
}
info, err := client.P2P.PeerInfo(cmd.Context(), pid)
formatter := func(data interface{}) interface{} {
peerAdd, ok := data.(peer.AddrInfo)
if !ok {
return peerAdd
}

peerAdd := data.(peer.AddrInfo)
ma := make([]string, len(info.Addrs))
for i := range peerAdd.Addrs {
ma[i] = peerAdd.Addrs[i].String()
Expand Down Expand Up @@ -294,7 +283,7 @@ var blockPeerCmd = &cobra.Command{
}
}

printOutput(nil, err, formatter)
printOutput(err, nil, formatter)
return nil
},
}
Expand Down Expand Up @@ -334,7 +323,7 @@ var unblockPeerCmd = &cobra.Command{
}
}

printOutput(nil, err, formatter)
printOutput(err, nil, formatter)
return nil
},
}
Expand Down Expand Up @@ -404,7 +393,7 @@ var protectCmd = &cobra.Command{
}
}

printOutput(nil, err, formatter)
printOutput(err, nil, formatter)
return nil
},
}
Expand All @@ -425,7 +414,7 @@ var unprotectCmd = &cobra.Command{
return err
}

result, err := client.P2P.Unprotect(cmd.Context(), pid, args[1])
_, err = client.P2P.Unprotect(cmd.Context(), pid, args[1])

formatter := func(data interface{}) interface{} {
err, ok := data.(error)
Expand All @@ -444,7 +433,7 @@ var unprotectCmd = &cobra.Command{
}
}

printOutput(result, err, formatter)
printOutput(err, nil, formatter)
return nil
},
}
Expand Down Expand Up @@ -575,7 +564,6 @@ var pubsubPeersCmd = &cobra.Command{
return err
}

//TODO: check
result, err := client.P2P.PubSubPeers(cmd.Context(), args[0])
peers := make([]string, len(result))

Expand All @@ -584,10 +572,7 @@ var pubsubPeersCmd = &cobra.Command{
}

formatter := func(data interface{}) interface{} {
conPeers, ok := data.([]string)
if !ok {
return data
}
conPeers := data.([]string)
return struct {
Peers []string `json:"peers"`
}{
Expand Down
8 changes: 4 additions & 4 deletions cmd/celestia/rpc.go
Original file line number Diff line number Diff line change
Expand Up @@ -404,10 +404,10 @@ func rpcClient(ctx context.Context) (*client.Client, error) {
func printOutput(data interface{}, err error, formatData func(interface{}) interface{}) {
if err != nil {
data = err
}

if formatData != nil {
data = formatData(data)
} else {
vgonkivs marked this conversation as resolved.
Show resolved Hide resolved
if formatData != nil {
data = formatData(data)
}
}

resp := struct {
Expand Down