photoprism/internal/query/geo.go

package query

import (
	"fmt"
	"strings"
	"time"

	"github.com/photoprism/photoprism/internal/form"
	"github.com/photoprism/photoprism/pkg/capture"
	"github.com/photoprism/photoprism/pkg/pluscode"
	"github.com/photoprism/photoprism/pkg/s2"
)

// GeoResult represents a photo for displaying it on a map.
type GeoResult struct {
	ID         string    `json:"ID"`
	PhotoLat   float64   `json:"Lat"`
	PhotoLng   float64   `json:"Lng"`
	PhotoUUID  string    `json:"PhotoUUID"`
	PhotoTitle string    `json:"PhotoTitle"`
	FileHash   string    `json:"FileHash"`
	FileWidth  int       `json:"FileWidth"`
	FileHeight int       `json:"FileHeight"`
	TakenAt    time.Time `json:"TakenAt"`
}

// Geo searches for photos based on a Form and returns a PhotoResult slice.
func (s *Repo) Geo(f form.GeoSearch) (results []GeoResult, err error) {
	if err := f.ParseQueryString(); err != nil {
		return results, err
	}

	defer log.Debug(capture.Time(time.Now(), fmt.Sprintf("search: %+v", f)))

	q := s.db.NewScope(nil).DB()

	q = q.Table("photos").
		Select(`photos.id, photos.photo_uuid, photos.photo_lat, photos.photo_lng, photos.photo_title, photos.taken_at, 
		files.file_hash, files.file_width, files.file_height`).
		Joins(`JOIN files ON files.photo_id = photos.id 
		AND files.file_missing = 0 AND files.file_primary AND files.deleted_at IS NULL`).
		Where("photos.deleted_at IS NULL").
		Where("photos.photo_lat <> 0").
		Group("photos.id, files.id")

	if f.Query != "" {
		q = q.Joins("LEFT JOIN photos_keywords ON photos_keywords.photo_id = photos.id").
			Joins("LEFT JOIN keywords ON photos_keywords.keyword_id = keywords.id").
			Where("keywords.keyword LIKE ?", strings.ToLower(f.Query)+"%")
	}

	if f.S2 != "" {
		s2Min, s2Max := s2.Range(f.S2, 7)
		q = q.Where("photos.location_id BETWEEN ? AND ?", s2Min, s2Max)
	} else if f.Olc != "" {
		s2Min, s2Max := s2.Range(pluscode.S2(f.Olc), 7)
		q = q.Where("photos.location_id BETWEEN ? AND ?", s2Min, s2Max)
	} else {
		// Inaccurate distance search, but probably 'good enough' for now
		if f.Lat > 0 {
			latMin := f.Lat - SearchRadius*float64(f.Dist)
			latMax := f.Lat + SearchRadius*float64(f.Dist)
			q = q.Where("photos.photo_lat BETWEEN ? AND ?", latMin, latMax)
		}

		if f.Lng > 0 {
			lngMin := f.Lng - SearchRadius*float64(f.Dist)
			lngMax := f.Lng + SearchRadius*float64(f.Dist)
			q = q.Where("photos.photo_lng BETWEEN ? AND ?", lngMin, lngMax)
		}
	}

	if !f.Before.IsZero() {
		q = q.Where("photos.taken_at <= ?", f.Before.Format("2006-01-02"))
	}

	if !f.After.IsZero() {
		q = q.Where("photos.taken_at >= ?", f.After.Format("2006-01-02"))
	}

	q = q.Order("taken_at, photos.photo_uuid")

	if result := q.Scan(&results); result.Error != nil {
		return results, result.Error
	}

	return results, nil
}