blob: 06a45071a3c3e538173d74d87e9203d61bf8919a (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
package request
import (
"context"
"net"
"net/http"
"gitlab.com/gitlab-org/gitlab-pages/internal/domain"
)
type ctxKey string
const (
ctxHTTPSKey ctxKey = "https"
ctxHostKey ctxKey = "host"
ctxDomainKey ctxKey = "domain"
)
// WithHTTPSFlag saves https flag in request's context
func WithHTTPSFlag(r *http.Request, https bool) *http.Request {
ctx := context.WithValue(r.Context(), ctxHTTPSKey, https)
return r.WithContext(ctx)
}
// IsHTTPS restores https flag from request's context
func IsHTTPS(r *http.Request) bool {
return r.Context().Value(ctxHTTPSKey).(bool)
}
// WithHostAndDomain saves host name and domain in the request's context
func WithHostAndDomain(r *http.Request, host string, domain *domain.Domain) *http.Request {
ctx := r.Context()
ctx = context.WithValue(ctx, ctxHostKey, host)
ctx = context.WithValue(ctx, ctxDomainKey, domain)
return r.WithContext(ctx)
}
// GetHost extracts the host from request's context
func GetHost(r *http.Request) string {
return r.Context().Value(ctxHostKey).(string)
}
// GetDomain extracts the domain from request's context
func GetDomain(r *http.Request) *domain.Domain {
return r.Context().Value(ctxDomainKey).(*domain.Domain)
}
// GetHostWithoutPort returns a host without the port. The host(:port) comes
// from a Host: header if it is provided, otherwise it is a server name.
func GetHostWithoutPort(r *http.Request) string {
host, _, err := net.SplitHostPort(r.Host)
if err != nil {
return r.Host
}
return host
}
|
