feat: Add nominatim
Signed-off-by: Tim Hårek Andreassen <tim@harek.no>
This commit is contained in:
parent
012bd95cbb
commit
dd25c4e6cb
3 changed files with 182 additions and 0 deletions
86
internal/nominatim/nominatim.go
Normal file
86
internal/nominatim/nominatim.go
Normal file
|
@ -0,0 +1,86 @@
|
|||
package nominatim
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"net/http"
|
||||
"strconv"
|
||||
)
|
||||
|
||||
type Nominatim struct {
|
||||
siteName string
|
||||
}
|
||||
|
||||
func New(siteName string) (*Nominatim, error) {
|
||||
if siteName == "" {
|
||||
return nil, fmt.Errorf("`siteName` must be defined.")
|
||||
}
|
||||
|
||||
return &Nominatim{
|
||||
siteName: siteName,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (n *Nominatim) Search(q string) (*SearchResults, error) {
|
||||
url := fmt.Sprintf("https://nominatim.openstreetmap.org/search?q=%s&format=jsonv2", q)
|
||||
|
||||
req, err := http.NewRequest("GET", url, nil)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
req.Header.Set("User-Agent", n.siteName)
|
||||
|
||||
resp, err := http.DefaultClient.Do(req)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
|
||||
if resp.StatusCode != http.StatusOK {
|
||||
return nil, fmt.Errorf("%s", resp.Status)
|
||||
}
|
||||
|
||||
body, err := io.ReadAll(resp.Body)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
result := SearchResults{}
|
||||
err = json.Unmarshal(body, &result)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &result, nil
|
||||
|
||||
}
|
||||
|
||||
func (n *Nominatim) Lookup(q string) (*Coordinates, error) {
|
||||
r, err := n.Search(q)
|
||||
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if len(*r) == 0 {
|
||||
return nil, fmt.Errorf("no result")
|
||||
}
|
||||
|
||||
firstResult := (*r)[0]
|
||||
lon, err := strconv.ParseFloat(*firstResult.Longitude, 64)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
lat, err := strconv.ParseFloat(*firstResult.Latitude, 64)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &Coordinates{
|
||||
Latitude: lat,
|
||||
Longitude: lon,
|
||||
}, nil
|
||||
}
|
57
internal/nominatim/nominatim_test.go
Normal file
57
internal/nominatim/nominatim_test.go
Normal file
|
@ -0,0 +1,57 @@
|
|||
package nominatim
|
||||
|
||||
import (
|
||||
"strconv"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func TestNewClient(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
c, err := New("my siteName")
|
||||
assert.NoError(err)
|
||||
|
||||
assert.Equal("my siteName", c.siteName)
|
||||
|
||||
_, err = New("")
|
||||
assert.Error(err)
|
||||
}
|
||||
|
||||
func TestSearch(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
c, err := New("my siteName")
|
||||
assert.NoError(err)
|
||||
|
||||
r, err := c.Search("bergen")
|
||||
assert.NoError(err)
|
||||
|
||||
assert.NotEmpty(r)
|
||||
firstResult := (*r)[0]
|
||||
lon, err := strconv.ParseFloat(*firstResult.Longitude, 64)
|
||||
assert.NoError(err)
|
||||
lat, err := strconv.ParseFloat(*firstResult.Latitude, 64)
|
||||
assert.NoError(err)
|
||||
|
||||
assert.Equal(5.3259192, lon)
|
||||
assert.Equal(60.3943055, lat)
|
||||
|
||||
assert.Nil(firstResult.Address)
|
||||
}
|
||||
|
||||
func TestLookup(t *testing.T) {
|
||||
assert := assert.New(t)
|
||||
c, err := New("my siteName")
|
||||
assert.NoError(err)
|
||||
|
||||
r, err := c.Lookup("bergen")
|
||||
assert.NoError(err)
|
||||
|
||||
assert.NotEmpty(r)
|
||||
|
||||
assert.Equal(5.3259192, r.Longitude)
|
||||
assert.Equal(60.3943055, r.Latitude)
|
||||
|
||||
_, err = c.Lookup("")
|
||||
assert.Error(err)
|
||||
}
|
39
internal/nominatim/types.go
Normal file
39
internal/nominatim/types.go
Normal file
|
@ -0,0 +1,39 @@
|
|||
package nominatim
|
||||
|
||||
type Address struct {
|
||||
ISO3166 string `json:"ISO3166-2-lvl4"`
|
||||
Borough string `json:"borough"`
|
||||
City string `json:"city"`
|
||||
Country string `json:"country"`
|
||||
CountryCode string `json:"country_code"`
|
||||
Neighbourhood string `json:"neighbourhood"`
|
||||
Postcode string `json:"postcode"`
|
||||
Road string `json:"road"`
|
||||
Shop string `json:"shop"`
|
||||
Suburb string `json:"suburb"`
|
||||
}
|
||||
|
||||
type SearchResult struct {
|
||||
Address *Address `json:"address"`
|
||||
AddressType *string `json:"addresstype"`
|
||||
BoundingBox *[]string `json:"boundingbox"`
|
||||
Category *string `json:"category"`
|
||||
DisplayName *string `json:"display_name"`
|
||||
Importance *float64 `json:"importance"`
|
||||
Latitude *string `json:"lat"`
|
||||
Longitude *string `json:"lon"`
|
||||
Licence *string `json:"licence"`
|
||||
Name *string `json:"name"`
|
||||
OsmId *int `json:"osm_id"`
|
||||
OsmType *string `json:"osm_type"`
|
||||
PlaceId *int `json:"place_id"`
|
||||
PlaceRank *int `json:"place_rank"`
|
||||
Type *string `json:"type"`
|
||||
}
|
||||
|
||||
type SearchResults = []SearchResult
|
||||
|
||||
type Coordinates struct {
|
||||
Latitude float64 `json:"latitude"`
|
||||
Longitude float64 `json:"longitude"`
|
||||
}
|
Loading…
Reference in a new issue