Don't split domain if we use standard HTTP ports

This commit is contained in:
Yohann D'ANELLO 2020-10-04 18:10:11 +02:00
parent d5ada0bce7
commit 7831238375
1 changed files with 9 additions and 4 deletions

View File

@ -7,6 +7,7 @@ import (
"log"
"net"
"net/http"
"strings"
"github.com/markbates/pkger"
"gitlab.crans.org/nounous/ghostream/internal/monitoring"
@ -49,10 +50,14 @@ func viewerGetHandler(w http.ResponseWriter, r *http.Request) {
// Get stream ID from URL, or from domain name
path := r.URL.Path[1:]
if cfg.OneStreamPerDomain {
host, _, err := net.SplitHostPort(r.Host)
if err != nil {
log.Printf("Failed to split host and port from %s", r.Host)
return
var host = r.Host
if strings.Contains(host, ":") {
realHost, _, err := net.SplitHostPort(r.Host)
if err != nil {
log.Printf("Failed to split host and port from %s", r.Host)
return
}
host = realHost
}
path = host
}