fixed favicon bug

zan8in · Mar 8, 2023 · 093ed9a · 093ed9a
1 parent 217c2cb
commit 093ed9a
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/pkg/favicon/favicon.go b/pkg/favicon/favicon.go
@@ -32,16 +32,18 @@ func HandleFaviconHash(target, body string) (string, error) {
 	if len(potentialURLs) > 0 {
 		for _, potentialURL := range potentialURLs {
 			if len(potentialURL) > 0 {
-				if potentialURL[:2] == "//" {
+				if strings.HasPrefix(potentialURL, "//") {
 					faviconPath = "http:" + potentialURL
-				} else {
-					if potentialURL[:4] == "http" {
-						faviconPath = potentialURL
-					} else {
-						faviconPath = target + "/" + strings.Trim(potentialURL, "/")
-					}
+					return faviconPath, nil
+				} else if strings.HasPrefix(potentialURL, "http") {
+					faviconPath = potentialURL
+					return faviconPath, nil
+				} else if strings.HasSuffix(potentialURL, ".ico") ||
+					strings.HasSuffix(potentialURL, ".png") ||
+					strings.HasSuffix(potentialURL, ".jpg") {
+					faviconPath = target + "/" + strings.Trim(potentialURL, "/")
+					return faviconPath, nil
 				}
-				return faviconPath, nil
 			}
 		}
 	}