-
Notifications
You must be signed in to change notification settings - Fork 0
/
trurl.go
70 lines (60 loc) · 1.12 KB
/
trurl.go
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
60
61
62
63
64
65
66
67
68
69
70
package trurl_go
import (
"net/url"
"strings"
)
const (
httpsPort = "443"
httpPort = "80"
https = "https"
http = "http"
)
type Trurl struct {
Url string
Scheme string
User string
Password string
Host string
Port string
Path string
Fragment string
RawUrl *url.URL
}
// Parse will give s as string and parse it into Trurl struct.
func Parse(s string) *Trurl {
// Parse URL.
u, err := url.Parse(s)
if err != nil {
return &Trurl{}
}
t := Trurl{
Url: strings.ToLower(s),
Port: u.Port(),
Host: u.Host,
Scheme: u.Scheme,
User: u.User.Username(),
Path: u.Path,
Fragment: u.Fragment,
RawUrl: u,
}
// Parse Password.
if pass, ok := u.User.Password(); ok {
t.Password = pass
}
// Parse port.
if t.Port == "" {
if strings.Contains(s, https) {
t.Port = httpsPort
} else if strings.Contains(s, http) {
t.Port = httpPort
}
}
return &t
}
// SetHost will set host on given url and Trurl.Host value.
func (t *Trurl) SetHost(h string) *Trurl {
t.Host = h
t.RawUrl.Host = h
t.Url = strings.ToLower(t.RawUrl.String())
return t
}