Add data
This commit is contained in:
6
data.go
Normal file
6
data.go
Normal file
@ -0,0 +1,6 @@
|
||||
package budgeteer
|
||||
|
||||
type Data interface {
|
||||
GetToken() Token
|
||||
GetBudgets() []Budget
|
||||
}
|
23
http/data.go
Normal file
23
http/data.go
Normal file
@ -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
|
||||
}
|
Reference in New Issue
Block a user