diff --git a/data.go b/data.go index 45f1284..19681c9 100644 --- a/data.go +++ b/data.go @@ -1,6 +1,5 @@ package budgeteer type Data interface { - GetToken() Token - GetBudgets() []Budget + GetBudgets() []*Budget } diff --git a/http/data.go b/http/data.go index 703fd9a..52b3690 100644 --- a/http/data.go +++ b/http/data.go @@ -1,17 +1,12 @@ package http -import "git.javil.eu/jacob1123/budgeteer/jwt" import "git.javil.eu/jacob1123/budgeteer" type TemplateData struct { - token jwt.Token + Token budgeteer.Token budgetService budgeteer.ModelService } -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) diff --git a/templates/dashboard.html b/templates/dashboard.html index a7ba99a..950812b 100644 --- a/templates/dashboard.html +++ b/templates/dashboard.html @@ -8,7 +8,7 @@
{{range .GetBudgets}}