From 903d4ca99b5cd0e7717f460edcb66cce0a20f3b1 Mon Sep 17 00:00:00 2001
From: rramiachraf <51409801+rramiachraf@users.noreply.github.com>
Date: Mon, 4 Mar 2024 19:49:03 +0100
Subject: [PATCH] fix images not working on annotations

---
 handlers/annotations.go | 17 +++++------------
 1 file changed, 5 insertions(+), 12 deletions(-)

diff --git a/handlers/annotations.go b/handlers/annotations.go
index 2696c98..a45785a 100644
--- a/handlers/annotations.go
+++ b/handlers/annotations.go
@@ -20,19 +20,11 @@ func Annotations(l *logrus.Logger) http.HandlerFunc {
 		id := mux.Vars(r)["id"]
 
 		if data, err := getCache[data.Annotation](id); err == nil {
+			encoder := json.NewEncoder(w)
 
-			response, err := json.Marshal(data)
-
-			if err != nil {
-				l.Errorf("could not marshal json: %s\n", err)
-				w.WriteHeader(http.StatusInternalServerError)
-				views.ErrorPage(500, "something went wrong").Render(context.Background(), w)
-				return
-			}
 			w.Header().Set("content-type", "application/json")
-			_, err = w.Write(response)
-			if err != nil {
-				l.Errorln("Error sending response: ", err)
+			if err = encoder.Encode(&data); err != nil {
+				l.Errorln(err)
 			}
 			return
 		}
@@ -96,7 +88,8 @@ func Annotations(l *logrus.Logger) http.HandlerFunc {
 }
 
 func cleanBody(body string) string {
-	var withCleanedImageLinks = strings.Replace(body, "https://images.rapgenius.com/", "/images/", -1)
+	var withCleanedImageLinks = strings.ReplaceAll(body, "https://images.rapgenius.com/", "/images/")
+	withCleanedImageLinks = strings.ReplaceAll(body, "https://images.genius.com/", "/images/")
 
 	var re = regexp.MustCompile(`https?:\/\/[a-z]*.?genius.com`)
 	var withCleanedLinks = re.ReplaceAllString(withCleanedImageLinks, "")