keep: ensure URLs have IA-supported schemes types

1 files changed, 3 insertions, 1 deletions

diff --git a/keep.go b/keep.go
index 75b0fcb..1305779 100644
--- a/keep.go
+++ b/keep.go
@@ -180,7 +180,9 @@ func messageCreate(s *discordgo.Session, m *discordgo.MessageCreate) {
 
 		// Assess whether message part looks like a valid URL
 		u, err := url.Parse(w)
-		if err != nil || !u.IsAbs() || strings.IndexByte(u.Host, '.') <= 0 {
+		if err != nil || !u.IsAbs() || strings.IndexByte(u.Host, '.') <= 0 ||
+			(u.Scheme != "https" && u.Scheme != "http") {
+
 			continue
 		}