58 lines
1.8 KiB
Go
58 lines
1.8 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
|
|
"gorm.io/gorm"
|
|
|
|
"github.com/gochat/gochat/internal/model"
|
|
)
|
|
|
|
// AttachmentRepo implements GORM repository for Attachment.
|
|
// Reference: Chatwoot app/models/attachment.rb
|
|
type AttachmentRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewAttachmentRepo creates a new Attachment repository.
|
|
func NewAttachmentRepo(db *gorm.DB) *AttachmentRepo {
|
|
return &AttachmentRepo{db: db}
|
|
}
|
|
|
|
// FindByID retrieves an attachment by primary key.
|
|
func (r *AttachmentRepo) FindByID(ctx context.Context, id uint) (*model.Attachment, error) {
|
|
var attachment model.Attachment
|
|
err := r.db.WithContext(ctx).First(&attachment, id).Error
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return &attachment, nil
|
|
}
|
|
|
|
// FindByMessage retrieves all attachments for a message.
|
|
func (r *AttachmentRepo) FindByMessage(ctx context.Context, messageID uint) ([]model.Attachment, error) {
|
|
var attachments []model.Attachment
|
|
err := r.db.WithContext(ctx).Where("message_id = ?", messageID).
|
|
Order("id ASC").Find(&attachments).Error
|
|
return attachments, err
|
|
}
|
|
|
|
// Create inserts a new attachment.
|
|
func (r *AttachmentRepo) Create(ctx context.Context, attachment *model.Attachment) error {
|
|
return r.db.WithContext(ctx).Create(attachment).Error
|
|
}
|
|
|
|
// Update modifies an existing attachment.
|
|
func (r *AttachmentRepo) Update(ctx context.Context, attachment *model.Attachment) error {
|
|
return r.db.WithContext(ctx).Save(attachment).Error
|
|
}
|
|
|
|
// Delete soft-deletes an attachment.
|
|
func (r *AttachmentRepo) Delete(ctx context.Context, id uint) error {
|
|
return r.db.WithContext(ctx).Delete(&model.Attachment{}, id).Error
|
|
}
|
|
|
|
// DeleteByMessage soft-deletes all attachments for a message.
|
|
func (r *AttachmentRepo) DeleteByMessage(ctx context.Context, messageID uint) error {
|
|
return r.db.WithContext(ctx).Where("message_id = ?", messageID).Delete(&model.Attachment{}).Error
|
|
} |