fix: bug fixes

cmd/add.go

@@ -18,6 +18,7 @@ package cmd
 
 import (
 	"bufio"
+	"bytes"
 	"fmt"
 	"io/ioutil"
 	"log"
@@ -44,7 +45,6 @@ func addRun(cmd *cobra.Command, args []string) {
 		gpm.CreateDatabase()
 	}
 	var account gpm.Account
-	var password string
 	var accounts []gpm.Account
 
 	data, err := ioutil.ReadFile(gpm.DatabaseFile)
@@ -54,21 +54,28 @@ func addRun(cmd *cobra.Command, args []string) {
 	}
 	if len(data) == 0 {
 		color.Red("Warning: If you forget your master password your data will be lost !!")
+		color.Yellow("Master password can contains characters and symbols.")
 		fmt.Println()
 	}
 	fmt.Print("password: ")
 	bytePassword, _ := terminal.ReadPassword(int(syscall.Stdin))
 	fmt.Println()
 
-	password = string(bytePassword)
-	if password == "" {
+	if string(bytePassword) == "" {
 		color.Red("Error: you haven't entered password")
 		if len(data) == 0 {
 			color.Red("Master password can't be empty")
 		}
 		os.Exit(0)
 	}
-	if len(data) == 0 && len(password) < 6 {
+
+	if len(data) == 0 && bytes.ContainsAny(bytePassword, "0123456789") {
+		color.Red("Error: master key can't have numbers !!")
+		color.Yellow("Tips: Use passphrases instead")
+		os.Exit(0)
+	}
+
+	if len(data) == 0 && len(string(bytePassword)) < 6 {
 		color.Red("Master password must be greater than 5")
 		os.Exit(0)
 	}

cmd/change.go

@@ -17,6 +17,7 @@ limitations under the License.
 package cmd
 
 import (
+	"bytes"
 	"fmt"
 	"io/ioutil"
 	"log"
@@ -69,12 +70,30 @@ func changeRun(cmd *cobra.Command, args []string) {
 			color.Red("Error: Password already in use !!")
 			os.Exit(0)
 		}
+
+		if string(byteNewPassword) == "" {
+			color.Red("Error: you haven't entered password")
+			os.Exit(0)
+		}
+
+		if bytes.ContainsAny(byteNewPassword, "0123456789") {
+			color.Red("Error: master key can't have numbers !!")
+			color.Yellow("Tips: Use passphrases instead")
+			os.Exit(0)
+		}
+
+		if len(string(byteNewPassword)) < 6 {
+			color.Red("Master password must be greater than 5")
+			os.Exit(0)
+		}
+
 		fmt.Print("Verify password: ")
 		byteVerifyPassword, _ := terminal.ReadPassword(int(syscall.Stdin))
 		fmt.Println()
 
 		if string(byteNewPassword) != string(byteVerifyPassword) {
 			color.Red("Error: both password is not same!")
+			os.Exit(0)
 		}
 
 		prompt := promptui.Select{

gpm/gpm.go

@@ -55,7 +55,7 @@ func GetHomeDir() string {
 
 // GetVersion returns version name, and code
 func GetVersion() string {
-	var version = "0.3.0"
+	var version = "0.3.1"
 	return version
 }