get forwarded url

This commit is contained in:
sentriz
2019-04-18 00:02:46 +01:00
parent 6120fc81ef
commit 7dbcb9ec5a
3 changed files with 31 additions and 5 deletions

1
go.mod
View File

@@ -2,6 +2,7 @@ module github.com/sentriz/gonic
require ( require (
cloud.google.com/go v0.37.1 // indirect cloud.google.com/go v0.37.1 // indirect
github.com/davecgh/go-spew v1.1.1
github.com/denisenkom/go-mssqldb v0.0.0-20190315220205-a8ed825ac853 // indirect github.com/denisenkom/go-mssqldb v0.0.0-20190315220205-a8ed825ac853 // indirect
github.com/dhowden/tag v0.0.0-20181104225729-a9f04c2798ca github.com/dhowden/tag v0.0.0-20181104225729-a9f04c2798ca
github.com/erikstmartin/go-testdb v0.0.0-20160219214506-8d10e4a1bae5 // indirect github.com/erikstmartin/go-testdb v0.0.0-20160219214506-8d10e4a1bae5 // indirect

View File

@@ -50,6 +50,9 @@ func (c *Controller) ServeHome(w http.ResponseWriter, r *http.Request) {
c.DB.Table("albums").Count(&data.AlbumCount) c.DB.Table("albums").Count(&data.AlbumCount)
c.DB.Table("tracks").Count(&data.TrackCount) c.DB.Table("tracks").Count(&data.TrackCount)
c.DB.Find(&data.AllUsers) c.DB.Find(&data.AllUsers)
var apiKey db.Setting
c.DB.Where("key = ?", "lastfm_api_key").First(&apiKey)
data.CurrentLastFMAPIKey = apiKey.Value
renderTemplate(w, r, "home", &data) renderTemplate(w, r, "home", &data)
} }

View File

@@ -142,6 +142,21 @@ func respondError(w http.ResponseWriter, r *http.Request,
)) ))
} }
func firstExisting(or string, strings ...string) string {
current := ""
for _, s := range strings {
if s == "" {
continue
}
current = s
break
}
if current == "" {
return or
}
return current
}
func renderTemplate(w http.ResponseWriter, r *http.Request, func renderTemplate(w http.ResponseWriter, r *http.Request,
name string, data *templateData) { name string, data *templateData) {
session := r.Context().Value("session").(*sessions.Session) session := r.Context().Value("session").(*sessions.Session)
@@ -154,11 +169,18 @@ func renderTemplate(w http.ResponseWriter, r *http.Request,
if ok { if ok {
data.User = user data.User = user
} }
if r.URL.Host == "" { scheme := firstExisting(
data.RequestRoot = "http://localhost" "http", // fallback
} else { r.Header.Get("X-Forwarded-Proto"),
data.RequestRoot = fmt.Sprintf("%s://%s", r.URL.Scheme, r.URL.Host) r.Header.Get("X-Forwarded-Scheme"),
} r.URL.Scheme,
)
host := firstExisting(
"localhost:7373", // fallback
r.Header.Get("X-Forwarded-Host"),
r.Host,
)
data.RequestRoot = fmt.Sprintf("%s://%s", scheme, host)
err := templates[name].ExecuteTemplate(w, "layout", data) err := templates[name].ExecuteTemplate(w, "layout", data)
if err != nil { if err != nil {
http.Error(w, fmt.Sprintf("500 when executing: %v", err), 500) http.Error(w, fmt.Sprintf("500 when executing: %v", err), 500)