nyne/nysoure
1
0
Fork 0
mirror of https://github.com/wgh136/nysoure.git synced 2025-09-27 04:17:23 +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

6
server/api/comment.go
View File

@@ -2,17 +2,19 @@ package api
import (
"net/url"
"nysoure/server/middleware"
"nysoure/server/model"
"nysoure/server/service"
"strconv"
"time"
"github.com/gofiber/fiber/v3"
)
func AddCommentRoutes(router fiber.Router) {
api := router.Group("/comments")
api.Post("/resource/:resourceID", createResourceComment)
api.Post("/reply/:commentID", createReplyComment)
api.Use(middleware.NewRequestLimiter(500, 24*time.Hour)).Post("/resource/:resourceID", createResourceComment)
api.Use(middleware.NewRequestLimiter(500, 24*time.Hour)).Post("/reply/:commentID", createReplyComment)
api.Get("/resource/:resourceID", listResourceComments)
api.Get("/reply/:commentID", listReplyComments)
api.Get("/user/:username", listCommentsByUser)

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
}

7
server/api/image.go
View File

@@ -1,12 +1,15 @@
package api
import (
"github.com/gofiber/fiber/v3"
"net/http"
"nysoure/server/middleware"
"nysoure/server/model"
"nysoure/server/service"
"strconv"
"strings"
"time"
"github.com/gofiber/fiber/v3"
)
func handleUploadImage(c fiber.Ctx) error {
@@ -94,7 +97,7 @@ func handleGetResampledImage(c fiber.Ctx) error {
func AddImageRoutes(api fiber.Router) {
image := api.Group("/image")
{
image.Put("/", handleUploadImage)
image.Use(middleware.NewRequestLimiter(50, time.Hour)).Put("/", handleUploadImage)
image.Get("/resampled/:id", handleGetResampledImage)
image.Get("/:id", handleGetImage)
image.Delete("/:id", handleDeleteImage)

4
server/api/user.go
View File

@@ -4,9 +4,11 @@ import (
"io"
"net/http"
"net/url"
"nysoure/server/middleware"
"nysoure/server/model"
"nysoure/server/service"
"strconv"
"time"
"github.com/gofiber/fiber/v3"
)
@@ -342,7 +344,7 @@ func handleGetMe(c fiber.Ctx) error {
func AddUserRoutes(r fiber.Router) {
u := r.Group("user")
u.Post("/register", handleUserRegister)
u.Use(middleware.NewRequestLimiter(5, time.Hour)).Post("/register", handleUserRegister)
u.Post("/login", handleUserLogin)
u.Put("/avatar", handleUserChangeAvatar)
u.Post("/password", handleUserChangePassword)