photoprism/internal/query/file_selection.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,
		Hidden:    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
}