doc_ai_backed/api/v1/pdf/handler.go

package pdf

import (
	"net/http"
	"path/filepath"
	"strings"

	pdfmodel "gitea.com/texpixel/document_ai/internal/model/pdf"
	"gitea.com/texpixel/document_ai/internal/service"
	"gitea.com/texpixel/document_ai/pkg/common"
	"gitea.com/texpixel/document_ai/pkg/constant"

	"github.com/gin-gonic/gin"
)

type PDFEndpoint struct {
	pdfService *service.PDFRecognitionService
}

func NewPDFEndpoint() *PDFEndpoint {
	return &PDFEndpoint{
		pdfService: service.NewPDFRecognitionService(),
	}
}

// CreateTask godoc
// @Summary Create a PDF recognition task
// @Description Create a new PDF recognition task (max 10 pages processed)
// @Tags PDF
// @Accept json
// @Produce json
// @Param request body pdfmodel.CreatePDFRecognitionRequest true "Create PDF task request"
// @Success 200 {object} common.Response{data=pdfmodel.CreatePDFTaskResponse}
// @Failure 400 {object} common.Response
// @Failure 500 {object} common.Response
// @Router /v1/pdf/recognition [post]
func (e *PDFEndpoint) CreateTask(c *gin.Context) {
	var req pdfmodel.CreatePDFRecognitionRequest
	if err := c.BindJSON(&req); err != nil {
		c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeParamError, "参数错误"))
		return
	}
	req.UserID = c.GetInt64(constant.ContextUserID)

	if strings.ToLower(filepath.Ext(req.FileName)) != ".pdf" {
		c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeParamError, "仅支持PDF文件"))
		return
	}

	task, err := e.pdfService.CreatePDFTask(c, &req)
	if err != nil {
		if bizErr, ok := err.(*common.BusinessError); ok {
			c.JSON(http.StatusOK, common.ErrorResponse(c, int(bizErr.Code), bizErr.Message))
			return
		}
		c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeSystemError, "创建任务失败"))
		return
	}

	c.JSON(http.StatusOK, common.SuccessResponse(c, &pdfmodel.CreatePDFTaskResponse{
		TaskNo: task.TaskUUID,
		Status: int(task.Status),
	}))
}

// GetTaskStatus godoc
// @Summary Get PDF recognition task status and results
// @Description Poll task status; pages field populated when status=2 (completed)
// @Tags PDF
// @Accept json
// @Produce json
// @Param task_no path string true "Task No"
// @Success 200 {object} common.Response{data=pdfmodel.GetPDFTaskResponse}
// @Failure 404 {object} common.Response
// @Failure 500 {object} common.Response
// @Router /v1/pdf/recognition/{task_no} [get]
func (e *PDFEndpoint) GetTaskStatus(c *gin.Context) {
	var req pdfmodel.GetPDFTaskRequest
	if err := c.ShouldBindUri(&req); err != nil {
		c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeParamError, "参数错误"))
		return
	}

	resp, err := e.pdfService.GetPDFTask(c, req.TaskNo)
	if err != nil {
		if bizErr, ok := err.(*common.BusinessError); ok {
			c.JSON(http.StatusOK, common.ErrorResponse(c, int(bizErr.Code), bizErr.Message))
			return
		}
		c.JSON(http.StatusOK, common.ErrorResponse(c, common.CodeSystemError, "查询任务失败"))
		return
	}

	c.JSON(http.StatusOK, common.SuccessResponse(c, resp))
}