(fix) #102: allow to specify peers.json in mpcium-cli

Author: anhthii

cmd/mpcium-cli/main.go

@@ -51,10 +51,10 @@ func main() {
 				Action: registerPeers,
 				Flags: []cli.Flag{
 					&cli.StringFlag{
-						Name:    "input",
-						Aliases: []string{"i"},
-						Usage:   "Input peers JSON file path (default: peers.json)",
-						Value:   peersFileName,
+						Name:     "peers",
+						Aliases:  []string{"p"},
+						Usage:    "Path to peers.json file",
+						Required: true,
 					},
 					&cli.StringFlag{
 						Name:    "environment",

cmd/mpcium-cli/register-peers.go

@@ -15,7 +15,7 @@ import (
 )
 
 func registerPeers(ctx context.Context, c *cli.Command) error {
-	inputPath := c.String("input")
+	inputPath := c.String("peers")
 	environment := c.String("environment")
 
 	// Hardcoded prefix for MPC peers in Consul
@@ -58,10 +58,26 @@ func registerPeers(ctx context.Context, c *cli.Command) error {
 	// Register peers in Consul
 	for nodeName, nodeID := range peerMap {
 		key := prefix + nodeName
+
+		// Check if the key already exists
+		existing, _, err := kv.Get(key, nil)
+		if err != nil {
+			return fmt.Errorf("failed to check existing key %s: %w", key, err)
+		}
+
+		if existing != nil {
+			existingID := string(existing.Value)
+			if existingID != nodeID {
+				return fmt.Errorf("conflict detected: peer %s already exists with ID %s, but trying to register with different ID %s", nodeName, existingID, nodeID)
+			}
+			fmt.Printf("Peer %s already registered with same ID %s, skipping\n", nodeName, nodeID)
+			continue
+		}
+
 		p := &api.KVPair{Key: key, Value: []byte(nodeID)}
 
 		// Store the key-value pair
-		_, err := kv.Put(p, nil)
+		_, err = kv.Put(p, nil)
 		if err != nil {
 			return fmt.Errorf("failed to store key %s: %w", key, err)
 		}