From 2fc3ccd24bc31000a44b10d4117042ea154c694c Mon Sep 17 00:00:00 2001 From: Mendel E Date: Sun, 8 Dec 2019 20:50:54 +0000 Subject: [PATCH] Try getting scheme from X-Forwarded-Proto --- main.go | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/main.go b/main.go index 929e67b..95a3008 100644 --- a/main.go +++ b/main.go @@ -9,6 +9,7 @@ import ( "math/rand" "net/http" "net/url" + "strings" "time" "unicode" @@ -160,11 +161,16 @@ func Bin(w http.ResponseWriter, r *http.Request, b *bitcask.Bitcask) { // return id u := url.URL{ - Scheme: "http", Host: r.Host, Path: "/", // todo: use config or dynamic } + if v, ok := r.Header["X-Forwarded-Proto"]; ok { + u.Scheme = strings.Join(v, "") + } else { + u.Scheme = "http" + } + q := url.Values{} q.Set("id", id) -- 2.26.2