InfantrySkillCalculator/main.go

package main

import (
	"InfantrySkillCalculator/controllers"
	"InfantrySkillCalculator/models"
	"InfantrySkillCalculator/utils"
	"github.com/gin-gonic/gin"
	_ "github.com/gorilla/sessions"
	"html/template"
	"io"
	"log"
	"os"
)

func init() {
	var err error
	utils.MainPageTemplates, err = template.ParseFiles(
		"./templates/index.html",
		"./templates/components/home_clan_bar.html",
		"./templates/components/opp_clan_bar.html",
		"./templates/components/home_player_list.html",
		"./templates/components/opp_player_list.html",
		"./templates/components/bottom_controls.html",
		"./templates/modals/delete_clan.html",
		"./templates/modals/add_clan.html",
		"./templates/modals/edit_clan.html",
		"./templates/modals/add_player.html",
		"./templates/modals/delete_player.html",
		"./templates/modals/edit_player.html",
		"./templates/modals/settings.html",
		"./templates/components/header.html",
	)
	if err != nil {
		log.Fatal(err)
	}

	utils.LoginPageTemplates, err = template.ParseFiles(
		"./templates/login.html",
		"./templates/components/header.html",
	)
	if err != nil {
		log.Fatal(err)
	}

	utils.RegisterPageTemplates, err = template.ParseFiles(
		"./templates/register.html",
		"./templates/components/header.html",
	)
	if err != nil {
		log.Fatal(err)
	}

	utils.PlayerItemTemplate, err = template.ParseFiles(
		"./templates/player_list_item.html",
	)
	if err != nil {
		log.Fatal(err)
	}

	controllers.LoadMetrics()
}

func main() {
	if os.Getenv("GO_ENV") == "production" {
		gin.SetMode(gin.ReleaseMode)
	}

	router := gin.New()
	err := router.SetTrustedProxies([]string{"127.0.0.1"})
	if err != nil {
		log.Fatal(err)
	}
	router.LoadHTMLGlob("templates/**/*")
	reader := router.Group("/")
	reader.Use(ReaderAuthRequired())
	author := router.Group("/")
	author.Use(AuthorAuthRequired())
	admin := router.Group("/admin")
	admin.Use(AdminAuthRequired())

	models.ConnectDatabase()
	models.ConnectCache(utils.PlayerCacheLifetime)

	var code models.ActivationCode
	if err := models.DB.First(&code).Error; err != nil {
		firstCode := utils.GenerateActivationCode()
		models.DB.Create(&models.ActivationCode{Code: firstCode, UserRole: models.AdminRole})
		log.Println("Created first activation code with ADMIN role:\n" + firstCode)
	}

	f, _ := os.OpenFile("isc_rest.log", os.O_RDWR|os.O_APPEND|os.O_CREATE, 0660)
	utils.GinWriter = io.MultiWriter(f, os.Stdout)
	router.Use(
		gin.LoggerWithWriter(utils.GinWriter, "/static"),
		gin.Recovery(),
	)
	reader.Use(
		gin.LoggerWithWriter(utils.GinWriter),
		gin.Recovery(),
	)
	author.Use(
		gin.LoggerWithWriter(utils.GinWriter),
		gin.Recovery(),
	)
	admin.Use(
		gin.LoggerWithWriter(utils.GinWriter),
		gin.Recovery(),
	)

	router.Static("/static", "./static")

	router.GET("/login", loginPage)
	router.POST("/login", loginPost)
	router.GET("/logout", logout)
	router.GET("/register", registerPage)
	router.POST("/register", registerPost)

	reader.GET("/", mainPage)

	reader.GET("/clans", controllers.GetAllClans)
	reader.GET("/clans_html", controllers.GetAllClansHTML)
	reader.GET("/clan/:id", controllers.GetClanByID)
	author.POST("/clan", controllers.AddClan)
	author.PATCH("/clan/:id", controllers.UpdateClanByID)
	author.DELETE("/clan/:id", controllers.DeleteClanByID)

	reader.GET("/players", controllers.GetAllPlayers)
	reader.GET("/players_html", controllers.GetPlayersByClanHTML)
	reader.GET("/player/:id", controllers.GetPlayerByID)
	reader.GET("/playerid/:name", controllers.GetPlayerIDByName)
	author.POST("/player", controllers.AddPlayer)
	author.PATCH("/player/:id", controllers.UpdatePlayerByID)
	author.DELETE("/player/:id", controllers.DeletePlayerByID)

	reader.GET("/cache/:player_id", controllers.GetCacheByPlayerID)

	reader.GET("/score/:player_id", controllers.GetScoreByPlayerID)
	reader.POST("/score/:player_name", controllers.GetScoreByPlayerName)

	reader.GET("/game", controllers.GetGames)
	reader.GET("/game_html", controllers.GetGamesHTML)

	reader.GET("/settings", controllers.GetSettings)
	reader.PATCH("/settings", controllers.UpdateSettings)

	admin.DELETE("/clear_cache", controllers.DeleteAllCaches)
	admin.DELETE("/purge_players", controllers.DeleteAllPlayers)
	admin.DELETE("/purge_clans", controllers.DeleteAllClans)
	admin.POST("/create_code", controllers.CreateCode)

	log.Println("Running on 8000...")
	log.Fatal(router.Run(":8000"))
}