feat: Add interval to forecast cmd

Signed-off-by: Tim Hårek Andreassen <tim@harek.no>

cmd/flags/flags.go

@@ -1,7 +1,8 @@
 package flags
 
 const (
-	LON  = "lon"
-	LAT  = "lat"
-	JSON = "json"
+	LON      = "lon"
+	LAT      = "lat"
+	JSON     = "json"
+	INTERVAL = "interval"
 )

cmd/forecast.go

@@ -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 {