#98 Adding confirmation before removing folder

Author: ggarri

cmd/lightchain/init.go

@@ -99,14 +99,19 @@ func newNodeCfgFromCmd(cmd *cobra.Command) (node.Config, network.Network, error)
 			return node.Config{}, "", fmt.Errorf("unable to initialize lightchain node. %s", err.Error())
 		}
 
-		if forceInit {
-			logger.Info(fmt.Sprintf("Forcing '%s' data dir removal...", dataDir))
-			if err := fs.RemoveAll(dataDir); err != nil {
-				return node.Config{}, "", fmt.Errorf("unable to remove data dir '%s'. %s", dataDir, err)
-			}
-		} else {
+		if !forceInit {
 			return node.Config{}, "", fmt.Errorf("unable to initialize lightchain node. %s already exists", dataDir)
 		}
+
+		isConfirmed := fs.AskForConfirmation(fmt.Sprintf("Are you sure to erase %s ?", dataDir))
+		if !isConfirmed {
+			return node.Config{}, "", fmt.Errorf("unable to initialize lightchain node. %s already exists", dataDir)
+		}
+
+		logger.Info(fmt.Sprintf("Removing '%s' folder ...", dataDir))
+		if err := fs.RemoveAll(dataDir); err != nil {
+			return node.Config{}, "", fmt.Errorf("unable to remove data dir '%s'. %s", dataDir, err)
+		}
 	}
 
 

fs/io.go

@@ -0,0 +1,30 @@
+package fs
+
+import (
+	"bufio"
+	"fmt"
+	"log"
+	"os"
+	"strings"
+)
+
+func AskForConfirmation(s string) bool {
+	reader := bufio.NewReader(os.Stdin)
+
+	for {
+		fmt.Printf("%s [y(yes)/n(no)]: ", s)
+
+		response, err := reader.ReadString('\n')
+		if err != nil {
+			log.Fatal(err)
+		}
+
+		response = strings.ToLower(strings.TrimSpace(response))
+
+		if response == "y" || response == "yes" {
+			return true
+		} else if response == "n" || response == "no" {
+			return false
+		}
+	}
+}