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