feat: show peer address on http root (closes #2)

Author: helder-moreira

cmd/cosmoseed/config.go

@@ -14,8 +14,8 @@ const (
 )
 
 var (
-	home, chainID, seeds, logLevel, nodeKeyFile string
-	showVersion, showNodeID, configReadOnly     bool
+	home, chainID, seeds, logLevel, nodeKeyFile, externalAddress string
+	showVersion, showNodeID, configReadOnly                      bool
 )
 
 func init() {
@@ -47,6 +47,11 @@ func init() {
 		utils.GetString("NODE_KEY_FILE", ""),
 		"override node key file on config.yaml",
 	)
+	flag.StringVar(&externalAddress,
+		"external-address",
+		utils.GetString("EXTERNAL_ADDRESS", ""),
+		"external address to use in format '<host>:<port>'",
+	)
 
 	flag.BoolVar(&showVersion, "version", false, "print version and exit")
 	flag.BoolVar(&showNodeID, "show-node-id", false, "print node ID and exit")

cmd/cosmoseed/main.go

@@ -53,6 +53,10 @@ func main() {
 		cfg.NodeKeyFile = nodeKeyFile
 	}
 
+	if externalAddress != "" {
+		cfg.ExternalAddress = externalAddress
+	}
+
 	seeder, err := cosmoseed2.NewSeeder(home, cfg)
 	if err != nil {
 		panic(err)

pkg/cosmoseed/config.go

@@ -23,8 +23,9 @@ type Config struct {
 	PeerQueueSize           int  `yaml:"peerQueueSize" default:"1000"`
 	DialWorkers             int  `yaml:"dialWorkers" default:"20"`
 
-	ChainID string `yaml:"chainID"`
-	Seeds   string `yaml:"seeds"`
+	ChainID         string `yaml:"chainID"`
+	Seeds           string `yaml:"seeds"`
+	ExternalAddress string `yaml:"externalAddress,omitempty"`
 
 	ApiAddr string `yaml:"apiAddr" default:"0.0.0.0:8080"`
 }

pkg/cosmoseed/http.go

@@ -6,9 +6,14 @@ import (
 )
 
 func (s *Seeder) registerRoutes(mux *http.ServeMux) {
+	mux.HandleFunc("/", s.seedAddress)
 	mux.HandleFunc("/peers", s.handlePeers)
 }
 
+func (s *Seeder) seedAddress(w http.ResponseWriter, r *http.Request) {
+	w.Write([]byte(s.GetFullAddress()))
+}
+
 func (s *Seeder) handlePeers(w http.ResponseWriter, r *http.Request) {
 	peers := s.pex.GetPeerSelection()
 

pkg/cosmoseed/node.go

@@ -8,6 +8,8 @@ import (
 	"os"
 	"os/signal"
 	"path"
+	"strconv"
+	"strings"
 	"syscall"
 	"time"
 
@@ -180,6 +182,48 @@ func (s *Seeder) GetNodeID() string {
 	return s.key.ID()
 }
 
+func (s *Seeder) GetP2pAddress() string {
+	if s.cfg.ExternalAddress != "" {
+		if parts := strings.Split(s.cfg.ExternalAddress, ":"); len(parts) == 2 {
+			return parts[0]
+		}
+	}
+
+	localIp, err := getLocalIP()
+	if err == nil {
+		return localIp
+	}
+
+	// If everything above fails just return a default
+	return "0.0.0.0"
+}
+
+func (s *Seeder) GetP2pPort() int {
+	if s.cfg.ExternalAddress != "" {
+		if parts := strings.Split(s.cfg.ExternalAddress, ":"); len(parts) == 2 {
+			port, err := strconv.Atoi(parts[1])
+			if err == nil {
+				return port
+			}
+		}
+	}
+
+	parts := strings.Split(s.cfg.ListenAddr, ":")
+	if len(parts) > 1 {
+		port, err := strconv.Atoi(parts[len(parts)-1])
+		if err == nil {
+			return port
+		}
+	}
+
+	// If everything above fails just return default port
+	return 26656
+}
+
+func (s *Seeder) GetFullAddress() string {
+	return fmt.Sprintf("%s@%s:%d", s.GetNodeID(), s.GetP2pAddress(), s.GetP2pPort())
+}
+
 func generateP2PConfig(home string, cfg *Config) *config.P2PConfig {
 	p2pConfig := config.DefaultP2PConfig()
 

pkg/cosmoseed/utils.go

@@ -2,6 +2,7 @@ package cosmoseed
 
 import (
 	"fmt"
+	"net"
 	"os"
 	"path/filepath"
 	"strings"
@@ -30,3 +31,43 @@ func splitAndTrimEmpty(s, sep, cutset string) []string {
 	}
 	return nonEmptyStrings
 }
+
+func getLocalIP() (string, error) {
+	ifaces, err := net.Interfaces()
+	if err != nil {
+		return "", err
+	}
+	for _, iface := range ifaces {
+		// Skip interfaces that are down or loopback
+		if iface.Flags&net.FlagUp == 0 || iface.Flags&net.FlagLoopback != 0 {
+			continue
+		}
+
+		addrs, err := iface.Addrs()
+		if err != nil {
+			continue
+		}
+
+		for _, addr := range addrs {
+			var ip net.IP
+
+			switch v := addr.(type) {
+			case *net.IPNet:
+				ip = v.IP
+			case *net.IPAddr:
+				ip = v.IP
+			}
+
+			if ip == nil || ip.IsLoopback() {
+				continue
+			}
+
+			// Return the first IPv4 address
+			ip = ip.To4()
+			if ip != nil {
+				return ip.String(), nil
+			}
+		}
+	}
+	return "", fmt.Errorf("no connected network interface found")
+}