Files
mproxy/pkg/servcli/referer.go
T
2026-05-30 12:04:50 +02:00

88 lines
1.7 KiB
Go

/*
* Copyright 2026 Oleg Borodin <onborodin@gmail.com>
*/
package servcli
import (
"net/url"
"path"
"strconv"
"strings"
)
const (
PathTypeIdentic = "identic"
PathTypePrefix = "prefix"
PathTypeRegexp = "regexp"
)
type Referer struct {
urlobj *url.URL
user, pass string
resource string
values url.Values
}
func ParsePath(rawpath string) (*Referer, error) {
ref := &Referer{
values: url.Values{},
}
if !strings.Contains(rawpath, "://") {
rawpath = "https://" + rawpath
}
urlobj, err := url.Parse(rawpath)
if err != nil {
return ref, err
}
if urlobj.User != nil {
ref.user = urlobj.User.Username()
ref.pass, _ = urlobj.User.Password()
urlobj.User = nil
}
ref.resource = path.Join("/", urlobj.Path)
urlobj.Path = "/"
ref.urlobj = urlobj
ref.values = urlobj.Query()
return ref, err
}
func (ref *Referer) Raw() string {
res := path.Join(ref.urlobj.Host, ref.resource)
query := ref.values.Encode()
if query != "" {
return res + "?" + query
}
return res
}
func (ref *Referer) SetResource(resource string) {
ref.resource = path.Join("/", resource)
}
func (ref *Referer) JoinResource(resource string) {
ref.resource = path.Join("/", ref.resource, resource)
}
func (ref *Referer) PathType(typ string) {
ref.values.Set("pathType", typ)
}
func (ref *Referer) DryRun(yesno bool) {
ref.values.Set("dryRun", strconv.FormatBool(yesno))
}
func (ref *Referer) HelloEP() string {
curl := ref.urlobj.JoinPath("/v3/api/service/hello")
return curl.String()
}
func (ref *Referer) Userinfo() (string, string) {
return ref.user, ref.pass
}
func (ref *Referer) SetUserinfo(user, pass string) {
if user != "" && pass != "" {
ref.user, ref.pass = user, pass
}
}