link tweaks

core.go

@@ -26,9 +26,7 @@ var ImageSuffixes = []string{
 	".bmp", ".webp",
 }
 
-var ImageBlackList = []string{
-
-}
+var ImageBlackList = []string{}
 
 var Rdb *redis.Client
 var Pgdb *pgx.ConnPool
@@ -97,22 +95,6 @@ func isHttpOk(code int) bool {
 	return code >= 200 && code < 300
 }
 
-func IsImageLink(link string) bool {
-	u, err := url.Parse(link)
-	if err != nil {
-		return false
-	}
-
-	path := strings.ToLower(u.Path)
-	for _, suffix := range ImageSuffixes {
-		if strings.HasSuffix(path, suffix) {
-			return true
-		}
-	}
-
-	return false
-}
-
 func FetchJson(link string, v interface{}, raw *[]byte, headers ...[]string) error {
 
 	body, err := Fetch(link, headers...)

link.go

@@ -2,7 +2,9 @@ package imhashdb
 
 import (
 	"go.uber.org/zap"
+	"net/url"
 	"regexp"
+	"strings"
 	"time"
 )
 
@@ -14,15 +16,36 @@ type ImgurImgResp struct {
 }
 
 type ImgurImg struct {
-	Link        string   `json:"link"`
+	Link string `json:"link"`
 }
 
 type ImgurAlbumResp struct {
 	Data struct {
-		Images      []ImgurImg `json:"images"`
+		Images []ImgurImg `json:"images"`
 	} `json:"data"`
 }
 
+func IsImageLink(link string) bool {
+
+	if strings.HasPrefix(link, "https://i.reddituploads.com") {
+		return true
+	}
+
+	u, err := url.Parse(link)
+	if err != nil {
+		return false
+	}
+
+	path := strings.ToLower(u.Path)
+	for _, suffix := range ImageSuffixes {
+		if strings.HasSuffix(path, suffix) {
+			return true
+		}
+	}
+
+	return false
+}
+
 func handleImgurLink(link string, meta *[]Meta) []string {
 
 	if ReImgurImg.MatchString(link) {