✨ Attachments

pkg/services/attachments.go

@@ -0,0 +1,40 @@
+package services
+
+import (
+	"code.smartsheep.studio/hydrogen/interactive/pkg/database"
+	"code.smartsheep.studio/hydrogen/interactive/pkg/models"
+	"github.com/google/uuid"
+	"mime/multipart"
+	"net/http"
+)
+
+func NewAttachment(user models.Account, header *multipart.FileHeader) (models.Attachment, error) {
+	attachment := models.Attachment{
+		FileID:   uuid.NewString(),
+		Filesize: header.Size,
+		Filename: header.Filename,
+		Mimetype: "unknown/unknown",
+		PostID:   nil,
+		AuthorID: user.ID,
+	}
+
+	// Open file
+	file, err := header.Open()
+	if err != nil {
+		return attachment, err
+	}
+	defer file.Close()
+
+	// Detect mimetype
+	fileHeader := make([]byte, 512)
+	_, err = file.Read(fileHeader)
+	if err != nil {
+		return attachment, err
+	}
+	attachment.Mimetype = http.DetectContentType(fileHeader)
+
+	// Save into database
+	err = database.C.Save(&attachment).Error
+
+	return attachment, err
+}

pkg/services/posts.go

@@ -18,10 +18,13 @@ func ListPost(tx *gorm.DB, take int, offset int) ([]*models.Post, error) {
 		Limit(take).
 		Offset(offset).
 		Preload("Author").
+		Preload("Attachments").
 		Preload("RepostTo").
 		Preload("ReplyTo").
 		Preload("RepostTo.Author").
 		Preload("ReplyTo.Author").
+		Preload("RepostTo.Attachments").
+		Preload("ReplyTo.Attachments").
 		Find(&posts).Error; err != nil {
 		return posts, err
 	}
@@ -66,6 +69,7 @@ WHERE t.id IN (?)`, prefix, prefix, prefix), postIds).Scan(&reactInfo)
 func NewPost(
 	user models.Account,
 	alias, title, content string,
+	attachments []models.Attachment,
 	categories []models.Category,
 	tags []models.Tag,
 	publishedAt *time.Time,
@@ -77,6 +81,7 @@ func NewPost(
 		alias,
 		title,
 		content,
+		attachments,
 		categories,
 		tags,
 		publishedAt,
@@ -89,6 +94,7 @@ func NewPostWithRealm(
 	user models.Account,
 	realm *models.Realm,
 	alias, title, content string,
+	attachments []models.Attachment,
 	categories []models.Category,
 	tags []models.Tag,
 	publishedAt *time.Time,
@@ -122,6 +128,7 @@ func NewPostWithRealm(
 		Alias:       alias,
 		Title:       title,
 		Content:     content,
+		Attachments: attachments,
 		Tags:        tags,
 		Categories:  categories,
 		AuthorID:    user.ID,