main.go

package main

import (
	"context"
	"musicboxapi/configuration"
	"musicboxapi/database"
	"musicboxapi/http"
	"musicboxapi/logging"
	"os"

	"github.com/gin-contrib/cors"
	"github.com/gin-gonic/gin"
	"github.com/lrstanley/go-ytdlp"
)

func main() {
	configuration.LoadConfiguration()

	database.CreateDatabasConnectionPool()
	defer database.DbInstance.Close()

	// If yt-dlp isn't installed yet, download and cache it for further use.
	ytdlp.MustInstall(context.TODO(), nil)

	setGinMode()

	ginEngine := gin.Default()

	// Trust nginx
	ginEngine.SetTrustedProxies([]string{"127.0.0.1"})

	ginEngine.Use(corsMiddelWare())

	// V1 API
	apiv1Group := ginEngine.Group(configuration.GetApiGroupUrl("v1"))
	http.V1Endpoints(apiv1Group)

	if configuration.Config.DevPort != "" {
		devPort := "127.0.0.1:" + configuration.Config.DevPort
		logging.Info("Running on development port")
		ginEngine.Run(devPort)
	} else {
		ginEngine.Run() // listen and serve on 0.0.0.0:8080
	}
}

func corsMiddelWare() gin.HandlerFunc {
	if configuration.Config.UseDevUrl {
		return cors.Default()
	} else {
		origin := os.Getenv("CORS_ORIGIN")
		// Use Default cors
		if len(origin) == 0 {
			logging.Warning("CORS is enabled for all origins")
			return cors.Default()
		} else {
			strictCors := cors.New(cors.Config{
				AllowAllOrigins: false,
				AllowOrigins:    []string{origin}, // move to env
			})
			return strictCors
		}
	}
}

func setGinMode() {
	if configuration.Config.UseDevUrl {
		gin.SetMode(gin.DebugMode)
	} else {
		gin.SetMode(gin.ReleaseMode)
	}
}