nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 12:17:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move request limiter to middleware.

Browse Source
This commit is contained in:
nyne
2025-07-09 17:00:39 +08:00
parent 0021a73951
commit b568b234c4
11 changed files with 73 additions and 71 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
server/api/file.go
View File

@@ -4,11 +4,14 @@ import (
"encoding/json"
"fmt"
"net/url"
"nysoure/server/config"
"nysoure/server/middleware"
"nysoure/server/model"
"nysoure/server/service"
"nysoure/server/utils"
"strconv"
"strings"
"time"
"github.com/gofiber/fiber/v3"
)
@@ -16,17 +19,17 @@ import (
func AddFileRoutes(router fiber.Router) {
fileGroup := router.Group("/files")
{
fileGroup.Post("/upload/init", initUpload)
fileGroup.Use(middleware.NewRequestLimiter(10, time.Hour)).Post("/upload/init", initUpload)
fileGroup.Post("/upload/block/:id/:index", uploadBlock)
fileGroup.Post("/upload/finish/:id", finishUpload)
fileGroup.Post("/upload/cancel/:id", cancelUpload)
fileGroup.Post("/redirect", createRedirectFile)
fileGroup.Use(middleware.NewRequestLimiter(50, time.Hour)).Post("/redirect", createRedirectFile)
fileGroup.Post("/upload/url", createServerDownloadTask)
fileGroup.Get("/:id", getFile)
fileGroup.Put("/:id", updateFile)
fileGroup.Delete("/:id", deleteFile)
fileGroup.Get("/download/local", downloadLocalFile)
fileGroup.Get("/download/:id", downloadFile)
fileGroup.Use(middleware.NewDynamicRequestLimiter(config.MaxDownloadsPerDayForSingleIP, 24*time.Hour)).Get("/download/:id", downloadFile)
}
}
@@ -201,8 +204,7 @@ func deleteFile(c fiber.Ctx) error {
func downloadFile(c fiber.Ctx) error {
cfToken := c.Query("cf_token")
ip := c.IP()
s, filename, err := service.DownloadFile(ip, c.Params("id"), cfToken)
s, filename, err := service.DownloadFile(c.Params("id"), cfToken)
if err != nil {
return err
}