91 lines
2.4 KiB
Go
91 lines
2.4 KiB
Go
package controllers
|
|
|
|
import (
|
|
"InfantrySkillCalculator/models"
|
|
"InfantrySkillCalculator/utils"
|
|
"github.com/gin-gonic/gin"
|
|
"session"
|
|
)
|
|
|
|
func CreateUser(username string, hashedPassword string, enabled bool, usedCode string) {
|
|
user := models.User{Username: username, Password: hashedPassword, Enabled: enabled}
|
|
var code models.ActivationCode
|
|
var bf2042 models.Game
|
|
|
|
if err := models.DB.Create(&user).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while creating user: %s", err.Error())
|
|
}
|
|
|
|
if err := models.DB.Where("code = ?", usedCode).First(&code).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while getting activation code: %s", err.Error())
|
|
}
|
|
|
|
code.UsedForUsername = username
|
|
if err := models.DB.Save(&code).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while updating activation code: %s", err.Error())
|
|
}
|
|
|
|
user.UserRole = code.UserRole
|
|
if err := models.DB.Save(&user).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while updating user role: %s", err.Error())
|
|
}
|
|
|
|
if err := models.DB.Where("tag = ?", "BF2042").First(&bf2042).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while getting game: %v", err)
|
|
}
|
|
|
|
userSettings := models.UserSettings{
|
|
Username: username,
|
|
ActiveGameID: bf2042.ID,
|
|
SquadColors: true,
|
|
CalcMedian: false,
|
|
UseCache: true,
|
|
}
|
|
|
|
if err := models.DB.Create(&userSettings).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while creating user settings: %s", err.Error())
|
|
}
|
|
|
|
utils.Logger.Infof("[USER] Created user: %s", username)
|
|
}
|
|
|
|
func IsUserEnabled(username string) bool {
|
|
var user models.User
|
|
|
|
if err := models.DB.Where("username = ?", username).First(&user).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while getting user: %s", err.Error())
|
|
}
|
|
|
|
return user.Enabled
|
|
}
|
|
|
|
func GetUserRole(username string) models.Role {
|
|
var user models.User
|
|
|
|
if err := models.DB.Where("username = ?", username).First(&user).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while getting user: %s", err.Error())
|
|
}
|
|
|
|
return user.UserRole
|
|
}
|
|
|
|
func GetUserRoleByCtx(c *gin.Context) models.Role {
|
|
username, ok := session.GetUsername(c)
|
|
|
|
if !ok {
|
|
return models.ReaderRole
|
|
}
|
|
|
|
return GetUserRole(username)
|
|
}
|
|
|
|
func IsUserAdmin(username string) bool {
|
|
var user models.User
|
|
|
|
if err := models.DB.Where("username = ?", username).First(&user).Error; err != nil {
|
|
utils.Logger.Fatalf("[USER] Error while getting user: %s", err.Error())
|
|
}
|
|
|
|
return user.UserRole == models.AdminRole
|
|
}
|