146 lines
3.7 KiB
Go
146 lines
3.7 KiB
Go
package entity
|
|
|
|
import (
|
|
"sort"
|
|
"time"
|
|
|
|
"github.com/jinzhu/gorm"
|
|
"github.com/photoprism/photoprism/internal/util"
|
|
)
|
|
|
|
// A photo can have multiple images and sidecar files
|
|
type Photo struct {
|
|
Model
|
|
PhotoUUID string `gorm:"unique_index;"`
|
|
PhotoToken string `gorm:"type:varchar(64);"`
|
|
PhotoPath string `gorm:"type:varchar(128);index;"`
|
|
PhotoName string
|
|
PhotoTitle string `json:"PhotoTitle"`
|
|
PhotoTitleChanged bool
|
|
PhotoDescription string `gorm:"type:text;"`
|
|
PhotoNotes string `gorm:"type:text;"`
|
|
PhotoArtist string `json:"PhotoArtist"`
|
|
PhotoFavorite bool `json:"PhotoFavorite"`
|
|
PhotoPrivate bool `json:"PhotoPrivate"`
|
|
PhotoNSFW bool `json:"PhotoNSFW"`
|
|
PhotoStory bool `json:"PhotoStory"`
|
|
PhotoLat float64 `gorm:"index;"`
|
|
PhotoLong float64 `gorm:"index;"`
|
|
PhotoAltitude int
|
|
PhotoFocalLength int
|
|
PhotoIso int
|
|
PhotoFNumber float64
|
|
PhotoExposure string
|
|
PhotoViews uint
|
|
Camera *Camera
|
|
CameraID uint `gorm:"index;"`
|
|
Lens *Lens
|
|
LensID uint `gorm:"index;"`
|
|
Country *Country
|
|
CountryID string `gorm:"index;"`
|
|
CountryChanged bool
|
|
Location *Location
|
|
LocationID uint
|
|
LocationChanged bool
|
|
LocationEstimated bool
|
|
TakenAt time.Time `gorm:"index;"`
|
|
TakenAtLocal time.Time
|
|
TakenAtChanged bool
|
|
TimeZone string
|
|
Files []File
|
|
Labels []Label
|
|
Keywords []Keyword
|
|
Albums []Album
|
|
}
|
|
|
|
func (m *Photo) BeforeCreate(scope *gorm.Scope) error {
|
|
if err := scope.SetColumn("PhotoUUID", util.UUID()); err != nil {
|
|
return err
|
|
}
|
|
|
|
if err := scope.SetColumn("PhotoToken", util.RandomToken(4)); 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, util.Keywords(m.PhotoTitle)...)
|
|
keywords = append(keywords, util.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)
|
|
}
|