[优化] 工具整理

This commit is contained in:
ryan
2026-03-14 15:03:05 +08:00
parent f1476610f6
commit 605a70b428
26 changed files with 316 additions and 240 deletions
+2 -1
View File
@@ -2,6 +2,7 @@ package model
import (
"atsflare/common"
"atsflare/utils/security"
"github.com/glebarez/sqlite"
"gorm.io/driver/mysql"
"gorm.io/gorm"
@@ -26,7 +27,7 @@ func createRootAccountIfNeed() error {
//if user.Status != common.UserStatusEnabled {
if err := DB.First(&user).Error; err != nil {
slog.Info("no user exists, create a root user", "username", "root")
hashedPassword, err := common.Password2Hash("123456")
hashedPassword, err := security.Password2Hash("123456")
if err != nil {
return err
}
+5 -4
View File
@@ -2,6 +2,7 @@ package model
import (
"atsflare/common"
"atsflare/utils/security"
"errors"
"strings"
)
@@ -63,7 +64,7 @@ func DeleteUserById(id int) (err error) {
func (user *User) Insert() error {
var err error
if user.Password != "" {
user.Password, err = common.Password2Hash(user.Password)
user.Password, err = security.Password2Hash(user.Password)
if err != nil {
return err
}
@@ -75,7 +76,7 @@ func (user *User) Insert() error {
func (user *User) Update(updatePassword bool) error {
var err error
if updatePassword {
user.Password, err = common.Password2Hash(user.Password)
user.Password, err = security.Password2Hash(user.Password)
if err != nil {
return err
}
@@ -102,7 +103,7 @@ func (user *User) ValidateAndFill() (err error) {
return errors.New("用户名或密码为空")
}
DB.Where(User{Username: user.Username}).First(user)
okay := common.ValidatePasswordAndHash(password, user.Password)
okay := security.ValidatePasswordAndHash(password, user.Password)
if !okay || user.Status != common.UserStatusEnabled {
return errors.New("用户名或密码错误,或用户已被封禁")
}
@@ -181,7 +182,7 @@ func ResetUserPasswordByEmail(email string, password string) error {
if email == "" || password == "" {
return errors.New("邮箱地址或密码为空!")
}
hashedPassword, err := common.Password2Hash(password)
hashedPassword, err := security.Password2Hash(password)
if err != nil {
return err
}