package api

import (
	handlers "github.com/suno-ai/glockenspiel/studio_api_flute/internal/api/handlers"

	"github.com/gin-gonic/gin"
)

func SetupRouter() *gin.Engine {
	r := gin.Default()

	// Health check endpoint
	r.GET("/health", func(c *gin.Context) {
		c.JSON(200, gin.H{
			"status": "OK",
		})
	})

	// WebSocket endpoint
	wsGroup := r.Group("/ws")
	{
		wsGroup.GET("/generate", func(c *gin.Context) {
			handlers.HandleWebSocket(c.Writer, c.Request)
		})
	}

	return r
}