package query import ( "fmt" "strings" "time" "github.com/photoprism/photoprism/pkg/fs" "github.com/jinzhu/gorm" "github.com/photoprism/photoprism/internal/entity" "github.com/photoprism/photoprism/internal/form" "github.com/photoprism/photoprism/pkg/capture" "github.com/photoprism/photoprism/pkg/pluscode" "github.com/photoprism/photoprism/pkg/s2" "github.com/photoprism/photoprism/pkg/txt" ) // Geo searches for photos based on Form values and returns GeoResults ([]GeoResult). func Geo(f form.GeoSearch) (results GeoResults, err error) { start := time.Now() if err := f.ParseQueryString(); err != nil { return results, err } defer log.Debug(capture.Time(time.Now(), fmt.Sprintf("geo: search %s", form.Serialize(f, true)))) s := UnscopedDb() // s.LogMode(true) s = s.Table("photos"). Select(`photos.id, photos.photo_uid, photos.photo_type, photos.photo_lat, photos.photo_lng, photos.photo_title, photos.photo_description, photos.photo_favorite, 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") f.Query = txt.Clip(f.Query, txt.ClipKeyword) if f.Query != "" { // Filter by label, label category and keywords. var categories []entity.Category var labels []entity.Label var labelIds []uint if len(f.Query) < 2 { return results, fmt.Errorf("query too short") } if err := Db().Where(AnySlug("custom_slug", f.Query, " ")).Find(&labels).Error; len(labels) == 0 || err != nil { log.Infof("search: label %s not found, using fuzzy search", txt.Quote(f.Query)) if likeAny := LikeAny("k.keyword", f.Query); likeAny != "" { s = s.Where("photos.id IN (SELECT pk.photo_id FROM keywords k JOIN photos_keywords pk ON k.id = pk.keyword_id WHERE (?))", gorm.Expr(likeAny)) } } else { for _, l := range labels { labelIds = append(labelIds, l.ID) Db().Where("category_id = ?", l.ID).Find(&categories) log.Infof("search: label %s includes %d categories", txt.Quote(l.LabelName), len(categories)) for _, category := range categories { labelIds = append(labelIds, category.LabelID) } } if likeAny := LikeAny("k.keyword", f.Query); likeAny != "" { s = s.Where("photos.id IN (SELECT pk.photo_id FROM keywords k JOIN photos_keywords pk ON k.id = pk.keyword_id WHERE (?)) OR "+ "photos.id IN (SELECT pl.photo_id FROM photos_labels pl WHERE pl.uncertainty < 100 AND pl.label_id IN (?))", gorm.Expr(likeAny), labelIds) } else { s = s.Where("photos.id IN (SELECT pl.photo_id FROM photos_labels pl WHERE pl.uncertainty < 100 AND pl.label_id IN (?))", labelIds) } } } if f.Album != "" { s = s.Joins("JOIN photos_albums ON photos_albums.photo_uid = photos.photo_uid").Where("photos_albums.hidden = 0 AND photos_albums.album_uid = ?", f.Album) } if f.Camera > 0 { s = s.Where("photos.camera_id = ?", f.Camera) } if f.Lens > 0 { s = s.Where("photos.lens_id = ?", f.Lens) } if (f.Year > 0 && f.Year <= txt.YearMax) || f.Year == entity.YearUnknown { s = s.Where("photos.photo_year = ?", f.Year) } if (f.Month >= txt.MonthMin && f.Month <= txt.MonthMax) || f.Month == entity.MonthUnknown { s = s.Where("photos.photo_month = ?", f.Month) } if f.Color != "" { s = s.Where("files.file_main_color IN (?)", strings.Split(strings.ToLower(f.Color), Or)) } if f.Favorite { s = s.Where("photos.photo_favorite = 1") } if f.Country != "" { s = s.Where("photos.photo_country IN (?)", strings.Split(strings.ToLower(f.Country), Or)) } // Filter by media type. if f.Type != "" { s = s.Where("photos.photo_type IN (?)", strings.Split(strings.ToLower(f.Type), Or)) } if f.Video { s = s.Where("photos.photo_type = 'video'") } else if f.Photo { s = s.Where("photos.photo_type IN ('image','raw','live')") } if f.Path != "" { p := f.Path if strings.HasPrefix(p, "/") { p = p[1:] } if strings.HasSuffix(p, "/") { s = s.Where("photos.photo_path = ?", p[:len(p)-1]) } else if strings.Contains(p, Or) { s = s.Where("photos.photo_path IN (?)", strings.Split(p, Or)) } else { s = s.Where("photos.photo_path LIKE ?", strings.ReplaceAll(p, "*", "%")) } } if strings.Contains(f.Name, Or) { s = s.Where("photos.photo_name IN (?)", strings.Split(f.Name, Or)) } else if f.Name != "" { s = s.Where("photos.photo_name LIKE ?", strings.ReplaceAll(fs.StripKnownExt(f.Name), "*", "%")) } // Filter by status. if f.Archived { s = s.Where("photos.photo_quality > -1") s = s.Where("photos.deleted_at IS NOT NULL") } else { s = s.Where("photos.deleted_at IS NULL") if f.Private { s = s.Where("photos.photo_private = 1") } else if f.Public { s = s.Where("photos.photo_private = 0") } if f.Review { s = s.Where("photos.photo_quality < 3") } else if f.Quality != 0 && f.Private == false { s = s.Where("photos.photo_quality >= ?", f.Quality) } } if f.Favorite { s = s.Where("photos.photo_favorite = 1") } if f.S2 != "" { s2Min, s2Max := s2.PrefixedRange(f.S2, 7) s = s.Where("photos.cell_id BETWEEN ? AND ?", s2Min, s2Max) } else if f.Olc != "" { s2Min, s2Max := s2.PrefixedRange(pluscode.S2(f.Olc), 7) s = s.Where("photos.cell_id BETWEEN ? AND ?", s2Min, s2Max) } else { // Inaccurate distance search, but probably 'good enough' for now if f.Lat > 0 { latMin := f.Lat - SearchRadius*float32(f.Dist) latMax := f.Lat + SearchRadius*float32(f.Dist) s = s.Where("photos.photo_lat BETWEEN ? AND ?", latMin, latMax) } if f.Lng > 0 { lngMin := f.Lng - SearchRadius*float32(f.Dist) lngMax := f.Lng + SearchRadius*float32(f.Dist) s = s.Where("photos.photo_lng BETWEEN ? AND ?", lngMin, lngMax) } } if !f.Before.IsZero() { s = s.Where("photos.taken_at <= ?", f.Before.Format("2006-01-02")) } if !f.After.IsZero() { s = s.Where("photos.taken_at >= ?", f.After.Format("2006-01-02")) } s = s.Order("taken_at, photos.photo_uid") if result := s.Scan(&results); result.Error != nil { return results, result.Error } log.Infof("geo: found %d photos for %s [%s]", len(results), f.SerializeAll(), time.Since(start)) return results, nil }