website/pkg/controller/photo/site_gallery.go

package photo

import (
	"net/http"

	"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/templates"
)

// SiteGallery controller (/photo/gallery) to view all members' public gallery pics.
func SiteGallery() http.HandlerFunc {
	tmpl := templates.Must("photo/gallery.html")

	// Whitelist for ordering options.
	var sortWhitelist = []string{
		"created_at desc",
		"created_at asc",
		"like_count desc",
		"comment_count desc",
		"views desc",
	}

	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// Query params.
		var (
			viewStyle = r.FormValue("view") // cards (default), full

			// Search filters.
			who              = r.FormValue("who")
			filterExplicit   = r.FormValue("explicit")
			filterVisibility = r.FormValue("visibility")
			adminView        = r.FormValue("admin_view") == "true"
			sort             = r.FormValue("sort")
			sortOK           bool
		)

		// Sort options.
		for _, v := range sortWhitelist {
			if sort == v {
				sortOK = true
				break
			}
		}
		if !sortOK {
			sort = sortWhitelist[0]
		}

		// Load the current user.
		currentUser, err := session.CurrentUser(r)
		if err != nil {
			session.FlashError(w, r, "Unexpected error: couldn't get CurrentUser")
		}

		// Is the current viewer shy?
		var (
			isShy         = currentUser.IsShy()
			myFriendCount = models.CountFriends(currentUser.ID)
		)

		// Defaults.
		if viewStyle != "full" {
			viewStyle = "cards"
		}
		if who == "" {
			// They didn't post a "Whose photos" filter, restore it from their last saved default.
			who = currentUser.GetProfileField("site_gallery_default")
		}
		if who != "friends" && who != "everybody" && who != "friends+private" && who != "likes" && who != "uncertified" {
			// Default Who setting should be Friends-only, unless you have no friends.
			if myFriendCount > 0 {
				who = "friends"
			} else {
				who = "everybody"
			}

			// Admin only who option.
			if who == "uncertified" && !currentUser.HasAdminScope(config.ScopePhotoModerator) {
				who = "friends"
			}
		}

		// Store their "Whose photos" filter on their page to default it for next time.
		currentUser.SetProfileField("site_gallery_default", who)

		// Admin scope warning.
		if adminView && !currentUser.HasAdminScope(config.ScopePhotoModerator) {
			session.FlashError(w, r, "Missing admin scope: %s", config.ScopePhotoModerator)
		}

		// Get the page of photos.
		pager := &models.Pagination{
			Page:    1,
			PerPage: config.PageSizeSiteGallery,
			Sort:    sort,
		}
		pager.ParsePage(r)
		photos, _ := models.PaginateGalleryPhotos(currentUser, models.Gallery{
			Explicit:    filterExplicit,
			Visibility:  filterVisibility,
			AdminView:   adminView,
			FriendsOnly: who == "friends",
			IsShy:       isShy || who == "friends+private",
			MyLikes:     who == "likes",
			Uncertified: who == "uncertified",
		}, pager)

		// Bulk load the users associated with these photos.
		var userIDs = []uint64{}
		for _, photo := range photos {
			userIDs = append(userIDs, photo.UserID)
		}
		userMap, err := models.MapUsers(currentUser, userIDs)
		if err != nil {
			session.FlashError(w, r, "Failed to MapUsers: %s", err)
		}

		// Get Likes information about these photos.
		var photoIDs = []uint64{}
		for _, p := range photos {
			photoIDs = append(photoIDs, p.ID)
		}
		likeMap := models.MapLikes(currentUser, "photos", photoIDs)
		commentMap := models.MapCommentCounts("photos", photoIDs)

		// Ping this user as having used the forums today.
		go func() {
			if err := models.LogDailyGalleryUser(currentUser); err != nil {
				log.Error("LogDailyGalleryUser(%s): error logging their usage statistic: %s", currentUser.Username, err)
			}
		}()

		var vars = map[string]interface{}{
			"IsSiteGallery": true,
			"Photos":        photos,
			"UserMap":       userMap,
			"LikeMap":       likeMap,
			"CommentMap":    commentMap,
			"Pager":         pager,
			"ViewStyle":     viewStyle,

			// Search filters
			"Sort":             sort,
			"FilterWho":        who,
			"FilterExplicit":   filterExplicit,
			"FilterVisibility": filterVisibility,
			"AdminView":        adminView,

			// Is the current user shy?
			"IsShyUser": isShy,
		}

		if err := tmpl.Execute(w, r, vars); err != nil {
			http.Error(w, err.Error(), http.StatusInternalServerError)
			return
		}
	})
}