h/projectreshoot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added configuration from environment variables

Browse Source
This commit is contained in:
Haelnorr
2025-02-08 16:27:33 +11:00
parent 1004dc9563
commit a842d09f96
7 changed files with 48 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
cookies/pagefrom.go
View File

@@ -3,7 +3,6 @@ package cookies
import (
"net/http"
"net/url"
"os"
"time"
)
@@ -18,15 +17,14 @@ func CheckPageFrom(w http.ResponseWriter, r *http.Request) string {
return pageFrom
}
func SetPageFrom(w http.ResponseWriter, r *http.Request) {
func SetPageFrom(w http.ResponseWriter, r *http.Request, trustedHost string) {
referer := r.Referer()
parsedURL, err := url.Parse(referer)
if err != nil {
return
}
var pageFrom string
expectedHost := os.Getenv("TRUSTED_HOST")
if parsedURL.Path == "" || parsedURL.Host != expectedHost {
if parsedURL.Path == "" || parsedURL.Host != trustedHost {
pageFrom = "/"
} else {
pageFrom = parsedURL.Path