-
Notifications
You must be signed in to change notification settings - Fork 232
/
stringz.go
executable file
·86 lines (74 loc) · 1.96 KB
/
stringz.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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
package stringz
import (
"net/url"
"strconv"
"strings"
"github.com/projectdiscovery/urlutil"
)
// TrimProtocol removes the HTTP scheme from an URI
func TrimProtocol(targetURL string, addDefaultPort bool) string {
URL := strings.TrimSpace(targetURL)
if strings.HasPrefix(strings.ToLower(URL), "http://") || strings.HasPrefix(strings.ToLower(URL), "https://") {
if addDefaultPort {
URL = AddURLDefaultPort(URL)
URL = URL[strings.Index(URL, "//")+2:]
}
}
return URL
}
// StringToSliceInt converts string to slice of ints
func StringToSliceInt(s string) ([]int, error) {
var r []int
if s == "" {
return r, nil
}
for _, v := range strings.Split(s, ",") {
vTrim := strings.TrimSpace(v)
if i, err := strconv.Atoi(vTrim); err == nil {
r = append(r, i)
} else {
return r, err
}
}
return r, nil
}
// SplitByCharAndTrimSpace splits string by a character and remove spaces
func SplitByCharAndTrimSpace(s, splitchar string) (result []string) {
for _, token := range strings.Split(s, splitchar) {
result = append(result, strings.TrimSpace(token))
}
return
}
// AddURLDefaultPort add url default port (80/443) from an URI
// eg:
// http://foo.com -> http://foo.com:80
// https://foo.com -> https://foo.com:443
func AddURLDefaultPort(rawURL string) string {
u, err := urlutil.Parse(rawURL)
if err != nil {
return rawURL
}
return u.String()
}
// RemoveURLDefaultPort remove url default port (80/443) from an URI
// eg:
// http://foo.com:80 -> http://foo.com
// https://foo.com:443 -> https://foo.com
func RemoveURLDefaultPort(rawURL string) string {
u, err := urlutil.Parse(rawURL)
if err != nil {
return rawURL
}
if u.Scheme == urlutil.HTTP && u.Port == "80" || u.Scheme == urlutil.HTTPS && u.Port == "443" {
u.Port = ""
}
return u.String()
}
func GetInvalidURI(rawURL string) (bool, string) {
if _, err := url.Parse(rawURL); err != nil {
if u, err := urlutil.Parse(rawURL); err == nil {
return true, u.RequestURI
}
}
return false, ""
}