From a3e779e08804024cfd6c1812bc6b11cd4b2b5df7 Mon Sep 17 00:00:00 2001
From: kolaente <k@knt.li>
Date: Wed, 25 Aug 2021 18:30:21 +0200
Subject: [PATCH] Add buckets to export

---
 pkg/models/export.go | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/pkg/models/export.go b/pkg/models/export.go
index 7f38ccbe6..5630f249d 100644
--- a/pkg/models/export.go
+++ b/pkg/models/export.go
@@ -56,7 +56,6 @@ func ExportUserData(u *user.User) (err error) {
 	if err != nil {
 		return err
 	}
-	// Buckets
 	// Task attachment files
 	// Saved filters
 	// Subscription Status
@@ -109,6 +108,16 @@ func exportListsAndTasks(s *xorm.Session, u *user.User, wr *zip.Writer) (err err
 		listMap[t.ListID].Tasks = append(listMap[t.ListID].Tasks, t)
 	}
 
+	buckets := []*Bucket{}
+	err = s.In("list_id", listIDs).Find(&buckets)
+	if err != nil {
+		return
+	}
+
+	for _, b := range buckets {
+		listMap[b.ListID].Buckets = append(listMap[b.ListID].Buckets, b)
+	}
+
 	data, err := json.Marshal(namespaces)
 	if err != nil {
 		return err