feat: refactor application to use Gin framework, update routing and middleware handling

This commit is contained in:
王锦强
2025-03-15 12:39:17 +08:00
parent cab289dabf
commit 6fa5c1f467
13 changed files with 314 additions and 279 deletions

View File

@@ -1,58 +1,58 @@
package middleware
import (
"net/http"
"strings"
"github.com/gin-gonic/gin"
)
// OpenAIAuth 中间件验证 OpenAI API 请求的令牌
func OpenAIAuth(apiToken string, next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
func OpenAIAuth(apiToken string) gin.HandlerFunc {
return func(c *gin.Context) {
// 如果没有配置令牌,跳过验证
if apiToken == "" {
next.ServeHTTP(w, r)
c.Next()
return
}
// 获取请求头中的 Authorization
authHeader := r.Header.Get("Authorization")
authHeader := c.GetHeader("Authorization")
if authHeader == "" {
http.Error(w, "未提供授权令牌", http.StatusUnauthorized)
c.AbortWithStatusJSON(401, gin.H{"error": "未提供授权令牌"})
return
}
// 验证格式是否为 "Bearer {token}"
parts := strings.SplitN(authHeader, " ", 2)
if len(parts) != 2 || parts[0] != "Bearer" {
http.Error(w, "授权格式无效", http.StatusUnauthorized)
c.AbortWithStatusJSON(401, gin.H{"error": "授权格式无效"})
return
}
// 验证令牌是否正确
if parts[1] != apiToken {
http.Error(w, "令牌无效", http.StatusUnauthorized)
c.AbortWithStatusJSON(401, gin.H{"error": "令牌无效"})
return
}
// 令牌验证通过,继续处理请求
next.ServeHTTP(w, r)
})
c.Next()
}
}
// TTSAuth 是用于验证 TTS API 接口的中间件
func TTSAuth(apiKey string, next http.Handler) http.Handler {
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
func TTSAuth(apiKey string) gin.HandlerFunc {
return func(c *gin.Context) {
// 从查询参数中获取 api_key
queryKey := r.URL.Query().Get("api_key")
queryKey := c.Query("api_key")
// 如果 apiKey 配置为空字符串,表示不需要验证
if apiKey != "" && queryKey != apiKey {
w.WriteHeader(http.StatusUnauthorized)
w.Write([]byte("未授权访问: 无效的 API 密钥"))
c.AbortWithStatusJSON(401, gin.H{"error": "未授权访问: 无效的 API 密钥"})
return
}
// 验证通过,继续处理请求
next.ServeHTTP(w, r)
})
c.Next()
}
}