119 lines
3.1 KiB
Go
119 lines
3.1 KiB
Go
package entity
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gosimple/slug"
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
|
)
|
|
|
|
// File represents an image or sidecar file that belongs to a photo
|
|
type File struct {
|
|
ID uint `gorm:"primary_key"`
|
|
Photo *Photo
|
|
PhotoID uint `gorm:"index;"`
|
|
PhotoUUID string `gorm:"type:varbinary(36);index;"`
|
|
FileUUID string `gorm:"type:varbinary(36);unique_index;"`
|
|
FileName string `gorm:"type:varbinary(768);unique_index"`
|
|
OriginalName string `gorm:"type:varbinary(768);"`
|
|
FileHash string `gorm:"type:varbinary(128);index"`
|
|
FileModified time.Time
|
|
FileSize int64
|
|
FileType string `gorm:"type:varbinary(32)"`
|
|
FileMime string `gorm:"type:varbinary(64)"`
|
|
FilePrimary bool
|
|
FileSidecar bool
|
|
FileVideo bool
|
|
FileMissing bool
|
|
FileDuplicate bool
|
|
FilePortrait bool
|
|
FileWidth int
|
|
FileHeight int
|
|
FileOrientation int
|
|
FileAspectRatio float32 `gorm:"type:FLOAT;"`
|
|
FileMainColor string `gorm:"type:varbinary(16);index;"`
|
|
FileColors string `gorm:"type:binary(9);"`
|
|
FileLuminance string `gorm:"type:binary(9);"`
|
|
FileDiff uint32
|
|
FileChroma uint8
|
|
FileNotes string `gorm:"type:text"`
|
|
FileError string `gorm:"type:varbinary(512)"`
|
|
Share []FileShare
|
|
Sync []FileSync
|
|
Links []Link `gorm:"foreignkey:ShareUUID;association_foreignkey:FileUUID"`
|
|
CreatedAt time.Time
|
|
CreatedIn int64
|
|
UpdatedAt time.Time
|
|
UpdatedIn int64
|
|
DeletedAt *time.Time `sql:"index"`
|
|
}
|
|
|
|
// FirstFileByHash gets a file in db from its hash
|
|
func FirstFileByHash(fileHash string) (File, error) {
|
|
var file File
|
|
|
|
q := Db().Unscoped().First(&file, "file_hash = ?", fileHash)
|
|
|
|
return file, q.Error
|
|
}
|
|
|
|
// BeforeCreate computes a random UUID when a new file is created in database
|
|
func (m *File) BeforeCreate(scope *gorm.Scope) error {
|
|
if rnd.IsPPID(m.FileUUID, 'f') {
|
|
return nil
|
|
}
|
|
|
|
return scope.SetColumn("FileUUID", rnd.PPID('f'))
|
|
}
|
|
|
|
// ShareFileName returns a meaningful file name useful for sharing.
|
|
func (m *File) ShareFileName() 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.MakeLang(m.Photo.PhotoTitle, "en"))
|
|
} else {
|
|
name = m.PhotoUUID
|
|
}
|
|
|
|
taken := m.Photo.TakenAtLocal.Format("20060102-150405")
|
|
token := rnd.Token(3)
|
|
|
|
result := fmt.Sprintf("%s-%s-%s.%s", taken, name, token, m.FileType)
|
|
|
|
return result
|
|
}
|
|
|
|
// Changed returns true if new and old file size or modified time are different.
|
|
func (m File) Changed(fileSize int64, fileModified time.Time) bool {
|
|
if m.DeletedAt != nil {
|
|
return true
|
|
}
|
|
|
|
if m.FileSize != fileSize {
|
|
return true
|
|
}
|
|
|
|
if m.FileModified.Round(time.Second).Equal(fileModified.Round(time.Second)) {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
// Purge removes a file from the index by marking it as missing.
|
|
func (m *File) Purge() error {
|
|
if err := Db().Unscoped().Model(m).Updates(map[string]interface{}{"file_missing": true, "file_primary": false}).Error; err != nil {
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|