From 9b23a43a76da26e455c65562049b38bc4b640187 Mon Sep 17 00:00:00 2001
From: Pavel Zbitskiy <pavel@algorand.com>
Date: Tue, 23 Jul 2024 14:53:17 -0400
Subject: [PATCH] cmd: Add a simple cli util to generate peer private key and
 print out peer ID

---
 cmd/p2pid/main.go | 65 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 65 insertions(+)
 create mode 100644 cmd/p2pid/main.go

diff --git a/cmd/p2pid/main.go b/cmd/p2pid/main.go
new file mode 100644
index 0000000000..f0f83b373c
--- /dev/null
+++ b/cmd/p2pid/main.go
@@ -0,0 +1,65 @@
+// 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 (
+	"flag"
+	"fmt"
+	"os"
+	"path"
+
+	"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"
+)
+
+var dataDirectory = flag.String("d", "", "Root Algorand daemon data path or a path to a directory where the private key will be stored. If not provided, the current directory will be used.")
+
+func main() {
+	flag.Parse()
+	dataDir := *dataDirectory
+	if dataDir == "" {
+		dataDir = "."
+	}
+
+	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)
+		os.Exit(1)
+	}
+	peerID, err := peer.IDFromPublicKey(peerKey.GetPublic())
+	if err != nil {
+		fmt.Fprintf(os.Stderr, "Error obtaining peerID from a key: %v\n", err)
+		os.Exit(1)
+	}
+
+	fmt.Printf("PeerID: %s\n", peerID.String())
+	if !exist {
+		fmt.Printf("Private key saved to %s\n", privKeyPath)
+	} else {
+		fmt.Printf("Used existing key from path %s\n", privKeyPath)
+	}
+}