init: 0.0.8

feat: 实现了分片上传
todo:
  1. 上传完成后 code 回显
  2. 清理不用的临时 code file
  3. 断点续传
  4. 多 chunk 并发上传

internal/handler/share.go

@@ -3,17 +3,23 @@ package handler
 import (
 	"fmt"
 	"github.com/loveuer/nf"
+	"github.com/loveuer/nf/nft/log"
+	"github.com/loveuer/ushare/internal/controller"
 	"github.com/loveuer/ushare/internal/model"
 	"github.com/loveuer/ushare/internal/opt"
 	"github.com/pkg/errors"
+	"github.com/spf13/cast"
 	"github.com/spf13/viper"
 	"net/http"
 	"os"
+	"regexp"
+	"strings"
 )
 
 func Fetch() nf.HandlerFunc {
 	return func(c *nf.Ctx) error {
-		code := c.Query("code")
+		code := c.Param("code")
+		log.Debug("handler.Fetch: code = %s", code)
 		info := new(model.Meta)
 		_, err := os.Stat(opt.MetaPath(code))
 		if err != nil {
@@ -40,3 +46,74 @@ func Fetch() nf.HandlerFunc {
 		return nil
 	}
 }
+
+func ShareNew() nf.HandlerFunc {
+	return func(c *nf.Ctx) error {
+
+		filename := strings.TrimSpace(c.Param("filename"))
+		if filename == "" {
+			return c.Status(http.StatusBadRequest).JSON(map[string]string{"msg": "filename required"})
+		}
+
+		size, err := cast.ToInt64E(c.Get(opt.HeaderSize))
+		if err != nil {
+			return c.Status(http.StatusBadRequest).JSON(map[string]string{"msg": "miss header: " + opt.HeaderSize})
+		}
+
+		code, err := controller.MetaManager.New(size, filename, c.IP())
+		if err != nil {
+			return c.Status(http.StatusInternalServerError).JSON(map[string]string{"msg": ""})
+		}
+
+		return c.Status(http.StatusOK).JSON(map[string]string{"code": code})
+	}
+}
+
+func ShareUpload() nf.HandlerFunc {
+	rangeValidator := regexp.MustCompile(`^bytes=\d+-\d+$`)
+	return func(c *nf.Ctx) error {
+		code := strings.TrimSpace(c.Param("code"))
+		if len(code) != 16 {
+			return c.Status(http.StatusBadRequest).JSON(map[string]string{"msg": "invalid file code"})
+		}
+
+		log.Debug("handler.ShareUpload: code = %s", code)
+
+		ranger := strings.TrimSpace(c.Get("Range"))
+		if ranger == "" {
+			return c.Status(http.StatusBadRequest).JSON(map[string]string{"msg": "miss header: Range"})
+		}
+
+		log.Debug("handler.ShareUpload: code = %s, ranger = %s", code, ranger)
+
+		if !rangeValidator.MatchString(ranger) {
+			log.Warn("handler.ShareUpload: invalid range, ranger = %s", ranger)
+			return c.Status(http.StatusBadRequest).JSON(map[string]string{"msg": "Range invalid(1)"})
+		}
+
+		strs := strings.Split(strings.TrimPrefix(ranger, "bytes="), "-")
+		if len(strs) != 2 {
+			return c.Status(http.StatusBadRequest).JSON(map[string]string{"msg": "Range invalid(2)"})
+		}
+
+		start, err := cast.ToInt64E(strs[0])
+		if err != nil {
+			return c.Status(http.StatusBadRequest).JSON(map[string]string{"msg": "Range invalid(3)"})
+		}
+
+		end, err := cast.ToInt64E(strs[1])
+		if err != nil {
+			return c.Status(http.StatusBadRequest).JSON(map[string]string{"msg": "Range invalid(4)"})
+		}
+
+		log.Debug("handler.ShareUpload: code = %s, start = %d, end = %d", code, start, end)
+
+		total, cursor, err := controller.MetaManager.Write(code, start, end, c.Request.Body)
+		if err != nil {
+			log.Error("handler.ShareUpload: write error: %s", err)
+			return c.Status(http.StatusInternalServerError).JSON(map[string]string{"msg": ""})
+		}
+
+		return c.Status(http.StatusOK).JSON(map[string]any{"size": total, "cursor": cursor})
+	}
+}