From 0ef3949b61e105283a3e234e53d177b22df52819 Mon Sep 17 00:00:00 2001 From: Andreas Mieke Date: Sun, 7 Mar 2021 23:23:59 +0100 Subject: [PATCH] Add XAccelPrefix config option --- internal/config/config.go | 1 + internal/web/routes.go | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/internal/config/config.go b/internal/config/config.go index 34318aa..4ca8c6d 100644 --- a/internal/config/config.go +++ b/internal/config/config.go @@ -17,6 +17,7 @@ func LoadConfig() { viper.SetDefault("ShortURL", "https://1750.st") viper.SetDefault("FileFolder", "/home/ezazzan/.shortdragon/files") viper.SetDefault("XAccel", false) + viper.SetDefault("XAccelPrefix", "") viper.SetDefault("MinAge", 30) // days viper.SetDefault("MaxAge", 365) // days diff --git a/internal/web/routes.go b/internal/web/routes.go index ad9f74d..ed513a0 100644 --- a/internal/web/routes.go +++ b/internal/web/routes.go @@ -200,7 +200,7 @@ func getFile(c *gin.Context) { if viper.GetBool("XAccel") { c.Header("Content-Disposition", "attatchment; filename="+parts[len(parts)-1]) - c.Header("X-Accel-Redirect", "/"+URL.FileName.String) + c.Header("X-Accel-Redirect", viper.GetString("XAccelPrefix")+"/"+URL.FileName.String) } else { c.FileAttachment(file, parts[len(parts)-1]) }