| package router |
|
|
| import ( |
| "github.com/QuantumNous/new-api/controller" |
| "github.com/QuantumNous/new-api/middleware" |
|
|
| |
| _ "github.com/QuantumNous/new-api/oauth" |
|
|
| "github.com/gin-contrib/gzip" |
| "github.com/gin-gonic/gin" |
| ) |
|
|
| func SetApiRouter(router *gin.Engine) { |
| apiRouter := router.Group("/api") |
| apiRouter.Use(middleware.RouteTag("api")) |
| apiRouter.Use(gzip.Gzip(gzip.DefaultCompression)) |
| apiRouter.Use(middleware.BodyStorageCleanup()) |
| apiRouter.Use(middleware.GlobalAPIRateLimit()) |
| { |
| apiRouter.GET("/setup", controller.GetSetup) |
| apiRouter.POST("/setup", controller.PostSetup) |
| apiRouter.GET("/status", controller.GetStatus) |
| apiRouter.GET("/uptime/status", controller.GetUptimeKumaStatus) |
| apiRouter.GET("/models", middleware.UserAuth(), controller.DashboardListModels) |
| apiRouter.GET("/status/test", middleware.AdminAuth(), controller.TestStatus) |
| apiRouter.GET("/notice", controller.GetNotice) |
| apiRouter.GET("/user-agreement", controller.GetUserAgreement) |
| apiRouter.GET("/privacy-policy", controller.GetPrivacyPolicy) |
| apiRouter.GET("/about", controller.GetAbout) |
| |
| apiRouter.GET("/home_page_content", controller.GetHomePageContent) |
| apiRouter.GET("/pricing", middleware.TryUserAuth(), controller.GetPricing) |
| apiRouter.GET("/verification", middleware.EmailVerificationRateLimit(), middleware.TurnstileCheck(), controller.SendEmailVerification) |
| apiRouter.GET("/reset_password", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.SendPasswordResetEmail) |
| apiRouter.POST("/user/reset", middleware.CriticalRateLimit(), controller.ResetPassword) |
| |
| apiRouter.GET("/oauth/state", middleware.CriticalRateLimit(), controller.GenerateOAuthCode) |
| apiRouter.GET("/oauth/email/bind", middleware.CriticalRateLimit(), controller.EmailBind) |
| |
| apiRouter.GET("/oauth/wechat", middleware.CriticalRateLimit(), controller.WeChatAuth) |
| apiRouter.GET("/oauth/wechat/bind", middleware.CriticalRateLimit(), controller.WeChatBind) |
| apiRouter.GET("/oauth/telegram/login", middleware.CriticalRateLimit(), controller.TelegramLogin) |
| apiRouter.GET("/oauth/telegram/bind", middleware.CriticalRateLimit(), controller.TelegramBind) |
| |
| apiRouter.GET("/oauth/:provider", middleware.CriticalRateLimit(), controller.HandleOAuth) |
| apiRouter.GET("/ratio_config", middleware.CriticalRateLimit(), controller.GetRatioConfig) |
|
|
| apiRouter.POST("/stripe/webhook", controller.StripeWebhook) |
| apiRouter.POST("/creem/webhook", controller.CreemWebhook) |
|
|
| |
| apiRouter.POST("/verify", middleware.UserAuth(), middleware.CriticalRateLimit(), controller.UniversalVerify) |
|
|
| userRoute := apiRouter.Group("/user") |
| { |
| userRoute.POST("/register", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.Register) |
| userRoute.POST("/login", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.Login) |
| userRoute.POST("/login/2fa", middleware.CriticalRateLimit(), controller.Verify2FALogin) |
| userRoute.POST("/passkey/login/begin", middleware.CriticalRateLimit(), controller.PasskeyLoginBegin) |
| userRoute.POST("/passkey/login/finish", middleware.CriticalRateLimit(), controller.PasskeyLoginFinish) |
| |
| userRoute.GET("/logout", controller.Logout) |
| userRoute.POST("/epay/notify", controller.EpayNotify) |
| userRoute.GET("/epay/notify", controller.EpayNotify) |
| userRoute.GET("/groups", controller.GetUserGroups) |
|
|
| selfRoute := userRoute.Group("/") |
| selfRoute.Use(middleware.UserAuth()) |
| { |
| selfRoute.GET("/self/groups", controller.GetUserGroups) |
| selfRoute.GET("/self", controller.GetSelf) |
| selfRoute.GET("/models", controller.GetUserModels) |
| selfRoute.PUT("/self", controller.UpdateSelf) |
| selfRoute.DELETE("/self", controller.DeleteSelf) |
| selfRoute.GET("/token", controller.GenerateAccessToken) |
| selfRoute.GET("/passkey", controller.PasskeyStatus) |
| selfRoute.POST("/passkey/register/begin", controller.PasskeyRegisterBegin) |
| selfRoute.POST("/passkey/register/finish", controller.PasskeyRegisterFinish) |
| selfRoute.POST("/passkey/verify/begin", controller.PasskeyVerifyBegin) |
| selfRoute.POST("/passkey/verify/finish", controller.PasskeyVerifyFinish) |
| selfRoute.DELETE("/passkey", controller.PasskeyDelete) |
| selfRoute.GET("/aff", controller.GetAffCode) |
| selfRoute.GET("/topup/info", controller.GetTopUpInfo) |
| selfRoute.GET("/topup/self", controller.GetUserTopUps) |
| selfRoute.POST("/topup", middleware.CriticalRateLimit(), controller.TopUp) |
| selfRoute.POST("/pay", middleware.CriticalRateLimit(), controller.RequestEpay) |
| selfRoute.POST("/amount", controller.RequestAmount) |
| selfRoute.POST("/stripe/pay", middleware.CriticalRateLimit(), controller.RequestStripePay) |
| selfRoute.POST("/stripe/amount", controller.RequestStripeAmount) |
| selfRoute.POST("/creem/pay", middleware.CriticalRateLimit(), controller.RequestCreemPay) |
| selfRoute.POST("/aff_transfer", controller.TransferAffQuota) |
| selfRoute.PUT("/setting", controller.UpdateUserSetting) |
|
|
| |
| selfRoute.GET("/2fa/status", controller.Get2FAStatus) |
| selfRoute.POST("/2fa/setup", controller.Setup2FA) |
| selfRoute.POST("/2fa/enable", controller.Enable2FA) |
| selfRoute.POST("/2fa/disable", controller.Disable2FA) |
| selfRoute.POST("/2fa/backup_codes", controller.RegenerateBackupCodes) |
|
|
| |
| selfRoute.GET("/checkin", controller.GetCheckinStatus) |
| selfRoute.POST("/checkin", middleware.TurnstileCheck(), controller.DoCheckin) |
|
|
| |
| selfRoute.GET("/oauth/bindings", controller.GetUserOAuthBindings) |
| selfRoute.DELETE("/oauth/bindings/:provider_id", controller.UnbindCustomOAuth) |
| } |
|
|
| adminRoute := userRoute.Group("/") |
| adminRoute.Use(middleware.AdminAuth()) |
| { |
| adminRoute.GET("/", controller.GetAllUsers) |
| adminRoute.GET("/topup", controller.GetAllTopUps) |
| adminRoute.POST("/topup/complete", controller.AdminCompleteTopUp) |
| adminRoute.GET("/search", controller.SearchUsers) |
| adminRoute.GET("/:id/oauth/bindings", controller.GetUserOAuthBindingsByAdmin) |
| adminRoute.DELETE("/:id/oauth/bindings/:provider_id", controller.UnbindCustomOAuthByAdmin) |
| adminRoute.DELETE("/:id/bindings/:binding_type", controller.AdminClearUserBinding) |
| adminRoute.GET("/:id", controller.GetUser) |
| adminRoute.POST("/", controller.CreateUser) |
| adminRoute.POST("/manage", controller.ManageUser) |
| adminRoute.PUT("/", controller.UpdateUser) |
| adminRoute.DELETE("/:id", controller.DeleteUser) |
| adminRoute.DELETE("/:id/reset_passkey", controller.AdminResetPasskey) |
|
|
| |
| adminRoute.GET("/2fa/stats", controller.Admin2FAStats) |
| adminRoute.DELETE("/:id/2fa", controller.AdminDisable2FA) |
| } |
| } |
|
|
| |
| subscriptionRoute := apiRouter.Group("/subscription") |
| subscriptionRoute.Use(middleware.UserAuth()) |
| { |
| subscriptionRoute.GET("/plans", controller.GetSubscriptionPlans) |
| subscriptionRoute.GET("/self", controller.GetSubscriptionSelf) |
| subscriptionRoute.PUT("/self/preference", controller.UpdateSubscriptionPreference) |
| subscriptionRoute.POST("/epay/pay", middleware.CriticalRateLimit(), controller.SubscriptionRequestEpay) |
| subscriptionRoute.POST("/stripe/pay", middleware.CriticalRateLimit(), controller.SubscriptionRequestStripePay) |
| subscriptionRoute.POST("/creem/pay", middleware.CriticalRateLimit(), controller.SubscriptionRequestCreemPay) |
| } |
| subscriptionAdminRoute := apiRouter.Group("/subscription/admin") |
| subscriptionAdminRoute.Use(middleware.AdminAuth()) |
| { |
| subscriptionAdminRoute.GET("/plans", controller.AdminListSubscriptionPlans) |
| subscriptionAdminRoute.POST("/plans", controller.AdminCreateSubscriptionPlan) |
| subscriptionAdminRoute.PUT("/plans/:id", controller.AdminUpdateSubscriptionPlan) |
| subscriptionAdminRoute.PATCH("/plans/:id", controller.AdminUpdateSubscriptionPlanStatus) |
| subscriptionAdminRoute.POST("/bind", controller.AdminBindSubscription) |
|
|
| |
| subscriptionAdminRoute.GET("/users/:id/subscriptions", controller.AdminListUserSubscriptions) |
| subscriptionAdminRoute.POST("/users/:id/subscriptions", controller.AdminCreateUserSubscription) |
| subscriptionAdminRoute.POST("/user_subscriptions/:id/invalidate", controller.AdminInvalidateUserSubscription) |
| subscriptionAdminRoute.DELETE("/user_subscriptions/:id", controller.AdminDeleteUserSubscription) |
| } |
|
|
| |
| apiRouter.POST("/subscription/epay/notify", controller.SubscriptionEpayNotify) |
| apiRouter.GET("/subscription/epay/notify", controller.SubscriptionEpayNotify) |
| apiRouter.GET("/subscription/epay/return", controller.SubscriptionEpayReturn) |
| apiRouter.POST("/subscription/epay/return", controller.SubscriptionEpayReturn) |
| optionRoute := apiRouter.Group("/option") |
| optionRoute.Use(middleware.RootAuth()) |
| { |
| optionRoute.GET("/", controller.GetOptions) |
| optionRoute.PUT("/", controller.UpdateOption) |
| optionRoute.GET("/channel_affinity_cache", controller.GetChannelAffinityCacheStats) |
| optionRoute.DELETE("/channel_affinity_cache", controller.ClearChannelAffinityCache) |
| optionRoute.POST("/rest_model_ratio", controller.ResetModelRatio) |
| optionRoute.POST("/migrate_console_setting", controller.MigrateConsoleSetting) |
| } |
|
|
| |
| customOAuthRoute := apiRouter.Group("/custom-oauth-provider") |
| customOAuthRoute.Use(middleware.RootAuth()) |
| { |
| customOAuthRoute.POST("/discovery", controller.FetchCustomOAuthDiscovery) |
| customOAuthRoute.GET("/", controller.GetCustomOAuthProviders) |
| customOAuthRoute.GET("/:id", controller.GetCustomOAuthProvider) |
| customOAuthRoute.POST("/", controller.CreateCustomOAuthProvider) |
| customOAuthRoute.PUT("/:id", controller.UpdateCustomOAuthProvider) |
| customOAuthRoute.DELETE("/:id", controller.DeleteCustomOAuthProvider) |
| } |
| performanceRoute := apiRouter.Group("/performance") |
| performanceRoute.Use(middleware.RootAuth()) |
| { |
| performanceRoute.GET("/stats", controller.GetPerformanceStats) |
| performanceRoute.DELETE("/disk_cache", controller.ClearDiskCache) |
| performanceRoute.POST("/reset_stats", controller.ResetPerformanceStats) |
| performanceRoute.POST("/gc", controller.ForceGC) |
| } |
| ratioSyncRoute := apiRouter.Group("/ratio_sync") |
| ratioSyncRoute.Use(middleware.RootAuth()) |
| { |
| ratioSyncRoute.GET("/channels", controller.GetSyncableChannels) |
| ratioSyncRoute.POST("/fetch", controller.FetchUpstreamRatios) |
| } |
| channelRoute := apiRouter.Group("/channel") |
| channelRoute.Use(middleware.AdminAuth()) |
| { |
| channelRoute.GET("/", controller.GetAllChannels) |
| channelRoute.GET("/search", controller.SearchChannels) |
| channelRoute.GET("/models", controller.ChannelListModels) |
| channelRoute.GET("/models_enabled", controller.EnabledListModels) |
| channelRoute.GET("/:id", controller.GetChannel) |
| channelRoute.POST("/:id/key", middleware.RootAuth(), middleware.CriticalRateLimit(), middleware.DisableCache(), middleware.SecureVerificationRequired(), controller.GetChannelKey) |
| channelRoute.GET("/test", controller.TestAllChannels) |
| channelRoute.GET("/test/:id", controller.TestChannel) |
| channelRoute.GET("/update_balance", controller.UpdateAllChannelsBalance) |
| channelRoute.GET("/update_balance/:id", controller.UpdateChannelBalance) |
| channelRoute.POST("/", controller.AddChannel) |
| channelRoute.PUT("/", controller.UpdateChannel) |
| channelRoute.DELETE("/disabled", controller.DeleteDisabledChannel) |
| channelRoute.POST("/tag/disabled", controller.DisableTagChannels) |
| channelRoute.POST("/tag/enabled", controller.EnableTagChannels) |
| channelRoute.PUT("/tag", controller.EditTagChannels) |
| channelRoute.DELETE("/:id", controller.DeleteChannel) |
| channelRoute.POST("/batch", controller.DeleteChannelBatch) |
| channelRoute.POST("/fix", controller.FixChannelsAbilities) |
| channelRoute.GET("/fetch_models/:id", controller.FetchUpstreamModels) |
| channelRoute.POST("/fetch_models", controller.FetchModels) |
| channelRoute.POST("/codex/oauth/start", controller.StartCodexOAuth) |
| channelRoute.POST("/codex/oauth/complete", controller.CompleteCodexOAuth) |
| channelRoute.POST("/:id/codex/oauth/start", controller.StartCodexOAuthForChannel) |
| channelRoute.POST("/:id/codex/oauth/complete", controller.CompleteCodexOAuthForChannel) |
| channelRoute.POST("/:id/codex/refresh", controller.RefreshCodexChannelCredential) |
| channelRoute.GET("/:id/codex/usage", controller.GetCodexChannelUsage) |
| channelRoute.POST("/ollama/pull", controller.OllamaPullModel) |
| channelRoute.POST("/ollama/pull/stream", controller.OllamaPullModelStream) |
| channelRoute.DELETE("/ollama/delete", controller.OllamaDeleteModel) |
| channelRoute.GET("/ollama/version/:id", controller.OllamaVersion) |
| channelRoute.POST("/batch/tag", controller.BatchSetChannelTag) |
| channelRoute.GET("/tag/models", controller.GetTagModels) |
| channelRoute.POST("/copy/:id", controller.CopyChannel) |
| channelRoute.POST("/multi_key/manage", controller.ManageMultiKeys) |
| channelRoute.POST("/upstream_updates/apply", controller.ApplyChannelUpstreamModelUpdates) |
| channelRoute.POST("/upstream_updates/apply_all", controller.ApplyAllChannelUpstreamModelUpdates) |
| channelRoute.POST("/upstream_updates/detect", controller.DetectChannelUpstreamModelUpdates) |
| channelRoute.POST("/upstream_updates/detect_all", controller.DetectAllChannelUpstreamModelUpdates) |
| } |
| tokenRoute := apiRouter.Group("/token") |
| tokenRoute.Use(middleware.UserAuth()) |
| { |
| tokenRoute.GET("/", controller.GetAllTokens) |
| tokenRoute.GET("/search", middleware.SearchRateLimit(), controller.SearchTokens) |
| tokenRoute.GET("/:id", controller.GetToken) |
| tokenRoute.POST("/:id/key", middleware.CriticalRateLimit(), middleware.DisableCache(), controller.GetTokenKey) |
| tokenRoute.POST("/", controller.AddToken) |
| tokenRoute.PUT("/", controller.UpdateToken) |
| tokenRoute.DELETE("/:id", controller.DeleteToken) |
| tokenRoute.POST("/batch", controller.DeleteTokenBatch) |
| } |
|
|
| usageRoute := apiRouter.Group("/usage") |
| usageRoute.Use(middleware.CORS(), middleware.CriticalRateLimit()) |
| { |
| tokenUsageRoute := usageRoute.Group("/token") |
| tokenUsageRoute.Use(middleware.TokenAuthReadOnly()) |
| { |
| tokenUsageRoute.GET("/", controller.GetTokenUsage) |
| } |
| } |
|
|
| redemptionRoute := apiRouter.Group("/redemption") |
| redemptionRoute.Use(middleware.AdminAuth()) |
| { |
| redemptionRoute.GET("/", controller.GetAllRedemptions) |
| redemptionRoute.GET("/search", controller.SearchRedemptions) |
| redemptionRoute.GET("/:id", controller.GetRedemption) |
| redemptionRoute.POST("/", controller.AddRedemption) |
| redemptionRoute.PUT("/", controller.UpdateRedemption) |
| redemptionRoute.DELETE("/invalid", controller.DeleteInvalidRedemption) |
| redemptionRoute.DELETE("/:id", controller.DeleteRedemption) |
| } |
| logRoute := apiRouter.Group("/log") |
| logRoute.GET("/", middleware.AdminAuth(), controller.GetAllLogs) |
| logRoute.DELETE("/", middleware.AdminAuth(), controller.DeleteHistoryLogs) |
| logRoute.GET("/stat", middleware.AdminAuth(), controller.GetLogsStat) |
| logRoute.GET("/self/stat", middleware.UserAuth(), controller.GetLogsSelfStat) |
| logRoute.GET("/channel_affinity_usage_cache", middleware.AdminAuth(), controller.GetChannelAffinityUsageCacheStats) |
| logRoute.GET("/search", middleware.AdminAuth(), controller.SearchAllLogs) |
| logRoute.GET("/self", middleware.UserAuth(), controller.GetUserLogs) |
| logRoute.GET("/self/search", middleware.UserAuth(), middleware.SearchRateLimit(), controller.SearchUserLogs) |
|
|
| dataRoute := apiRouter.Group("/data") |
| dataRoute.GET("/", middleware.AdminAuth(), controller.GetAllQuotaDates) |
| dataRoute.GET("/self", middleware.UserAuth(), controller.GetUserQuotaDates) |
|
|
| logRoute.Use(middleware.CORS(), middleware.CriticalRateLimit()) |
| { |
| logRoute.GET("/token", middleware.TokenAuthReadOnly(), controller.GetLogByKey) |
| } |
| groupRoute := apiRouter.Group("/group") |
| groupRoute.Use(middleware.AdminAuth()) |
| { |
| groupRoute.GET("/", controller.GetGroups) |
| } |
|
|
| prefillGroupRoute := apiRouter.Group("/prefill_group") |
| prefillGroupRoute.Use(middleware.AdminAuth()) |
| { |
| prefillGroupRoute.GET("/", controller.GetPrefillGroups) |
| prefillGroupRoute.POST("/", controller.CreatePrefillGroup) |
| prefillGroupRoute.PUT("/", controller.UpdatePrefillGroup) |
| prefillGroupRoute.DELETE("/:id", controller.DeletePrefillGroup) |
| } |
|
|
| mjRoute := apiRouter.Group("/mj") |
| mjRoute.GET("/self", middleware.UserAuth(), controller.GetUserMidjourney) |
| mjRoute.GET("/", middleware.AdminAuth(), controller.GetAllMidjourney) |
|
|
| taskRoute := apiRouter.Group("/task") |
| { |
| taskRoute.GET("/self", middleware.UserAuth(), controller.GetUserTask) |
| taskRoute.GET("/", middleware.AdminAuth(), controller.GetAllTask) |
| } |
|
|
| vendorRoute := apiRouter.Group("/vendors") |
| vendorRoute.Use(middleware.AdminAuth()) |
| { |
| vendorRoute.GET("/", controller.GetAllVendors) |
| vendorRoute.GET("/search", controller.SearchVendors) |
| vendorRoute.GET("/:id", controller.GetVendorMeta) |
| vendorRoute.POST("/", controller.CreateVendorMeta) |
| vendorRoute.PUT("/", controller.UpdateVendorMeta) |
| vendorRoute.DELETE("/:id", controller.DeleteVendorMeta) |
| } |
|
|
| modelsRoute := apiRouter.Group("/models") |
| modelsRoute.Use(middleware.AdminAuth()) |
| { |
| modelsRoute.GET("/sync_upstream/preview", controller.SyncUpstreamPreview) |
| modelsRoute.POST("/sync_upstream", controller.SyncUpstreamModels) |
| modelsRoute.GET("/missing", controller.GetMissingModels) |
| modelsRoute.GET("/", controller.GetAllModelsMeta) |
| modelsRoute.GET("/search", controller.SearchModelsMeta) |
| modelsRoute.GET("/:id", controller.GetModelMeta) |
| modelsRoute.POST("/", controller.CreateModelMeta) |
| modelsRoute.PUT("/", controller.UpdateModelMeta) |
| modelsRoute.DELETE("/:id", controller.DeleteModelMeta) |
| } |
|
|
| |
| deploymentsRoute := apiRouter.Group("/deployments") |
| deploymentsRoute.Use(middleware.AdminAuth()) |
| { |
| deploymentsRoute.GET("/settings", controller.GetModelDeploymentSettings) |
| deploymentsRoute.POST("/settings/test-connection", controller.TestIoNetConnection) |
| deploymentsRoute.GET("/", controller.GetAllDeployments) |
| deploymentsRoute.GET("/search", controller.SearchDeployments) |
| deploymentsRoute.POST("/test-connection", controller.TestIoNetConnection) |
| deploymentsRoute.GET("/hardware-types", controller.GetHardwareTypes) |
| deploymentsRoute.GET("/locations", controller.GetLocations) |
| deploymentsRoute.GET("/available-replicas", controller.GetAvailableReplicas) |
| deploymentsRoute.POST("/price-estimation", controller.GetPriceEstimation) |
| deploymentsRoute.GET("/check-name", controller.CheckClusterNameAvailability) |
| deploymentsRoute.POST("/", controller.CreateDeployment) |
|
|
| deploymentsRoute.GET("/:id", controller.GetDeployment) |
| deploymentsRoute.GET("/:id/logs", controller.GetDeploymentLogs) |
| deploymentsRoute.GET("/:id/containers", controller.ListDeploymentContainers) |
| deploymentsRoute.GET("/:id/containers/:container_id", controller.GetContainerDetails) |
| deploymentsRoute.PUT("/:id", controller.UpdateDeployment) |
| deploymentsRoute.PUT("/:id/name", controller.UpdateDeploymentName) |
| deploymentsRoute.POST("/:id/extend", controller.ExtendDeployment) |
| deploymentsRoute.DELETE("/:id", controller.DeleteDeployment) |
| } |
| } |
| } |
|
|
