website/pkg/models/comment_photo.go

package models

import (
	"time"
)

// CommentPhoto table associates a photo attachment to a (forum) comment.
type CommentPhoto struct {
	ID        uint64 `gorm:"primaryKey"`
	UserID    uint64 `gorm:"index"`
	CommentID uint64 `gorm:"index"`
	Filename  string
	Filesize  int64
	CreatedAt time.Time
	UpdatedAt time.Time
	ExpiredAt time.Time
}

// CreateCommentPhoto with most of the settings you want (not ID or timestamps) in the database.
func CreateCommentPhoto(tmpl CommentPhoto) (*CommentPhoto, error) {
	p := &CommentPhoto{
		CommentID: tmpl.CommentID,
		UserID:    tmpl.UserID,
		Filename:  tmpl.Filename,
		Filesize:  tmpl.Filesize,
	}

	result := DB.Create(p)
	return p, result.Error
}

// GetCommentPhoto by ID.
func GetCommentPhoto(id uint64) (*CommentPhoto, error) {
	p := &CommentPhoto{}
	result := DB.First(&p, id)
	return p, result.Error
}

// GetPhotos returns the comment photos for a given comment.
func (c *Comment) GetPhotos() ([]*CommentPhoto, error) {
	mapping, err := MapCommentPhotos([]*Comment{c})
	if err != nil {
		return nil, err
	}

	return mapping.Get(c.ID), nil
}

/*
PaginateUserCommentPhotos gets a page of all CommentPhotos by the user.
*/
func PaginateUserCommentPhotos(userID uint64, pager *Pagination) ([]*CommentPhoto, error) {
	var p = []*CommentPhoto{}

	query := DB.Where(
		"user_id = ?",
		userID,
	).Order(
		pager.Sort,
	)

	// Get the total count.
	query.Model(&CommentPhoto{}).Count(&pager.Total)

	result := query.Offset(
		pager.GetOffset(),
	).Limit(pager.PerPage).Find(&p)

	return p, result.Error
}

// CommentPhotoMap maps comment IDs to CommentPhotos.
type CommentPhotoMap map[uint64][]*CommentPhoto

// Get like stats from the map.
func (lm CommentPhotoMap) Get(id uint64) []*CommentPhoto {
	if stats, ok := lm[id]; ok {
		return stats
	}
	return nil
}

// MapCommentPhotos returns a map of photo attachments to a series of comments.
func MapCommentPhotos(comments []*Comment) (CommentPhotoMap, error) {
	var (
		result = CommentPhotoMap{} // map[uint64][]*CommentPhoto{}
		ps     = []*CommentPhoto{}
		IDs    = []uint64{}
	)

	for _, c := range comments {
		IDs = append(IDs, c.ID)
	}

	res := DB.Model(&CommentPhoto{}).Where("comment_id IN ?", IDs).Find(&ps)
	if res.Error != nil {
		return nil, res.Error
	}

	for _, row := range ps {
		if _, ok := result[row.CommentID]; !ok {
			result[row.CommentID] = []*CommentPhoto{}
		}
		result[row.CommentID] = append(result[row.CommentID], row)
	}

	return result, nil
}

// Save CommentPhoto.
func (p *CommentPhoto) Save() error {
	result := DB.Save(p)
	return result.Error
}

// Delete CommentPhoto.
func (p *CommentPhoto) Delete() error {
	result := DB.Delete(p)
	return result.Error
}

// GetOrphanedCommentPhotos gets all (up to 500) photos having a blank CommentID older than 24 hours.
func GetOrphanedCommentPhotos() ([]*CommentPhoto, int64, error) {
	var (
		count  int64
		cutoff = time.Now().Add(-24 * time.Hour)
		ps     = []*CommentPhoto{}
	)

	query := DB.Model(&CommentPhoto{}).Where("comment_id = 0 AND created_at < ?", cutoff)
	query.Count(&count)
	res := query.Limit(500).Find(&ps)
	if res.Error != nil {
		return nil, 0, res.Error
	}

	return ps, count, res.Error
}