package models import ( "errors" "fmt" "strings" "time" "code.nonshy.com/nonshy/website/pkg/config" "code.nonshy.com/nonshy/website/pkg/log" "code.nonshy.com/nonshy/website/pkg/redis" "gorm.io/gorm" ) // Photo table. type Photo struct { ID uint64 `gorm:"primaryKey"` UserID uint64 `gorm:"index"` Filename string CroppedFilename string // if cropped, e.g. for profile photo Filesize int64 Caption string AltText string Flagged bool // photo has been reported by the community AdminLabel string // admin label(s) on this photo (e.g. verified non-explicit) Visibility PhotoVisibility `gorm:"index"` Gallery bool `gorm:"index"` // photo appears in the public gallery (if public) Explicit bool `gorm:"index"` // is an explicit photo Pinned bool `gorm:"index"` // user pins it to the front of their gallery LikeCount int64 `gorm:"index"` // cache of 'likes' count CommentCount int64 `gorm:"index"` // cache of comments count Views uint64 `gorm:"index"` // view count CreatedAt time.Time `gorm:"index"` UpdatedAt time.Time } // PhotoVisibility settings. type PhotoVisibility string const ( PhotoPublic PhotoVisibility = "public" // on profile page and/or public gallery PhotoFriends PhotoVisibility = "friends" // only friends can see it PhotoPrivate PhotoVisibility = "private" // private // Special visibility in case, on User Gallery view, user applies a filter // for friends-only picture but they are not friends with the user. PhotoNotAvailable PhotoVisibility = "not_available" ) // PhotoVisibility preset settings. var ( PhotoVisibilityAll = []PhotoVisibility{ PhotoPublic, PhotoFriends, PhotoPrivate, } // Site Gallery visibility for when your friends show up in the gallery. // Or: "Friends + Gallery" photos can appear to your friends in the Site Gallery. PhotoVisibilityFriends = []string{ string(PhotoPublic), string(PhotoFriends), } ) // CreatePhoto with most of the settings you want (not ID or timestamps) in the database. func CreatePhoto(tmpl Photo) (*Photo, error) { if tmpl.UserID == 0 { return nil, errors.New("UserID required") } p := &Photo{ UserID: tmpl.UserID, Filename: tmpl.Filename, CroppedFilename: tmpl.CroppedFilename, Filesize: tmpl.Filesize, Caption: tmpl.Caption, AltText: tmpl.AltText, Visibility: tmpl.Visibility, Gallery: tmpl.Gallery, Pinned: tmpl.Pinned, Explicit: tmpl.Explicit, } result := DB.Create(p) return p, result.Error } // GetPhoto by ID. func GetPhoto(id uint64) (*Photo, error) { p := &Photo{} result := DB.First(&p, id) return p, result.Error } // GetPhotos by an array of IDs, mapped to their IDs. func GetPhotos(IDs []uint64) (map[uint64]*Photo, error) { var ( mp = map[uint64]*Photo{} ps = []*Photo{} ) result := DB.Model(&Photo{}).Where("id IN ?", IDs).Find(&ps) for _, row := range ps { mp[row.ID] = row } return mp, result.Error } // CanBeEditedBy checks whether a photo can be edited by the current user. // // Admins with PhotoModerator scope can always edit. func (p *Photo) CanBeEditedBy(currentUser *User) bool { if currentUser.HasAdminScope(config.ScopePhotoModerator) { return true } return p.UserID == currentUser.ID } // CanBeSeenBy checks whether a photo can be seen by the current user. // // An admin user with omni photo view permission can always see the photo. // // Note: this function incurs several DB queries to look up the photo's owner and block lists. func (p *Photo) CanBeSeenBy(currentUser *User) (bool, error) { // Admins with photo moderator ability can always see. if currentUser.HasAdminScope(config.ScopePhotoModerator) { return true, nil } return p.ShouldBeSeenBy(currentUser) } // ShouldBeSeenBy checks whether a photo should be seen by the current user. // // Even if the current user is an admin with photo moderator ability, this function will return // whether the admin 'should' be able to see if not for their admin status. Example: a private // photo may be shown to the admin so they can moderate it, but they shouldn't be able to "like" // it or mark it as "viewed." // // Note: this function incurs several DB queries to look up the photo's owner and block lists. func (p *Photo) ShouldBeSeenBy(currentUser *User) (bool, error) { // Find the photo's owner. user, err := GetUser(p.UserID) if err != nil { return false, err } var isOwnPhoto = currentUser.ID == user.ID // Is either one blocking? if IsBlocking(currentUser.ID, user.ID) { return false, errors.New("is blocking") } // Is this user private and we're not friends? var ( areFriends = AreFriends(user.ID, currentUser.ID) isPrivate = user.Visibility == UserVisibilityPrivate && !areFriends ) if isPrivate && !isOwnPhoto { return false, errors.New("user is private and we aren't friends") } // Is this a private photo and are we allowed to see? isGranted := IsPrivateUnlocked(user.ID, currentUser.ID) if p.Visibility == PhotoPrivate && !isGranted && !isOwnPhoto { return false, errors.New("photo is private") } return true, nil } // UserGallery configuration for filtering gallery pages. type UserGallery struct { Explicit string // "", "true", "false" Visibility []PhotoVisibility } /* PaginateUserPhotos gets a page of photos belonging to a user ID. */ func PaginateUserPhotos(userID uint64, conf UserGallery, pager *Pagination) ([]*Photo, error) { var ( p = []*Photo{} wheres = []string{} placeholders = []interface{}{} ) var explicit = []bool{} switch conf.Explicit { case "true": explicit = []bool{true} case "false": explicit = []bool{false} } wheres = append(wheres, "user_id = ? AND visibility IN ?") placeholders = append(placeholders, userID, conf.Visibility) if len(explicit) > 0 { wheres = append(wheres, "explicit = ?") placeholders = append(placeholders, explicit[0]) } query := DB.Where( strings.Join(wheres, " AND "), placeholders..., ).Order( pager.Sort, ) // Get the total count. query.Model(&Photo{}).Count(&pager.Total) result := query.Offset( pager.GetOffset(), ).Limit(pager.PerPage).Find(&p) return p, result.Error } // View a photo, incrementing its Views count but not its UpdatedAt. // Debounced with a Redis key. func (p *Photo) View(user *User) error { // The owner of the photo does not count views. if p.UserID == user.ID { return nil } // Should the viewer be able to see this, regardless of their admin ability? if ok, err := p.ShouldBeSeenBy(user); !ok { return err } // Debounce this. var redisKey = fmt.Sprintf(config.PhotoViewDebounceRedisKey, user.ID, p.ID) if redis.Exists(redisKey) { return nil } redis.Set(redisKey, nil, config.PhotoViewDebounceCooldown) return DB.Model(&Photo{}).Where( "id = ?", p.ID, ).Updates(map[string]interface{}{ "views": p.Views + 1, "updated_at": p.UpdatedAt, }).Error } // CountPhotos returns the total number of photos on a user's account. func CountPhotos(userID uint64) int64 { var count int64 result := DB.Where( "user_id = ?", userID, ).Model(&Photo{}).Count(&count) if result.Error != nil { log.Error("CountPhotos(%d): %s", userID, result.Error) } return count } // GetOrphanedPhotos gets all photos having no user ID associated. func GetOrphanedPhotos() ([]*Photo, int64, error) { var ( count int64 ps = []*Photo{} ) query := DB.Model(&Photo{}).Where(` NOT EXISTS ( SELECT 1 FROM users WHERE users.id = photos.user_id ) OR photos.user_id = 0 `) query.Count(&count) res := query.Find(&ps) if res.Error != nil { return nil, 0, res.Error } return ps, count, res.Error } // HasAdminLabelNonExplicit checks if the non-explicit admin label is applied to this photo. func (p *Photo) HasAdminLabelNonExplicit() bool { return config.HasAdminLabel(config.AdminLabelPhotoNonExplicit, p.AdminLabel) } // HasAdminLabelForceExplicit checks if the force-explicit admin label is applied to this photo. func (p *Photo) HasAdminLabelForceExplicit() bool { return config.HasAdminLabel(config.AdminLabelPhotoForceExplicit, p.AdminLabel) } // HasAdminLabel checks if the photo has an admin label (for convenient front-end access on the Edit page). func (p *Photo) HasAdminLabel(label string) bool { return config.HasAdminLabel(label, p.AdminLabel) } // PhotoMap helps map a set of users to look up by ID. type PhotoMap map[uint64]*Photo // MapPhotos looks up a set of photos IDs in bulk and returns a PhotoMap suitable for templates. func MapPhotos(photoIDs []uint64) (PhotoMap, error) { var ( photoMap = PhotoMap{} set = map[uint64]interface{}{} distinct = []uint64{} ) // Uniqueify the IDs. for _, uid := range photoIDs { if _, ok := set[uid]; ok { continue } set[uid] = nil distinct = append(distinct, uid) } var ( photos = []*Photo{} result = DB.Model(&Photo{}).Where("id IN ?", distinct).Find(&photos) ) if result.Error == nil { for _, row := range photos { photoMap[row.ID] = row } } return photoMap, result.Error } // Has a photo ID in the map? func (pm PhotoMap) Has(id uint64) bool { _, ok := pm[id] return ok } // Get a photo from the PhotoMap. func (pm PhotoMap) Get(id uint64) *Photo { if photo, ok := pm[id]; ok { return photo } return nil } /* IsSiteGalleryThrottled returns whether the user is throttled from marking additional pictures for the Site Gallery. The thresholds are in pkg/config but the idea is a user can only upload (say) 5 Site Gallery photos within a 24 hour time span, so that new users who sign up and immediately max out their full gallery don't end up spamming the Site Gallery for pages and pages. If the user has too many recent Site Gallery pictures: - Newly uploaded photos can NOT check the Gallery box. - Editing any existing photo which is NOT in the Gallery: you can not mark the box either. - Existing Gallery photos CAN be un-marked for the gallery, which (if it is one of the 5 recent photos) may put the user below the threshold again. If the user is on the Edit page for an existing photo, provide the Photo; otherwise leave it nil if the user is uploading a new photo for the first time. */ func IsSiteGalleryThrottled(user *User, editPhoto *Photo) bool { // If the editing photo is already in the gallery, allow the user to keep or remove it. if editPhoto != nil && editPhoto.Gallery { return false } var count = CountRecentGalleryPhotos(user, config.SiteGalleryRateLimitInterval) log.Debug("IsSiteGalleryThrottled(%s): they have %d recent Gallery photos", user.Username, count) return count >= config.SiteGalleryRateLimitMax } // CountRecentGalleryPhotos returns the count of recently uploaded Site Gallery photos for a user, // within the past 24 hours, to rate limit spammy bulk uploads that will flood the gallery. func CountRecentGalleryPhotos(user *User, duration time.Duration) (count int64) { result := DB.Where( "user_id = ? AND created_at >= ? AND gallery IS TRUE", user.ID, time.Now().Add(-duration), ).Model(&Photo{}).Count(&count) if result.Error != nil { log.Error("CountRecentGalleryPhotos(%d): %s", user.ID, result.Error) } return } // AllFriendsOnlyPhotoIDs returns the listing of all friends-only photo IDs belonging to the user(s) given. func AllFriendsOnlyPhotoIDs(users ...*User) ([]uint64, error) { var userIDs = []uint64{} for _, user := range users { userIDs = append(userIDs, user.ID) } if len(userIDs) == 0 { return nil, errors.New("no user IDs given") } var photoIDs = []uint64{} err := DB.Table( "photos", ).Select( "photos.id AS id", ).Where( "user_id IN ? AND visibility = ?", userIDs, PhotoFriends, ).Scan(&photoIDs) if err.Error != nil { return photoIDs, fmt.Errorf("AllFriendsOnlyPhotoIDs(%+v): %s", userIDs, err.Error) } return photoIDs, nil } // CountPhotosICanSee returns the number of photos on an account which can be seen by the given viewer. func CountPhotosICanSee(user *User, viewer *User) int64 { // Visibility filters to query by. var visibilities = []PhotoVisibility{ PhotoPublic, } // Is the viewer friends with the target? if AreFriends(user.ID, viewer.ID) { visibilities = append(visibilities, PhotoFriends) } // Is the viewer granted private access? if IsPrivateUnlocked(user.ID, viewer.ID) { visibilities = append(visibilities, PhotoPrivate) } // Get the photo count now. var count int64 result := DB.Where( "user_id = ? AND visibility IN ?", user.ID, visibilities, ).Model(&Photo{}).Count(&count) if result.Error != nil { log.Error("CountPhotosICanSee(%d, %d): %s", user.ID, viewer.ID, result.Error) } return count } // MapPhotoCounts returns a mapping of user ID to the CountPhotos()-equivalent result for each. // It's used on the member directory to show photo counts on each user card. func MapPhotoCounts(users []*User) PhotoCountMap { return MapPhotoCountsByVisibility(users, PhotoPublic) } // MapPhotoCountsByVisibility returns a mapping of user ID to the CountPhotos()-equivalent result for each. func MapPhotoCountsByVisibility(users []*User, visibility PhotoVisibility) PhotoCountMap { var ( userIDs = []uint64{} result = PhotoCountMap{} ) for _, user := range users { userIDs = append(userIDs, user.ID) } type group struct { UserID uint64 PhotoCount int64 } var groups = []group{} if res := DB.Table( "photos", ).Select( "user_id, count(id) AS photo_count", ).Where( "user_id IN ? AND visibility = ?", userIDs, visibility, ).Group("user_id").Scan(&groups); res.Error != nil { log.Error("CountPhotosForUsers: %s", res.Error) } // Map the results in. for _, row := range groups { result[row.UserID] = row.PhotoCount } return result } // MapPhotoCounts returns a mapping of user ID to the CountPhotosICanSee()-equivalent result for each. // It's used on the member directory to show photo counts on each user card. /* TODO: under construction.. func MapPhotoCounts(users []*User, viewer *User) PhotoCountMap { var ( userIDs = []uint64{} result = PhotoCountMap{} wheres = []string{} placeholders = []interface{}{} // User ID filters for the viewer's context. myFriendIDs = FriendIDs(viewer.ID) myPrivateGrantedIDs = PrivateGrantedUserIDs(viewer.ID) ) // Define "all photos visibilities" var ( photosPublic = []PhotoVisibility{ PhotoPublic, } photosFriends = []PhotoVisibility{ PhotoPublic, PhotoFriends, } photosPrivate = []PhotoVisibility{ PhotoPublic, PhotoPrivate, } ) // Flatten the userIDs of all passed in users. for _, user := range users { userIDs = append(userIDs, user.ID) } // Build the where clause. wheres = append(wheres, "user_id IN ?") placeholders = append(placeholders, userIDs) log.Error("FRIEND IDS: %+v", myFriendIDs) // Filter by which photos are visible to us. wheres = append(wheres, "((user_id IN ? AND visibility IN ?) OR "+ "(user_id IN ? AND visibility IN ?) OR "+ "(user_id NOT IN ? AND visibility IN ?))", ) placeholders = append(placeholders, myFriendIDs, photosFriends, myPrivateGrantedIDs, photosPrivate, myFriendIDs, photosPublic, ) type group struct { UserID uint64 PhotoCount int64 } var groups = []group{} if res := DB.Table( "photos", ).Select( "user_id, count(id) AS photo_count", ).Where( strings.Join(wheres, " AND "), placeholders..., ).Group("user_id").Scan(&groups); res.Error != nil { log.Error("CountPhotosForUsers: %s", res.Error) } // Map the results in. for _, row := range groups { result[row.UserID] = row.PhotoCount } return result } */ type PhotoCountMap map[uint64]int64 // Get a photo count for the given user ID from the map. func (pc PhotoCountMap) Get(id uint64) int64 { if value, ok := pc[id]; ok { return value } return 0 } // CountExplicitPhotos returns the number of explicit photos a user has (so non-explicit viewers can see some do exist) func CountExplicitPhotos(userID uint64, visibility []PhotoVisibility) (int64, error) { query := DB.Where( "user_id = ? AND visibility IN ? AND explicit = ?", userID, visibility, true, ) var count int64 result := query.Model(&Photo{}).Count(&count) return count, result.Error } // CountPublicPhotos returns the number of public photos on a user's page. func CountPublicPhotos(userID uint64) int64 { return CountUserPhotosByVisibility(userID, PhotoPublic) } // CountUserPhotosByVisibility returns the number of a user's photos by visibility. func CountUserPhotosByVisibility(userID uint64, visibility PhotoVisibility) int64 { query := DB.Where( "user_id = ? AND visibility = ?", userID, visibility, ) var count int64 result := query.Model(&Photo{}).Count(&count) if result.Error != nil { log.Error("CountUserPhotosByVisibility(%d, %s): %s", userID, visibility, result.Error) } return count } // DistinctPhotoTypes returns types of photos the user has: a set of public, friends, or private. // // The result is cached on the User the first time it's queried. func (u *User) DistinctPhotoTypes() (result map[PhotoVisibility]struct{}) { if u.cachePhotoTypes != nil { return u.cachePhotoTypes } result = map[PhotoVisibility]struct{}{} var results = []*Photo{} query := DB.Model(&Photo{}). Select("DISTINCT photos.visibility"). Where("user_id = ?", u.ID). Group("photos.visibility"). Find(&results) if query.Error != nil { log.Error("User.DistinctPhotoTypes(%s): %s", u.Username, query.Error) return } for _, row := range results { log.Warn("DistinctPhotoTypes(%s): got %+v", u.Username, row) result[row.Visibility] = struct{}{} } u.cachePhotoTypes = result return } // FlushCaches clears any cached attributes (such as distinct photo types) for the user. func (u *User) FlushCaches() { u.cachePhotoTypes = nil u.cacheBlockedUserIDs = nil u.cachePhotoIDs = nil } // Gallery config for the main Gallery paginator. type Gallery struct { Explicit string // Explicit filter Visibility string // Visibility filter AdminView bool // Show all images IsShy bool // Current user is like a Shy Account (or: show self/friends and private photo grants only) FriendsOnly bool // Only show self/friends instead of everybody's pics MyLikes bool // Filter to photos I have liked Uncertified bool // Filter for non-certified members only } /* PaginateGalleryPhotos gets a page of all public user photos for the site gallery. Admin view returns ALL photos regardless of Gallery status. */ func PaginateGalleryPhotos(user *User, conf Gallery, pager *Pagination) ([]*Photo, error) { var ( filterExplicit = conf.Explicit filterVisibility = conf.Visibility adminView = conf.AdminView friendsOnly = conf.FriendsOnly // Show only self and friends pictures isShy = conf.IsShy // Self, friends, and private photo grants only p = []*Photo{} query *gorm.DB // Get the user ID and their preferences. userID = user.ID explicitOK = user.Explicit // User opted-in for explicit content blocklist = BlockedUserIDs(user) mutelist = MutedUserIDs(user, MutedUserContextSiteGallery) privateUserIDs = PrivateGrantedUserIDs(userID) privateUserIDsAreFriends = PrivateGrantedUserIDsAreFriends(user) wheres = []string{} placeholders = []interface{}{} ) // Define "all photos visibilities" var ( photosPublic = []PhotoVisibility{ PhotoPublic, } photosFriends = []PhotoVisibility{ PhotoPublic, PhotoFriends, } photosPrivate = []PhotoVisibility{ PhotoPrivate, } ) // Admins see everything on the site (only an admin user can get an admin view). adminView = user.HasAdminScope(config.ScopePhotoModerator) && adminView // Friend IDs subquery, used in a "WHERE user_id IN ?" clause. friendsQuery := fmt.Sprintf(`( SELECT target_user_id FROM friends WHERE source_user_id = %d AND approved IS TRUE )`, userID) // What sets of User ID * Visibility filters to query under? var ( visOrs = []string{} visPlaceholders = []interface{}{} ) // Whose photos can you see on the Site Gallery? if isShy { // Shy users can only see their Friends photos (public or friends visibility) // and any Private photos to whom they were granted access. visOrs = append(visOrs, fmt.Sprintf("(photos.user_id IN %s AND photos.visibility IN ?)", friendsQuery), "(photos.user_id IN ? AND photos.visibility IN ?)", "photos.user_id = ?", ) visPlaceholders = append(visPlaceholders, photosFriends, privateUserIDs, photosPrivate, userID, ) } else if friendsOnly { // User wants to see only self and friends photos. visOrs = append(visOrs, fmt.Sprintf("(photos.user_id IN %s AND photos.visibility IN ?)", friendsQuery), "photos.user_id = ?", ) visPlaceholders = append(visPlaceholders, photosFriends, userID) // If their friends granted private photos, include those too. if len(privateUserIDsAreFriends) > 0 { visOrs = append(visOrs, "(photos.user_id IN ? AND photos.visibility IN ?)") visPlaceholders = append(visPlaceholders, privateUserIDsAreFriends, photosPrivate) } } else { // You can see friends' Friend photos but only public for non-friends. visOrs = append(visOrs, fmt.Sprintf("(photos.user_id IN %s AND photos.visibility IN ?)", friendsQuery), "(photos.user_id IN ? AND photos.visibility IN ?)", fmt.Sprintf("(photos.user_id NOT IN %s AND photos.visibility IN ?)", friendsQuery), "photos.user_id = ?", ) visPlaceholders = append(placeholders, photosFriends, privateUserIDs, photosPrivate, photosPublic, userID, ) } // Join the User ID * Visibility filters into a nested "OR" wheres = append(wheres, fmt.Sprintf("(%s)", strings.Join(visOrs, " OR "))) placeholders = append(placeholders, visPlaceholders...) // Gallery photos only. wheres = append(wheres, "photos.gallery = ?") placeholders = append(placeholders, true) // Filter by photos the user has liked. if conf.MyLikes { wheres = append(wheres, ` EXISTS ( SELECT 1 FROM likes WHERE likes.user_id = ? AND likes.table_name = 'photos' AND likes.table_id = photos.id ) `) placeholders = append(placeholders, user.ID) } // Filter blocked users. if len(blocklist) > 0 { wheres = append(wheres, "photos.user_id NOT IN ?") placeholders = append(placeholders, blocklist) } // Filter Site Gallery muted users. if len(mutelist) > 0 { wheres = append(wheres, "photos.user_id NOT IN ?") placeholders = append(placeholders, mutelist) } // Non-explicit pics unless the user opted in. Allow explicit filter setting to override. if filterExplicit != "" { wheres = append(wheres, "photos.explicit = ?") placeholders = append(placeholders, filterExplicit == "true") } else if !explicitOK { wheres = append(wheres, "photos.explicit = ?") placeholders = append(placeholders, false) } // Is the user furthermore clamping the visibility filter? if filterVisibility != "" { wheres = append(wheres, "photos.visibility = ?") placeholders = append(placeholders, filterVisibility) } // Only certified (and not banned) user photos. if conf.Uncertified { wheres = append(wheres, "EXISTS (SELECT 1 FROM users WHERE id = photos.user_id AND users.certified IS NOT true AND users.status='active')", ) } else { wheres = append(wheres, "EXISTS (SELECT 1 FROM users WHERE id = photos.user_id AND users.certified = true AND users.status='active')", ) } // Exclude private users' photos. wheres = append(wheres, "NOT EXISTS (SELECT 1 FROM users WHERE id = photos.user_id AND photos.visibility = 'private')", ) // Admin view: get ALL PHOTOS on the site, period. if adminView { query = DB // Admin may filter too. if filterVisibility != "" { query = query.Where("photos.visibility = ?", filterVisibility) } if filterExplicit != "" { query = query.Where("photos.explicit = ?", filterExplicit == "true") } if conf.Uncertified { query = query.Where( "EXISTS (SELECT 1 FROM users WHERE id = photos.user_id AND users.certified IS NOT true AND users.status='active')", ) } } else { query = DB.Where( strings.Join(wheres, " AND "), placeholders..., ) } query = query.Order(pager.Sort) query.Model(&Photo{}).Count(&pager.Total) result := query.Offset(pager.GetOffset()).Limit(pager.PerPage).Find(&p) return p, result.Error } // UpdatePhotoCachedCounts will refresh the cached like/comment count on the photos table. func UpdatePhotoCachedCounts(photoID uint64) error { res := DB.Exec(` UPDATE photos SET like_count = ( SELECT count(id) FROM likes WHERE table_name='photos' AND table_id=photos.id ), comment_count = ( SELECT count(id) FROM comments WHERE table_name='photos' AND table_id=photos.id ) WHERE photos.id = ?; `, photoID) return res.Error } // Save photo. func (p *Photo) Save() error { result := DB.Save(p) return result.Error } // Delete photo. func (p *Photo) Delete() error { result := DB.Delete(p) return result.Error }