1 files changed, 23 insertions, 0 deletions

diff --git a/modules/grils/gril.go b/modules/grils/gril.go
index edf25a9..a450647 100644
--- a/modules/grils/gril.go
+++ b/modules/grils/gril.go
@@ -2,7 +2,9 @@ package grils
 
 import (
         "database/sql"
+        "fmt"
         "github.com/lib/pq"
+        "os"
         "time"
 )
 
@@ -25,3 +27,24 @@ type Gril struct {
         UpdatedAt  time.Time
         Lists      []int
 }
+
+func (g *Gril) ImagePath() 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 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
+                        }
+                }
+        }
+        return ""
+}