mirror of
https://github.com/wgh136/nysoure.git
synced 2025-09-27 12:17:24 +00:00
add RealUserMiddleware and update resource retrieval to track real users
This commit is contained in:
@@ -79,7 +79,7 @@ func serveIndexHtml(c fiber.Ctx) error {
|
||||
idStr := strings.TrimPrefix(path, "/resources/")
|
||||
id, err := strconv.Atoi(idStr)
|
||||
if err == nil {
|
||||
r, err := service.GetResource(uint(id), c.Hostname())
|
||||
r, err := service.GetResource(uint(id), c.Hostname(), c)
|
||||
if err == nil {
|
||||
if len(r.Images) > 0 {
|
||||
preview = fmt.Sprintf("%s/api/image/%d", serverBaseURL, r.Images[0].ID)
|
||||
|
17
server/middleware/real_user_middleware.go
Normal file
17
server/middleware/real_user_middleware.go
Normal file
@@ -0,0 +1,17 @@
|
||||
package middleware
|
||||
|
||||
import (
|
||||
"strings"
|
||||
|
||||
"github.com/gofiber/fiber/v3"
|
||||
)
|
||||
|
||||
func RealUserMiddleware(c fiber.Ctx) error {
|
||||
userAgent := c.Get("User-Agent")
|
||||
if strings.Contains(userAgent, "Mozilla") || strings.Contains(userAgent, "AppleWebKit") {
|
||||
c.Locals("real_user", true)
|
||||
} else {
|
||||
c.Locals("real_user", false)
|
||||
}
|
||||
return c.Next()
|
||||
}
|
Reference in New Issue
Block a user