25'ten fazla konu seçemezsiniz Konular bir harf veya rakamla başlamalı, kısa çizgiler ('-') içerebilir ve en fazla 35 karakter uzunluğunda olabilir.
 
 
 

451 satır
24 KiB

  1. package router
  2. import (
  3. "github.com/QuantumNous/new-api/controller"
  4. "github.com/QuantumNous/new-api/middleware"
  5. // Import oauth package to register providers via init()
  6. _ "github.com/QuantumNous/new-api/oauth"
  7. "github.com/gin-contrib/gzip"
  8. "github.com/gin-gonic/gin"
  9. )
  10. func SetApiRouter(router *gin.Engine) {
  11. apiRouter := router.Group("/api")
  12. apiRouter.Use(gzip.Gzip(gzip.DefaultCompression))
  13. apiRouter.Use(middleware.BodyStorageCleanup()) // 清理请求体存储
  14. apiRouter.Use(middleware.GlobalAPIRateLimit())
  15. {
  16. apiRouter.GET("/setup", controller.GetSetup)
  17. apiRouter.POST("/setup", controller.PostSetup)
  18. apiRouter.GET("/status", controller.GetStatus)
  19. apiRouter.GET("/uptime/status", controller.GetUptimeKumaStatus)
  20. apiRouter.GET("/models", middleware.UserAuth(), controller.DashboardListModels)
  21. apiRouter.GET("/status/test", middleware.AdminAuth(), controller.TestStatus)
  22. apiRouter.GET("/notice", controller.GetNotice)
  23. apiRouter.GET("/user-agreement", controller.GetUserAgreement)
  24. apiRouter.GET("/privacy-policy", controller.GetPrivacyPolicy)
  25. apiRouter.GET("/terms", controller.GetTermsOfService)
  26. apiRouter.GET("/usage-policy", controller.GetUsagePolicy)
  27. apiRouter.GET("/about", controller.GetAbout)
  28. //apiRouter.GET("/midjourney", controller.GetMidjourney)
  29. apiRouter.GET("/home_page_content", controller.GetHomePageContent)
  30. apiRouter.GET("/pricing", middleware.TryUserAuth(), controller.GetPricing)
  31. apiRouter.GET("/pricing/user/*model", middleware.TryUserAuth(), controller.GetUserPricing)
  32. apiRouter.GET("/captcha", controller.GetCaptcha)
  33. apiRouter.GET("/verification", middleware.EmailVerificationRateLimit(), middleware.TurnstileCheck(), controller.SendEmailVerification)
  34. apiRouter.GET("/reset_password", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.SendPasswordResetEmail)
  35. apiRouter.POST("/user/reset", middleware.CriticalRateLimit(), controller.ResetPassword)
  36. // OAuth routes - specific routes must come before :provider wildcard
  37. apiRouter.GET("/oauth/state", middleware.CriticalRateLimit(), controller.GenerateOAuthCode)
  38. apiRouter.GET("/oauth/email/bind", middleware.CriticalRateLimit(), controller.EmailBind)
  39. // Non-standard OAuth (WeChat, Telegram) - keep original routes
  40. apiRouter.GET("/oauth/wechat", middleware.CriticalRateLimit(), controller.WeChatAuth)
  41. apiRouter.GET("/oauth/wechat/bind", middleware.CriticalRateLimit(), controller.WeChatBind)
  42. apiRouter.GET("/oauth/telegram/login", middleware.CriticalRateLimit(), controller.TelegramLogin)
  43. apiRouter.GET("/oauth/telegram/bind", middleware.CriticalRateLimit(), controller.TelegramBind)
  44. // Standard OAuth providers (GitHub, Discord, OIDC, LinuxDO) - unified route
  45. apiRouter.GET("/oauth/:provider", middleware.CriticalRateLimit(), controller.HandleOAuth)
  46. apiRouter.GET("/ratio_config", middleware.CriticalRateLimit(), controller.GetRatioConfig)
  47. apiRouter.GET("/playground/config", controller.GetPlaygroundConfig)
  48. assetRoute := apiRouter.Group("/v1/volcengine")
  49. assetRoute.Use(middleware.SystemPerformanceCheck(), middleware.TokenAuth())
  50. {
  51. assetRoute.POST("/asset", controller.DoubaoAssetProxy)
  52. }
  53. apiRouter.POST("/stripe/webhook", controller.StripeWebhook)
  54. apiRouter.POST("/creem/webhook", controller.CreemWebhook)
  55. apiRouter.POST("/wechat/pay/webhook", controller.WechatPayWebhook)
  56. apiRouter.POST("/alipay/pay/webhook", controller.AlipayPayWebhook)
  57. // Universal secure verification routes
  58. apiRouter.POST("/verify", middleware.UserAuth(), middleware.CriticalRateLimit(), controller.UniversalVerify)
  59. userRoute := apiRouter.Group("/user")
  60. userRoute.Use(middleware.CORS())
  61. {
  62. userRoute.POST("/register", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.Register)
  63. userRoute.POST("/login", middleware.CriticalRateLimit(), middleware.TurnstileCheck(), controller.Login)
  64. userRoute.POST("/login/2fa", middleware.CriticalRateLimit(), controller.Verify2FALogin)
  65. userRoute.POST("/passkey/login/begin", middleware.CriticalRateLimit(), controller.PasskeyLoginBegin)
  66. userRoute.POST("/passkey/login/finish", middleware.CriticalRateLimit(), controller.PasskeyLoginFinish)
  67. //userRoute.POST("/tokenlog", middleware.CriticalRateLimit(), controller.TokenLog)
  68. userRoute.GET("/logout", controller.Logout)
  69. userRoute.POST("/epay/notify", controller.EpayNotify)
  70. userRoute.GET("/epay/notify", controller.EpayNotify)
  71. userRoute.GET("/groups", controller.GetUserGroups)
  72. selfRoute := userRoute.Group("/")
  73. selfRoute.Use(middleware.UserAuth())
  74. {
  75. selfRoute.GET("/self/groups", controller.GetUserGroups)
  76. selfRoute.GET("/self", controller.GetSelf)
  77. selfRoute.GET("/models", controller.GetUserModels)
  78. selfRoute.PUT("/self", controller.UpdateSelf)
  79. selfRoute.DELETE("/self", controller.DeleteSelf)
  80. selfRoute.GET("/token", controller.GenerateAccessToken)
  81. selfRoute.GET("/passkey", controller.PasskeyStatus)
  82. selfRoute.POST("/passkey/register/begin", controller.PasskeyRegisterBegin)
  83. selfRoute.POST("/passkey/register/finish", controller.PasskeyRegisterFinish)
  84. selfRoute.POST("/passkey/verify/begin", controller.PasskeyVerifyBegin)
  85. selfRoute.POST("/passkey/verify/finish", controller.PasskeyVerifyFinish)
  86. selfRoute.DELETE("/passkey", controller.PasskeyDelete)
  87. selfRoute.GET("/aff", controller.GetAffCode)
  88. selfRoute.GET("/topup/info", controller.GetTopUpInfo)
  89. selfRoute.GET("/topup/self", controller.GetUserTopUps)
  90. selfRoute.POST("/topup", middleware.CriticalRateLimit(), controller.TopUp)
  91. selfRoute.POST("/pay", middleware.CriticalRateLimit(), controller.RequestEpay)
  92. selfRoute.POST("/amount", controller.RequestAmount)
  93. selfRoute.POST("/stripe/pay", middleware.CriticalRateLimit(), controller.RequestStripePay)
  94. selfRoute.POST("/stripe/amount", controller.RequestStripeAmount)
  95. selfRoute.POST("/creem/pay", middleware.CriticalRateLimit(), controller.RequestCreemPay)
  96. selfRoute.POST("/wechat/pay/amount", controller.RequestWechatPayAmount)
  97. selfRoute.POST("/wechat/pay", controller.RequestWechatPay)
  98. selfRoute.GET("/wechat/pay/status", controller.WechatPayStatus)
  99. selfRoute.POST("/alipay/pay/amount", controller.RequestAlipayPayAmount)
  100. selfRoute.POST("/alipay/pay", controller.RequestAlipayPay)
  101. selfRoute.GET("/alipay/pay/status", controller.AlipayPayStatus)
  102. selfRoute.POST("/aff_transfer", controller.TransferAffQuota)
  103. selfRoute.PUT("/setting", controller.UpdateUserSetting)
  104. // 2FA routes
  105. selfRoute.GET("/2fa/status", controller.Get2FAStatus)
  106. selfRoute.POST("/2fa/setup", controller.Setup2FA)
  107. selfRoute.POST("/2fa/enable", controller.Enable2FA)
  108. selfRoute.POST("/2fa/disable", controller.Disable2FA)
  109. selfRoute.POST("/2fa/backup_codes", controller.RegenerateBackupCodes)
  110. // Check-in routes
  111. selfRoute.GET("/checkin", controller.GetCheckinStatus)
  112. selfRoute.POST("/checkin", middleware.TurnstileCheck(), controller.DoCheckin)
  113. // Custom OAuth bindings
  114. selfRoute.GET("/oauth/bindings", controller.GetUserOAuthBindings)
  115. selfRoute.DELETE("/oauth/bindings/:provider_id", controller.UnbindCustomOAuth)
  116. }
  117. adminRoute := userRoute.Group("/")
  118. adminRoute.Use(middleware.AdminAuth())
  119. {
  120. adminRoute.GET("/", controller.GetAllUsers)
  121. adminRoute.GET("/topup", controller.GetAllTopUps)
  122. adminRoute.POST("/topup/complete", controller.AdminCompleteTopUp)
  123. adminRoute.GET("/search", controller.SearchUsers)
  124. adminRoute.GET("/:id/oauth/bindings", controller.GetUserOAuthBindingsByAdmin)
  125. adminRoute.DELETE("/:id/oauth/bindings/:provider_id", controller.UnbindCustomOAuthByAdmin)
  126. adminRoute.DELETE("/:id/bindings/:binding_type", controller.AdminClearUserBinding)
  127. adminRoute.GET("/:id/video-channel-bindings", controller.GetUserVideoChannelBindings)
  128. adminRoute.PUT("/:id/video-channel-bindings", controller.SetUserVideoChannelBindings)
  129. adminRoute.GET("/:id", controller.GetUser)
  130. adminRoute.POST("/", controller.CreateUser)
  131. adminRoute.POST("/manage", controller.ManageUser)
  132. adminRoute.PUT("/", controller.UpdateUser)
  133. adminRoute.DELETE("/:id", controller.DeleteUser)
  134. adminRoute.DELETE("/:id/reset_passkey", controller.AdminResetPasskey)
  135. // Admin 2FA routes
  136. adminRoute.GET("/2fa/stats", controller.Admin2FAStats)
  137. adminRoute.DELETE("/:id/2fa", controller.AdminDisable2FA)
  138. // User model RPM rate limit routes
  139. adminRoute.GET("/:id/rate-limits", controller.GetUserRateLimits)
  140. adminRoute.PUT("/:id/rate-limits", controller.SetUserRateLimits)
  141. }
  142. }
  143. // Subscription billing (plans, purchase, admin management)
  144. subscriptionRoute := apiRouter.Group("/subscription")
  145. subscriptionRoute.Use(middleware.UserAuth())
  146. {
  147. subscriptionRoute.GET("/plans", controller.GetSubscriptionPlans)
  148. subscriptionRoute.GET("/self", controller.GetSubscriptionSelf)
  149. subscriptionRoute.PUT("/self/preference", controller.UpdateSubscriptionPreference)
  150. subscriptionRoute.POST("/epay/pay", middleware.CriticalRateLimit(), controller.SubscriptionRequestEpay)
  151. subscriptionRoute.POST("/stripe/pay", middleware.CriticalRateLimit(), controller.SubscriptionRequestStripePay)
  152. subscriptionRoute.POST("/creem/pay", middleware.CriticalRateLimit(), controller.SubscriptionRequestCreemPay)
  153. subscriptionRoute.POST("/wechat/pay", middleware.CriticalRateLimit(), controller.SubscriptionRequestWechatPay)
  154. }
  155. subscriptionAdminRoute := apiRouter.Group("/subscription/admin")
  156. subscriptionAdminRoute.Use(middleware.AdminAuth())
  157. {
  158. subscriptionAdminRoute.GET("/plans", controller.AdminListSubscriptionPlans)
  159. subscriptionAdminRoute.POST("/plans", controller.AdminCreateSubscriptionPlan)
  160. subscriptionAdminRoute.PUT("/plans/:id", controller.AdminUpdateSubscriptionPlan)
  161. subscriptionAdminRoute.PATCH("/plans/:id", controller.AdminUpdateSubscriptionPlanStatus)
  162. subscriptionAdminRoute.POST("/bind", controller.AdminBindSubscription)
  163. // User subscription management (admin)
  164. subscriptionAdminRoute.GET("/users/:id/subscriptions", controller.AdminListUserSubscriptions)
  165. subscriptionAdminRoute.POST("/users/:id/subscriptions", controller.AdminCreateUserSubscription)
  166. subscriptionAdminRoute.POST("/user_subscriptions/:id/invalidate", controller.AdminInvalidateUserSubscription)
  167. subscriptionAdminRoute.DELETE("/user_subscriptions/:id", controller.AdminDeleteUserSubscription)
  168. }
  169. // Subscription payment callbacks (no auth)
  170. apiRouter.POST("/subscription/epay/notify", controller.SubscriptionEpayNotify)
  171. apiRouter.GET("/subscription/epay/notify", controller.SubscriptionEpayNotify)
  172. apiRouter.GET("/subscription/epay/return", controller.SubscriptionEpayReturn)
  173. apiRouter.POST("/subscription/epay/return", controller.SubscriptionEpayReturn)
  174. optionRoute := apiRouter.Group("/option")
  175. optionRoute.Use(middleware.RootAuth())
  176. {
  177. optionRoute.GET("/", controller.GetOptions)
  178. optionRoute.PUT("/", controller.UpdateOption)
  179. optionRoute.GET("/model_pricing", controller.GetModelPricingRules)
  180. optionRoute.GET("/model_pricing/*model", controller.GetModelPricingRule)
  181. optionRoute.PUT("/model_pricing/*model", controller.UpdateModelPricingRule)
  182. optionRoute.DELETE("/model_pricing/*model", controller.DeleteModelPricingRule)
  183. optionRoute.GET("/model_display_pricing", controller.GetModelDisplayPricingRules)
  184. optionRoute.GET("/model_display_pricing/*model", controller.GetModelDisplayPricingRule)
  185. optionRoute.PUT("/model_display_pricing/*model", controller.UpdateModelDisplayPricingRule)
  186. optionRoute.DELETE("/model_display_pricing/*model", controller.DeleteModelDisplayPricingRule)
  187. optionRoute.GET("/channel_affinity_cache", controller.GetChannelAffinityCacheStats)
  188. optionRoute.DELETE("/channel_affinity_cache", controller.ClearChannelAffinityCache)
  189. optionRoute.POST("/rest_model_ratio", controller.ResetModelRatio)
  190. optionRoute.POST("/migrate_console_setting", controller.MigrateConsoleSetting) // 用于迁移检测的旧键,下个版本会删除
  191. }
  192. // 邮箱后缀额度规则路由(管理员权限)
  193. emailQuotaRuleRoute := apiRouter.Group("/email_quota_rule")
  194. emailQuotaRuleRoute.Use(middleware.AdminAuth())
  195. {
  196. emailQuotaRuleRoute.GET("/", controller.GetAllEmailQuotaRules)
  197. emailQuotaRuleRoute.POST("/", controller.CreateEmailQuotaRule)
  198. emailQuotaRuleRoute.PUT("/:id", controller.UpdateEmailQuotaRule)
  199. emailQuotaRuleRoute.DELETE("/:id", controller.DeleteEmailQuotaRule)
  200. }
  201. // Custom OAuth provider management (root only)
  202. customOAuthRoute := apiRouter.Group("/custom-oauth-provider")
  203. customOAuthRoute.Use(middleware.RootAuth())
  204. {
  205. customOAuthRoute.POST("/discovery", controller.FetchCustomOAuthDiscovery)
  206. customOAuthRoute.GET("/", controller.GetCustomOAuthProviders)
  207. customOAuthRoute.GET("/:id", controller.GetCustomOAuthProvider)
  208. customOAuthRoute.POST("/", controller.CreateCustomOAuthProvider)
  209. customOAuthRoute.PUT("/:id", controller.UpdateCustomOAuthProvider)
  210. customOAuthRoute.DELETE("/:id", controller.DeleteCustomOAuthProvider)
  211. }
  212. performanceRoute := apiRouter.Group("/performance")
  213. performanceRoute.Use(middleware.RootAuth())
  214. {
  215. performanceRoute.GET("/stats", controller.GetPerformanceStats)
  216. performanceRoute.DELETE("/disk_cache", controller.ClearDiskCache)
  217. performanceRoute.POST("/reset_stats", controller.ResetPerformanceStats)
  218. performanceRoute.POST("/gc", controller.ForceGC)
  219. }
  220. ratioSyncRoute := apiRouter.Group("/ratio_sync")
  221. ratioSyncRoute.Use(middleware.RootAuth())
  222. {
  223. ratioSyncRoute.GET("/channels", controller.GetSyncableChannels)
  224. ratioSyncRoute.POST("/fetch", controller.FetchUpstreamRatios)
  225. }
  226. channelRoute := apiRouter.Group("/channel")
  227. channelRoute.Use(middleware.AdminAuth())
  228. {
  229. channelRoute.GET("/", controller.GetAllChannels)
  230. channelRoute.GET("/search", controller.SearchChannels)
  231. channelRoute.GET("/models", controller.ChannelListModels)
  232. channelRoute.GET("/models_enabled", controller.EnabledListModels)
  233. channelRoute.GET("/:id", controller.GetChannel)
  234. channelRoute.POST("/:id/key", middleware.RootAuth(), middleware.CriticalRateLimit(), middleware.DisableCache(), middleware.SecureVerificationRequired(), controller.GetChannelKey)
  235. channelRoute.GET("/test", controller.TestAllChannels)
  236. channelRoute.GET("/test/:id", controller.TestChannel)
  237. channelRoute.GET("/update_balance", controller.UpdateAllChannelsBalance)
  238. channelRoute.GET("/update_balance/:id", controller.UpdateChannelBalance)
  239. channelRoute.POST("/", controller.AddChannel)
  240. channelRoute.PUT("/", controller.UpdateChannel)
  241. channelRoute.DELETE("/disabled", controller.DeleteDisabledChannel)
  242. channelRoute.POST("/tag/disabled", controller.DisableTagChannels)
  243. channelRoute.POST("/tag/enabled", controller.EnableTagChannels)
  244. channelRoute.PUT("/tag", controller.EditTagChannels)
  245. channelRoute.DELETE("/:id", controller.DeleteChannel)
  246. channelRoute.POST("/batch", controller.DeleteChannelBatch)
  247. channelRoute.POST("/fix", controller.FixChannelsAbilities)
  248. channelRoute.GET("/fetch_models/:id", controller.FetchUpstreamModels)
  249. channelRoute.POST("/fetch_models", middleware.RootAuth(), controller.FetchModels)
  250. channelRoute.POST("/codex/oauth/start", controller.StartCodexOAuth)
  251. channelRoute.POST("/codex/oauth/complete", controller.CompleteCodexOAuth)
  252. channelRoute.POST("/:id/codex/oauth/start", controller.StartCodexOAuthForChannel)
  253. channelRoute.POST("/:id/codex/oauth/complete", controller.CompleteCodexOAuthForChannel)
  254. channelRoute.POST("/:id/codex/refresh", controller.RefreshCodexChannelCredential)
  255. channelRoute.GET("/:id/codex/usage", controller.GetCodexChannelUsage)
  256. channelRoute.POST("/ollama/pull", controller.OllamaPullModel)
  257. channelRoute.POST("/ollama/pull/stream", controller.OllamaPullModelStream)
  258. channelRoute.DELETE("/ollama/delete", controller.OllamaDeleteModel)
  259. channelRoute.GET("/ollama/version/:id", controller.OllamaVersion)
  260. channelRoute.POST("/batch/tag", controller.BatchSetChannelTag)
  261. channelRoute.GET("/tag/models", controller.GetTagModels)
  262. channelRoute.POST("/copy/:id", controller.CopyChannel)
  263. channelRoute.POST("/multi_key/manage", controller.ManageMultiKeys)
  264. }
  265. tokenRoute := apiRouter.Group("/token")
  266. tokenRoute.Use(middleware.UserAuth())
  267. {
  268. tokenRoute.GET("/", controller.GetAllTokens)
  269. tokenRoute.GET("/search", middleware.SearchRateLimit(), controller.SearchTokens)
  270. tokenRoute.GET("/:id", controller.GetToken)
  271. tokenRoute.POST("/", controller.AddToken)
  272. tokenRoute.PUT("/", controller.UpdateToken)
  273. tokenRoute.DELETE("/:id", controller.DeleteToken)
  274. tokenRoute.POST("/batch", controller.DeleteTokenBatch)
  275. }
  276. usageRoute := apiRouter.Group("/usage")
  277. usageRoute.Use(middleware.CORS(), middleware.CriticalRateLimit())
  278. {
  279. tokenUsageRoute := usageRoute.Group("/token")
  280. tokenUsageRoute.Use(middleware.TokenAuthReadOnly())
  281. {
  282. tokenUsageRoute.GET("/", controller.GetTokenUsage)
  283. }
  284. }
  285. redemptionRoute := apiRouter.Group("/redemption")
  286. redemptionRoute.Use(middleware.AdminAuth())
  287. {
  288. redemptionRoute.GET("/", controller.GetAllRedemptions)
  289. redemptionRoute.GET("/search", controller.SearchRedemptions)
  290. redemptionRoute.GET("/:id", controller.GetRedemption)
  291. redemptionRoute.POST("/", controller.AddRedemption)
  292. redemptionRoute.PUT("/", controller.UpdateRedemption)
  293. redemptionRoute.DELETE("/invalid", controller.DeleteInvalidRedemption)
  294. redemptionRoute.DELETE("/:id", controller.DeleteRedemption)
  295. }
  296. logRoute := apiRouter.Group("/log")
  297. logRoute.GET("/", middleware.AdminAuth(), controller.GetAllLogs)
  298. logRoute.DELETE("/", middleware.AdminAuth(), controller.DeleteHistoryLogs)
  299. logRoute.GET("/stat", middleware.AdminAuth(), controller.GetLogsStat)
  300. logRoute.GET("/self/stat", middleware.UserAuth(), controller.GetLogsSelfStat)
  301. logRoute.GET("/channel_affinity_usage_cache", middleware.AdminAuth(), controller.GetChannelAffinityUsageCacheStats)
  302. logRoute.GET("/search", middleware.AdminAuth(), controller.SearchAllLogs)
  303. logRoute.GET("/self", middleware.UserAuth(), controller.GetUserLogs)
  304. logRoute.GET("/self/search", middleware.UserAuth(), middleware.SearchRateLimit(), controller.SearchUserLogs)
  305. dataRoute := apiRouter.Group("/data")
  306. dataRoute.GET("/", middleware.AdminAuth(), controller.GetAllQuotaDates)
  307. dataRoute.GET("/self", middleware.UserAuth(), controller.GetUserQuotaDates)
  308. logRoute.Use(middleware.CORS(), middleware.CriticalRateLimit())
  309. {
  310. logRoute.GET("/token", middleware.TokenAuthReadOnly(), controller.GetLogByKey)
  311. }
  312. groupRoute := apiRouter.Group("/group")
  313. groupRoute.Use(middleware.AdminAuth())
  314. {
  315. groupRoute.GET("/", controller.GetGroups)
  316. }
  317. prefillGroupRoute := apiRouter.Group("/prefill_group")
  318. prefillGroupRoute.Use(middleware.AdminAuth())
  319. {
  320. prefillGroupRoute.GET("/", controller.GetPrefillGroups)
  321. prefillGroupRoute.POST("/", controller.CreatePrefillGroup)
  322. prefillGroupRoute.PUT("/", controller.UpdatePrefillGroup)
  323. prefillGroupRoute.DELETE("/:id", controller.DeletePrefillGroup)
  324. }
  325. mjRoute := apiRouter.Group("/mj")
  326. mjRoute.GET("/self", middleware.UserAuth(), controller.GetUserMidjourney)
  327. mjRoute.GET("/", middleware.AdminAuth(), controller.GetAllMidjourney)
  328. taskRoute := apiRouter.Group("/task")
  329. {
  330. taskRoute.GET("/self", middleware.UserAuth(), controller.GetUserTask)
  331. taskRoute.GET("/", middleware.AdminAuth(), controller.GetAllTask)
  332. }
  333. vendorRoute := apiRouter.Group("/vendors")
  334. vendorRoute.Use(middleware.AdminAuth())
  335. {
  336. vendorRoute.GET("/", controller.GetAllVendors)
  337. vendorRoute.GET("/search", controller.SearchVendors)
  338. vendorRoute.GET("/:id", controller.GetVendorMeta)
  339. vendorRoute.POST("/", controller.CreateVendorMeta)
  340. vendorRoute.PUT("/", controller.UpdateVendorMeta)
  341. vendorRoute.PUT("/reorder", controller.ReorderVendors)
  342. vendorRoute.DELETE("/:id", controller.DeleteVendorMeta)
  343. }
  344. modelsRoute := apiRouter.Group("/models")
  345. modelsRoute.Use(middleware.AdminAuth())
  346. {
  347. modelsRoute.GET("/sync_upstream/preview", controller.SyncUpstreamPreview)
  348. modelsRoute.POST("/sync_upstream", controller.SyncUpstreamModels)
  349. modelsRoute.GET("/missing", controller.GetMissingModels)
  350. modelsRoute.GET("/", controller.GetAllModelsMeta)
  351. modelsRoute.GET("/search", controller.SearchModelsMeta)
  352. modelsRoute.GET("/:id", controller.GetModelMeta)
  353. modelsRoute.POST("/", controller.CreateModelMeta)
  354. modelsRoute.PUT("/", controller.UpdateModelMeta)
  355. modelsRoute.PUT("/reorder", controller.ReorderModels)
  356. modelsRoute.DELETE("/:id", controller.DeleteModelMeta)
  357. }
  358. // Deployments (model deployment management)
  359. deploymentsRoute := apiRouter.Group("/deployments")
  360. deploymentsRoute.Use(middleware.AdminAuth())
  361. {
  362. deploymentsRoute.GET("/settings", controller.GetModelDeploymentSettings)
  363. deploymentsRoute.POST("/settings/test-connection", controller.TestIoNetConnection)
  364. deploymentsRoute.GET("/", controller.GetAllDeployments)
  365. deploymentsRoute.GET("/search", controller.SearchDeployments)
  366. deploymentsRoute.POST("/test-connection", controller.TestIoNetConnection)
  367. deploymentsRoute.GET("/hardware-types", controller.GetHardwareTypes)
  368. deploymentsRoute.GET("/locations", controller.GetLocations)
  369. deploymentsRoute.GET("/available-replicas", controller.GetAvailableReplicas)
  370. deploymentsRoute.POST("/price-estimation", controller.GetPriceEstimation)
  371. deploymentsRoute.GET("/check-name", controller.CheckClusterNameAvailability)
  372. deploymentsRoute.POST("/", controller.CreateDeployment)
  373. deploymentsRoute.GET("/:id", controller.GetDeployment)
  374. deploymentsRoute.GET("/:id/logs", controller.GetDeploymentLogs)
  375. deploymentsRoute.GET("/:id/containers", controller.ListDeploymentContainers)
  376. deploymentsRoute.GET("/:id/containers/:container_id", controller.GetContainerDetails)
  377. deploymentsRoute.PUT("/:id", controller.UpdateDeployment)
  378. deploymentsRoute.PUT("/:id/name", controller.UpdateDeploymentName)
  379. deploymentsRoute.POST("/:id/extend", controller.ExtendDeployment)
  380. deploymentsRoute.DELETE("/:id", controller.DeleteDeployment)
  381. }
  382. // Region sync API routes (for inter-node communication)
  383. syncRoute := apiRouter.Group("/internal/sync")
  384. syncRoute.Use(middleware.SyncAuth())
  385. {
  386. syncRoute.POST("/user/create", controller.ReceiveSyncedUserCreate)
  387. syncRoute.POST("/quota/update", controller.ReceiveQuotaUpdate)
  388. syncRoute.POST("/quota/query", controller.QueryUserQuota)
  389. syncRoute.POST("/quota/batch-query", controller.BatchQueryUserQuota)
  390. syncRoute.POST("/quota/batch-deduct", controller.BatchDeductQuota)
  391. syncRoute.GET("/config", controller.GetSyncConfig)
  392. }
  393. migrationRoute := apiRouter.Group("/internal/migration")
  394. migrationRoute.Use(middleware.SyncAuth())
  395. {
  396. migrationRoute.GET("/users/check-synced-copy", controller.CheckSyncedCopy)
  397. migrationRoute.GET("/users", controller.ListMigrationUsers)
  398. migrationRoute.POST("/users/query", controller.QueryMigrationUsers)
  399. migrationRoute.GET("/users/:id", controller.GetMigrationUser)
  400. migrationRoute.POST("/users/:id/convert-to-synced", controller.ConvertMigrationUserToSynced)
  401. }
  402. userMigrationRoute := apiRouter.Group("/user-migrations")
  403. userMigrationRoute.Use(middleware.RootAuth(), middleware.MasterNodeOnly())
  404. {
  405. userMigrationRoute.GET("/candidate-users", controller.ListMigrationCandidateUsers)
  406. userMigrationRoute.POST("/batches", controller.CreateUserMigrationBatch)
  407. userMigrationRoute.GET("/batches", controller.ListUserMigrationBatches)
  408. userMigrationRoute.GET("/batches/:id", controller.GetUserMigrationBatch)
  409. userMigrationRoute.POST("/batches/:id/scan", controller.ScanUserMigrationBatch)
  410. userMigrationRoute.POST("/batches/:id/execute", controller.ExecuteUserMigrationBatch)
  411. userMigrationRoute.POST("/batches/:id/retry", controller.RetryUserMigrationBatch)
  412. userMigrationRoute.POST("/batches/:id/cancel", controller.CancelUserMigrationBatch)
  413. userMigrationRoute.POST("/batches/:id/verify", controller.VerifyUserMigrationBatch)
  414. userMigrationRoute.POST("/items/:id/resolve", controller.ResolveUserMigrationItem)
  415. }
  416. }
  417. }