117 lines
3.3 KiB
Go
117 lines
3.3 KiB
Go
package templates
|
|
|
|
import (
|
|
"net/http"
|
|
"time"
|
|
|
|
"code.nonshy.com/nonshy/website/pkg/config"
|
|
"code.nonshy.com/nonshy/website/pkg/log"
|
|
"code.nonshy.com/nonshy/website/pkg/models"
|
|
"code.nonshy.com/nonshy/website/pkg/session"
|
|
)
|
|
|
|
// MergeVars mixes in globally available template variables. The http.Request is optional.
|
|
func MergeVars(r *http.Request, m map[string]interface{}) {
|
|
m["Title"] = config.Title
|
|
m["BuildHash"] = config.RuntimeBuild
|
|
m["BuildDate"] = config.RuntimeBuildDate
|
|
m["Subtitle"] = config.Subtitle
|
|
m["YYYY"] = time.Now().Year()
|
|
|
|
if r == nil {
|
|
return
|
|
}
|
|
|
|
m["Request"] = r
|
|
}
|
|
|
|
// MergeUserVars mixes in global template variables: LoggedIn and CurrentUser. The http.Request is optional.
|
|
func MergeUserVars(r *http.Request, m map[string]interface{}) {
|
|
// Defaults
|
|
m["LoggedIn"] = false
|
|
m["CurrentUser"] = nil
|
|
m["SessionImpersonated"] = false
|
|
|
|
// User notification counts for nav bar.
|
|
m["NavUnreadMessages"] = 0 // New messages
|
|
m["NavFriendRequests"] = 0 // Friend requests
|
|
m["NavUnreadNotifications"] = 0 // general notifications
|
|
m["NavTotalNotifications"] = 0 // Total of above
|
|
|
|
// Admin notification counts for nav bar.
|
|
m["NavCertificationPhotos"] = 0 // Cert. photos needing approval
|
|
m["NavAdminFeedback"] = 0 // Unacknowledged feedback
|
|
m["NavAdminNotifications"] = 0 // Total of above
|
|
|
|
if r == nil {
|
|
return
|
|
}
|
|
|
|
m["SessionImpersonated"] = session.Impersonated(r)
|
|
|
|
if user, err := session.CurrentUser(r); err == nil {
|
|
m["LoggedIn"] = true
|
|
m["CurrentUser"] = user
|
|
|
|
// Get user recent notifications.
|
|
/*notifPager := &models.Pagination{
|
|
Page: 1,
|
|
PerPage: 10,
|
|
}
|
|
if notifs, err := models.PaginateNotifications(user, notifPager); err == nil {
|
|
m["Notifications"] = notifs
|
|
}*/
|
|
|
|
// Collect notification counts.
|
|
var (
|
|
// For users
|
|
countMessages int64
|
|
countFriendReqs int64
|
|
countNotifications int64
|
|
|
|
// For admins
|
|
countCertPhotos int64
|
|
countFeedback int64
|
|
)
|
|
|
|
// Get unread message count.
|
|
if count, err := models.CountUnreadMessages(user.ID); err == nil {
|
|
m["NavUnreadMessages"] = count
|
|
countMessages = count
|
|
} else {
|
|
log.Error("MergeUserVars: couldn't CountUnreadMessages for %d: %s", user.ID, err)
|
|
}
|
|
|
|
// Get friend request count.
|
|
if count, err := models.CountFriendRequests(user.ID); err == nil {
|
|
m["NavFriendRequests"] = count
|
|
countFriendReqs = count
|
|
} else {
|
|
log.Error("MergeUserVars: couldn't CountFriendRequests for %d: %s", user.ID, err)
|
|
}
|
|
|
|
// Count other notifications.
|
|
if count, err := models.CountUnreadNotifications(user.ID); err == nil {
|
|
m["NavUnreadNotifications"] = count
|
|
countNotifications = count
|
|
} else {
|
|
log.Error("MergeUserVars: couldn't CountFriendRequests for %d: %s", user.ID, err)
|
|
}
|
|
|
|
// Are we admin?
|
|
if user.IsAdmin {
|
|
// Any pending certification photos or feedback?
|
|
countCertPhotos = models.CountCertificationPhotosNeedingApproval()
|
|
countFeedback = models.CountUnreadFeedback()
|
|
m["NavCertificationPhotos"] = countCertPhotos
|
|
m["NavAdminFeedback"] = countFeedback
|
|
|
|
// Total notification count for admin actions.
|
|
m["NavAdminNotifications"] = countCertPhotos + countFeedback
|
|
}
|
|
|
|
// Total count for user notifications.
|
|
m["NavTotalNotifications"] = countMessages + countFriendReqs + countNotifications + countCertPhotos + countFeedback
|
|
}
|
|
}
|