90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
package search
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/gin-gonic/gin"
|
|
geojson "github.com/paulmach/go.geojson"
|
|
|
|
"github.com/photoprism/photoprism/internal/entity"
|
|
)
|
|
|
|
// GeoResult represents a photo geo search result.
|
|
type GeoResult struct {
|
|
ID string `json:"-" select:"photos.id"`
|
|
PhotoUID string `json:"UID" select:"photos.photo_uid"`
|
|
PhotoType string `json:"Type,omitempty" select:"photos.photo_type"`
|
|
PhotoLat float32 `json:"Lat" select:"photos.photo_lat"`
|
|
PhotoLng float32 `json:"Lng" select:"photos.photo_lng"`
|
|
PhotoTitle string `json:"Title" select:"photos.photo_title"`
|
|
PhotoDescription string `json:"Description,omitempty" select:"photos.photo_description"`
|
|
PhotoFavorite bool `json:"Favorite,omitempty" select:"photos.photo_favorite"`
|
|
FileHash string `json:"Hash" select:"files.file_hash"`
|
|
FileWidth int `json:"Width" select:"files.file_width"`
|
|
FileHeight int `json:"Height" select:"files.file_height"`
|
|
TakenAt time.Time `json:"TakenAt" select:"photos.taken_at"`
|
|
TakenAtLocal time.Time `json:"TakenAtLocal" select:"photos.taken_at_local"`
|
|
}
|
|
|
|
// Lat returns the position latitude.
|
|
func (photo GeoResult) Lat() float64 {
|
|
return float64(photo.PhotoLat)
|
|
}
|
|
|
|
// Lng returns the position longitude.
|
|
func (photo GeoResult) Lng() float64 {
|
|
return float64(photo.PhotoLng)
|
|
}
|
|
|
|
// GeoResults represents a list of geo search results.
|
|
type GeoResults []GeoResult
|
|
|
|
// GeoJSON returns results as specified on https://geojson.org/.
|
|
func (photos GeoResults) GeoJSON() ([]byte, error) {
|
|
fc := geojson.NewFeatureCollection()
|
|
|
|
bbox := make([]float64, 4)
|
|
|
|
bboxMin := func(pos int, val float64) {
|
|
if bbox[pos] == 0.0 || bbox[pos] > val {
|
|
bbox[pos] = val
|
|
}
|
|
}
|
|
|
|
bboxMax := func(pos int, val float64) {
|
|
if bbox[pos] == 0.0 || bbox[pos] < val {
|
|
bbox[pos] = val
|
|
}
|
|
}
|
|
|
|
for _, p := range photos {
|
|
bboxMin(0, p.Lng())
|
|
bboxMin(1, p.Lat())
|
|
bboxMax(2, p.Lng())
|
|
bboxMax(3, p.Lat())
|
|
|
|
props := gin.H{
|
|
"UID": p.PhotoUID,
|
|
"Hash": p.FileHash,
|
|
"TakenAt": p.TakenAt,
|
|
"Title": p.PhotoTitle,
|
|
}
|
|
|
|
if p.PhotoType != entity.TypeImage && p.PhotoType != entity.TypeDefault {
|
|
props["Type"] = p.PhotoType
|
|
}
|
|
|
|
if p.PhotoFavorite {
|
|
props["Favorite"] = true
|
|
}
|
|
|
|
feat := geojson.NewPointFeature([]float64{p.Lng(), p.Lat()})
|
|
feat.ID = p.ID
|
|
feat.Properties = props
|
|
fc.AddFeature(feat)
|
|
}
|
|
|
|
fc.BoundingBox = bbox
|
|
|
|
return fc.MarshalJSON()
|
|
}
|