cmd/get-peer-list: allow filtering out lists

Signed-off-by: Ciro S. Costa <[email protected]>

cmd/monero/commands/daemon/get_peer_list.go

@@ -15,6 +15,9 @@ import (
 
 type getPeerListCommand struct {
 	JSON bool
+
+	White bool
+	Gray  bool
 }
 
 func (c *getPeerListCommand) Cmd() *cobra.Command {
@@ -27,6 +30,11 @@ func (c *getPeerListCommand) Cmd() *cobra.Command {
 	cmd.Flags().BoolVar(&c.JSON, "json",
 		false, "whether or not to output the result as json")
 
+	cmd.Flags().BoolVar(&c.White, "white",
+		true, "whether or not to show the white list")
+	cmd.Flags().BoolVar(&c.Gray, "gray",
+		false, "whether or not show gray list")
+
 	return cmd
 }
 
@@ -39,6 +47,10 @@ func (c *getPeerListCommand) RunE(_ *cobra.Command, _ []string) error {
 		return fmt.Errorf("client: %w", err)
 	}
 
+	if !c.White && !c.Gray {
+		return fmt.Errorf("either white or gray (or both) must be set")
+	}
+
 	resp, err := client.GetPeerList(ctx)
 	if err != nil {
 		return fmt.Errorf("get block count: %w", err)
@@ -58,15 +70,27 @@ func (c *getPeerListCommand) pretty(v *daemon.GetPeerListResult) {
 
 	table.AddRow("TYPE", "HOST", "PORT ", "RPC", "SINCE")
 
-	for _, peer := range v.GrayList {
-		table.AddRow("GRAY", peer.Host, peer.Port, peer.RPCPort, "")
-	}
-
 	sort.Slice(v.WhiteList, func(i, j int) bool {
 		return v.WhiteList[i].LastSeen < v.WhiteList[j].LastSeen
 	})
-	for _, peer := range v.WhiteList {
-		table.AddRow("WHITE", peer.Host, peer.Port, peer.RPCPort, humanize.Time(time.Unix(peer.LastSeen, 0)))
+
+	if c.Gray {
+		for _, peer := range v.GrayList {
+			table.AddRow("gray",
+				peer.Host, peer.Port,
+				peer.RPCPort, "")
+		}
+	}
+
+	if c.White {
+		for _, peer := range v.WhiteList {
+			table.AddRow("white",
+				peer.Host, peer.Port,
+				peer.RPCPort,
+				humanize.Time(time.Unix(peer.LastSeen, 0)),
+			)
+		}
+
 	}
 
 	fmt.Println(table)