package services import ( "fmt" "mime" "mime/multipart" "net/http" "path/filepath" "git.solsynth.dev/hydrogen/paperclip/pkg/internal/database" "git.solsynth.dev/hydrogen/paperclip/pkg/internal/models" "github.com/google/uuid" "gorm.io/gorm" ) const metadataCacheLimit = 512 var metadataCache = make(map[uint]models.Attachment) func GetAttachmentByID(id uint) (models.Attachment, error) { if val, ok := metadataCache[id]; ok { return val, nil } var attachment models.Attachment if err := database.C.Where(models.Attachment{ BaseModel: models.BaseModel{ID: id}, }).Preload("Account").First(&attachment).Error; err != nil { return attachment, err } else { if len(metadataCache) > metadataCacheLimit { clear(metadataCache) } metadataCache[id] = attachment } 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 exists.Alternative = attachment.Alternative exists.Usage = attachment.Usage exists.Metadata = attachment.Metadata attachment = exists attachment.ID = 0 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 the user didn't provide file mimetype manually, we have 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) } else { if len(metadataCache) > metadataCacheLimit { clear(metadataCache) } metadataCache[attachment.ID] = attachment } return attachment, linked, nil } func UpdateAttachment(item models.Attachment) (models.Attachment, error) { if err := database.C.Save(&item).Error; err != nil { return item, err } else { if len(metadataCache) > metadataCacheLimit { clear(metadataCache) } metadataCache[item.ID] = item } return item, 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 } else { delete(metadataCache, item.ID) } if dupeCount != -1 && dupeCount <= 1 { return DeleteFile(item) } return nil }