yr/cmd/forecast_helper.go
Tim Hårek Andreassen bfbd031e41
refactor: Swap logic order for UTC time convertion
Signed-off-by: Tim Hårek Andreassen <tim@harek.no>
2024-10-04 23:38:23 +02:00

88 lines
1.9 KiB
Go

package cmd
import (
"encoding/json"
"fmt"
"os"
"git.sr.ht/~timharek/yr/cmd/flags"
"git.sr.ht/~timharek/yr/pkg/nominatim"
"git.sr.ht/~timharek/yr/yr"
"github.com/pkg/browser"
"github.com/spf13/cobra"
)
type forecastH struct {
isJson bool
isDebug bool
isUTC bool
isWeb bool
f *yr.ForecastResult
}
func forecastHelper(cmd *cobra.Command, args []string) *forecastH {
isJson, err := cmd.Flags().GetBool(flags.JSON)
cobra.CheckErr(err)
isDebug, err := cmd.Flags().GetBool(flags.DEBUG)
cobra.CheckErr(err)
isUTC, err := cmd.Flags().GetBool(flags.UTC)
cobra.CheckErr(err)
isWeb, err := cmd.Flags().GetBool(flags.WEB)
cobra.CheckErr(err)
lon, _ := cmd.Flags().GetFloat64(flags.LON)
lat, _ := cmd.Flags().GetFloat64(flags.LAT)
c, err := yr.New()
cobra.CheckErr(err)
if len(args) == 0 && (lon == 0 || lat == 0) {
fmt.Fprintln(os.Stderr, "No location or coordinates provided.")
os.Exit(1)
}
f := &yr.ForecastResult{}
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 isDebug {
if !isUTC {
f.Expires = f.Expires.Local()
f.LastModified = f.LastModified.Local()
}
fmt.Printf("Expires: %v\n", f.Expires)
fmt.Printf("LastModified: %v\n", f.LastModified)
fmt.Printf("Default cache-dir: %v\n", os.TempDir())
}
return &forecastH{
isJson,
isDebug,
isUTC,
isWeb,
f,
}
}
// Opens u in browser with added coordinates and exits
func openBrowser(u string, f *yr.ForecastResult) {
url := u
if f != nil {
url = fmt.Sprintf(u, f.Coordinates.Latitude, f.Coordinates.Longitude)
}
err := browser.OpenURL(url)
cobra.CheckErr(err)
os.Exit(0)
}
// Returns forecast in pretty printed JSON and exits
func outputJson(v any) {
j, err := json.MarshalIndent(v, "", " ")
cobra.CheckErr(err)
fmt.Printf("%s\n", j)
os.Exit(0)
}