xinyin025/one-api
1
0
Fork 0
mirror of https://github.com/songquanpeng/one-api.git synced 2025-11-16 21:23:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

⚡️ improve: Prohibit users from deleting accounts (#132)

Browse Source
This commit is contained in:
MartialBE
2024-03-30 19:49:42 +08:00
parent a3503ba3d9
commit 89a84e2386
4 changed files with 39 additions and 67 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
model/common.go
View File

@@ -140,3 +140,13 @@ func assembleSumSelectStr(selectStr string) string {
return sumSelectStr
}
func RecordExists(table interface{}, fieldName string, fieldValue interface{}, excludeID interface{}) bool {
var count int64
query := DB.Model(table).Where(fmt.Sprintf("%s = ?", fieldName), fieldValue)
if excludeID != nil {
query = query.Not("id", excludeID)
}
query.Count(&count)
return count > 0
}

46
model/user.go
View File

@@ -5,6 +5,7 @@ import (
"fmt"
"one-api/common"
"strings"
"time"
"gorm.io/gorm"
)
@@ -12,25 +13,26 @@ import (
// User if you add sensitive fields, don't forget to clean them in setupLogin function.
// Otherwise, the sensitive information will be saved on local storage in plain text!
type User struct {
Id int `json:"id"`
Username string `json:"username" gorm:"unique;index" validate:"max=12"`
Password string `json:"password" gorm:"not null;" validate:"min=8,max=20"`
DisplayName string `json:"display_name" gorm:"index" validate:"max=20"`
Role int `json:"role" gorm:"type:int;default:1"` // admin, common
Status int `json:"status" gorm:"type:int;default:1"` // enabled, disabled
Email string `json:"email" gorm:"index" validate:"max=50"`
GitHubId string `json:"github_id" gorm:"column:github_id;index"`
WeChatId string `json:"wechat_id" gorm:"column:wechat_id;index"`
TelegramId int64 `json:"telegram_id" gorm:"bigint,column:telegram_id;default:0;"`
VerificationCode string `json:"verification_code" gorm:"-:all"` // this field is only for Email verification, don't save it to database!
AccessToken string `json:"access_token" gorm:"type:char(32);column:access_token;uniqueIndex"` // this token is for system management
Quota int `json:"quota" gorm:"type:int;default:0"`
UsedQuota int `json:"used_quota" gorm:"type:int;default:0;column:used_quota"` // used quota
RequestCount int `json:"request_count" gorm:"type:int;default:0;"` // request number
Group string `json:"group" gorm:"type:varchar(32);default:'default'"`
AffCode string `json:"aff_code" gorm:"type:varchar(32);column:aff_code;uniqueIndex"`
InviterId int `json:"inviter_id" gorm:"type:int;column:inviter_id;index"`
CreatedTime int64 `json:"created_time" gorm:"bigint"`
Id int `json:"id"`
Username string `json:"username" gorm:"unique;index" validate:"max=12"`
Password string `json:"password" gorm:"not null;" validate:"min=8,max=20"`
DisplayName string `json:"display_name" gorm:"index" validate:"max=20"`
Role int `json:"role" gorm:"type:int;default:1"` // admin, common
Status int `json:"status" gorm:"type:int;default:1"` // enabled, disabled
Email string `json:"email" gorm:"index" validate:"max=50"`
GitHubId string `json:"github_id" gorm:"column:github_id;index"`
WeChatId string `json:"wechat_id" gorm:"column:wechat_id;index"`
TelegramId int64 `json:"telegram_id" gorm:"bigint,column:telegram_id;default:0;"`
VerificationCode string `json:"verification_code" gorm:"-:all"` // this field is only for Email verification, don't save it to database!
AccessToken string `json:"access_token" gorm:"type:char(32);column:access_token;uniqueIndex"` // this token is for system management
Quota int `json:"quota" gorm:"type:int;default:0"`
UsedQuota int `json:"used_quota" gorm:"type:int;default:0;column:used_quota"` // used quota
RequestCount int `json:"request_count" gorm:"type:int;default:0;"` // request number
Group string `json:"group" gorm:"type:varchar(32);default:'default'"`
AffCode string `json:"aff_code" gorm:"type:varchar(32);column:aff_code;uniqueIndex"`
InviterId int `json:"inviter_id" gorm:"type:int;column:inviter_id;index"`
CreatedTime int64 `json:"created_time" gorm:"bigint"`
DeletedAt *time.Time `gorm:"index"`
}
type UserUpdates func(*User)
@@ -102,6 +104,9 @@ func DeleteUserById(id int) (err error) {
}
func (user *User) Insert(inviterId int) error {
if RecordExists(&User{}, "username", user.Username, nil) {
return errors.New("用户名已存在！")
}
var err error
if user.Password != "" {
user.Password, err = common.Password2Hash(user.Password)
@@ -134,6 +139,9 @@ func (user *User) Insert(inviterId int) error {
}
func (user *User) Update(updatePassword bool) error {
if RecordExists(&User{}, "username", user.Username, user.Id) {
return errors.New("用户名已存在！")
}
var err error
if updatePassword {
user.Password, err = common.Password2Hash(user.Password)