Update database and file handling

database/db.go

@@ -41,6 +41,11 @@ func CreateDBEngine() error {
 	if err != nil {
 		return err
 	}
+
+	Engine.SetMaxIdleConns(20)
+	Engine.SetMaxOpenConns(20)
+	Engine.SetConnMaxLifetime(time.Minute * 5)
+
 	if err := Engine.Ping(); err != nil {
 		return err
 	}

file/download.go

@@ -4,7 +4,6 @@ import (
 	"log"
 	"net/url"
 	"strings"
-	"time"
 
 	"github.com/gin-gonic/gin"
 	"github.com/tempfiles-Team/tempfiles-backend/database"
@@ -70,7 +69,7 @@ func DownloadHandler(c *gin.Context) {
 		return
 	}
 
-	if (FileTracking.DownloadLimit != 0 && FileTracking.DownloadCount >= FileTracking.DownloadLimit) || !FileTracking.ExpireTime.Before(time.Now()) || FileTracking.IsDeleted {
+	if FileTracking.DownloadLimit != 0 && FileTracking.DownloadCount >= FileTracking.DownloadLimit {
 
 		FileTracking.IsDeleted = true
 

file/file.go

@@ -40,7 +40,12 @@ func FileHandler(c *gin.Context) {
 		return
 	}
 
-	baseUrl := c.Request.Host
+	scheme := "http"
+	if c.Request.TLS != nil || c.Request.Header.Get("X-Forwarded-Proto") == "https" {
+		scheme = "https"
+	}
+
+	baseUrl := scheme + "://" + c.Request.Host
 
 	if files, err := GetFiles(FileTracking.FolderId, baseUrl); err != nil {
 		c.JSON(500, gin.H{

file/id_generator.go

@@ -35,5 +35,5 @@ func GenerateFolderId(files []*multipart.FileHeader) (string, error) {
 
 	combinedHash := sha1.Sum(bytes.Join(hashes, nil))
 
-	return base64.StdEncoding.EncodeToString(combinedHash[:]), nil
+	return base64.RawURLEncoding.EncodeToString(combinedHash[:]), nil
 }

main.go

@@ -81,64 +81,70 @@ func main() {
 
 	app.GET("/info", func(c *gin.Context) {
 		apiName := c.Query("api")
-		backendUrl := c.Request.Host
+
+		scheme := "http"
+		if c.Request.TLS != nil || c.Request.Header.Get("X-Forwarded-Proto") == "https" {
+			scheme = "https"
+		}
+
+		baseUrl := scheme + "://" + c.Request.Host
+
 		switch apiName {
 		case "upload":
 			c.JSON(200, gin.H{
 				"apiName": "/upload",
 				"method":  "POST",
 				"desc":    "특정 파일을 서버에 업로드합니다.",
-				"command": "curl -X POST -F 'file=@[filepath or filename]' " + backendUrl + "/upload",
+				"command": "curl -LX POST -F 'file=@[filepath or filename]' " + baseUrl + "/upload",
 			})
 		case "list":
 			c.JSON(200, gin.H{
 				"apiName": "/list",
 				"method":  "GET",
 				"desc":    "서버에 존재하는 파일 리스트를 반환합니다.",
-				"command": "curl " + backendUrl + "/list",
+				"command": "curl -L " + baseUrl + "/list",
 			})
 		case "file":
 			c.JSON(200, gin.H{
 				"apiName": "/file/[file_id]",
 				"method":  "GET",
 				"desc":    "서버에 존재하는 특정 파일에 대한 세부 정보를 반환합니다.",
-				"command": "curl " + backendUrl + "/file/[file_id]",
+				"command": "curl -L " + baseUrl + "/file/[file_id]",
 			})
 		case "del":
 			c.JSON(200, gin.H{
 				"apiName": "/del/[file_id]",
 				"method":  "DELETE",
 				"desc":    "서버에 존재하는 특정 파일을 삭제합니다.",
-				"command": "curl -X DELETE " + backendUrl + "/del/[file_id]",
+				"command": "curl -LX DELETE " + baseUrl + "/del/[file_id]",
 			})
 		case "dl":
 			c.JSON(200, gin.H{
 				"apiName": "/dl/[file_id]",
 				"method":  "GET",
 				"desc":    "서버에 존재하는 특정 파일을 다운로드 합니다.",
-				"command": "curl -O " + backendUrl + "/dl/[file_id]",
+				"command": "curl -LO " + baseUrl + "/dl/[file_id]",
 			})
 		default:
 			c.JSON(200, []gin.H{
 				{
-					"apiUrl":     backendUrl + "/upload",
+					"apiUrl":     baseUrl + "/upload",
 					"apiHandler": "upload",
 				},
 				{
-
-					"apiUrl":     backendUrl + "/list",
+					"apiUrl":     baseUrl + "/list",
 					"apiHandler": "list",
 				},
 				{
-					"apiUrl":     backendUrl + "/file/[file_id]",
+					"apiUrl":     baseUrl + "/file/[file_id]",
 					"apiHandler": "file",
 				},
 				{
-					"apiUrl":     backendUrl + "/del/[file_id]",
+					"apiUrl":     baseUrl + "/del/[file_id]",
 					"apiHandler": "del",
 				},
 				{
-					"apiUrl":     backendUrl + "/dl/[file_id]",
+					"apiUrl":     baseUrl + "/dl/[file_id]",
 					"apiHandler": "dl",
 				},
 			})