Use Token field instead of GetToken

This commit is contained in:
Jan Bader 2016-12-27 00:47:36 +01:00
parent e525fdc928
commit 1ee88466f2
3 changed files with 3 additions and 9 deletions

View File

@ -1,6 +1,5 @@
package budgeteer package budgeteer
type Data interface { type Data interface {
GetToken() Token GetBudgets() []*Budget
GetBudgets() []Budget
} }

View File

@ -1,17 +1,12 @@
package http package http
import "git.javil.eu/jacob1123/budgeteer/jwt"
import "git.javil.eu/jacob1123/budgeteer" import "git.javil.eu/jacob1123/budgeteer"
type TemplateData struct { type TemplateData struct {
token jwt.Token Token budgeteer.Token
budgetService budgeteer.ModelService budgetService budgeteer.ModelService
} }
func (d *TemplateData) GetToken() jwt.Token {
return d.token
}
func (d TemplateData) GetBudgets() []*budgeteer.Budget { func (d TemplateData) GetBudgets() []*budgeteer.Budget {
userID := d.Token.GetID() userID := d.Token.GetID()
budgets, err := d.budgetService.BudgetsForUser(userID) budgets, err := d.budgetService.BudgetsForUser(userID)

View File

@ -8,7 +8,7 @@
</head> </head>
<body> <body>
<div class="container" id="head"> <div class="container" id="head">
Budgeteer - {{.GetName}} Budgeteer - {{.Token.GetName}}
</div> </div>
<div class="container col-lg-12" id="content"> <div class="container col-lg-12" id="content">
{{range .GetBudgets}} {{range .GetBudgets}}