Paperclip/pkg/services/attachments.go

package services

import (
	"fmt"
	"mime"
	"mime/multipart"
	"net/http"
	"path/filepath"

	"git.solsynth.dev/hydrogen/paperclip/pkg/database"
	"git.solsynth.dev/hydrogen/paperclip/pkg/models"
	"github.com/google/uuid"
	"gorm.io/gorm"
)

func GetAttachmentByID(id uint) (models.Attachment, error) {
	var attachment models.Attachment
	if err := database.C.Where(models.Attachment{
		BaseModel: models.BaseModel{ID: id},
	}).First(&attachment).Error; err != nil {
		return attachment, err
	}
	return attachment, nil
}

func GetAttachmentByUUID(id string) (models.Attachment, error) {
	var attachment models.Attachment
	if err := database.C.Where(models.Attachment{
		Uuid: id,
	}).First(&attachment).Error; err != nil {
		return attachment, err
	}
	return attachment, nil
}

func GetAttachmentByHash(hash string) (models.Attachment, error) {
	var attachment models.Attachment
	if err := database.C.Where(models.Attachment{
		HashCode: hash,
	}).First(&attachment).Error; err != nil {
		return attachment, err
	}
	return attachment, nil
}

func NewAttachmentMetadata(tx *gorm.DB, user models.Account, file *multipart.FileHeader, attachment models.Attachment) (models.Attachment, bool, error) {
	linked := false
	exists, pickupErr := GetAttachmentByHash(attachment.HashCode)
	if pickupErr == nil {
		linked = true
		attachment = exists
		attachment.ID = 0
		attachment.Uuid = uuid.NewString()
		attachment.AccountID = user.ID
	} else {
		// Upload the new file
		attachment.Uuid = uuid.NewString()
		attachment.Size = file.Size
		attachment.Name = file.Filename
		attachment.AccountID = user.ID

		// If user didn't provide file mimetype manually, we gotta to detect it
		if len(attachment.MimeType) == 0 {
			if ext := filepath.Ext(attachment.Name); len(ext) > 0 {
				// Detect mimetype by file extensions
				attachment.MimeType = mime.TypeByExtension(ext)
			} else {
				// Detect mimetype by file header
				// This method as a fallback method, because this isn't pretty accurate
				header, err := file.Open()
				if err != nil {
					return attachment, false, fmt.Errorf("failed to read file header: %v", err)
				}
				defer header.Close()

				fileHeader := make([]byte, 512)
				_, err = header.Read(fileHeader)
				if err != nil {
					return attachment, false, err
				}
				attachment.MimeType = http.DetectContentType(fileHeader)
			}
		}
	}

	if err := tx.Save(&attachment).Error; err != nil {
		return attachment, linked, fmt.Errorf("failed to save attachment record: %v", err)
	}

	return attachment, linked, nil
}

func DeleteAttachment(item models.Attachment) error {
	var dupeCount int64
	if err := database.C.
		Where(&models.Attachment{HashCode: item.HashCode}).
		Model(&models.Attachment{}).
		Count(&dupeCount).Error; err != nil {
		dupeCount = -1
	}

	if err := database.C.Delete(&item).Error; err != nil {
		return err
	}

	if dupeCount != -1 && dupeCount <= 1 {
		return DeleteFile(item)
	}

	return nil
}