diff --git a/pkg/provider/main.go b/pkg/provider/main.go index 4ad19a2..e3f92a1 100644 --- a/pkg/provider/main.go +++ b/pkg/provider/main.go @@ -138,15 +138,7 @@ func (p *provider) retrieveSubdomainIP(addr string) (string, error) { return "", fmt.Errorf("%v: %v", utils.ErrIpLenght, ips) } - ip := ips[0].String() - if strings.Contains(ip, ":") { - ip, _, err = net.SplitHostPort(ip) - if err != nil { - return "", fmt.Errorf("%v: %v", utils.ErrSplitAddr, ip) - } - } - - return ip, nil + return ips[0].String(), nil } func (p *provider) checkResponse(body []byte, tokenBased bool, ip string) error { diff --git a/pkg/watcher/watcher.go b/pkg/watcher/watcher.go index 89e4f6d..662c52d 100644 --- a/pkg/watcher/watcher.go +++ b/pkg/watcher/watcher.go @@ -120,7 +120,7 @@ func (w *watcher) retrieveServerIP() (string, error) { } var ip string - if rsp.Header.Get("content-type") == "application/json" { + if strings.Contains(rsp.Header.Get("content-type"), "application/json") { body := make(map[string]interface{}) if err := json.Unmarshal(b, &body); err != nil { return "", fmt.Errorf("failed to unmarshal JSON body: %v", err)