package main

import (
	"fmt"
	"net/http"
	"os"

	"github.com/gin-gonic/gin"
	socketio "github.com/googollee/go-socket.io"
)

func Setup(basePath string, server *socketio.Server) *gin.Engine {

	r := gin.Default()

	if os.Getenv("NODE") == "production" {
		gin.SetMode(gin.ReleaseMode)
	}

	r.GET(basePath+"/check", func(c *gin.Context) {
		c.JSON(http.StatusAccepted, gin.H{"check": "good"})
	})

	//logger.RegisterLoggerRoutes(r, basePath, db)
	r.POST(basePath+"/update", func(c *gin.Context) {
		var payload UpdatePayload
		if err := c.BindJSON(&payload); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
			return
		}

		c.Writer.Header().Set("Content-Type", "text/event-stream")
		c.Writer.Header().Set("Cache-Control", "no-cache")
		c.Writer.Header().Set("Connection", "keep-alive")

		flusher, ok := c.Writer.(http.Flusher)
		if !ok {
			c.String(http.StatusInternalServerError, "Streaming not supported")
			return
		}

		// Start the update and get the channel
		updates := UpdateApp(server)

		for msg := range updates {
			//fmt.Fprintf(c.Writer, "event: log\ndata: %s\n\n", msg)
			fmt.Fprintf(c.Writer, "%s\n\n", msg)
			flusher.Flush()
		}

		fmt.Fprintf(c.Writer, "Update process finished\n\n")
		flusher.Flush()

	})

	r.Any(basePath+"/", func(c *gin.Context) { errorApiLoc(c) })

	return r
}

func errorApiLoc(c *gin.Context) {

	// log.Error("Api endpoint hit that dose not exist", "system", map[string]interface{}{
	// 	"endpoint":   c.Request.URL.Path,
	// 	"client_ip":  c.ClientIP(),
	// 	"user_agent": c.Request.UserAgent(),
	// })
	c.JSON(http.StatusBadRequest, gin.H{"message": "looks like you have encountered a route that dose not exist"})
}