103 lines
1.8 KiB
Go
103 lines
1.8 KiB
Go
package helpers
|
|
|
|
import (
|
|
"html/template"
|
|
"net/http"
|
|
"strings"
|
|
"synlotto-website/models"
|
|
|
|
"github.com/gorilla/csrf"
|
|
)
|
|
|
|
func TemplateFuncs() template.FuncMap {
|
|
return template.FuncMap{
|
|
"plus1": func(i int) int { return i + 1 },
|
|
"minus1": func(i int) int {
|
|
if i > 1 {
|
|
return i - 1
|
|
}
|
|
return 0
|
|
},
|
|
"mul": func(a, b int) int { return a * b },
|
|
"add": func(a, b int) int { return a + b },
|
|
"min": func(a, b int) int {
|
|
if a < b {
|
|
return a
|
|
}
|
|
return b
|
|
},
|
|
"intVal": func(p *int) int {
|
|
if p == nil {
|
|
return 0
|
|
}
|
|
return *p
|
|
},
|
|
"inSlice": InSlice,
|
|
"lower": lower,
|
|
"rangeClass": rangeClass,
|
|
}
|
|
}
|
|
|
|
func TemplateContext(w http.ResponseWriter, r *http.Request) map[string]interface{} {
|
|
session, _ := GetSession(w, r)
|
|
|
|
var flash string
|
|
if f, ok := session.Values["flash"].(string); ok {
|
|
flash = f
|
|
delete(session.Values, "flash")
|
|
session.Save(r, w)
|
|
}
|
|
|
|
var currentUser *models.User
|
|
var isAdmin bool
|
|
|
|
switch v := session.Values["user_id"].(type) {
|
|
case int:
|
|
currentUser = models.GetUserByID(v)
|
|
case int64:
|
|
currentUser = models.GetUserByID(int(v))
|
|
}
|
|
|
|
if currentUser != nil {
|
|
isAdmin = currentUser.IsAdmin
|
|
}
|
|
|
|
return map[string]interface{}{
|
|
"CSRFField": csrf.TemplateField(r),
|
|
"Flash": flash,
|
|
"User": currentUser,
|
|
"IsAdmin": isAdmin,
|
|
}
|
|
}
|
|
|
|
func InSlice(n int, list []int) bool {
|
|
for _, v := range list {
|
|
if v == n {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func lower(input string) string {
|
|
return strings.ToLower(input)
|
|
}
|
|
|
|
func rangeClass(n int) string {
|
|
switch {
|
|
case n >= 1 && n <= 9:
|
|
return "01-09"
|
|
case n >= 10 && n <= 19:
|
|
return "10-19"
|
|
case n >= 20 && n <= 29:
|
|
return "20-29"
|
|
case n >= 30 && n <= 39:
|
|
return "30-39"
|
|
case n >= 40 && n <= 49:
|
|
return "40-49"
|
|
default:
|
|
return "50-plus"
|
|
}
|
|
}
|