accept flags and -d for a box
This commit is contained in:
35
cmd/run.go
35
cmd/run.go
@@ -3,18 +3,30 @@ package cmd
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"path"
|
||||
)
|
||||
|
||||
func Run(ctx context.Context) error {
|
||||
m, err := NewMapsOf(ctx, os.Args[1])
|
||||
fs := flag.NewFlagSet(os.Args[0], flag.ContinueOnError)
|
||||
searchRadius := fs.Float64("r", 5.0, "search radius in miles")
|
||||
area := fs.Float64("a", 0.5, "result radius in miles")
|
||||
doArea := fs.Bool("d", false, "do result radius in miles")
|
||||
if err := fs.Parse(os.Args[1:]); err != nil {
|
||||
panic(err)
|
||||
}
|
||||
radius := *area / convertToMiles
|
||||
radiusX := radius / 2
|
||||
radiusY := radius / 3
|
||||
|
||||
m, err := NewMapsOf(ctx, fs.Args()[0])
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
results, err := m.Search(ctx, os.Args[2], 5.0)
|
||||
results, err := m.Search(ctx, fs.Args()[1], *searchRadius)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
@@ -25,16 +37,27 @@ func Run(ctx context.Context) error {
|
||||
Name string `json:"name"`
|
||||
} `json:"properties"`
|
||||
Geometry struct {
|
||||
Type string `json:"type"`
|
||||
Coordinates []float64 `json:"coordinates"`
|
||||
Type string `json:"type"`
|
||||
Coordinates []any `json:"coordinates"`
|
||||
} `json:"geometry"`
|
||||
}
|
||||
geoJsons := make([]geoJson, len(results))
|
||||
for i := range results {
|
||||
geoJsons[i].Type = "Feature"
|
||||
geoJsons[i].Properties.Name = path.Join(os.Args[2], results[i].Name)
|
||||
geoJsons[i].Properties.Name = path.Join(fs.Args()[1], results[i].Name)
|
||||
geoJsons[i].Geometry.Type = "Point"
|
||||
geoJsons[i].Geometry.Coordinates = []float64{results[i].Lng, results[i].Lat}
|
||||
geoJsons[i].Geometry.Coordinates = []any{results[i].Lng, results[i].Lat}
|
||||
if *doArea {
|
||||
geoJsons[i].Geometry.Type = "Polygon"
|
||||
x, y := results[i].Lng, results[i].Lat
|
||||
geoJsons[i].Geometry.Coordinates = []any{[]any{
|
||||
[2]float64{x - radiusX, y + radiusY}, // top left
|
||||
[2]float64{x + radiusX, y + radiusY}, // top righ
|
||||
[2]float64{x + radiusX, y - radiusY}, // bot righ
|
||||
[2]float64{x - radiusX, y - radiusY}, // bot left
|
||||
[2]float64{x - radiusX, y + radiusY}, // top left
|
||||
}}
|
||||
}
|
||||
}
|
||||
b, _ := json.Marshal(map[string]any{
|
||||
"features": geoJsons,
|
||||
|
||||
Reference in New Issue
Block a user