1 files changed, 13 insertions, 10 deletions

diff --git a/modules/grils/gril.go b/modules/grils/gril.go
index a450647..fe316cd 100644
--- a/modules/grils/gril.go
+++ b/modules/grils/gril.go
@@ -28,23 +28,26 @@ type Gril struct {
         Lists      []int
 }
 
-func (g *Gril) ImagePath() string {
+func (g *Gril) ImagePath(prioritizeThumbnail bool) string {
+        var big string
         exts := []string{"png", "jpg", "gif", "jpeg"}
-        var imagePath string
         for _, ext := range exts {
                 path := fmt.Sprintf("assets/img/gril/%d/0.%s", g.ID, ext)
                 if _, err := os.Stat(path); err == nil {
-                        return path
+                        if prioritizeThumbnail {
+                                big = path
+                        } else {
+                                return path
+                        }
                 }
         }
 
-        if len(imagePath) == 0 {
-                for _, ext := range exts {
-                        path := fmt.Sprintf("assets/img/gril/%d/1.%s", g.ID, ext)
-                        if _, err := os.Stat(path); err == nil {
-                                return path
-                        }
+        for _, ext := range exts {
+                path := fmt.Sprintf("assets/img/gril/%d/1.%s", g.ID, ext)
+                if _, err := os.Stat(path); err == nil {
+                        return path
                 }
         }
-        return ""
+
+        return big
 }