Merge branch 'master' of projekte.fagott.pw:grilist

2 files changed, 25 insertions, 4 deletions

diff --git a/modules/grils/gril.go b/modules/grils/gril.go
index c8c6538..89dd7c4 100644
--- a/modules/grils/gril.go
+++ b/modules/grils/gril.go
@@ -3,6 +3,7 @@ package grils
 import (
         "database/sql"
         "fmt"
+        "os"
         "strconv"
         "strings"
         "time"
@@ -39,3 +40,27 @@ func (g *Gril) Slug() string {
                 g.ID,
                 strings.Replace(g.RomajiName, " ", "", -1))
 }
+
+func (g *Gril) ImagePath(prioritizeThumbnail bool) string {
+        var big string
+        exts := []string{"png", "jpg", "gif", "jpeg"}
+        for _, ext := range exts {
+                path := fmt.Sprintf("assets/img/gril/%d/0.%s", g.ID, ext)
+                if _, err := os.Stat(path); err == nil {
+                        if prioritizeThumbnail {
+                                big = path
+                        } else {
+                                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 big
+}
diff --git a/modules/grils/grils.go b/modules/grils/grils.go
index 46d0fb3..0df93a0 100644
--- a/modules/grils/grils.go
+++ b/modules/grils/grils.go
@@ -44,10 +44,6 @@ func (m *GrilsModule) Init(g *grilist.Grilist) {
         m.g.Router.GET("/gril/:id/*rest", m.viewGril)
 }
 
-func (m *GrilsModule) Interface() interface{} {
-        return m
-}
-
 func (m *GrilsModule) getGrils(whereClause string, params ...interface{}) ([]*Gril, error) {
         var grils []*Gril