feat: Add interval to forecast cmd

Signed-off-by: Tim Hårek Andreassen <tim@harek.no>
This commit is contained in:
Tim Hårek Andreassen 2024-10-02 20:40:49 +02:00
parent 65dd6ad1f5
commit 6cb3994ae9
No known key found for this signature in database
GPG key ID: E59C7734F0E10EB5
2 changed files with 9 additions and 4 deletions

View file

@ -4,4 +4,5 @@ const (
LON = "lon"
LAT = "lat"
JSON = "json"
INTERVAL = "interval"
)

View file

@ -24,6 +24,7 @@ var forecastCmd = &cobra.Command{
func init() {
rootCmd.AddCommand(forecastCmd)
rootCmd.PersistentFlags().IntP(flags.INTERVAL, "i", 0, "Number of hours to output")
}
func forecast(cmd *cobra.Command, args []string) {
@ -31,6 +32,7 @@ func forecast(cmd *cobra.Command, args []string) {
cobra.CheckErr(err)
lon, _ := cmd.Flags().GetFloat64(flags.LON)
lat, _ := cmd.Flags().GetFloat64(flags.LAT)
interval, _ := cmd.Flags().GetInt(flags.INTERVAL)
c, err := yr.New()
cobra.CheckErr(err)
@ -43,12 +45,14 @@ func forecast(cmd *cobra.Command, args []string) {
if len(args) == 0 {
f, err = c.ForecastCoords(&nominatim.Coordinates{Longitude: lon, Latitude: lat}, nil)
cobra.CheckErr(err)
} else {
location := args[0]
f, err = c.Forecast(location)
cobra.CheckErr(err)
}
if interval > 0 {
f = f[:interval]
}
if isJson {