2 files changed, 23 insertions, 23 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 ""
+}
diff --git a/modules/grils/grils.go b/modules/grils/grils.go
index 9bfd6e6..a1c84c7 100644
--- a/modules/grils/grils.go
+++ b/modules/grils/grils.go
@@ -8,7 +8,6 @@ import (
         "github.com/julienschmidt/httprouter"
         "log"
         "net/http"
-        "os"
         "regexp"
         "strconv"
         "strings"
@@ -174,28 +173,6 @@ func (m *GrilsModule) viewGril(w http.ResponseWriter, r *http.Request, p httprou
         data["user"] = user
         data["gril"] = gril
 
-        // get the file ending
-        exts := []string{"png", "jpg", "gif", "jpeg"}
-        var imagePath string
-        for _, ext := range exts {
-                path := fmt.Sprintf("assets/img/gril/%d/0.%s", gril.ID, ext)
-                if _, err := os.Stat(path); err == nil {
-                        imagePath = path
-                        break
-                }
-        }
-
-        if len(imagePath) == 0 {
-                for _, ext := range exts {
-                        path := fmt.Sprintf("assets/img/gril/%d/1.%s", gril.ID, ext)
-                        if _, err := os.Stat(path); err == nil {
-                                imagePath = path
-                                break
-                        }
-                }
-        }
-        data["imagePath"] = "/" + imagePath
-
         m.g.Renderer.RenderPage("gril", w, data)
 }