Merge branch 'master' of ssh://mc.telamitto.org/grilist

1 files changed, 12 insertions, 2 deletions

diff --git a/modules/grils/gril.go b/modules/grils/gril.go
index af84d04..7b782b3 100644
--- a/modules/grils/gril.go
+++ b/modules/grils/gril.go
@@ -16,6 +16,7 @@ type DataSource int
 const (
         DataSourceACD DataSource = iota
         DataSourceAniDB
+        DataSourceAnilist
 )
 
 type Gril struct {
@@ -42,10 +43,19 @@ func (g *Gril) Slug() string {
 }
 
 func ImagePath(id int, prioritizeThumbnail bool) string {
+        // Image Priority
+        // without prioritizeThumbnail: Anilist > ACD (big) > ACD (thumbnail)
+        // with prioritizeThumbnail: Anilist > ACD (thumbnail) > ACD (big)
+        
+        anilistPath := fmt.Sprintf("assets/img/gril/%d/%d.jpg", DataSourceAnilist, id)
+        if _, err := os.Stat(anilistPath); err == nil {
+                return fmt.Sprintf(anilistPath)
+        }
+        
         var big string
         exts := []string{"png", "jpg", "gif", "jpeg"}
         for _, ext := range exts {
-                path := fmt.Sprintf("assets/img/gril/%d/0.%s", id, ext)
+                path := fmt.Sprintf("assets/img/gril/%d/%d/0.%s", DataSourceACD, id, ext)
                 if _, err := os.Stat(path); err == nil {
                         if prioritizeThumbnail {
                                 big = path
@@ -56,7 +66,7 @@ func ImagePath(id int, prioritizeThumbnail bool) string {
         }
 
         for _, ext := range exts {
-                path := fmt.Sprintf("assets/img/gril/%d/1.%s", id, ext)
+                path := fmt.Sprintf("assets/img/gril/%d/%d/1.%s", DataSourceACD, id, ext)
                 if _, err := os.Stat(path); err == nil {
                         return path
                 }