Interactive/pkg/services/attachments.go

package services

import (
	"mime/multipart"
	"net/http"
	"strings"

	"code.smartsheep.studio/hydrogen/interactive/pkg/database"
	"code.smartsheep.studio/hydrogen/interactive/pkg/models"
	"github.com/google/uuid"
)

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",
		Type:     models.AttachmentOthers,
		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)

	switch strings.Split(attachment.Mimetype, "/")[0] {
	case "image":
		attachment.Type = models.AttachmentPhoto
	case "video":
		attachment.Type = models.AttachmentVideo
	case "audio":
		attachment.Type = models.AttachmentAudio
	default:
		attachment.Type = models.AttachmentOthers
	}

	// Save into database
	err = database.C.Save(&attachment).Error

	return attachment, err
}