fix localhost trick from 8aaf5b.

normalize.go

@@ -15,13 +15,12 @@ func NormalizeURL(u string) string {
 	u = strings.TrimSpace(u)
 	u = strings.ToLower(u)
 
-	if strings.HasPrefix(u, "localhost") == true {
+	if strings.Split(u, ":")[0] == "localhost" {
 		u = "ws://" + u
-	}
-
-	if !strings.HasPrefix(u, "http") && !strings.HasPrefix(u, "ws") {
+	} else if !strings.HasPrefix(u, "http") && !strings.HasPrefix(u, "ws") {
 		u = "wss://" + u
 	}
+
 	p, err := url.Parse(u)
 	if err != nil {
 		return ""

normalize_test.go

@@ -23,11 +23,11 @@ var urlTests = []urlTest{
 	{"x.com/?x=23", "wss://x.com?x=23"},
 	{"localhost:4036", "ws://localhost:4036"},
 	{"localhost:4036/relay", "ws://localhost:4036/relay"},
+	{"localhostmagnanimus.com", "wss://localhostmagnanimus.com"},
 	{NormalizeURL("localhost:4036/relay"), "ws://localhost:4036/relay"},
 }
 
 func TestNormalizeURL(t *testing.T) {
-
 	for _, test := range urlTests {
 		if output := NormalizeURL(test.url); output != test.expected {
 			t.Errorf("Output '%s' not equal to expected '%s'", output, test.expected)