diff --git a/main.go b/main.go index 18d252f..d39cb6a 100644 --- a/main.go +++ b/main.go @@ -47,20 +47,18 @@ func main() { tg.SendMessage(chatId, "failed getting domains: " + err.Error()) } - polish := false - ndc := false + match := false for _, domain := range allDomains { - if strings.HasSuffix(domain, ".pl") { - polish = true + for _, tld := range []string{".pl", ".app", ".dev", ".cloud"} { + if strings.HasSuffix(domain, tld) { + match = true + } } if strings.HasSuffix(domain, "ndc.pl") { - ndc = true + match = false } } - if !polish { - continue - } - if ndc { + if !match { continue }