From 35d96e497979609c0158092412714ac4ec4adc8f Mon Sep 17 00:00:00 2001 From: Bel LaPointe Date: Fri, 12 Oct 2018 11:07:08 -0600 Subject: [PATCH] toWhitelist gets base domain --- main.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/main.go b/main.go index 723f4b4..78886db 100644 --- a/main.go +++ b/main.go @@ -134,6 +134,12 @@ func fixScheme(u *url.URL) { func toWhitelist(okay []string, host string) bool { host = strings.Split(host, ":")[0] host = strings.Replace(host, "www.", "", -1) + host = strings.Replace(host, "http://", "", -1) + host = strings.Replace(host, "https://", "", -1) + hosts := strings.Split(host, ".") + if len(hosts) > 1 { + host = hosts[len(hosts)-2] + "." + hosts[len(hosts)-1] + } for i := range okay { if strings.Contains(okay[i], host) { return true