提交 v1.3.0 beta

server/service/cli.go

@@ -0,0 +1,100 @@
+package service
+
+import (
+	"context"
+	"crypto/md5"
+	"fmt"
+	"next-terminal/server/common/nt"
+
+	"next-terminal/server/env"
+	"next-terminal/server/model"
+	"next-terminal/server/repository"
+	"next-terminal/server/utils"
+
+	"gorm.io/gorm"
+)
+
+type Cli struct {
+}
+
+func NewCli() *Cli {
+	return &Cli{}
+}
+
+func (cli Cli) ResetPassword(username string) error {
+	user, err := repository.UserRepository.FindByUsername(context.TODO(), username)
+	if err != nil {
+		return err
+	}
+	password := "next-terminal"
+	passwd, err := utils.Encoder.Encode([]byte(password))
+	if err != nil {
+		return err
+	}
+	u := &model.User{
+		Password: string(passwd),
+		ID:       user.ID,
+	}
+	if err := repository.UserRepository.Update(context.TODO(), u); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (cli Cli) ResetTotp(username string) error {
+	user, err := repository.UserRepository.FindByUsername(context.TODO(), username)
+	if err != nil {
+		return err
+	}
+	u := &model.User{
+		TOTPSecret: "-",
+		ID:         user.ID,
+	}
+	if err := repository.UserRepository.Update(context.TODO(), u); err != nil {
+		return err
+	}
+	return nil
+}
+
+func (cli Cli) ChangeEncryptionKey(oldEncryptionKey, newEncryptionKey string) error {
+
+	oldPassword := []byte(fmt.Sprintf("%x", md5.Sum([]byte(oldEncryptionKey))))
+	newPassword := []byte(fmt.Sprintf("%x", md5.Sum([]byte(newEncryptionKey))))
+
+	return env.GetDB().Transaction(func(tx *gorm.DB) error {
+		c := context.WithValue(context.TODO(), nt.DB, tx)
+		credentials, err := repository.CredentialRepository.FindAll(c)
+		if err != nil {
+			return err
+		}
+		for i := range credentials {
+			credential := credentials[i]
+			if err := CredentialService.Decrypt(&credential, oldPassword); err != nil {
+				return err
+			}
+			if err := CredentialService.Encrypt(&credential, newPassword); err != nil {
+				return err
+			}
+			if err := repository.CredentialRepository.UpdateById(c, &credential, credential.ID); err != nil {
+				return err
+			}
+		}
+		assets, err := repository.AssetRepository.FindAll(c)
+		if err != nil {
+			return err
+		}
+		for i := range assets {
+			asset := assets[i]
+			if err := AssetService.Decrypt(&asset, oldPassword); err != nil {
+				return err
+			}
+			if err := AssetService.Encrypt(&asset, newPassword); err != nil {
+				return err
+			}
+			if err := repository.AssetRepository.UpdateById(c, &asset, asset.ID); err != nil {
+				return err
+			}
+		}
+		return nil
+	})
+}