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]) }