main.go

package main

import (
	"game-stats-api/pkg/database"
	"game-stats-api/pkg/server"
	"log"
	"net/http"
	"os"
	"time"

	_ "game-stats-api/docs"

	"github.com/go-chi/chi/middleware"
	"github.com/gorilla/handlers"
	"github.com/gorilla/mux"
	"github.com/joho/godotenv"
	httpSwagger "github.com/swaggo/http-swagger"
)

// @Title Game Stats API
// @Version 1.0
// @License.name MIT
// @License.url https://opensource.org/licenses/MIT
// @BasePath /api/v1
func main() {
	godotenv.Load()

	address := os.Getenv("ADDRESS")
	if address == "" {
		address = "127.0.0.1"
	}
	port := os.Getenv("PORT")
	if port == "" {
		port = "3000"
	}

	err := database.Initialize("data.db")
	if err != nil {
		log.Println(err)
		os.Exit(1)
	}

	router := mux.NewRouter()
	router.StrictSlash(true)
	router.HandleFunc("/api/v1/games/", server.CreateGameHandler).Methods("POST")
	router.HandleFunc("/api/v1/games/", server.GetAllGamesHandler).Methods("GET")
	router.HandleFunc("/api/v1/games/{id:[0-9]+}/", server.GetGameHandler).Methods("GET")
	router.HandleFunc("/api/v1/games/{id:[0-9]+}/", server.DeleteGameHandler).Methods("DELETE")
	router.HandleFunc("/api/v1/games/{id:[0-9]+}/players/", server.CreateGamePlayerHandler).Methods("POST")
	router.HandleFunc("/api/v1/games/{gid:[0-9]+}/players/{pid:[0-9]+}/", server.DeleteGamePlayerHandler).Methods("DELETE")
	router.HandleFunc("/api/v1/games/{gid:[0-9]+}/players/{pid:[0-9]+}/stats/", server.AddGamePlayerStatHandler).Methods("POST")
	router.HandleFunc("/api/v1/games/{gid:[0-9]+}/players/{pid:[0-9]+}/laststat/", server.DeleteGamePlayerLastStatHandler).Methods("DELETE")
	router.Use(middleware.Logger)
	router.Use(middleware.AllowContentEncoding("application/json"))

	router.PathPrefix("/swagger/").Handler(httpSwagger.WrapHandler)

	spa := server.NewSpaHandler("ui/dist/spa", "index.html")
	router.PathPrefix("/").Handler(spa)

	allowedHeaders := handlers.AllowedHeaders([]string{"X-Requested-With", "content-type"})
	allowedOrigins := handlers.AllowedOrigins([]string{"*"})
	allowedMethods := handlers.AllowedMethods([]string{"GET", "POST", "DELETE", "OPTIONS"})

	log.Printf("Listening on %v:%v", address, port)

	srv := &http.Server{
		Handler:      handlers.CORS(allowedOrigins, allowedHeaders, allowedMethods)(router),
		Addr:         address + ":" + port,
		WriteTimeout: 15 * time.Second,
		ReadTimeout:  15 * time.Second,
	}
	srv.ListenAndServe()
}