yoge
9d712c921a
- Migrate recognition_results table to JSON schema (meta_data + content),
replacing flat latex/markdown/mathml/mml columns
- Add TaskTypePDF constant and update all formula read/write paths
- Add PDFRecognitionService using pdftoppm (Poppler) for CGO-free page
rendering; limits processing to first 10 pages (pre-hook)
- Reuse existing downstream OCR endpoint (cloud.texpixel.com) for each
page image; stores results as [{page_number, markdown}] JSON array
- Add Redis queue + distributed lock for PDF worker goroutine
- Add REST endpoints: POST /v1/pdf/recognition, GET /v1/pdf/recognition/:task_no
- Add .pdf to OSS upload file type whitelist
- Add migrations/pdf_recognition.sql for safe data migration
Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
75 lines
2.5 KiB
Go
75 lines
2.5 KiB
Go
package api
|
|
|
|
import (
|
|
"gitea.com/texpixel/document_ai/api/v1/analytics"
|
|
"gitea.com/texpixel/document_ai/api/v1/formula"
|
|
"gitea.com/texpixel/document_ai/api/v1/oss"
|
|
"gitea.com/texpixel/document_ai/api/v1/pdf"
|
|
"gitea.com/texpixel/document_ai/api/v1/task"
|
|
"gitea.com/texpixel/document_ai/api/v1/user"
|
|
"gitea.com/texpixel/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)
|
|
formulaRouter.POST("/test_process_mathpix_task", endpoint.TestProcessMathpixTask)
|
|
}
|
|
|
|
taskRouter := v1.Group("/task", common.GetAuthMiddleware())
|
|
{
|
|
endpoint := task.NewTaskEndpoint()
|
|
taskRouter.POST("/evaluate", endpoint.EvaluateTask)
|
|
taskRouter.GET("/list", common.MustAuthMiddleware(), endpoint.GetTaskList)
|
|
taskRouter.POST("/export", endpoint.ExportTask)
|
|
}
|
|
|
|
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)
|
|
}
|
|
|
|
userEndpoint := user.NewUserEndpoint()
|
|
|
|
userRouter := v1.Group("/user")
|
|
{
|
|
userRouter.POST("/sms", userEndpoint.SendVerificationCode)
|
|
userRouter.POST("/email/code", userEndpoint.SendEmailVerifyCode)
|
|
userRouter.POST("/register", userEndpoint.RegisterByEmail)
|
|
userRouter.POST("/login", userEndpoint.LoginByEmail)
|
|
userRouter.GET("/oauth/google/url", userEndpoint.GetGoogleOAuthUrl)
|
|
userRouter.POST("/oauth/google/callback", userEndpoint.GoogleOAuthCallback)
|
|
}
|
|
|
|
userAuthRouter := v1.Group("/user", common.GetAuthMiddleware())
|
|
{
|
|
userAuthRouter.GET("/info", common.MustAuthMiddleware(), userEndpoint.GetUserInfo)
|
|
}
|
|
|
|
pdfRouter := v1.Group("/pdf", common.GetAuthMiddleware())
|
|
{
|
|
endpoint := pdf.NewPDFEndpoint()
|
|
pdfRouter.POST("/recognition", endpoint.CreateTask)
|
|
pdfRouter.GET("/recognition/:task_no", endpoint.GetTaskStatus)
|
|
}
|
|
|
|
// 数据埋点路由
|
|
analyticsRouter := v1.Group("/analytics", common.GetAuthMiddleware())
|
|
{
|
|
analyticsHandler := analytics.NewAnalyticsHandler()
|
|
analyticsRouter.POST("/track", analyticsHandler.TrackEvent)
|
|
}
|
|
}
|
|
|
|
}
|