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" "code.nonshy.com/nonshy/website/pkg/worker" ) // 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 m["NavChatStatistics"] = worker.GetChatStatistics() // 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); 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); 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 } }