155 lines
3.1 KiB
Go
155 lines
3.1 KiB
Go
package templateHelper
|
|
|
|
import (
|
|
"html/template"
|
|
"net/http"
|
|
"strings"
|
|
"time"
|
|
|
|
"synlotto-website/internal/models"
|
|
|
|
"github.com/alexedwards/scs/v2"
|
|
"github.com/justinas/nosurf"
|
|
)
|
|
|
|
// ToDo should these structs be here?
|
|
type siteMeta struct {
|
|
Name string
|
|
CopyrightYearStart int
|
|
}
|
|
|
|
var meta siteMeta
|
|
|
|
func InitSiteMeta(name string, yearStart, yearEnd int) {
|
|
meta = siteMeta{
|
|
Name: name,
|
|
CopyrightYearStart: yearStart,
|
|
}
|
|
}
|
|
|
|
var sm *scs.SessionManager
|
|
|
|
func InitSessionManager(manager *scs.SessionManager) {
|
|
sm = manager
|
|
}
|
|
|
|
func TemplateContext(w http.ResponseWriter, r *http.Request, data models.TemplateData) map[string]interface{} {
|
|
return map[string]interface{}{
|
|
"CSRFToken": nosurf.Token(r),
|
|
"User": data.User,
|
|
"IsAdmin": data.IsAdmin,
|
|
"NotificationCount": data.NotificationCount,
|
|
"Notifications": data.Notifications,
|
|
"MessageCount": data.MessageCount,
|
|
"Messages": data.Messages,
|
|
"SiteName": meta.Name,
|
|
"CopyrightYearStart": meta.CopyrightYearStart,
|
|
}
|
|
}
|
|
|
|
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 },
|
|
"sub": 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,
|
|
"truncate": func(s string, max int) string {
|
|
if len(s) <= max {
|
|
return s
|
|
}
|
|
return s[:max] + "..."
|
|
},
|
|
"PageRange": PageRange,
|
|
"now": time.Now,
|
|
"humanTime": func(v interface{}) string {
|
|
switch t := v.(type) {
|
|
case time.Time:
|
|
return t.Local().Format("02 Jan 2006 15:04")
|
|
case string:
|
|
parsed, err := time.Parse(time.RFC3339, t)
|
|
if err == nil {
|
|
return parsed.Local().Format("02 Jan 2006 15:04")
|
|
}
|
|
return t
|
|
default:
|
|
return ""
|
|
}
|
|
},
|
|
"rangeClass": rangeClass,
|
|
}
|
|
}
|
|
|
|
func LoadTemplateFiles(name string, files ...string) *template.Template {
|
|
shared := []string{
|
|
"web/templates/main/layout.html",
|
|
"web/templates/main/topbar.html",
|
|
"web/templates/main/footer.html",
|
|
}
|
|
all := append(shared, files...)
|
|
return template.Must(template.New(name).Funcs(TemplateFuncs()).ParseFiles(all...))
|
|
}
|
|
|
|
func SetFlash(r *http.Request, message string) {
|
|
if sm != nil {
|
|
sm.Put(r.Context(), "flash", message)
|
|
}
|
|
}
|
|
|
|
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 PageRange(current, total int) []int {
|
|
var pages []int
|
|
for i := 1; i <= total; i++ {
|
|
pages = append(pages, i)
|
|
}
|
|
return pages
|
|
}
|
|
|
|
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"
|
|
}
|
|
}
|