72 lines
2.5 KiB
Go
72 lines
2.5 KiB
Go
package routes
|
||
|
||
import (
|
||
"github.com/gin-gonic/gin"
|
||
"hr_receiver/controllers"
|
||
"hr_receiver/middleware"
|
||
"net/http"
|
||
)
|
||
|
||
func SetupRouter() *gin.Engine {
|
||
jwtService := middleware.NewJWTService(middleware.ApiSecret, middleware.TokenExp)
|
||
r := gin.Default()
|
||
r.Use(middleware.GzipMiddleware())
|
||
trainingController := controllers.NewTrainingController()
|
||
stepTrainController := controllers.NewStepTrainingController()
|
||
lessonPlanController := controllers.NewLessonPlanController()
|
||
|
||
v1 := r.Group("/api/v1")
|
||
{
|
||
records := v1.Group("/train-records") //.Use(middleware.AuthMiddleware())
|
||
{
|
||
records.POST("", trainingController.CreateTrainingRecord)
|
||
records.GET("/analysis", trainingController.HandleCurveAnalysis)
|
||
records.POST("/analysis-by-ai", trainingController.AnalyzeByAI)
|
||
// 可扩展其他路由：GET, PUT, DELETE等
|
||
}
|
||
steps := v1.Group("/step").Use(middleware.JWTAuth(), middleware.RequireStepTrainingAccess())
|
||
{
|
||
steps.POST("", stepTrainController.CreateTrainingRecord)
|
||
steps.GET("train-records", stepTrainController.GetTrainingRecords)
|
||
steps.GET("train-data/:trainId", stepTrainController.GetTrainingRecordByTrainId)
|
||
steps.GET("train-rank/:trainId", stepTrainController.GetTrainingRank)
|
||
// 可扩展其他路由：GET, PUT, DELETE等
|
||
}
|
||
lessonPlans := v1.Group("/lesson-plans").Use(middleware.JWTAuth())
|
||
{
|
||
lessonPlans.POST("/upload", middleware.RequireHeartRateOperatorOrHigher(), lessonPlanController.Upload)
|
||
lessonPlans.GET("", middleware.RequireHeartRateOperatorOrHigher(), lessonPlanController.List)
|
||
lessonPlans.GET("/page", middleware.RequireHeartRateOperatorOrHigher(), lessonPlanController.Page)
|
||
lessonPlans.GET("/:id/download", lessonPlanController.Download)
|
||
lessonPlans.POST("/:id/share-code", lessonPlanController.GenerateShareCode)
|
||
lessonPlans.DELETE("/:id", lessonPlanController.Delete)
|
||
}
|
||
v1.GET("/lesson-plans/share/:code/download", lessonPlanController.DownloadByShareCode)
|
||
public := v1.Group("")
|
||
{
|
||
public.POST("/register", controllers.Register)
|
||
public.POST("/login", controllers.Login)
|
||
}
|
||
auth := v1.Group("/auth")
|
||
{
|
||
auth.GET("/token", func(c *gin.Context) {
|
||
|
||
clientSecret := c.GetHeader("X-API-Key")
|
||
if clientSecret != middleware.ApiSecret {
|
||
c.JSON(http.StatusUnauthorized, gin.H{"error": "invalid secret"})
|
||
return
|
||
}
|
||
|
||
token, err := jwtService.GenerateToken()
|
||
if err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"error": "failed to generate token"})
|
||
return
|
||
}
|
||
|
||
c.JSON(http.StatusOK, gin.H{"token": token})
|
||
})
|
||
}
|
||
}
|
||
return r
|
||
}
|