52 lines
1.5 KiB
Go
52 lines
1.5 KiB
Go
package proxy
|
||
|
||
import (
|
||
"errors"
|
||
"net/url"
|
||
"strings"
|
||
)
|
||
|
||
// normalizeURL 规范化URL格式，处理缺少斜杠的情况
|
||
func normalizeURL(rawURL string) string {
|
||
if strings.HasPrefix(rawURL, "https:/") && !strings.HasPrefix(rawURL, "https://") {
|
||
return strings.Replace(rawURL, "https:/", "https://", 1)
|
||
}
|
||
if strings.HasPrefix(rawURL, "http:/") && !strings.HasPrefix(rawURL, "http://") {
|
||
return strings.Replace(rawURL, "http:/", "http://", 1)
|
||
}
|
||
return rawURL
|
||
}
|
||
|
||
// BuildFromProxyPath 构建 /proxy/*path 形式传入的 URL
|
||
func BuildFromProxyPath(pathPart string, originalQuery url.Values) (string, error) {
|
||
pathPart = strings.TrimPrefix(pathPart, "/")
|
||
if pathPart == "" { return "", errors.New("目标为空") }
|
||
pathPart = normalizeURL(pathPart)
|
||
return mergeQuery(pathPart, originalQuery)
|
||
}
|
||
|
||
// BuildFromProtocol 构建 /:protocol/*remainder 形式
|
||
func BuildFromProtocol(protocol, remainder string, originalQuery url.Values) (string, error) {
|
||
if protocol != "http" && protocol != "https" {
|
||
return "", errors.New("不支持的协议")
|
||
}
|
||
full := protocol + ":/" + remainder
|
||
full = normalizeURL(full)
|
||
return mergeQuery(full, originalQuery)
|
||
}
|
||
|
||
func mergeQuery(raw string, original url.Values) (string, error) {
|
||
parsed, err := url.Parse(raw)
|
||
if err != nil { return "", err }
|
||
// 合并 query
|
||
q := parsed.Query()
|
||
for k, vs := range original {
|
||
for _, v := range vs { q.Add(k, v) }
|
||
}
|
||
parsed.RawQuery = q.Encode()
|
||
if _, err := url.ParseRequestURI(parsed.String()); err != nil {
|
||
return "", err
|
||
}
|
||
return parsed.String(), nil
|
||
}
|