website/pkg/controller/photo/user_gallery.go

package photo

import (
	"net/http"
	"regexp"

	"code.nonshy.com/nonshy/website/pkg/config"
	"code.nonshy.com/nonshy/website/pkg/models"
	"code.nonshy.com/nonshy/website/pkg/session"
	"code.nonshy.com/nonshy/website/pkg/templates"
)

var UserPhotosRegexp = regexp.MustCompile(`^/photo/u/([^@]+?)$`)

// UserPhotos controller (/photo/u/:username) to view a user's gallery or manage if it's yourself.
func UserPhotos() http.HandlerFunc {
	tmpl := templates.Must("photo/gallery.html")
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// Query params.
		var (
			viewStyle = r.FormValue("view") // cards (default), full
		)
		if viewStyle != "full" {
			viewStyle = "cards"
		}

		// Parse the username out of the URL parameters.
		var username string
		m := UserPhotosRegexp.FindStringSubmatch(r.URL.Path)
		if m != nil {
			username = m[1]
		}

		// Find this user.
		user, err := models.FindUser(username)
		if err != nil {
			templates.NotFoundPage(w, r)
			return
		}

		// Load the current user in case they are viewing their own page.
		currentUser, err := session.CurrentUser(r)
		if err != nil {
			session.FlashError(w, r, "Unexpected error: couldn't get CurrentUser")
		}
		var isOwnPhotos = currentUser.ID == user.ID

		// Is either one blocking?
		if models.IsBlocking(currentUser.ID, user.ID) && !currentUser.IsAdmin {
			templates.NotFoundPage(w, r)
			return
		}

		// Is this user private and we're not friends?
		var (
			areFriends = models.AreFriends(user.ID, currentUser.ID)
			isPrivate  = user.Visibility == models.UserVisibilityPrivate && !areFriends
		)
		if isPrivate && !currentUser.IsAdmin && !isOwnPhotos {
			session.FlashError(w, r, "This user's profile page and photo gallery are private.")
			templates.Redirect(w, "/u/"+user.Username)
			return
		}

		// Has this user granted access to see their privates?
		var (
			isGrantee = models.IsPrivateUnlocked(user.ID, currentUser.ID) // THEY have granted US access
			isGranted = models.IsPrivateUnlocked(currentUser.ID, user.ID) // WE have granted THEM access
		)

		// What set of visibilities to query?
		visibility := []models.PhotoVisibility{models.PhotoPublic}
		if isOwnPhotos || isGrantee || currentUser.IsAdmin {
			visibility = append(visibility, models.PhotoFriends, models.PhotoPrivate)
		} else if models.AreFriends(user.ID, currentUser.ID) {
			visibility = append(visibility, models.PhotoFriends)
		}

		// Explicit photo filter?
		explicit := currentUser.Explicit
		if isOwnPhotos {
			explicit = true
		}

		// Get the page of photos.
		pager := &models.Pagination{
			Page:    1,
			PerPage: config.PageSizeUserGallery,
			Sort:    "created_at desc",
		}
		pager.ParsePage(r)
		photos, err := models.PaginateUserPhotos(user.ID, visibility, explicit, pager)

		// Get the count of explicit photos if we are not viewing explicit photos.
		var explicitCount int64
		if !explicit {
			explicitCount, _ = models.CountExplicitPhotos(user.ID, visibility)
		}

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

		var vars = map[string]interface{}{
			"IsOwnPhotos":            currentUser.ID == user.ID,
			"IsMyPrivateUnlockedFor": isGranted, // have WE granted THIS USER to see our private pics?
			"User":                   user,
			"Photos":                 photos,
			"PhotoCount":             models.CountPhotos(user.ID),
			"Pager":                  pager,
			"LikeMap":                likeMap,
			"CommentMap":             commentMap,
			"ViewStyle":              viewStyle,
			"ExplicitCount":          explicitCount,
		}

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