website/pkg/controller/account/profile.go

package account

import (
	"net/http"
	"net/url"
	"regexp"

	"code.nonshy.com/nonshy/website/pkg/log"
	"code.nonshy.com/nonshy/website/pkg/middleware"
	"code.nonshy.com/nonshy/website/pkg/models"
	"code.nonshy.com/nonshy/website/pkg/session"
	"code.nonshy.com/nonshy/website/pkg/templates"
	"code.nonshy.com/nonshy/website/pkg/worker"
)

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

// User profile page (/u/username)
func Profile() http.HandlerFunc {
	tmpl := templates.Must("account/profile.html")
	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		// Parse the username out of the URL parameters.
		var username string
		m := ProfileRegexp.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
		}

		// Get the current user (if logged in). If not, check for external view.
		currentUser, err := session.CurrentUser(r)
		if err != nil {
			// The viewer is not logged in, bail now with the basic profile page. If this
			// user doesn't allow external viewers, redirect to login page.
			if user.Visibility != models.UserVisibilityExternal {
				session.FlashError(w, r, "You must be signed in to view this page.")
				templates.Redirect(w, "/login?next="+url.QueryEscape(r.URL.String()))
				return
			}

			vars := map[string]interface{}{
				"User": user,
			}
			if err := tmpl.Execute(w, r, vars); err != nil {
				http.Error(w, err.Error(), http.StatusInternalServerError)
				return
			}
			return
		}

		// Is the site under a Maintenance Mode restriction?
		if middleware.MaintenanceMode(currentUser, w, r) {
			return
		}

		// Forcing an external view? (preview of logged-out profile view for visibility=external accounts)
		// You must be logged-in actually to see this.
		if r.FormValue("view") == "external" {
			vars := map[string]interface{}{
				"User":           user,
				"IsPrivate":      true,
				"IsExternalView": true,
			}
			if err := tmpl.Execute(w, r, vars); err != nil {
				http.Error(w, err.Error(), http.StatusInternalServerError)
				return
			}
			return
		}

		// Inject relationship booleans for profile picture display.
		models.SetUserRelationships(currentUser, []*models.User{user})

		// Admin user can always see the profile pic - but only on this page. Other avatar displays
		// will show the yellow or pink shy.png if the admin is not friends or not granted.
		if currentUser.IsAdmin {
			user.UserRelationship.IsFriend = true
			user.UserRelationship.IsPrivateGranted = true
		}

		var isSelf = currentUser.ID == user.ID

		// Banned or disabled? Only admin can view then.
		if user.Status != models.UserStatusActive && !currentUser.IsAdmin {
			templates.NotFoundPage(w, r)
			return
		}

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

		// Are they friends? And/or is this user private?
		var (
			isFriend  = models.FriendStatus(currentUser.ID, user.ID)
			isPrivate = !currentUser.IsAdmin && !isSelf && user.Visibility == models.UserVisibilityPrivate && isFriend != "approved"
		)

		// Get Likes for this profile.
		likeMap := models.MapLikes(currentUser, "users", []uint64{user.ID})

		// Get the summary of WHO liked this picture.
		likeExample, likeRemainder, err := models.WhoLikes("users", user.ID)
		if err != nil {
			log.Error("WhoLikes(user %d): %s", user.ID, err)
		}

		vars := map[string]interface{}{
			"User":       user,
			"LikeMap":    likeMap,
			"IsFriend":   isFriend,
			"IsPrivate":  isPrivate,
			"PhotoCount": models.CountPhotosICanSee(user, currentUser),
			"NoteCount":  models.CountNotesAboutUser(currentUser, user),
			"OnChat":     worker.GetChatStatistics().IsOnline(user.Username),

			// Details on who likes the photo.
			"LikeExample":   likeExample,
			"LikeRemainder": likeRemainder,
			"LikeTableName": "users",
			"LikeTableID":   user.ID,
		}

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