From f36fe1b1c0ba467c3babb03317111ed059caf4bb Mon Sep 17 00:00:00 2001
From: Jan Bader <jan.bader@ub-software.de>
Date: Tue, 27 Dec 2016 00:27:52 +0100
Subject: [PATCH] Use id to load budgets

---
 budget.go                 | 2 +-
 postgres/budgetservice.go | 5 ++++-
 token.go                  | 1 +
 3 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/budget.go b/budget.go
index c0084a3..4bf8a22 100644
--- a/budget.go
+++ b/budget.go
@@ -20,5 +20,5 @@ type UserBudget struct {
 // BudgetService provides Methods for CRUD of Budgets
 type BudgetService interface {
 	Budget(id string) (Budget, error)
-	BudgetsForUser(user User) ([]Budget, error)
+	BudgetsForUser(id string) ([]Budget, error)
 }
diff --git a/postgres/budgetservice.go b/postgres/budgetservice.go
index d3d8d5b..3da9cb3 100644
--- a/postgres/budgetservice.go
+++ b/postgres/budgetservice.go
@@ -12,7 +12,10 @@ func (s *Repository) Budget(id string) (*budgeteer.Budget, error) {
 	return b, nil
 }
 
-func (s *Repository) BudgetsForUser(user budgeteer.User) ([]*budgeteer.Budget, error) {
+func (s *Repository) BudgetsForUser(id string) ([]*budgeteer.Budget, error) {
+	user := &budgeteer.User{
+		ID: id,
+	}
 	err := s.DB.Model(&user).
 		Column("user.*", "Budgets").
 		Select()
diff --git a/token.go b/token.go
index b5eea8c..8f40ba9 100644
--- a/token.go
+++ b/token.go
@@ -5,6 +5,7 @@ type Token interface {
 	GetUsername() string
 	GetName() string
 	GetExpiry() float64
+	GetID() string
 }
 
 // TokenVerifier verifies a Token