nysoure/server/model/comment.go

package model

import (
	"time"

	"gorm.io/gorm"
)

type Comment struct {
	gorm.Model
	Content string      `gorm:"not null"`
	RefID   uint        `gorm:"not null;index:idx_refid_type,priority:1"`
	Type    CommentType `gorm:"not null;index:idx_refid_type,priority:2"`
	UserID  uint        `gorm:"not null"`
	User    User        `gorm:"foreignKey:UserID"`
	Images  []Image     `gorm:"many2many:comment_images;"`
}

type CommentType uint

const (
	CommentTypeResource CommentType = iota + 1
	CommentTypeReply
)

type CommentView struct {
	ID        uint        `json:"id"`
	Content   string      `json:"content"`
	CreatedAt time.Time   `json:"created_at"`
	User      UserView    `json:"user"`
	Images    []ImageView `json:"images"`
}

func (c *Comment) ToView() *CommentView {
	imageViews := make([]ImageView, 0, len(c.Images))
	for _, img := range c.Images {
		imageViews = append(imageViews, img.ToView())
	}

	return &CommentView{
		ID:        c.ID,
		Content:   c.Content,
		CreatedAt: c.CreatedAt,
		User:      c.User.ToView(),
		Images:    imageViews,
	}
}

type CommentWithResourceView struct {
	ID        uint         `json:"id"`
	Content   string       `json:"content"`
	CreatedAt time.Time    `json:"created_at"`
	Resource  ResourceView `json:"resource"`
	User      UserView     `json:"user"`
	Images    []ImageView  `json:"images"`
}

func (c *Comment) ToViewWithResource(r *Resource) *CommentWithResourceView {
	imageViews := make([]ImageView, 0, len(c.Images))
	for _, img := range c.Images {
		imageViews = append(imageViews, img.ToView())
	}

	return &CommentWithResourceView{
		ID:        c.ID,
		Content:   c.Content,
		CreatedAt: c.CreatedAt,
		Resource:  r.ToView(),
		User:      c.User.ToView(),
		Images:    imageViews,
	}
}