complete backend

backend/app/http/admin/uploads.go

@@ -14,10 +14,13 @@ import (
 	"quyun/providers/ali"
 	"quyun/providers/app"
 
+	"github.com/go-jet/jet/v2/qrm"
 	"github.com/gofiber/fiber/v3"
 	log "github.com/sirupsen/logrus"
 )
 
+const UPLOAD_PATH = "quyun"
+
 // @provider
 type uploads struct {
 	app *app.Config
@@ -143,5 +146,42 @@ func (up *uploads) Complete(ctx fiber.Ctx, md5 string, body *UploadFileInfo) err
 // Token
 // @Router			/v1/admin/uploads/token [get]
 func (up *uploads) Token(ctx fiber.Ctx) (*ali.PolicyToken, error) {
-	return up.ali.GetToken("quyun")
+	return up.ali.GetToken(UPLOAD_PATH)
+}
+
+// PreUploadCheck
+// @Router			/v1/admin/uploads/pre-uploaded-check/:md5 [get]
+// @Bind md5 path
+func (up *uploads) PreUploadCheck(ctx fiber.Ctx, md5 string) error {
+	_, err := models.Medias.GetByHash(ctx.Context(), md5)
+	if err != nil && errors.Is(err, qrm.ErrNoRows) {
+		return ctx.SendString("ok")
+	}
+	return ctx.SendString("exists")
+}
+
+type PostUploadedForm struct {
+	OriginalName string `json:"originalName"`
+	Md5          string `json:"md5"`
+	MimeType     string `json:"mimeType"`
+	Size         int64  `json:"size"`
+}
+
+// PostUploadedAction
+// @Router			/v1/admin/uploads/post-uploaded-action [post]
+// @Bind body body
+func (up *uploads) PostUploadedAction(ctx fiber.Ctx, body *PostUploadedForm) error {
+	m, err := models.Medias.GetByHash(ctx.Context(), body.Md5)
+	if err != nil && !errors.Is(err, qrm.ErrNoRows) {
+		return err
+	}
+
+	m = &model.Medias{
+		Name:     body.OriginalName,
+		MimeType: body.MimeType,
+		Size:     body.Size,
+		Hash:     body.Md5,
+		Path:     filepath.Join(UPLOAD_PATH, body.Md5+filepath.Ext(body.OriginalName)),
+	}
+	return models.Medias.Create(ctx.Context(), m)
 }