11c3ed70e3
Signed-off-by: Michael Mayer <michael@liquidbytes.net>
199 lines
5.8 KiB
Go
199 lines
5.8 KiB
Go
package entity
|
|
|
|
import (
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/photoprism/photoprism/pkg/rnd"
|
|
"github.com/photoprism/photoprism/pkg/txt"
|
|
)
|
|
|
|
// A photo can have multiple images and sidecar files
|
|
type Photo struct {
|
|
ID uint `gorm:"primary_key"`
|
|
TakenAt time.Time `gorm:"type:datetime;index:idx_photos_taken_uuid;" json:"TakenAt"`
|
|
PhotoUUID string `gorm:"type:varbinary(36);unique_index;index:idx_photos_taken_uuid;"`
|
|
PhotoPath string `gorm:"type:varbinary(512);index;"`
|
|
PhotoName string `gorm:"type:varbinary(256);"`
|
|
PhotoTitle string `json:"PhotoTitle"`
|
|
PhotoDescription string `gorm:"type:text;" json:"PhotoDescription"`
|
|
PhotoNotes string `gorm:"type:text;" json:"PhotoNotes"`
|
|
PhotoArtist string `json:"PhotoArtist"`
|
|
PhotoCopyright string `json:"PhotoCopyright"`
|
|
PhotoFavorite bool `json:"PhotoFavorite"`
|
|
PhotoPrivate bool `json:"PhotoPrivate"`
|
|
PhotoNSFW bool `json:"PhotoNSFW"`
|
|
PhotoStory bool `json:"PhotoStory"`
|
|
PhotoLat float64 `gorm:"index;" json:"PhotoLat"`
|
|
PhotoLng float64 `gorm:"index;" json:"PhotoLng"`
|
|
PhotoAltitude int `json:"PhotoAltitude"`
|
|
PhotoFocalLength int `json:"PhotoFocalLength"`
|
|
PhotoIso int `json:"PhotoIso"`
|
|
PhotoFNumber float64 `json:"PhotoFNumber"`
|
|
PhotoExposure string `gorm:"type:varbinary(64);" json:"PhotoExposure"`
|
|
CameraID uint `gorm:"index:idx_photos_camera_lens;" json:"CameraID"`
|
|
LensID uint `gorm:"index:idx_photos_camera_lens;" json:"LensID"`
|
|
LocationID string `gorm:"type:varbinary(16);index;" json:"LocationID"`
|
|
PlaceID string `gorm:"type:varbinary(16);index;" json:"PlaceID"`
|
|
LocationEstimated bool `json:"LocationEstimated"`
|
|
PhotoCountry string `gorm:"index:idx_photos_country_year_month;" json:"PhotoCountry"`
|
|
PhotoYear int `gorm:"index:idx_photos_country_year_month;"`
|
|
PhotoMonth int `gorm:"index:idx_photos_country_year_month;"`
|
|
TimeZone string `gorm:"type:varbinary(64);" json:"TimeZone"`
|
|
TakenAtLocal time.Time `gorm:"type:datetime;"`
|
|
ModifiedTitle bool `json:"ModifiedTitle"`
|
|
ModifiedDetails bool `json:"ModifiedDetails"`
|
|
ModifiedLocation bool `json:"ModifiedLocation"`
|
|
ModifiedDate bool `json:"ModifiedDate"`
|
|
Camera *Camera `json:"Camera"`
|
|
Lens *Lens `json:"Lens"`
|
|
Location *Location `json:"-"`
|
|
Place *Place `json:"-"`
|
|
Files []File
|
|
Labels []PhotoLabel
|
|
Keywords []Keyword `json:"-"`
|
|
Albums []Album `json:"-"`
|
|
CreatedAt time.Time
|
|
UpdatedAt time.Time
|
|
DeletedAt *time.Time `sql:"index"`
|
|
}
|
|
|
|
func (m *Photo) BeforeCreate(scope *gorm.Scope) error {
|
|
if err := scope.SetColumn("PhotoUUID", rnd.PPID('p')); err != nil {
|
|
return err
|
|
}
|
|
|
|
if m.TakenAt.IsZero() || m.TakenAtLocal.IsZero() {
|
|
now := time.Now()
|
|
|
|
if err := scope.SetColumn("TakenAt", now); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := scope.SetColumn("TakenAtLocal", now); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Photo) BeforeSave(scope *gorm.Scope) error {
|
|
if m.TakenAt.IsZero() || m.TakenAtLocal.IsZero() {
|
|
now := time.Now()
|
|
|
|
if err := scope.SetColumn("TakenAt", now); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := scope.SetColumn("TakenAtLocal", now); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Photo) IndexKeywords(keywords []string, db *gorm.DB) {
|
|
var keywordIds []uint
|
|
|
|
// Index title and description
|
|
keywords = append(keywords, txt.Keywords(m.PhotoTitle)...)
|
|
keywords = append(keywords, txt.Keywords(m.PhotoDescription)...)
|
|
last := ""
|
|
|
|
sort.Strings(keywords)
|
|
|
|
for _, w := range keywords {
|
|
if len(w) < 3 || w == last {
|
|
continue
|
|
}
|
|
|
|
last = w
|
|
kw := NewKeyword(w).FirstOrCreate(db)
|
|
|
|
if kw.Skip {
|
|
continue
|
|
}
|
|
|
|
keywordIds = append(keywordIds, kw.ID)
|
|
|
|
NewPhotoKeyword(m.ID, kw.ID).FirstOrCreate(db)
|
|
}
|
|
|
|
db.Where("photo_id = ? AND keyword_id NOT IN (?)", m.ID, keywordIds).Delete(&PhotoKeyword{})
|
|
}
|
|
|
|
func (m *Photo) PreloadFiles(db *gorm.DB) {
|
|
q := db.NewScope(nil).DB().
|
|
Table("files").
|
|
Select(`files.*`).
|
|
Where("files.photo_id = ?", m.ID).
|
|
Order("files.file_primary DESC")
|
|
|
|
logError(q.Scan(&m.Files))
|
|
}
|
|
|
|
/* func (m *Photo) PreloadLabels(db *gorm.DB) {
|
|
q := db.NewScope(nil).DB().
|
|
Table("labels").
|
|
Select(`labels.*`).
|
|
Joins("JOIN photos_labels ON photos_labels.label_id = labels.id AND photos_labels.photo_id = ?", m.ID).
|
|
Where("labels.deleted_at IS NULL").
|
|
Order("labels.label_name ASC")
|
|
|
|
logError(q.Scan(&m.Labels))
|
|
} */
|
|
|
|
func (m *Photo) PreloadKeywords(db *gorm.DB) {
|
|
q := db.NewScope(nil).DB().
|
|
Table("keywords").
|
|
Select(`keywords.*`).
|
|
Joins("JOIN photos_keywords ON photos_keywords.keyword_id = keywords.id AND photos_keywords.photo_id = ?", m.ID).
|
|
Order("keywords.keyword ASC")
|
|
|
|
logError(q.Scan(&m.Keywords))
|
|
}
|
|
|
|
func (m *Photo) PreloadAlbums(db *gorm.DB) {
|
|
q := db.NewScope(nil).DB().
|
|
Table("albums").
|
|
Select(`albums.*`).
|
|
Joins("JOIN photos_albums ON photos_albums.album_uuid = albums.album_uuid AND photos_albums.photo_uuid = ?", m.PhotoUUID).
|
|
Where("albums.deleted_at IS NULL").
|
|
Order("albums.album_name ASC")
|
|
|
|
logError(q.Scan(&m.Albums))
|
|
}
|
|
|
|
func (m *Photo) PreloadMany(db *gorm.DB) {
|
|
m.PreloadFiles(db)
|
|
// m.PreloadLabels(db)
|
|
m.PreloadKeywords(db)
|
|
m.PreloadAlbums(db)
|
|
}
|
|
|
|
func (m *Photo) NoLocation() bool {
|
|
return m.LocationID == ""
|
|
}
|
|
|
|
func (m *Photo) HasLocation() bool {
|
|
return m.LocationID != ""
|
|
}
|
|
|
|
func (m *Photo) NoPlace() bool {
|
|
return len(m.PlaceID) < 2
|
|
}
|
|
|
|
func (m *Photo) HasPlace() bool {
|
|
return len(m.PlaceID) >= 2
|
|
}
|
|
|
|
func (m *Photo) NoTitle() bool {
|
|
return m.PhotoTitle == ""
|
|
}
|
|
|
|
func (m *Photo) HasTitle() bool {
|
|
return m.PhotoTitle != ""
|
|
}
|