photoprism/internal/query/photo_results.go

137 lines
4.3 KiB
Go
Raw Normal View History

package query
import (
"fmt"
"strings"
"time"
"github.com/gosimple/slug"
"github.com/photoprism/photoprism/internal/entity"
"github.com/photoprism/photoprism/pkg/rnd"
"github.com/ulule/deepcopier"
)
// Default photo result slice for simple use cases.
type Photos []entity.Photo
// PhotoResult contains found photos and their main file plus other meta data.
type PhotoResult struct {
ID uint `json:"-"`
DocumentID string `json:"DocumentID,omitempty"`
PhotoUID string `json:"UID"`
PhotoType string `json:"Type"`
TakenAt time.Time `json:"TakenAt"`
TakenAtLocal time.Time `json:"TakenAtLocal"`
TakenSrc string `json:"TakenSrc"`
TimeZone string `json:"TimeZone"`
PhotoPath string `json:"Path"`
PhotoName string `json:"Name"`
PhotoTitle string `json:"Title"`
PhotoDescription string `json:"Description"`
PhotoYear int `json:"Year"`
PhotoMonth int `json:"Month"`
PhotoCountry string `json:"Country"`
PhotoFavorite bool `json:"Favorite"`
PhotoPrivate bool `json:"Private"`
PhotoLat float32 `json:"Lat"`
PhotoLng float32 `json:"Lng"`
PhotoAltitude int `json:"Altitude"`
PhotoIso int `json:"Iso"`
PhotoFocalLength int `json:"FocalLength"`
PhotoFNumber float32 `json:"FNumber"`
PhotoExposure string `json:"Exposure"`
PhotoQuality int `json:"Quality"`
PhotoResolution int `json:"Resolution"`
CameraID uint `json:"CameraID"` // Camera
CameraModel string `json:"CameraModel"`
CameraMake string `json:"CameraMake"`
LensID uint `json:"LensID"` // Lens
LensModel string `json:"LensModel"`
LensMake string `json:"LensMake"`
PlaceUID string `json:"PlaceUID"`
LocUID string `json:"LocUID"` // Location
LocLabel string `json:"LocLabel"`
LocCity string `json:"LocCity"`
LocState string `json:"LocState"`
LocCountry string `json:"LocCountry"`
FileID uint `json:"-"` // File
FileUID string `json:"FileUID"`
FileRoot string `json:"FileRoot"`
FileName string `json:"FileName"`
FileHash string `json:"Hash"`
FileWidth int `json:"Width"`
FileHeight int `json:"Height"`
FilePrimary bool `json:"-"`
FileMissing bool `json:"-"`
FileVideo bool `json:"-"`
FileDuration time.Duration `json:"-"`
FileCodec string `json:"-"`
FileType string `json:"-"`
FileMime string `json:"-"`
FileSize int64 `json:"-"`
FileOrientation int `json:"-"`
FileAspectRatio float32 `json:"-"`
FileColors string `json:"-"`
FileChroma uint8 `json:"-"`
FileLuminance string `json:"-"`
FileDiff uint32 `json:"-"`
Merged bool `json:"Merged"`
CreatedAt time.Time `json:"CreatedAt"`
UpdatedAt time.Time `json:"UpdatedAt"`
DeletedAt time.Time `json:"DeletedAt,omitempty"`
Files []entity.File `json:"Files"`
}
type PhotoResults []PhotoResult
func (m PhotoResults) Merged() (PhotoResults, int, error) {
count := len(m)
merged := make([]PhotoResult, 0, count)
var lastId uint
var i int
for _, res := range m {
file := entity.File{}
if err := deepcopier.Copy(&file).From(res); err != nil {
return merged, count, err
}
file.ID = res.FileID
if lastId == res.ID && i > 0 {
merged[i-1].Files = append(merged[i-1].Files, file)
merged[i-1].Merged = true
continue
}
lastId = res.ID
res.Files = append(res.Files, file)
merged = append(merged, res)
i++
}
return merged, count, nil
}
func (m *PhotoResult) ShareFileName() string {
var name string
if m.PhotoTitle != "" {
name = strings.Title(slug.MakeLang(m.PhotoTitle, "en"))
} else {
name = m.PhotoUID
}
taken := m.TakenAtLocal.Format("20060102-150405")
token := rnd.Token(3)
result := fmt.Sprintf("%s-%s-%s.%s", taken, name, token, m.FileType)
return result
}