package controllers
import (
"InfantrySkillCalculator/models"
"InfantrySkillCalculator/utils"
"fmt"
"github.com/gin-gonic/gin"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"net/http"
)
type AddClanInput struct {
Name string `json:"name" binding:"required"`
Tag string `json:"tag" binding:"required"`
KeepUpdated bool `json:"keep_updated" gorm:"default:false"`
}
type UpdateClanInput struct {
Name string `json:"name" binding:"required"`
Tag string `json:"tag" binding:"required"`
KeepUpdated bool `json:"keep_updated" gorm:"default:false"`
}
// GetAllClans GET /clan
func GetAllClans(c *gin.Context) {
var clans []models.Clan
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)
}
// GetAllClansHTML GET /clans_html
func GetAllClansHTML(c *gin.Context) {
var clans []models.Clan
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 = ``
for _, clan := range clans {
htmlOptions += fmt.Sprintf(``, clan.ID, clan.Tag, clan.Name)
}
c.Header("Content-Type", "text/html")
c.String(http.StatusOK, htmlOptions)
}
// 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()})
utils.Logger.Errorf("[CLAN] Error while binding JSON: %s", err.Error())
return
}
clan = models.Clan{Name: input.Name, Tag: input.Tag, KeepUpdated: input.KeepUpdated}
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("[CLAN] Added clan: [%s] %s", clan.Tag, clan.Name)
}
// GetClanByID GET /clan/:id
func GetClanByID(c *gin.Context) {
var clan models.Clan
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
}
c.JSON(http.StatusOK, clan)
}
// 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
}
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("[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 := 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
}
c.JSON(http.StatusOK, true)
utils.Logger.Infof("[CLAN] Deleted clan: [%s] %s", clan.Tag, clan.Name)
}
// DeleteAllClans DELETE /admin/clan
func DeleteAllClans(c *gin.Context) {
var clans []models.Clan
if err := models.DB.
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!")
utils.Logger.Infof("[CLAN] Purged %d clans!", len(clans))
}