optimize: safe.String and Encryption logic

Signed-off-by: 孙林耀 <sunlinyao@vip.qq.com>

safe/encrypt.go

@@ -168,6 +168,13 @@ func Encrypt(originalBytes []byte, key string, o *EncryptOptions) (string, error
 const ciphertextPrefix = "{CRYPT}"
 
 func Decrypt(cipherString, key string) ([]byte, error) {
+	switch len(key) {
+	case 8, 16, 24, 32:
+	default:
+		hash := sha256.New()
+		hash.Write([]byte(key))
+		key = string(hash.Sum(nil))
+	}
 	if len(cipherString) <= len(ciphertextPrefix)+5 {
 		return nil, errors.New("invalid ciphertext")
 	}

safe/string.go

@@ -106,7 +106,6 @@ func (e String) UnsafeString() (string, error) {
 		return "", nil
 	}
 	if strings.HasPrefix(e.value, ciphertextPrefix) {
-		fmt.Println(e.getSecret(), e.value)
 		if secret := e.getSecret(); secret != "" {
 			decrypt, err := Decrypt(e.value, secret)
 			return string(decrypt), err
@@ -134,7 +133,7 @@ func (e *String) SetValue(value string) (err error) {
 	return nil
 }
 
-func (e *String) SetSecret(secret string) {
+func (e *String) UpdateSecret(secret string) {
 	plain, err := e.UnsafeString()
 	if err != nil {
 		return
@@ -147,6 +146,10 @@ func (e *String) SetSecret(secret string) {
 	e.secret = secret
 }
 
+func (e *String) SetSecret(secret string) {
+	e.secret = secret
+}
+
 // Scan implements the Scanner interface.
 func (e *String) Scan(value any) error {
 	switch vt := value.(type) {