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

2 files changed, 25 insertions, 0 deletions

diff --git a/modules/grils/gril.go b/modules/grils/gril.go
index 9fd3be5..edf25a9 100644
--- a/modules/grils/gril.go
+++ b/modules/grils/gril.go
@@ -23,4 +23,5 @@ type Gril struct {
         Tags       []string
         ForeignIDs map[DataSource]int
         UpdatedAt  time.Time
+        Lists      []int
 }
diff --git a/modules/grils/grils.go b/modules/grils/grils.go
index e2eb543..a38f41f 100644
--- a/modules/grils/grils.go
+++ b/modules/grils/grils.go
@@ -74,6 +74,26 @@ func (m *GrilsModule) getGrils(whereClause string, params ...interface{}) ([]*Gr
         return grils, nil
 }
 
+func (m *GrilsModule) GetListsOfGril(gril *Gril) error {
+        rows, err := m.g.DB.Query(`SELECT list_id FROM grilist.lists_grils WHERE gril_id = $1`, gril.ID)
+        if err != nil {
+                return err
+        }
+
+        defer rows.Close()
+        for rows.Next() {
+                var listID int
+                if err := rows.Scan(&listID); err != nil {
+                        log.Println(err)
+                        continue
+                }
+
+                gril.Lists = append(gril.Lists, listID)
+        }
+
+        return nil
+}
+
 func (m *GrilsModule) ProvideDashboardData(user *charakterin.User) []grilist.DashboardCategory {
         var categories []grilist.DashboardCategory
 
@@ -137,6 +157,10 @@ func (m *GrilsModule) viewGril(w http.ResponseWriter, r *http.Request, p httprou
                 return
         }
 
+        if err := m.GetListsOfGril(gril); err != nil {
+                log.Println(err)
+        }
+
         data := m.g.Renderer.DefaultData()
         data["user"] = user
         data["gril"] = gril