92 lines
1.6 KiB
Go
92 lines
1.6 KiB
Go
|
package urlx
|
||
|
|
|
||
|
|
import (
|
||
|
|
"net/url"
|
||
|
|
"strings"
|
||
|
|
|
||
|
|
"git.cloudyne.io/go/hiscaler-gox/isx"
|
||
|
|
)
|
||
|
|
|
||
|
|
type URL struct {
|
||
|
|
Path string // URL path
|
||
|
|
URL *url.URL // A url.URL represents
|
||
|
|
Invalid bool // Path is a valid url
|
||
|
|
values url.Values // Query values
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewURL(path string) *URL {
|
||
|
|
u := &URL{
|
||
|
|
Path: path,
|
||
|
|
Invalid: false,
|
||
|
|
values: url.Values{},
|
||
|
|
}
|
||
|
|
if v, err := url.Parse(u.Path); err == nil {
|
||
|
|
u.URL = v
|
||
|
|
u.Invalid = true
|
||
|
|
if values, err := url.ParseQuery(v.RawQuery); err == nil {
|
||
|
|
u.values = values
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return u
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u URL) GetValue(key, defaultValue string) string {
|
||
|
|
v := u.values.Get(key)
|
||
|
|
if v == "" {
|
||
|
|
v = defaultValue
|
||
|
|
}
|
||
|
|
return v
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u URL) SetValue(key, value string) URL {
|
||
|
|
u.values.Set(key, value)
|
||
|
|
return u
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u URL) AddValue(key, value string) URL {
|
||
|
|
u.values.Add(key, value)
|
||
|
|
return u
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u URL) DelKey(key string) URL {
|
||
|
|
u.values.Del(key)
|
||
|
|
return u
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u URL) HasKey(key string) bool {
|
||
|
|
return u.values.Has(key)
|
||
|
|
}
|
||
|
|
|
||
|
|
func (u URL) String() string {
|
||
|
|
s := u.URL.String()
|
||
|
|
rawQuery := u.URL.RawQuery
|
||
|
|
if rawQuery == "" {
|
||
|
|
if len(u.values) > 0 {
|
||
|
|
s += "?" + u.values.Encode()
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
s = strings.Replace(s, rawQuery, u.values.Encode(), 1)
|
||
|
|
}
|
||
|
|
return s
|
||
|
|
}
|
||
|
|
|
||
|
|
// IsAbsolute 是否为绝对地址
|
||
|
|
func IsAbsolute(s string) bool {
|
||
|
|
if strings.HasPrefix(s, "//") {
|
||
|
|
s = "http:" + s
|
||
|
|
}
|
||
|
|
if isx.HttpURL(s) {
|
||
|
|
if u, err := url.Parse(s); err == nil {
|
||
|
|
if u.IsAbs() && len(u.Host) > 2 && strings.Index(u.Host, ".") > 0 {
|
||
|
|
return true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return false
|
||
|
|
}
|
||
|
|
|
||
|
|
// IsRelative 是否为相对地址
|
||
|
|
func IsRelative(url string) bool {
|
||
|
|
return !IsAbsolute(url)
|
||
|
|
}
|