From d6fea884cf8b6b71cb990f93c15a9847cc2e456f Mon Sep 17 00:00:00 2001
From: nyne <me@nyne.dev>
Date: Thu, 15 May 2025 10:47:18 +0800
Subject: [PATCH] fix downloading file

---
 main.go            | 11 +++++++++--
 server/api/file.go |  3 ++-
 2 files changed, 11 insertions(+), 3 deletions(-)

diff --git a/main.go b/main.go
index 201c676..6859ef2 100644
--- a/main.go
+++ b/main.go
@@ -10,9 +10,14 @@ import (
 	"github.com/gofiber/fiber/v3/middleware/logger"
 )
 
+const (
+	debugMode = true
+)
+
 func main() {
 	app := fiber.New(fiber.Config{
-		BodyLimit: 8 * 1024 * 1024,
+		BodyLimit:  8 * 1024 * 1024,
+		TrustProxy: true,
 	})
 
 	app.Use(logger.New(logger.Config{
@@ -23,7 +28,9 @@ func main() {
 
 	app.Use(middleware.JwtMiddleware)
 
-	app.Use(cors.New(cors.ConfigDefault))
+	if debugMode {
+		app.Use(cors.New(cors.ConfigDefault))
+	}
 
 	apiG := app.Group("/api")
 	{
diff --git a/server/api/file.go b/server/api/file.go
index d30735f..d3576e0 100644
--- a/server/api/file.go
+++ b/server/api/file.go
@@ -190,7 +190,8 @@ func deleteFile(c fiber.Ctx) error {
 }
 
 func downloadFile(c fiber.Ctx) error {
-	s, filename, err := service.DownloadFile(c.Params("id"))
+	ip := c.IP()
+	s, filename, err := service.DownloadFile(ip, c.Params("id"))
 	if err != nil {
 		return err
 	}