Maxi/InfantrySkillCalculator
1
0
Fork 0
Code Issues 6 Pull Requests Actions Packages Projects 1 Releases Activity

Entire logging rework. Full error handling. Small improvements.

Browse Source
This commit is contained in:
MaxJa4
2024-01-22 17:24:17 +01:00
parent da1ff4e4e5
commit ca697da0da
18 changed files with 278 additions and 205 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
controllers/cache_controller.go
View File

@@ -24,12 +24,14 @@ func GetCacheByPlayerID(c *gin.Context) {
game, err := GetActiveGame(c)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "No active game available!"})
utils.Logger.Warnf("[CACHE] No active game available! Error: %s", err.Error())
return
}
val, err := models.PlayerCache.GetScore(playerId, game.Tag)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Warnf("[CACHE] Record not found! Error: %s", err.Error())
} else {
c.JSON(http.StatusOK, val)
}
@@ -38,20 +40,24 @@ func GetCacheByPlayerID(c *gin.Context) {
// AddCache POST /cache
func AddCache(c *gin.Context) {
var input AddCacheInput
var game models.Game
if err := c.BindJSON(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Warnf("[CACHE] Failed to bind JSON! Error: %s", err.Error())
return
}
var game models.Game
if err := FindGameByTag(&game, input.GameTag).Error; err != nil {
if err := models.DB.Where("tag = ?", input.GameTag).First(&game).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Game not found!"})
utils.Logger.Warnf("[CACHE] Game not found! Error: %s", err.Error())
return
}
err := models.PlayerCache.SetScore(input.PlayerID, input.GameTag, input.Score)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Cache update failed! Error: " + err.Error()})
utils.Logger.Warnf("[CACHE] Cache update failed! Error: %s", err.Error())
return
}
@@ -67,6 +73,7 @@ func UpdateCacheByPlayerID(c *gin.Context) {
err := models.PlayerCache.SetScore(playerID, gameTag, score)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Cache update failed! Error: " + err.Error()})
utils.Logger.Warnf("[CACHE] Cache update failed! Error: %s", err.Error())
} else {
c.JSON(http.StatusOK, nil)
}
@@ -79,6 +86,7 @@ func DeleteCacheByPlayerID(c *gin.Context) {
if err := models.PlayerCache.DeleteScore(playerID, gameTag); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Warnf("[CACHE] Cache deletion failed! Error: %s", err.Error())
} else {
c.JSON(http.StatusOK, nil)
}
@@ -88,13 +96,14 @@ func DeleteCacheByPlayerID(c *gin.Context) {
func DeleteAllCaches(c *gin.Context) {
if err := models.PlayerCache.PurgeCache(); err != nil {
c.String(http.StatusBadRequest, err.Error())
utils.Logger.Warnf("[CACHE] Cache purge failed! Error: %s", err.Error())
} else {
c.String(http.StatusOK, "Purged all caches!")
}
}
func UpdateCacheAfterExpiry(key string) {
utils.Logger.Println("[KeepUpdated] Cache expired: " + key)
utils.Logger.Infof("[KeepUpdated] Updating cache for key %s", key)
playerId, err := models.PlayerCache.GetPlayerIdFromCacheKey(key)
if err != nil {
@@ -107,19 +116,20 @@ func UpdateCacheAfterExpiry(key string) {
Where("id = ?", playerId).
First(&player).Error; err != nil {
utils.Logger.Println("Failed to find player: " + err.Error())
utils.Logger.Errorf("[KeepUpdated] Failed to find player with ID %d! Error: %s", playerId, err.Error())
return
}
if player.Clan.KeepUpdated {
score := CalcPlayerScore(player.Name, "BF2042")
if score == score && score != -1 { // not NaN
if err := models.PlayerCache.SetScore(player.ID, "BF2042", score); err != nil {
utils.Logger.Println("Failed to update cache: " + err.Error())
utils.Logger.Warnf("[KeepUpdated] Failed to update cache for player %s! Error: %s", player.Name, err.Error())
} else {
utils.Logger.Println("[KeepUpdated] Updated cache for player " + player.Name)
utils.Logger.Infof("[KeepUpdated] Updated cache for player %s", player.Name)
}
} else {
utils.Logger.Println("[KeepUpdated] Received NaN for player " + player.Name)
utils.Logger.Warnf("[KeepUpdated] Failed to calculate score for player %s", player.Name)
}
}
}

97
controllers/clan_controller.go
View File

@@ -25,7 +25,11 @@ type UpdateClanInput struct {
// GetAllClans GET /clan
func GetAllClans(c *gin.Context) {
var clans []models.Clan
models.DB.Find(&clans)
if err := models.DB.Find(&clans).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[CLAN] Error while getting all clans: %s", err.Error())
return
}
c.JSON(http.StatusOK, clans)
}
@@ -33,7 +37,11 @@ func GetAllClans(c *gin.Context) {
// GetAllClansHTML GET /clans_html
func GetAllClansHTML(c *gin.Context) {
var clans []models.Clan
models.DB.Find(&clans)
if err := models.DB.Find(&clans).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[CLAN] Error while getting all clans: %s", err.Error())
return
}
var htmlOptions string
htmlOptions = `<option disabled selected value>Auswählen...</option>`
@@ -48,34 +56,34 @@ func GetAllClansHTML(c *gin.Context) {
// AddClan POST /clan
func AddClan(c *gin.Context) {
var input AddClanInput
var clan models.Clan
if err := c.BindJSON(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
var clan models.Clan
if err := FindClanByName(&clan, input.Name).Error; err == nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Clan with this name already exists!"})
return
} else if err := FindClanByTag(&clan, input.Tag).Error; err == nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Clan with this tag already exists!"})
utils.Logger.Errorf("[CLAN] Error while binding JSON: %s", err.Error())
return
}
clan = models.Clan{Name: input.Name, Tag: input.Tag, KeepUpdated: input.KeepUpdated}
models.DB.Create(&clan)
if err := models.DB.Create(&clan).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[CLAN] Error while adding clan '[%s] %s': %s", input.Tag, input.Name, err.Error())
return
}
c.JSON(http.StatusOK, clan)
utils.Logger.Infof("Added clan '%s' with tag '%s'", clan.Name, clan.Tag)
utils.Logger.Infof("[CLAN] Added clan: [%s] %s", clan.Tag, clan.Name)
}
// GetClanByID GET /clan/:id
func GetClanByID(c *gin.Context) {
var clan models.Clan
if err := FindClanByID(&clan, c).Error; err != nil {
if err := models.DB.Where("id = ?", c.Param("id")).First(&clan).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[CLAN] Error while getting clan by ID: %s", err.Error())
return
}
@@ -85,41 +93,57 @@ func GetClanByID(c *gin.Context) {
// UpdateClanByID PATCH /clan/:id
func UpdateClanByID(c *gin.Context) {
var input UpdateClanInput
var clan models.Clan
if err := c.ShouldBindJSON(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[CLAN] Error while binding JSON: %s", err.Error())
return
}
res := models.DB.Model(&models.Clan{}).
Where("id = ?", c.Param("id")).
Updates(map[string]interface{}{
"Name": input.Name,
"Tag": input.Tag,
"KeepUpdated": input.KeepUpdated,
})
if res.Error != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": res.Error.Error()})
if err := models.DB.First(&clan, c.Param("id")).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[CLAN] Error while getting clan by ID: %s", err.Error())
return
}
oldValues := models.Clan{
Name: clan.Name,
Tag: clan.Tag,
KeepUpdated: clan.KeepUpdated,
}
clan.Name = input.Name
clan.Tag = input.Tag
clan.KeepUpdated = input.KeepUpdated
if err := models.DB.Save(&clan).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[CLAN] Error while updating clan '%s': %s", input.Name, err.Error())
return
}
c.JSON(http.StatusOK, nil)
utils.Logger.Infof("Updated clan '%s' with tag '%s'", input.Name, input.Tag)
utils.Logger.Infof("[CLAN] Updated clan: [%s] %s -> [%s] %s", oldValues.Tag, oldValues.Name, clan.Tag, clan.Name)
}
// DeleteClanByID DELETE /clan/:id
func DeleteClanByID(c *gin.Context) {
var clan models.Clan
if err := FindClanByID(&clan, c).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
if err := models.DB.Clauses(clause.Returning{}).
Where("id = ?", c.Param("id")).
Delete(&clan).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[CLAN] Error while deleting clan '%s': %s", clan.Name, err.Error())
return
}
models.DB.Delete(&clan)
c.JSON(http.StatusOK, true)
utils.Logger.Infof("Deleted clan '%s' with tag '%s'", clan.Name, clan.Tag)
utils.Logger.Infof("[CLAN] Deleted clan: [%s] %s", clan.Tag, clan.Name)
}
// DeleteAllClans DELETE /admin/clan
@@ -129,21 +153,12 @@ func DeleteAllClans(c *gin.Context) {
Session(&gorm.Session{AllowGlobalUpdate: true}).
Clauses(clause.Returning{}).
Delete(&clans).Error; err != nil {
c.String(http.StatusBadRequest, "Purge failed! Error: "+err.Error())
utils.Logger.Errorf("[CLAN] Error while purging all clans: %s", err.Error())
return
}
c.String(http.StatusOK, "Purged "+utils.UintToString(uint(len(clans)))+" clans!")
}
func FindClanByName(out interface{}, name string) *gorm.DB {
return models.DB.Where("name = ?", name).First(out)
}
func FindClanByTag(out interface{}, tag string) *gorm.DB {
return models.DB.Where("tag = ?", tag).First(out)
}
func FindClanByID(out interface{}, c *gin.Context) *gorm.DB {
return models.DB.Where("id = ?", c.Param("id")).First(out)
utils.Logger.Infof("[CLAN] Purged %d clans!", len(clans))
}

6
controllers/code_controller.go
View File

@@ -5,6 +5,7 @@ import (
"InfantrySkillCalculator/utils"
"github.com/gin-gonic/gin"
"net/http"
"strings"
)
// CreateCode POST /code
@@ -12,6 +13,7 @@ func CreateCode(c *gin.Context) {
userRole, ok := c.GetPostForm("user_role")
if !ok {
c.String(http.StatusBadRequest, "Missing user role")
utils.Logger.Error("Missing user role")
return
}
var role models.Role
@@ -24,14 +26,18 @@ func CreateCode(c *gin.Context) {
role = models.ReaderRole
default:
c.String(http.StatusInternalServerError, "Invalid user role: "+userRole)
utils.Logger.Error("Invalid user role: " + userRole)
return
}
newCode := utils.GenerateActivationCode()
newCodeObj := models.ActivationCode{Code: newCode, UserRole: role}
if err := models.DB.Create(&newCodeObj).Error; err != nil {
c.String(http.StatusInternalServerError, "Failed to create new code: "+err.Error())
utils.Logger.Error("Failed to create new code: " + err.Error())
return
}
c.String(http.StatusOK, "Activation code for role '"+string(role)+"': "+newCode)
utils.Logger.Info("Activation code for role '" + string(role) + "' created: " + newCode[:4] + strings.Repeat("*", 28))
}

42
controllers/game_controller.go
View File

@@ -2,9 +2,10 @@ package controllers
import (
"InfantrySkillCalculator/models"
"InfantrySkillCalculator/utils"
"fmt"
"github.com/gin-gonic/gin"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"net/http"
)
@@ -21,7 +22,11 @@ type UpdateGameInput struct {
// GetGames GET /game
func GetGames(c *gin.Context) {
var games []models.Game
models.DB.Find(&games)
if err := models.DB.Find(&games).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[GAME] Record not found: %v", err)
return
}
c.JSON(http.StatusOK, games)
}
@@ -29,7 +34,11 @@ func GetGames(c *gin.Context) {
// GetGamesHTML GET /game_html
func GetGamesHTML(c *gin.Context) {
var games []models.Game
models.DB.Find(&games)
if err := models.DB.Find(&games).Error; err != nil {
c.String(http.StatusBadRequest, "")
utils.Logger.Errorf("[GAME] Record not found: %v", err)
return
}
var htmlOptions string
htmlOptions = `<option disabled selected value>Auswählen...</option>`
@@ -46,6 +55,7 @@ func GetGameByID(c *gin.Context) {
var game models.Game
if err := models.DB.Where("id = ?", c.Param("id")).First(&game).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[GAME] Record not found: %v", err)
return
}
@@ -57,11 +67,16 @@ func AddGame(c *gin.Context) {
var input AddGameInput
if err := c.BindJSON(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[GAME] Record not found: %v", err)
return
}
game := models.Game{Name: input.Name, Tag: input.Tag}
models.DB.Create(&game)
if err := models.DB.Create(&game).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[GAME] Record not found: %v", err)
return
}
c.JSON(http.StatusOK, game)
}
@@ -71,12 +86,14 @@ func UpdateGameByID(c *gin.Context) {
var game models.Game
if err := models.DB.Where("id = ?", c.Param("id")).First(&game).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[GAME] Record not found: %v", err)
return
}
var input UpdateGameInput
if err := c.ShouldBindJSON(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[GAME] Could not parse input: %v", err)
return
}
@@ -88,20 +105,13 @@ func UpdateGameByID(c *gin.Context) {
// DeleteGameByID DELETE /game/:id
func DeleteGameByID(c *gin.Context) {
var game models.Game
if err := models.DB.Where("id = ?", c.Param("id")).First(&game).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
if err := models.DB.Clauses(clause.Returning{}).Delete(&game).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[GAME] Error while deleting game: %v", err)
return
}
models.DB.Delete(&game)
c.JSON(http.StatusOK, true)
}
func FindGame(out interface{}, c *gin.Context) *gorm.DB {
return models.DB.Where("id = ?", c.Param("id")).First(&out)
}
func FindGameByTag(out interface{}, tag string) *gorm.DB {
return models.DB.Where("tag = ?", tag).First(out)
utils.Logger.Infof("[GAME] Deleted game: %s", game.Name)
}

6
controllers/metrics_controller.go
View File

@@ -13,7 +13,7 @@ var gameMetrics models.GameMetrics
func LoadMetrics() {
f, err := os.Open("./config/metrics.json")
if err != nil {
utils.Logger.Fatal("Failed to open metrics.json: ", err)
utils.Logger.Fatal("[METRICS] Failed to open metrics.json: ", err)
}
defer func(f *os.File) {
_ = f.Close()
@@ -21,12 +21,12 @@ func LoadMetrics() {
data, err := io.ReadAll(f)
if err != nil {
utils.Logger.Fatal("Failed to read metrics.json: ", err)
utils.Logger.Fatal("[METRICS] Failed to read metrics.json: ", err)
}
var metrics models.GameMetrics
if err := json.Unmarshal(data, &metrics); err != nil {
utils.Logger.Fatal("Failed to deserialize metrics.json: ", err)
utils.Logger.Fatal("[METRICS] Failed to deserialize metrics.json: ", err)
}
gameMetrics = metrics

108
controllers/player_controller.go
View File

@@ -24,7 +24,11 @@ type UpdatePlayerInput struct {
// GetAllPlayers GET /player
func GetAllPlayers(c *gin.Context) {
var players []models.Player
models.DB.Find(&players)
if err := models.DB.Find(&players).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[PLAYER] Record not found! Error: %s", err.Error())
return
}
c.JSON(http.StatusOK, players)
}
@@ -38,14 +42,14 @@ func GetPlayersByClanHTML(c *gin.Context) {
Find(&players).Error; err != nil {
c.String(http.StatusBadRequest, "")
utils.Logger.Fatal(err)
utils.Logger.Errorf("[PLAYER] Clan not found! Error: %s", err.Error())
return
}
game, err := GetActiveGame(c)
if err != nil {
c.String(http.StatusBadRequest, "")
utils.Logger.Fatal(err)
utils.Logger.Errorf("[PLAYER] No active game not found! Error: %s", err.Error())
return
}
@@ -57,7 +61,8 @@ func GetPlayersByClanHTML(c *gin.Context) {
scores, err := models.PlayerCache.GetScores(playerIDs, game.Tag)
if err != nil {
c.String(http.StatusBadRequest, "")
utils.Logger.Fatal(err)
utils.Logger.Errorf("[PLAYER] Could not get scores! Error: %s", err.Error())
return
}
userRole := GetUserRoleByCtx(c)
@@ -83,92 +88,95 @@ func GetPlayersByClanHTML(c *gin.Context) {
err = utils.PlayerItemTemplate.Execute(c.Writer, data)
if err != nil {
utils.Logger.Fatal(err)
utils.Logger.Errorf("[PLAYER] Could not execute template! Error: %s", err.Error())
}
}
// AddPlayer POST /player
func AddPlayer(c *gin.Context) {
var input AddPlayerInput
var player models.Player
if err := c.BindJSON(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
return
}
var player models.Player
if err := models.DB.Where("name = ?", input.Name).First(&player).Error; err == nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Player with this name already exists!"})
utils.Logger.Errorf("[PLAYER] Could not bind JSON! Error: %s", err.Error())
return
}
player = models.Player{Name: input.Name, ClanID: input.ClanID}
models.DB.Create(&player)
if err := models.DB.Create(&player); err.Error != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error.Error()})
utils.Logger.Errorf("[PLAYER] Could not create player! Error: %s", err.Error.Error())
return
}
c.JSON(http.StatusOK, player)
utils.Logger.Infof("Added player '%s'", input.Name)
utils.Logger.Infof("[PLAYER] Added player '%s' in clan %d", input.Name, input.ClanID)
}
// GetPlayerByID GET /player/:id
func GetPlayerByID(c *gin.Context) {
player := FindPlayerByID(utils.StringToUint(c.Param("id")))
if player == nil {
var player models.Player
if err := models.DB.Where("id = ?", c.Param("id")).First(&player).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[PLAYER] Could not find player! Error: %s", err.Error())
return
}
c.JSON(http.StatusOK, player)
}
// GetPlayerIDByName GET /playerid/:name
func GetPlayerIDByName(c *gin.Context) {
var player models.Player
if err := FindPlayerByName(&player, c).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
return
}
c.JSON(http.StatusOK, player.ID)
}
// UpdatePlayerByID PATCH /player/:id
func UpdatePlayerByID(c *gin.Context) {
var input UpdatePlayerInput
var player models.Player
if err := c.ShouldBindJSON(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[PLAYER] Could not bind JSON! Error: %s", err.Error())
return
}
res := models.DB.Model(&models.Player{}).
Where("id = ?", utils.StringToUint(c.Param("id"))).
Updates(map[string]interface{}{
"Name": input.Name,
"ClanID": input.ClanID,
})
if res.Error != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": res.Error.Error()})
if err := models.DB.First(&player, c.Param("id")).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
utils.Logger.Errorf("[PLAYER] Could not find player! Error: %s", err.Error())
return
}
oldValues := models.Player{
Name: player.Name,
ClanID: player.ClanID,
}
player.Name = input.Name
player.ClanID = input.ClanID
if err := models.DB.Save(&player).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[PLAYER] Could not update player! Error: %s", err.Error())
return
}
c.JSON(http.StatusOK, nil)
utils.Logger.Infof("Updated player '%s'", input.Name)
utils.Logger.Infof("[PLAYER] Updated player: %s -> %s", oldValues.Name, player.Name)
}
// DeletePlayerByID DELETE /player/:id
func DeletePlayerByID(c *gin.Context) {
player := FindPlayerByID(utils.StringToUint(c.Param("id")))
if player == nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Record not found!"})
var player models.Player
if err := models.DB.Clauses(clause.Returning{}).Delete(&player).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[PLAYER] Could not delete player! Error: %s", err.Error())
return
}
models.DB.Delete(&player)
c.JSON(http.StatusOK, true)
utils.Logger.Infof("Deleted player '%s'", player.Name)
utils.Logger.Infof("[PLAYER] Deleted player: %s", player.Name)
}
// DeleteAllPlayers DELETE /admin/player
@@ -178,22 +186,12 @@ func DeleteAllPlayers(c *gin.Context) {
Session(&gorm.Session{AllowGlobalUpdate: true}).
Clauses(clause.Returning{}).
Delete(&players).Error; err != nil {
c.String(http.StatusBadRequest, "Purge failed! Error: "+err.Error())
utils.Logger.Errorf("[PLAYER] Could not purge players! Error: %s", err.Error())
return
}
c.String(http.StatusOK, "Purged "+utils.UintToString(uint(len(players)))+" players!")
}
func FindPlayerByID(id uint) *models.Player {
var res models.Player
if err := models.DB.Where("id = ?", id).First(&res).Error; err != nil {
return nil
} else {
return &res
}
}
func FindPlayerByName(out interface{}, c *gin.Context) *gorm.DB {
return models.DB.Where("name = ?", c.Param("name")).First(out)
utils.Logger.Infof("[PLAYER] Purged %d players!", len(players))
}

36
controllers/score_controller.go
View File

@@ -15,20 +15,17 @@ import (
// GetScoreByPlayerID GET /score/:player_id
func GetScoreByPlayerID(c *gin.Context) {
var player models.Player
var playerId = utils.StringToUint(c.Param("player_id"))
if err := models.DB.
Model(&models.Player{}).
Where("id = ?", playerId).
First(&player).Error; err != nil {
if err := models.DB.Where("id = ?", c.Param("player_id")).First(&player).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "Player not found!"})
utils.Logger.Errorf("[SCORE] Player not found: %s", err.Error())
return
}
game, err := GetActiveGame(c)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "No active game available!"})
utils.Logger.Errorf("[SCORE] No active game available: %s", err.Error())
return
}
@@ -37,7 +34,8 @@ func GetScoreByPlayerID(c *gin.Context) {
score = CalcPlayerScore(player.Name, game.Tag)
if score == score && score != -1 { // not NaN
if err := models.PlayerCache.SetScore(player.ID, game.Tag, score); err != nil {
utils.Logger.Fatal(err)
utils.Logger.Errorf("[SCORE] Failed to cache player score: %s", err.Error())
return
}
}
}
@@ -48,19 +46,20 @@ func GetScoreByPlayerID(c *gin.Context) {
c.String(http.StatusOK, fmt.Sprintf("%.2f", score))
} else {
c.JSON(http.StatusBadRequest, "Invalid request!")
utils.Logger.Warnf("[SCORE] Invalid request! Player: %s, Game: %s | Score: %f", player.Name, game.Tag, score)
}
}
// GetScoreByPlayerName POST /score/:player_name
func GetScoreByPlayerName(c *gin.Context) {
playerName := c.Param("player_name")
game, err := GetActiveGame(c)
if err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "No active game available!"})
utils.Logger.Errorf("[SCORE] No active game available: %s", err.Error())
return
}
score := CalcPlayerScore(playerName, game.Tag)
score := CalcPlayerScore(c.Param("player_name"), game.Tag)
if score != score || score == -1 { // NaN
c.String(http.StatusNotFound, "Spieler nicht gefunden!")
@@ -68,18 +67,19 @@ func GetScoreByPlayerName(c *gin.Context) {
c.String(http.StatusOK, fmt.Sprintf("%.2f", score))
} else {
c.String(http.StatusBadRequest, "Ungültige Abfrage!")
utils.Logger.Warnf("[SCORE] Invalid request! Player: %s, Game: %s | Score: %f", c.Param("player_name"), game.Tag, score)
}
}
func CalcPlayerScore(playerName string, gameTag string) float32 {
if gameTag != "BF5" && gameTag != "BF2042" {
utils.Logger.Errorf("Invalid game tag '%s'", gameTag)
utils.Logger.Errorf("[SCORE] Invalid game tag '%s'", gameTag)
return -1
}
gameMetrics := GetGameMetric(gameTag)
if gameMetrics == nil {
utils.Logger.Errorf("No game metrics specified for '%s'", gameTag)
utils.Logger.Errorf("[SCORE] No game metrics specified for '%s'", gameTag)
return -1
}
@@ -97,7 +97,7 @@ func CalcPlayerScore(playerName string, gameTag string) float32 {
req, err := http.NewRequest("GET", reqUri, nil)
if err != nil {
utils.Logger.Errorf("Failed to create request: %s", err)
utils.Logger.Errorf("[SCORE] Failed to create request: %s", err.Error())
return -1
}
@@ -105,7 +105,7 @@ func CalcPlayerScore(playerName string, gameTag string) float32 {
res, err := c.Do(req)
if err != nil {
utils.Logger.Errorf("Failed to send request: %s", err)
utils.Logger.Errorf("[SCORE] Failed to send request: %s", err.Error())
return -1
}
@@ -113,16 +113,18 @@ func CalcPlayerScore(playerName string, gameTag string) float32 {
_ = Body.Close()
}(res.Body)
if res.StatusCode == 404 {
utils.Logger.Errorf("User '%s' does not exist!", playerName)
utils.Logger.Errorf("[SCORE] User '%s' does not exist!", playerName)
return -1
} else if res.StatusCode != 200 {
utils.Logger.Fatalf("Status code error: %d %s", res.StatusCode, res.Status)
utils.Logger.Errorf("[SCORE] Status code error: %d %s", res.StatusCode, res.Status)
return -1
}
var data models.TrackerWeaponJSON
var body, _ = io.ReadAll(res.Body)
if err := json.Unmarshal(body, &data); err != nil {
utils.Logger.Fatalf("Failed to deserialize tracker API response: %s", err)
utils.Logger.Errorf("Failed to deserialize tracker API response: %s", err)
return -1
}
sort.SliceStable(data.Weapons, func(i, j int) bool { return data.Weapons[i].Kills > data.Weapons[j].Kills })
@@ -141,7 +143,7 @@ func CalcPlayerScore(playerName string, gameTag string) float32 {
weaponMetrics := FindWeaponMetric(gameMetrics.WeaponMetrics, weapon.Type)
if weaponMetrics == nil {
utils.Logger.Error("No weapon metrics specified for '" + gameTag + "', WType '" + weapon.Type + "'! Skipping...")
utils.Logger.Errorf("[SCORE] No weapon metrics specified for '%s'", weapon.Type)
continue
}

66
controllers/user_controller.go
View File

@@ -9,39 +9,31 @@ import (
func CreateUser(username string, hashedPassword string, enabled bool, usedCode string) {
user := models.User{Username: username, Password: hashedPassword, Enabled: enabled}
err := models.DB.Create(&user).Error
if err != nil {
utils.Logger.Fatalf("Error while creating user: %v", err)
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())
}
var code models.ActivationCode
err = models.DB.
Model(&models.ActivationCode{}).
Where("code = ?", usedCode).
First(&code).Error
if err != nil {
utils.Logger.Fatalf("Error while getting activation code: %v", err)
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
err = models.DB.Save(&code).Error
if err != nil {
utils.Logger.Fatalf("Error while updating activation code: %v", err)
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
err = models.DB.Save(&user).Error
if err != nil {
utils.Logger.Fatalf("Error while updating user role: %v", err)
if err := models.DB.Save(&user).Error; err != nil {
utils.Logger.Fatalf("[USER] Error while updating user role: %s", err.Error())
}
var bf2042 models.Game
err = models.DB.
Where("tag = ?", "BF2042").
First(&bf2042).Error
if err != nil {
utils.Logger.Fatalf("Error while getting game: %v", err)
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,
@@ -49,38 +41,50 @@ func CreateUser(username string, hashedPassword string, enabled bool, usedCode s
CalcMedian: false,
UseCache: true,
}
models.DB.Create(&userSettings)
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
models.DB.Where("username = ?", username).First(&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
err := models.DB.Where("username = ?", username).First(&user).Error
if err != nil {
utils.Logger.Fatal(err)
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
err := models.DB.Where("username = ?", username).First(&user).Error
if err != nil {
utils.Logger.Fatal(err)
return false
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
}

15
controllers/user_settings_controller.go
View File

@@ -2,6 +2,7 @@ package controllers
import (
"InfantrySkillCalculator/models"
"InfantrySkillCalculator/utils"
"errors"
"github.com/gin-gonic/gin"
"net/http"
@@ -22,11 +23,13 @@ func GetSettings(c *gin.Context) {
username, ok := session.GetUsername(c)
if !ok {
c.JSON(http.StatusBadRequest, gin.H{"error": "Not logged in!"})
utils.Logger.Errorf("[SETTINGS] User not logged in: %s", username)
return
}
if err := models.DB.Where("username = ?", username).First(&settings).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "No settings available!"})
utils.Logger.Errorf("[SETTINGS] No settings available for user %s", username)
return
}
@@ -48,16 +51,19 @@ func GetActiveGame(c *gin.Context) (models.Game, error) {
if !ok {
err := errors.New("not logged in")
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[SETTINGS] User not logged in: %s", username)
return models.Game{}, err
}
if err := models.DB.Where("username = ?", username).First(&settings).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "No settings available!"})
utils.Logger.Errorf("[SETTINGS] No settings available for user %s", username)
return models.Game{}, err
}
if err := models.DB.Where("id = ?", settings.ActiveGameID).First(&game).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "No active game available!"})
utils.Logger.Errorf("[SETTINGS] No active game available for user %s", username)
return models.Game{}, err
}
@@ -71,21 +77,28 @@ func UpdateSettings(c *gin.Context) {
username, ok := session.GetUsername(c)
if !ok {
c.JSON(http.StatusBadRequest, gin.H{"error": "Not logged in!"})
utils.Logger.Errorf("[SETTINGS] User not logged in: %s", username)
return
}
if err := models.DB.Where("username = ?", username).First(&settings).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": "No settings available!"})
utils.Logger.Errorf("[SETTINGS] No settings available for user %s", username)
return
}
var input UpdateSettingsInput
if err := c.ShouldBindJSON(&input); err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[SETTINGS] Failed to bind JSON: %s", err.Error())
return
}
models.DB.Model(&settings).Updates(input)
if err := models.DB.Model(&settings).Updates(input).Error; err != nil {
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
utils.Logger.Errorf("[SETTINGS] Failed to update settings: %s", err.Error())
return
}
c.JSON(http.StatusOK, nil)
}