ianshaw
9780f0fd9d
- 更新 middleware import 路径到 internal/server/middleware - 修复 api_key_auth_google.go 使用正确的 service 类型 - 更新 router.go 和 http.go 支持 Gemini v1beta 路由 - 在 routes/gateway.go 中添加 Gemini v1beta API 端点 - 在 routes/admin.go 中添加 Gemini OAuth 路由 - 更新 wire.go 添加 GeminiOAuthService cleanup - 重新生成 wire_gen.go
44 lines
1.5 KiB
Go
44 lines
1.5 KiB
Go
package routes
|
|
|
|
import (
|
|
"github.com/Wei-Shaw/sub2api/internal/handler"
|
|
"github.com/Wei-Shaw/sub2api/internal/server/middleware"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
// RegisterGatewayRoutes 注册 API 网关路由（Claude/OpenAI/Gemini 兼容）
|
|
func RegisterGatewayRoutes(
|
|
r *gin.Engine,
|
|
h *handler.Handlers,
|
|
apiKeyAuth middleware.ApiKeyAuthMiddleware,
|
|
apiKeyService *service.ApiKeyService,
|
|
subscriptionService *service.SubscriptionService,
|
|
) {
|
|
// API网关（Claude API兼容）
|
|
gateway := r.Group("/v1")
|
|
gateway.Use(gin.HandlerFunc(apiKeyAuth))
|
|
{
|
|
gateway.POST("/messages", h.Gateway.Messages)
|
|
gateway.POST("/messages/count_tokens", h.Gateway.CountTokens)
|
|
gateway.GET("/models", h.Gateway.Models)
|
|
gateway.GET("/usage", h.Gateway.Usage)
|
|
// OpenAI Responses API
|
|
gateway.POST("/responses", h.OpenAIGateway.Responses)
|
|
}
|
|
|
|
// Gemini 原生 API 兼容层（Gemini SDK/CLI 直连）
|
|
gemini := r.Group("/v1beta")
|
|
gemini.Use(middleware.ApiKeyAuthWithSubscriptionGoogle(apiKeyService, subscriptionService))
|
|
{
|
|
gemini.GET("/models", h.Gateway.GeminiV1BetaListModels)
|
|
gemini.GET("/models/:model", h.Gateway.GeminiV1BetaGetModel)
|
|
// Gin treats ":" as a param marker, but Gemini uses "{model}:{action}" in the same segment.
|
|
gemini.POST("/models/*modelAction", h.Gateway.GeminiV1BetaModels)
|
|
}
|
|
|
|
// OpenAI Responses API（不带v1前缀的别名）
|
|
r.POST("/responses", gin.HandlerFunc(apiKeyAuth), h.OpenAIGateway.Responses)
|
|
}
|