File size: 2,520 Bytes
6bc074c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 | package handler
import (
"aurora/httpclient/bogdanfinn"
"aurora/internal/accounts"
"aurora/internal/chatgpt"
"aurora/internal/config"
"aurora/middlewares"
"github.com/gin-gonic/gin"
)
func optionsHandler(c *gin.Context) {
c.Header("Access-Control-Allow-Origin", "*")
c.Header("Access-Control-Allow-Methods", "POST")
c.Header("Access-Control-Allow-Headers", "*")
c.JSON(200, gin.H{
"message": "pong",
})
}
func RegisterRouter(accountPool *accounts.Pool, cfg *config.Config) *gin.Engine {
chatHandler := NewChatHandler(accountPool, cfg)
imageHandler := NewImageHandler(accountPool, cfg)
audioHandler := NewAudioHandler(accountPool, cfg)
authHandler := NewAuthHandler(accountPool)
modelsHandler := NewModelsHandler()
// 初始化基础前置参数（DPL、BasicCookies 等）
proxyUrl := ""
client := bogdanfinn.NewStdClient()
chatgpt.GetDpl(client, proxyUrl)
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"}) })
router.POST("/auth/session", authHandler.Session)
router.POST("/auth/refresh", authHandler.Refresh)
router.OPTIONS("/v1/chat/completions", optionsHandler)
router.OPTIONS("/v1/models", optionsHandler)
router.OPTIONS("/v1/responses", optionsHandler)
router.OPTIONS("/v1/images/generations", optionsHandler)
router.OPTIONS("/v1/images/edits", optionsHandler)
router.OPTIONS("/v1/images/variations", optionsHandler)
router.OPTIONS("/v1/files", optionsHandler)
router.OPTIONS("/v1/audio/speech", optionsHandler)
router.OPTIONS("/v1/audio/transcriptions", optionsHandler)
router.OPTIONS("/v1/audio/translations", optionsHandler)
authGroup := router.Group("").Use(middlewares.Authorization)
authGroup.POST("/v1/chat/completions", chatHandler.Nightmare)
authGroup.POST("/v1/responses", chatHandler.Responses)
authGroup.POST("/v1/files", chatHandler.Files)
authGroup.GET("/v1/models", modelsHandler.ListModels)
authGroup.POST("/backend-api/conversation", chatHandler.ChatGPTConversation)
authGroup.POST("/v1/images/generations", imageHandler.Generations)
authGroup.POST("/v1/images/edits", imageHandler.Edits)
authGroup.POST("/v1/images/variations", imageHandler.Variations)
authGroup.POST("/v1/audio/speech", audioHandler.TTS)
authGroup.POST("/v1/audio/transcriptions", audioHandler.Transcriptions)
authGroup.POST("/v1/audio/translations", audioHandler.Translations)
return router
}
|