1 files changed, 38 insertions, 12 deletions

diff --git a/modules/lists/lists.go b/modules/lists/lists.go
index 24d774e..343ba62 100644
--- a/modules/lists/lists.go
+++ b/modules/lists/lists.go
@@ -1,6 +1,7 @@
 package lists
 
 import (
+        "database/sql"
         "fagott.pw/charakterin"
         "fagott.pw/grilist/frontend"
         "fagott.pw/grilist/grilist"
@@ -90,17 +91,9 @@ func mkCard(title, description string, actions ...frontend.Action) frontend.Card
         return card
 }
 
-func (m *Module) ProvideDashboardData(user *charakterin.User) grilist.DashboardCategory {
-        data := grilist.DashboardCategory{
-                Title: "Meine Listen",
-        }
-
-        rows, err := m.g.DB.Query(`SELECT id, name, description FROM grilist.lists WHERE user_id = $1`, user.ID)
-        if err != nil {
-                return data
-        }
+func rowsToCard(rows *sql.Rows) []frontend.Card {
+        var cards []frontend.Card
 
-        defer rows.Close()
         for rows.Next() {
                 var id int
                 var title, description string
@@ -109,12 +102,45 @@ func (m *Module) ProvideDashboardData(user *charakterin.User) grilist.DashboardC
                         continue
                 }
 
-                data.Cards = append(data.Cards, mkCard(title, description, frontend.Action{"anguckieren", fmt.Sprintf("/lists/%d", id)}))
+                cards = append(cards, mkCard(title, description, frontend.Action{"anguckieren", fmt.Sprintf("/lists/%d", id)}))
         }
+        return cards
+}
+
+// ProvideDashboardData gibt Daten für das Dashboard bezogen auf den Benutzer zurück
+func (m *Module) ProvideDashboardData(user *charakterin.User) []grilist.DashboardCategory {
+        var categories []grilist.DashboardCategory
+
+        // Listen des Benutzers
+        rows, err := m.g.DB.Query(`SELECT id, name, description FROM grilist.lists WHERE user_id = $1`, user.ID)
+        if err != nil {
+                log.Println(err)
+                return []grilist.DashboardCategory{}
+        }
+        defer rows.Close()
+
+        categories = append(categories, grilist.DashboardCategory{
+                Title: "Meine Listen",
+                Cards: rowsToCard(rows),
+        })
+
+        // Neue Listen
+        rows, err = m.g.DB.Query(`SELECT id, name, description FROM grilist.lists ORDER BY id DESC LIMIT 5`)
+        if err != nil {
+                log.Println(err)
+                return categories
+        }
+        defer rows.Close()
+
+        categories = append(categories, grilist.DashboardCategory{
+                Title: "Neueste Listen",
+                Cards: rowsToCard(rows),
+        })
 
-        return data
+        return categories
 }
 
+// New erstellt eine neue Instanz des Modules
 func New() *Module {
         return &Module{}
 }