Jay
53cb949d2b
Before, the entire url was being lowercased, which mutilates various case-sensitive parts. Now, only the host gets lowercased, and the test verify that the userinfo, path, and query parts preserve their original case.
47 lines
842 B
Go
47 lines
842 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(input)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// lowercase host
|
|
parsed.Host = strings.ToLower(parsed.Host)
|
|
|
|
host := parsed.Hostname()
|
|
port := parsed.Port()
|
|
|
|
// strip default ports
|
|
if (parsed.Scheme == "wss" && port == "443") ||
|
|
(parsed.Scheme == "ws" && port == "80") {
|
|
parsed.Host = host
|
|
}
|
|
|
|
parsed.Path = strings.TrimRight(parsed.Path, "/")
|
|
|
|
return parsed.String(), nil
|
|
|
|
}
|