Skip grid if grid image only has 1 img

Wikidepia · Jul 6, 2024 · a0d9c7f · a0d9c7f
1 parent 47fc5af
commit a0d9c7f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/handlers/grid.go b/handlers/grid.go
@@ -141,10 +141,6 @@ func Grid() fiber.Handler {
 			return err
 		}
 
-		if len(item.Medias) == 1 {
-			return c.Redirect("/images/" + postID + "/1")
-		}
-
 		// Filter media only include image
 		var mediaURLs []string
 		for _, media := range item.Medias {
@@ -154,6 +150,10 @@ func Grid() fiber.Handler {
 			mediaURLs = append(mediaURLs, media.URL)
 		}
 
+		if len(item.Medias) == 1 || len(mediaURLs) == 1 {
+			return c.Redirect("/images/" + postID + "/1")
+		}
+
 		var wg sync.WaitGroup
 		images := make([]image.Image, len(mediaURLs))
 		client := http.Client{Transport: transport, Timeout: timeout}