cmd: Add goal node subcommand to generate peer private key (algorand#6078)

Author: algorandskiy

cmd/goal/node.go

@@ -79,6 +79,7 @@ func init() {
 	nodeCmd.AddCommand(catchupCmd)
 	// Once the server-side implementation of the shutdown command is ready, we should enable this one.
 	//nodeCmd.AddCommand(shutdownCmd)
+	nodeCmd.AddCommand(p2pID)
 
 	startCmd.Flags().StringVarP(&peerDial, "peer", "p", "", "Peer address to dial for initial connection")
 	startCmd.Flags().StringVarP(&listenIP, "listen", "l", "", "Endpoint / REST address to listen on")

cmd/goal/p2pid.go

@@ -0,0 +1,72 @@
+// Copyright (C) 2019-2024 Algorand, Inc.
+// This file is part of go-algorand
+//
+// go-algorand is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as
+// published by the Free Software Foundation, either version 3 of the
+// License, or (at your option) any later version.
+//
+// go-algorand is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with go-algorand.  If not, see <https://www.gnu.org/licenses/>.
+
+// generate a new p2p private key and print out peerID to stdout
+
+package main
+
+import (
+	"fmt"
+	"os"
+	"path"
+
+	"github.com/algorand/go-algorand/cmd/util/datadir"
+	"github.com/algorand/go-algorand/config"
+	"github.com/algorand/go-algorand/network/p2p"
+	"github.com/algorand/go-algorand/util"
+	"github.com/libp2p/go-libp2p/core/peer"
+	"github.com/spf13/cobra"
+)
+
+var p2pID = &cobra.Command{
+	Use:   "generate-p2pid",
+	Short: "Generate a new p2p private key",
+	Long:  "Generate a new p2p private key (saved to " + p2p.DefaultPrivKeyPath + ") and print out peerID to stdout",
+	Args:  validateNoPosArgsFn,
+	Run: func(cmd *cobra.Command, args []string) {
+		anyError := false
+		datadir.OnDataDirs(func(dataDir string) {
+			exist := false
+			privKeyPath := path.Join(dataDir, p2p.DefaultPrivKeyPath)
+			if util.FileExists(privKeyPath) {
+				exist = true
+			}
+
+			peerKey, err := p2p.GetPrivKey(config.Local{P2PPersistPeerID: true}, dataDir)
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "Error obtaining private key: %v\n", err)
+				anyError = true
+				return
+			}
+			peerID, err := peer.IDFromPublicKey(peerKey.GetPublic())
+			if err != nil {
+				fmt.Fprintf(os.Stderr, "Error obtaining peerID from a key: %v\n", err)
+				anyError = true
+				return
+			}
+
+			fmt.Printf("PeerID: %s\n", peerID.String())
+			if !exist {
+				fmt.Printf("Private key saved to %s\n", privKeyPath)
+			} else {
+				fmt.Printf("Used existing key %s\n", privKeyPath)
+			}
+		})
+		if anyError {
+			os.Exit(1)
+		}
+	},
+}