photoprism/internal/query/file_selection.go
Michael Mayer 97c9962053 Index: Add experimental EPS and SVG vector graphics support #1177 #2207
Signed-off-by: Michael Mayer <michael@photoprism.app>
2023-02-11 20:18:04 +01:00

186 lines
4.8 KiB
Go

package query
import (
"errors"
"fmt"
"github.com/photoprism/photoprism/pkg/fs"
"github.com/photoprism/photoprism/internal/entity"
"github.com/photoprism/photoprism/internal/form"
"github.com/photoprism/photoprism/pkg/media"
)
const MegaByte = 1024 * 1024
// FileSelection represents a selection filter to include/exclude certain files.
type FileSelection struct {
MaxSize int
Media []string
OmitMedia []string
Types []string
OmitTypes []string
Primary bool
Originals bool
Hidden bool
Private bool
Archived bool
}
// DownloadSelection selects files to download.
func DownloadSelection(mediaRaw, mediaSidecar, originals bool) FileSelection {
omitMedia := make([]string, 0, 2)
if !mediaRaw {
omitMedia = append(omitMedia, media.Raw.String())
}
if !mediaSidecar {
omitMedia = append(omitMedia, media.Sidecar.String())
}
return FileSelection{
OmitMedia: omitMedia,
Originals: originals,
Private: true,
Archived: true,
}
}
// ShareSelection selects files to share, for example for upload via WebDAV.
func ShareSelection(originals bool) FileSelection {
var omitMedia []string
var omitTypes []string
if !originals {
omitMedia = []string{
media.Unknown.String(),
media.Raw.String(),
media.Sidecar.String(),
media.Text.String(),
media.Other.String(),
}
omitTypes = []string{
fs.ImagePNG.String(),
fs.ImageWebP.String(),
fs.ImageTIFF.String(),
fs.ImageAVIF.String(),
fs.ImageHEIC.String(),
fs.ImageBMP.String(),
fs.ImageGIF.String(),
}
}
return FileSelection{
Originals: originals,
OmitMedia: omitMedia,
OmitTypes: omitTypes,
Hidden: false,
Private: false,
Archived: false,
MaxSize: 1024 * MegaByte,
}
}
// SelectedFiles finds files based on the given selection form, e.g. for downloading or sharing.
func SelectedFiles(f form.Selection, o FileSelection) (results entity.Files, err error) {
if f.Empty() {
return results, errors.New("no items selected")
}
// Resolve photos in smart albums.
if photoIds, err := AlbumsPhotoUIDs(f.Albums, false, o.Private); err != nil {
log.Warnf("query: %s", err.Error())
} else if len(photoIds) > 0 {
f.Photos = append(f.Photos, photoIds...)
}
var concat string
switch DbDialect() {
case MySQL:
concat = "CONCAT(a.path, '/%')"
case SQLite3:
concat = "a.path || '/%'"
default:
return results, fmt.Errorf("unknown sql dialect: %s", DbDialect())
}
// Search condition.
where := fmt.Sprintf(`photos.photo_uid IN (?)
OR photos.place_id IN (?)
OR photos.photo_uid IN (SELECT photo_uid FROM files WHERE file_uid IN (?))
OR photos.photo_path IN (
SELECT a.path FROM folders a WHERE a.folder_uid IN (?) UNION
SELECT b.path FROM folders a JOIN folders b ON b.path LIKE %s WHERE a.folder_uid IN (?))
OR photos.photo_uid IN (SELECT photo_uid FROM photos_albums WHERE hidden = 0 AND album_uid IN (?))
OR files.file_uid IN (SELECT file_uid FROM %s m WHERE m.subj_uid IN (?))
OR photos.id IN (SELECT pl.photo_id FROM photos_labels pl JOIN labels l ON pl.label_id = l.id AND pl.uncertainty < 100 AND l.deleted_at IS NULL WHERE l.label_uid IN (?))
OR photos.id IN (SELECT pl.photo_id FROM photos_labels pl JOIN categories c ON c.label_id = pl.label_id AND pl.uncertainty < 100 JOIN labels lc ON lc.id = c.category_id AND lc.deleted_at IS NULL WHERE lc.label_uid IN (?))`,
concat, entity.Marker{}.TableName())
// Build search query.
s := UnscopedDb().Table("files").
Select("files.*").
Joins("JOIN photos ON photos.id = files.photo_id").
Where("files.file_missing = 0 AND files.file_name <> '' AND files.file_hash <> ''").
Where(where, f.Photos, f.Places, f.Files, f.Files, f.Files, f.Albums, f.Subjects, f.Labels, f.Labels).
Group("files.id")
// File size limit?
if o.MaxSize > 0 {
s = s.Where("files.file_size < ?", o.MaxSize)
}
// Specific media types only?
if len(o.Media) > 0 {
s = s.Where("files.media_type IN (?)", o.Media)
}
// Exclude media types?
if len(o.OmitMedia) > 0 {
s = s.Where("files.media_type NOT IN (?)", o.OmitMedia)
}
// Specific file types only?
if len(o.Types) > 0 {
s = s.Where("files.file_type IN (?)", o.Types)
}
// Exclude file types?
if len(o.OmitTypes) > 0 {
s = s.Where("files.file_type NOT IN (?)", o.OmitTypes)
}
// Previews files only?
if o.Primary {
s = s.Where("files.file_primary = 1")
}
// Files in originals only?
if o.Originals {
s = s.Where("files.file_root = '/'")
}
// Exclude private?
if !o.Private {
s = s.Where("photos.photo_private <> 1")
}
// Exclude hidden photos?
if !o.Hidden {
s = s.Where("photos.photo_quality > -1")
}
// Exclude archived photos?
if !o.Archived {
s = s.Where("photos.deleted_at IS NULL")
}
// Find and return.
if result := s.Scan(&results); result.Error != nil {
return results, result.Error
}
return results, nil
}