init: 0.0.8

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

internal/handler/share.go

@@ -0,0 +1,119 @@
+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.Param("code")
+		log.Debug("handler.Fetch: code = %s", code)
+		info := new(model.Meta)
+		_, err := os.Stat(opt.MetaPath(code))
+		if err != nil {
+			if errors.Is(err, os.ErrNotExist) {
+				return c.Status(http.StatusNotFound).JSON(map[string]string{"msg": "文件不存在"})
+			}
+
+			return c.SendStatus(http.StatusInternalServerError)
+		}
+
+		viper.SetConfigFile(opt.MetaPath(code))
+		viper.SetConfigType("env")
+		if err = viper.ReadInConfig(); err != nil {
+			return c.SendStatus(http.StatusInternalServerError)
+		}
+
+		if err = viper.Unmarshal(info); err != nil {
+			return c.SendStatus(http.StatusInternalServerError)
+		}
+
+		c.SetHeader("Content-Disposition", fmt.Sprintf(`attachment; filename="%s"`, info.Filename))
+		http.ServeFile(c.Writer, c.Request, opt.FilePath(code))
+
+		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})
+	}
+}