photoprism/internal/models/file.go
2019-06-09 05:22:53 +02:00

60 lines
1.4 KiB
Go

package models
import (
"fmt"
"strings"
"github.com/gosimple/slug"
"github.com/jinzhu/gorm"
uuid "github.com/satori/go.uuid"
)
// An image or sidecar file that belongs to a photo
type File struct {
Model
Photo *Photo
PhotoID uint
FileUUID string `gorm:"unique_index;"`
FileName string `gorm:"type:varchar(512);unique_index"` // max 3072 bytes / 4 bytes for utf8mb4 = 768 chars
FileHash string `gorm:"type:varchar(128);unique_index"`
FileOriginalName string
FileType string `gorm:"type:varchar(32)"`
FileMime string `gorm:"type:varchar(64)"`
FilePrimary bool
FileMissing bool
FileDuplicate bool
FilePortrait bool
FileWidth int
FileHeight int
FileOrientation int
FileAspectRatio float64
FileMainColor string
FileColors string
FileLuminance string
FileChroma uint
FileNotes string `gorm:"type:text"`
}
func (m *File) BeforeCreate(scope *gorm.Scope) error {
return scope.SetColumn("FileUUID", uuid.NewV4().String())
}
func (m *File) DownloadFileName() string {
if m.Photo == nil {
return fmt.Sprintf("%s.%s", m.FileHash, m.FileType)
}
var name string
if m.Photo.PhotoTitle != "" {
name = strings.Title(slug.Make(m.Photo.PhotoTitle))
} else {
name = string(m.PhotoID)
}
taken := m.Photo.TakenAt.Format("20060102-150405")
result := fmt.Sprintf("%s-%s.%s", taken, name, m.FileType)
return result
}