feat(files): enable upload limiter

src/handlers/fileshdr/handlers.go

@@ -10,6 +10,7 @@ import (
 	"os"
 	"path"
 	"path/filepath"
+	"strconv"
 	"strings"
 	"time"
 
@@ -289,6 +290,21 @@ func (h *FileHandlers) UploadChunk(c *gin.Context) {
 		return
 	}
 
+	userIDInt, err := strconv.ParseUint(userID, 10, 64)
+	if err != nil {
+		c.JSON(q.ErrResp(c, 500, err))
+		return
+	}
+
+	ok, err := h.deps.Limiter().CanUpload(userIDInt, len([]byte(req.Content)))
+	if err != nil {
+		c.JSON(q.ErrResp(c, 500, err))
+		return
+	} else if !ok {
+		c.JSON(q.ErrResp(c, 503, errors.New("retry later")))
+		return
+	}
+
 	tmpFilePath := q.UploadPath(userID, req.Path)
 	locker := h.NewAutoLocker(c, lockName(tmpFilePath))
 	locker.Exec(func() {

src/handlers/util.go

@@ -7,6 +7,8 @@ import (
 	"path/filepath"
 
 	"github.com/gin-gonic/gin"
+
+	"github.com/ihexxa/quickshare/src/cryptoutil"
 )
 
 var (
@@ -137,3 +139,22 @@ func UploadPath(userID, relFilePath string) string {
 func UploadFolder(userID string) string {
 	return filepath.Join(userID, UploadDir)
 }
+
+func GetUserInfo(tokenStr string, tokenEncDec cryptoutil.ITokenEncDec) (map[string]string, error) {
+	claims, err := tokenEncDec.FromToken(
+		tokenStr,
+		map[string]string{
+			UserIDParam: "",
+			UserParam:   "",
+			RoleParam:   "",
+			ExpireParam: "",
+		},
+	)
+	if err != nil {
+		return nil, err
+	} else if claims[UserIDParam] == "" || claims[UserParam] == "" {
+		return nil, errors.New("empty user id or name")
+	}
+
+	return claims, nil
+}