119 lines
3.5 KiB
Go
119 lines
3.5 KiB
Go
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) != opt.CodeLength {
|
|
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})
|
|
}
|
|
}
|