From 5754b97e4da34cdba86ea049ebcbdb0469222f6a Mon Sep 17 00:00:00 2001
From: Jan Bader <jan.bader@ub-software.de>
Date: Tue, 27 Dec 2016 00:28:52 +0100
Subject: [PATCH] Add data

---
 data.go      |  6 ++++++
 http/data.go | 23 +++++++++++++++++++++++
 2 files changed, 29 insertions(+)
 create mode 100644 data.go
 create mode 100644 http/data.go

diff --git a/data.go b/data.go
new file mode 100644
index 0000000..45f1284
--- /dev/null
+++ b/data.go
@@ -0,0 +1,6 @@
+package budgeteer
+
+type Data interface {
+	GetToken() Token
+	GetBudgets() []Budget
+}
diff --git a/http/data.go b/http/data.go
new file mode 100644
index 0000000..b866764
--- /dev/null
+++ b/http/data.go
@@ -0,0 +1,23 @@
+package http
+
+import "git.javil.eu/jacob1123/budgeteer/jwt"
+import "git.javil.eu/jacob1123/budgeteer"
+
+type TemplateData struct {
+	token         jwt.Token
+	budgetService budgeteer.BudgetService
+}
+
+func (d *TemplateData) GetToken() jwt.Token {
+	return d.token
+}
+
+func (d *TemplateData) GetBudgets() []budgeteer.Budget {
+	userID := d.token.GetID()
+	budgets, err := d.budgetService.BudgetsForUser(userID)
+	if err != nil {
+		return nil
+	}
+
+	return budgets
+}