Implement KeepUpdated. Switch to Logrus logger. Relocate score-related functions.
This commit is contained in:
@@ -2,8 +2,8 @@ package controllers
|
||||
|
||||
import (
|
||||
"InfantrySkillCalculator/models"
|
||||
"InfantrySkillCalculator/utils"
|
||||
"github.com/gin-gonic/gin"
|
||||
"log"
|
||||
"session"
|
||||
)
|
||||
|
||||
@@ -11,7 +11,7 @@ func CreateUser(username string, hashedPassword string, enabled bool, usedCode s
|
||||
user := models.User{Username: username, Password: hashedPassword, Enabled: enabled}
|
||||
err := models.DB.Create(&user).Error
|
||||
if err != nil {
|
||||
log.Fatalf("Error while creating user: %v", err)
|
||||
utils.Logger.Fatalf("Error while creating user: %v", err)
|
||||
}
|
||||
|
||||
var code models.ActivationCode
|
||||
@@ -20,19 +20,19 @@ func CreateUser(username string, hashedPassword string, enabled bool, usedCode s
|
||||
Where("code = ?", usedCode).
|
||||
First(&code).Error
|
||||
if err != nil {
|
||||
log.Fatalf("Error while getting activation code: %v", err)
|
||||
utils.Logger.Fatalf("Error while getting activation code: %v", err)
|
||||
}
|
||||
|
||||
code.UsedForUsername = username
|
||||
err = models.DB.Save(&code).Error
|
||||
if err != nil {
|
||||
log.Fatalf("Error while updating activation code: %v", err)
|
||||
utils.Logger.Fatalf("Error while updating activation code: %v", err)
|
||||
}
|
||||
|
||||
user.UserRole = code.UserRole
|
||||
err = models.DB.Save(&user).Error
|
||||
if err != nil {
|
||||
log.Fatalf("Error while updating user role: %v", err)
|
||||
utils.Logger.Fatalf("Error while updating user role: %v", err)
|
||||
}
|
||||
|
||||
var bf2042 models.Game
|
||||
@@ -40,7 +40,7 @@ func CreateUser(username string, hashedPassword string, enabled bool, usedCode s
|
||||
Where("tag = ?", "BF2042").
|
||||
First(&bf2042).Error
|
||||
if err != nil {
|
||||
log.Fatalf("Error while getting game: %v", err)
|
||||
utils.Logger.Fatalf("Error while getting game: %v", err)
|
||||
}
|
||||
userSettings := models.UserSettings{
|
||||
Username: username,
|
||||
@@ -62,7 +62,7 @@ func GetUserRole(username string) models.Role {
|
||||
var user models.User
|
||||
err := models.DB.Where("username = ?", username).First(&user).Error
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
utils.Logger.Fatal(err)
|
||||
}
|
||||
return user.UserRole
|
||||
}
|
||||
@@ -79,7 +79,7 @@ func IsUserAdmin(username string) bool {
|
||||
var user models.User
|
||||
err := models.DB.Where("username = ?", username).First(&user).Error
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
utils.Logger.Fatal(err)
|
||||
return false
|
||||
}
|
||||
return user.UserRole == models.AdminRole
|
||||
|
||||
Reference in New Issue
Block a user