mirror of
https://github.com/chaitin/SafeLine.git
synced 2026-02-04 07:43:27 +08:00
30 lines
675 B
Go
30 lines
675 B
Go
package model
|
|
|
|
import (
|
|
"gorm.io/gorm/clause"
|
|
|
|
"chaitin.cn/patronus/safeline-2/management/webserver/pkg/constants"
|
|
"chaitin.cn/patronus/safeline-2/management/webserver/pkg/database"
|
|
)
|
|
|
|
type User struct {
|
|
Base
|
|
Username string `gorm:"uniqueIndex;not null"`
|
|
Password string
|
|
Comment string
|
|
|
|
TFAEnabled bool `gorm:"column:tfa_enabled;default:true"`
|
|
TFASecret string `gorm:"column:tfa_secret"`
|
|
LastLoginTime int64 `gorm:"default:0"`
|
|
IsEnabled bool `gorm:"default:true"`
|
|
}
|
|
|
|
func initAdminUser() error {
|
|
db := database.GetDB()
|
|
user := User{
|
|
Username: constants.SuperUser,
|
|
}
|
|
db.Clauses(clause.OnConflict{DoNothing: true}).Create(&user)
|
|
return nil
|
|
}
|