94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package nws
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"code.jhot.me/jhot/hats/internal/util"
|
|
"github.com/go-resty/resty/v2"
|
|
"github.com/samber/lo"
|
|
)
|
|
|
|
const (
|
|
NWS_BASE_URL = "https://api.weather.gov"
|
|
)
|
|
|
|
type NwsClient struct {
|
|
restClient *resty.Client
|
|
}
|
|
|
|
func New() *NwsClient {
|
|
return &NwsClient{
|
|
restClient: resty.New().
|
|
SetBaseURL(NWS_BASE_URL).
|
|
SetHeader("Accept", "application/geo+json").
|
|
SetHeader("User-Agent", util.GetEnvWithDefault("NWS_USER_AGENT", "(HATS-client, hats@jhot.me)")),
|
|
}
|
|
}
|
|
|
|
func (c *NwsClient) CoordinatesToPoint(lat float64, lon float64) (Point, error) {
|
|
var p Point
|
|
_, err := c.restClient.R().SetResult(&p).Get(fmt.Sprintf("points/%f,%f", lat, lon))
|
|
return p, err
|
|
}
|
|
|
|
func (c *NwsClient) GetHourlyForecast(p Point) (HourlyProps, error) {
|
|
var data struct {
|
|
Properties HourlyProps `json:"properties"`
|
|
}
|
|
_, err := util.CheckSuccess(c.restClient.R().SetResult(&data).
|
|
Get(removeBase(p.Properties.ForecastHourly)))
|
|
if err != nil {
|
|
return HourlyProps{}, nil
|
|
}
|
|
|
|
return data.Properties, nil
|
|
}
|
|
|
|
func (p HourlyProps) GetHighLow(hours int) (high int, low int) {
|
|
if hours > len(p.Periods) {
|
|
hours = len(p.Periods)
|
|
}
|
|
|
|
temps := lo.Map(p.Periods[0:hours-1], func(p HourlyPeriod, _ int) int {
|
|
return p.Temperature
|
|
})
|
|
|
|
return lo.Max(temps), lo.Min(temps)
|
|
}
|
|
|
|
func (c *NwsClient) GetStations(p Point) ([]string, error) {
|
|
var data struct {
|
|
ObservationStations []string `json:"observationStations"`
|
|
}
|
|
_, err := util.CheckSuccess(c.restClient.R().SetResult(&data).
|
|
Get(removeBase(p.Properties.ObservationStations)))
|
|
if err != nil {
|
|
return []string{}, err
|
|
}
|
|
|
|
return data.ObservationStations, nil
|
|
}
|
|
|
|
func (c *NwsClient) GetLatestObservations(p Point) (Observations, error) {
|
|
stations, err := c.GetStations(p)
|
|
if err != nil {
|
|
return Observations{}, fmt.Errorf("%w: error getting stations", err)
|
|
}
|
|
|
|
var data struct {
|
|
Properties Observations `json:"properties"`
|
|
}
|
|
_, err = util.CheckSuccess(c.restClient.R().SetResult(&data).
|
|
Get(removeBase(stations[0] + "/observations/latest")))
|
|
if err != nil {
|
|
return Observations{}, err
|
|
}
|
|
|
|
return data.Properties, nil
|
|
}
|
|
|
|
func removeBase(longUrl string) string {
|
|
return strings.TrimPrefix(longUrl, NWS_BASE_URL+"/")
|
|
}
|