init repo

api/v1/oss/handler.go

@@ -0,0 +1,106 @@
+package oss
+
+import (
+	"fmt"
+	"net/http"
+	"os"
+	"path/filepath"
+	"time"
+
+	"gitea.com/bitwsd/document_ai/config"
+	"gitea.com/bitwsd/document_ai/internal/storage/dao"
+	"gitea.com/bitwsd/document_ai/pkg/common"
+	"gitea.com/bitwsd/document_ai/pkg/constant"
+	"gitea.com/bitwsd/document_ai/pkg/oss"
+	"gitea.com/bitwsd/document_ai/pkg/utils"
+	"github.com/gin-gonic/gin"
+	"gorm.io/gorm"
+)
+
+func GetPostObjectSignature(ctx *gin.Context) {
+	policyToken, err := oss.GetPolicyToken()
+	if err != nil {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeSystemError, err.Error()))
+		return
+	}
+	ctx.JSON(http.StatusOK, common.SuccessResponse(ctx, policyToken))
+}
+
+// GetSignatureURL handles the request to generate a signed URL for a file upload.
+// @Summary Generate signed URL for file upload
+// @Description This endpoint generates a signed URL for uploading a file to the OSS (Object Storage Service).
+// @Tags file
+// @Accept json
+// @Produce json
+// @Param hash query string true "Hash value of the file"
+// @Success 200 {object} common.Response{data=map[string]string{"sign_url":string, "repeat":bool, "path":string}} "Signed URL generated successfully"
+// @Failure 200 {object} common.Response "Error response"
+// @Router /signature_url [get]
+func GetSignatureURL(ctx *gin.Context) {
+	userID := ctx.GetInt64(constant.ContextUserID)
+	type Req struct {
+		FileHash string `json:"file_hash" binding:"required"`
+		FileName string `json:"file_name" binding:"required"`
+	}
+	req := Req{}
+	err := ctx.BindJSON(&req)
+	if err != nil {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeParamError, "param error"))
+		return
+	}
+	taskDao := dao.NewRecognitionTaskDao()
+	sess := dao.DB.WithContext(ctx)
+	task, err := taskDao.GetTaskByFileURL(sess, userID, req.FileHash)
+	if err != nil && err != gorm.ErrRecordNotFound {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeDBError, "failed to get task"))
+		return
+	}
+	if task.ID != 0 {
+		ctx.JSON(http.StatusOK, common.SuccessResponse(ctx, gin.H{"sign_url": "", "repeat": true, "path": task.FileURL}))
+		return
+	}
+	extend := filepath.Ext(req.FileName)
+	if extend == "" {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeParamError, "invalid file name"))
+		return
+	}
+	if !utils.InArray(extend, []string{".jpg", ".jpeg", ".png", ".gif", ".bmp", ".tiff", ".webp"}) {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeParamError, "invalid file type"))
+		return
+	}
+	path := filepath.Join(oss.FormulaDir, fmt.Sprintf("%s%s", utils.NewUUID(), extend))
+	url, err := oss.GetPolicyURL(ctx, path)
+	if err != nil {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeSystemError, err.Error()))
+		return
+	}
+	ctx.JSON(http.StatusOK, common.SuccessResponse(ctx, gin.H{"sign_url": url, "repeat": false, "path": path}))
+}
+
+func UploadFile(ctx *gin.Context) {
+	if err := os.MkdirAll(config.GlobalConfig.UploadDir, 0755); err != nil {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeSystemError, "Failed to create upload directory"))
+		return
+	}
+
+	// Get file from form
+	file, err := ctx.FormFile("file")
+	if err != nil {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeParamError, "File is required"))
+		return
+	}
+
+	// Generate unique filename with timestamp
+	ext := filepath.Ext(file.Filename)
+	filename := fmt.Sprintf("%d%s", time.Now().UnixNano(), ext)
+	filepath := filepath.Join(config.GlobalConfig.UploadDir, filename)
+
+	// Save file
+	if err := ctx.SaveUploadedFile(file, filepath); err != nil {
+		ctx.JSON(http.StatusOK, common.ErrorResponse(ctx, common.CodeSystemError, "Failed to save file"))
+		return
+	}
+
+	// Return success with file path
+	ctx.JSON(http.StatusOK, common.SuccessResponse(ctx, gin.H{"sign_url": filepath}))
+}

api/v1/oss/router.go

@@ -0,0 +1,12 @@
+package oss
+
+import "github.com/gin-gonic/gin"
+
+func SetupRouter(parent *gin.RouterGroup) {
+	router := parent.Group("oss")
+	{
+		router.POST("/signature", GetPostObjectSignature)
+		router.POST("/signature_url", GetSignatureURL)
+		router.POST("/file/upload", UploadFile)
+	}
+}