photoprism/internal/maps/osm/location.go
Michael Mayer 366c70d992 Optimize performance and data structures
Signed-off-by: Michael Mayer <michael@liquidbytes.net>
2019-12-27 05:18:52 +01:00

140 lines
2.9 KiB
Go

package osm
import (
"encoding/json"
"fmt"
"net/http"
"strconv"
"strings"
"time"
"github.com/melihmucuk/geocache"
"github.com/photoprism/photoprism/internal/util"
)
type Location struct {
PlaceID int `json:"place_id"`
LocLat string `json:"lat"`
LocLng string `json:"lon"`
LocName string `json:"name"`
LocCategory string `json:"category"`
LocType string `json:"type"`
LocDisplayName string `json:"display_name"`
Address Address `json:"address"`
Cached bool
}
var ReverseLookupURL = "https://nominatim.openstreetmap.org/reverse?lat=%f&lon=%f&format=jsonv2&accept-language=en&zoom=18"
// API docs see https://wiki.openstreetmap.org/wiki/Nominatim#Reverse_Geocoding
func FindLocation(lat, lng float64) (result Location, err error) {
if lat == 0.0 || lng == 0.0 {
return result, fmt.Errorf("osm: skipping lat %f, lng %f", lat, lng)
}
point := geocache.GeoPoint{Latitude: lat, Longitude: lng}
if hit, ok := geoCache.Get(point); ok {
log.Debugf("osm: cache hit for lat %f, lng %f", lat, lng)
result = hit.(Location)
result.Cached = true
return result, nil
}
url := fmt.Sprintf(ReverseLookupURL, lat, lng)
log.Debugf("osm: query %s", url)
r, err := http.Get(url)
if err != nil {
log.Errorf("osm: %s", err.Error())
return result, err
}
err = json.NewDecoder(r.Body).Decode(&result)
if err != nil {
log.Errorf("osm: %s", err.Error())
return result, err
}
geoCache.Set(point, result, time.Hour)
result.Cached = false
return result, nil
}
func (o Location) State() (result string) {
result = o.Address.State
return strings.TrimSpace(result)
}
func (o Location) City() (result string) {
if o.Address.City != "" {
result = o.Address.City
} else if o.Address.Town != "" {
result = o.Address.Town
} else if o.Address.Village != "" {
result = o.Address.Village
} else if o.Address.County != "" {
result = o.Address.County
} else if o.Address.State != "" {
result = o.Address.State
}
return strings.TrimSpace(result)
}
func (o Location) Suburb() (result string) {
result = o.Address.Suburb
return strings.TrimSpace(result)
}
func (o Location) CountryCode() (result string) {
result = o.Address.CountryCode
return strings.ToLower(strings.TrimSpace(result))
}
func (o Location) Latitude() (result float64) {
if o.LocLat == "" {
log.Warn("osm: no latitude")
return 0.0
}
result, err := strconv.ParseFloat(o.LocLat, 64)
if err != nil {
log.Errorf("osm: %s", err.Error())
}
return result
}
func (o Location) Longitude() (result float64) {
if o.LocLng == "" {
log.Warn("osm: no longitude")
return 0.0
}
result, err := strconv.ParseFloat(o.LocLng, 64)
if err != nil {
log.Errorf("osm: %s", err.Error())
}
return result
}
func (o Location) Keywords() (result []string) {
return util.Keywords(o.LocDisplayName)
}
func (o Location) Source() string {
return "osm"
}