feat: dht ping (#50)

lib/dht.go

@@ -107,6 +107,25 @@ func DhtGetClosestPeers(ctx context.Context, key []byte, proto protocol.ID, ma m
 	return ais, nil
 }
 
+func DhtPing(ctx context.Context, proto protocol.ID, ma multiaddr.Multiaddr) error {
+	ai, err := peer.AddrInfoFromP2pAddr(ma)
+	if err != nil {
+		return err
+	}
+
+	m, err := DhtProtocolMessenger(ctx, proto, ai)
+	if err != nil {
+		return err
+	}
+
+	err = m.Ping(ctx, ai.ID)
+	if err != nil {
+		return err
+	}
+
+	return nil
+}
+
 // dhtMsgSender handles sending dht wire protocol messages to a given peer
 type dhtMsgSender struct {
 	h         host.Host

main.go

@@ -234,6 +234,40 @@ func main() {
 							},
 						},
 					},
+					{
+						Name:        "ping",
+						ArgsUsage:   "<multiaddr>",
+						Usage:       "ping a DHT node",
+						Description: "sends a DHT ping to the target",
+						Action: func(c *cli.Context) error {
+							if c.NArg() != 1 {
+								return fmt.Errorf("invalid number of arguments")
+							}
+							maStr := c.Args().Get(0)
+							protoID := c.String("protocolID")
+
+							ma, err := multiaddr.NewMultiaddr(maStr)
+							if err != nil {
+								return err
+							}
+
+							err = vole.DhtPing(c.Context, protocol.ID(protoID), ma)
+							if err != nil {
+								return err
+							}
+
+							fmt.Println("Ok")
+							return nil
+						},
+						Flags: []cli.Flag{
+							&cli.StringFlag{
+								Name:        "protocolID",
+								Usage:       "the protocol ID",
+								DefaultText: "/ipfs/kad/1.0.0",
+								Value:       "/ipfs/kad/1.0.0",
+							},
+						},
+					},
 				},
 			},
 			{