Created

net/urlx/url.go

@@ -0,0 +1,91 @@
+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)
+}