42 lines
770 B
Go
42 lines
770 B
Go
package transport
|
|
|
|
import (
|
|
"errors"
|
|
"net/url"
|
|
"strings"
|
|
)
|
|
|
|
var InvalidProtocol = errors.New("URL must use ws:// or wss:// scheme")
|
|
|
|
func ParseURL(urlStr string) (*url.URL, error) {
|
|
parsedURL, err := url.Parse(urlStr)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if parsedURL.Scheme != "ws" && parsedURL.Scheme != "wss" {
|
|
return nil, InvalidProtocol
|
|
}
|
|
|
|
return parsedURL, nil
|
|
}
|
|
|
|
func NormalizeURL(input string) (string, error) {
|
|
parsed, err := ParseURL(strings.ToLower(input))
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
host := parsed.Hostname()
|
|
port := parsed.Port()
|
|
if (parsed.Scheme == "wss" && port == "443") ||
|
|
(parsed.Scheme == "ws" && port == "80") {
|
|
parsed.Host = host
|
|
}
|
|
|
|
parsed.Path = strings.TrimRight(parsed.Path, "/")
|
|
|
|
return parsed.String(), nil
|
|
|
|
}
|