budgeteer/http/data.go
2016-12-27 00:45:30 +01:00

24 lines
454 B
Go

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
}