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+"/")
}