package main

import (
	"janex/admin-dashboard-backend/grouptasks"
	"janex/admin-dashboard-backend/modules/config"
	"janex/admin-dashboard-backend/modules/database"
	"janex/admin-dashboard-backend/modules/logger"
	"janex/admin-dashboard-backend/modules/structs"
	"janex/admin-dashboard-backend/modules/utils"
	"janex/admin-dashboard-backend/routers/router"
	"janex/admin-dashboard-backend/socketserver"

	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/fiber/v2/middleware/cors"
	flogger "github.com/gofiber/fiber/v2/middleware/logger"
	"github.com/gofiber/websocket/v2"
	"github.com/rs/zerolog/log"
)

func init() {
	config.LoadConfig()
	logger.InitLogger()
	utils.ValidatorInit()
	database.InitDatabase()
}

func main() {
	app := fiber.New()

	app.Use(cors.New())

	if config.Cfg.Debug {
		app.Use(flogger.New(flogger.Config{
			Format: "${pid} ${locals:requestid} ${status} - ${latency} ${method} ${path}​\n",
		}))
	}

	router.SetupRoutes(app)

	app.Use("/ws", func(c *fiber.Ctx) error {
		// IsWebSocketUpgrade returns true if the client
		// requested upgrade to the WebSocket protocol.
		if websocket.IsWebSocketUpgrade(c) {
			sessionId := c.Query("auth")

			if len(sessionId) != utils.LenHeaderXAuthorization {
				return c.SendStatus(fiber.StatusUnauthorized)
			}

			// validate ws session

			var userSession structs.UserSession

			database.DB.First(&userSession, "id = ?", sessionId)

			if userSession.Id != "" {
				log.Info().Msg("session id: " + userSession.Id + " user: " + userSession.Id)

				var user structs.User

				database.DB.First(&user, "id = ?", userSession.UserId)

				if user.Id != "" {
					log.Info().Msg("user " + user.Id + user.Username)

					c.Locals("sessionId", sessionId)
					c.Locals("userId", user.Id)
				}
			}

			return c.Next()
		}

		return fiber.ErrUpgradeRequired
	})

	grouptasks.ReadGroups()

	go socketserver.RunHub()
	socketserver.WebSocketServer(app)

	app.Listen(config.Cfg.Host + ":" + config.Cfg.Port)
}

/*cmd, err := exec.Command("python3", "./tasks/test.py", "--url", "test", "--name", "test").Output()

if err != nil {
	log.Fatal(err)
}

fmt.Println(string(cmd)) */