From ca3b05b398c156dca819b312ae8340baec12c12f Mon Sep 17 00:00:00 2001
From: vimiix <i@vimiix.com>
Date: Wed, 10 Jul 2024 18:09:24 +0800
Subject: [PATCH] fix: make sure password is set before validate repo

---
 ssx/ssx.go | 43 ++++++++++++++++++++++++++++++++-----------
 1 file changed, 32 insertions(+), 11 deletions(-)

diff --git a/ssx/ssx.go b/ssx/ssx.go
index cb22ea4..cb98aff 100644
--- a/ssx/ssx.go
+++ b/ssx/ssx.go
@@ -74,6 +74,9 @@ func NewSSX(opt *CmdOption) (*SSX, error) {
 		return nil, err
 	}
 
+	if err := ssx.MakeSurePasswordIsSet(); err != nil {
+		return nil, err
+	}
 	validate, err := ssx.ValidateRepo()
 	if err != nil {
 		return nil, err
@@ -95,12 +98,16 @@ var (
 	Password = []byte("password")
 )
 
-func (s *SSX) ChallengeDBPassword() error {
+func (s *SSX) MakeSurePasswordIsSet() error {
 	v, err := s.repo.GetMetadata(Password)
 	if err != nil {
 		return err
 	}
-
+	if len(v) > 0 {
+		return nil
+	}
+	lg.Warn("db password not set, please supplement it")
+	// not set password, fill it
 	prompt := promptui.Prompt{
 		Label: "Input db password",
 		Mask:  '*',
@@ -111,22 +118,36 @@ func (s *SSX) ChallengeDBPassword() error {
 			return nil
 		},
 	}
-	if len(v) == 0 {
-		lg.Warn("db password not set, please supplement it")
-		// not set password, fill it
-		password, err := prompt.Run()
-		if err != nil {
-			return err
-		}
-		encrypt := utils.HashWithSHA256(password)
-		return s.repo.SetMetadata(Password, []byte(encrypt))
+	password, err := prompt.Run()
+	if err != nil {
+		return err
 	}
+	encrypt := utils.HashWithSHA256(password)
+	return s.repo.SetMetadata(Password, []byte(encrypt))
+}
 
+func (s *SSX) ChallengeDBPassword() error {
+	v, err := s.repo.GetMetadata(Password)
+	if err != nil {
+		return err
+	}
+
+	prompt := promptui.Prompt{
+		Label: "Input db password",
+		Mask:  '*',
+		Validate: func(s string) error {
+			if len(s) == 0 {
+				return errors.New("password can't be empty")
+			}
+			return nil
+		},
+	}
 	password, err := prompt.Run()
 	if err != nil {
 		return err
 	}
 	encrypt := utils.HashWithSHA256(password)
+	// v is not empty, we have ensure it in MakeSurePasswordIsSet
 	if encrypt == string(v) {
 		return s.updateDeviceID()
 	}