package initialize

import (
	"aurora/middlewares"
	"os"

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

func RegisterRouter() *gin.Engine {
	handler := NewHandle(
		checkProxy(),
	)

	router := gin.Default()
	router.Use(middlewares.Cors)

	router.GET("/", func(c *gin.Context) {
		c.JSON(200, gin.H{
			"message": "Hello, world!",
		})
	})

	router.GET("/ping", func(c *gin.Context) {
		c.JSON(200, gin.H{
			"message": "pong",
		})
	})

	prefixGroup := os.Getenv("PREFIX")
	if prefixGroup != "" {
		prefixRouter := router.Group(prefixGroup)
		{
			// 现有的路由
			prefixRouter.OPTIONS("/v1/chat/completions", optionsHandler)
			prefixRouter.OPTIONS("/v1/chat/models", optionsHandler)
			prefixRouter.POST("/v1/chat/completions", middlewares.Authorization, handler.duckduckgo)
			prefixRouter.GET("/v1/models", middlewares.Authorization, handler.engines)

			// 新增的路由
			prefixRouter.POST("/api/v1/chat/completions", middlewares.Authorization, handler.duckduckgo)
			prefixRouter.POST("/completions", middlewares.Authorization, handler.duckduckgo)
			prefixRouter.GET("/api/v1/models", middlewares.Authorization, handler.engines)
			prefixRouter.GET("/models", middlewares.Authorization, handler.engines)
		}
	}

	router.OPTIONS("/v1/chat/completions", optionsHandler)
	router.OPTIONS("/v1/chat/models", optionsHandler)
	authGroup := router.Group("").Use(middlewares.Authorization)

	// authGroup 下的路由
	authGroup.POST("/v1/chat/completions", handler.duckduckgo)
	authGroup.POST("/api/v1/chat/completions", handler.duckduckgo)
	authGroup.POST("/completions", handler.duckduckgo)

	authGroup.GET("/v1/models", handler.engines)
	authGroup.GET("/api/v1/models", handler.engines)
	authGroup.GET("/models", handler.engines)

	return router
}