92e6c4fe1e
Extends DownloadSettings with 4 additional options: - Name: File name pattern for downloaded files (existed) - Disabled: Disables downloads - Originals: Only download files stored in "originals" folder - MediaRaw: Include RAW image files - MediaSidecar: Include metadata sidecar files (JSON, XMP, YAML)
151 lines
3.9 KiB
Go
151 lines
3.9 KiB
Go
package query
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"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(primary bool) FileSelection {
|
|
return FileSelection{
|
|
Originals: !primary,
|
|
Primary: primary,
|
|
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")
|
|
}
|
|
|
|
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 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 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("file_size < ?", o.MaxSize)
|
|
}
|
|
|
|
// Specific media types only?
|
|
if len(o.Media) > 0 {
|
|
s = s.Where("media_type IN (?)", o.Media)
|
|
}
|
|
|
|
// Exclude media types?
|
|
if len(o.OmitMedia) > 0 {
|
|
s = s.Where("media_type NOT IN (?)", o.OmitMedia)
|
|
}
|
|
|
|
// Specific file types only?
|
|
if len(o.Types) > 0 {
|
|
s = s.Where("file_type IN (?)", o.Types)
|
|
}
|
|
|
|
// Exclude file types?
|
|
if len(o.OmitTypes) > 0 {
|
|
s = s.Where("file_type NOT IN (?)", o.OmitTypes)
|
|
}
|
|
|
|
// Primary files only?
|
|
if o.Primary {
|
|
s = s.Where("file_primary = 1")
|
|
}
|
|
|
|
// Files in originals only?
|
|
if o.Originals {
|
|
s = s.Where("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
|
|
}
|