51 lines
1.6 KiB
Go
51 lines
1.6 KiB
Go
package api
|
|
|
|
import (
|
|
"gitea.com/bitwsd/document_ai/api/v1/formula"
|
|
"gitea.com/bitwsd/document_ai/api/v1/oss"
|
|
"gitea.com/bitwsd/document_ai/api/v1/task"
|
|
"gitea.com/bitwsd/document_ai/api/v1/user"
|
|
"gitea.com/bitwsd/document_ai/pkg/common"
|
|
"github.com/gin-gonic/gin"
|
|
)
|
|
|
|
func SetupRouter(engine *gin.RouterGroup) {
|
|
v1 := engine.Group("/v1")
|
|
{
|
|
formulaRouter := v1.Group("/formula", common.GetAuthMiddleware())
|
|
{
|
|
endpoint := formula.NewFormulaEndpoint()
|
|
formulaRouter.POST("/recognition", endpoint.CreateTask)
|
|
formulaRouter.POST("/ai_enhance", endpoint.AIEnhanceRecognition)
|
|
formulaRouter.GET("/recognition/:task_no", endpoint.GetTaskStatus)
|
|
}
|
|
|
|
taskRouter := v1.Group("/task", common.GetAuthMiddleware())
|
|
{
|
|
endpoint := task.NewTaskEndpoint()
|
|
taskRouter.POST("/evaluate", endpoint.EvaluateTask)
|
|
taskRouter.GET("/list", common.MustAuthMiddleware(), endpoint.GetTaskList)
|
|
}
|
|
|
|
ossRouter := v1.Group("/oss", common.GetAuthMiddleware())
|
|
{
|
|
endpoint := oss.NewOSSEndpoint()
|
|
ossRouter.POST("/signature", endpoint.GetPostObjectSignature)
|
|
ossRouter.POST("/signature_url", endpoint.GetSignatureURL)
|
|
ossRouter.POST("/file/upload", endpoint.UploadFile)
|
|
}
|
|
|
|
userRouter := v1.Group("/user", common.GetAuthMiddleware())
|
|
{
|
|
userEndpoint := user.NewUserEndpoint()
|
|
{
|
|
userRouter.POST("/sms", userEndpoint.SendVerificationCode)
|
|
userRouter.POST("/register", userEndpoint.RegisterByEmail)
|
|
userRouter.POST("/login", userEndpoint.LoginByEmail)
|
|
userRouter.GET("/info", common.MustAuthMiddleware(), userEndpoint.GetUserInfo)
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|