package main import ( "context" "encoding/json" "fmt" "math" "net/http" "net/url" "strconv" "strings" "time" "go.elara.ws/infinitime" "go.elara.ws/infinitime/weather" "go.elara.ws/logger/log" ) // METResponse represents a response from // the MET Norway API type METResponse struct { Properties struct { Timeseries []struct { Time time.Time Data METData } } } // METData represents data in a METResponse type METData struct { Instant struct { Details struct { AirPressure float32 `json:"air_pressure_at_sea_level"` AirTemperature float32 `json:"air_temperature"` DewPoint float32 `json:"dew_point_temperature"` CloudAreaFraction float32 `json:"cloud_area_fraction"` FogAreaFraction float32 `json:"fog_area_fraction"` RelativeHumidity float32 `json:"relative_humidity"` UVIndex float32 `json:"ultraviolet_index_clear_sky"` WindDirection float32 `json:"wind_from_direction"` WindSpeed float32 `json:"wind_speed"` } } NextHour struct { Summary struct { SymbolCode string `json:"symbol_code"` } Details struct { PrecipitationAmount float32 `json:"precipitation_amount"` } } `json:"next_1_hours"` } // OSMData represents lat/long data from // OpenStreetMap Nominatim type OSMData []struct { Lat string `json:"lat"` Lon string `json:"lon"` } var sendWeatherCh = make(chan struct{}, 1) func sleepCtx(ctx context.Context, d time.Duration) { select { case <-time.After(d): case <-ctx.Done(): } } func initWeather(ctx context.Context, wg WaitGroup, dev *infinitime.Device) error { if !k.Bool("weather.enabled") { return nil } // Get location based on string in config lat, lon, err := getLocation(ctx, k.String("weather.location")) if err != nil { return err } timer := time.NewTimer(time.Hour) wg.Add(1) go func() { defer wg.Done("weather") for { select { case _, ok := <-ctx.Done(): if !ok { return } default: } // Attempt to get weather data, err := getWeather(ctx, lat, lon) if err != nil { log.Warn("Error getting weather data").Err(err).Send() // Wait 15 minutes before retrying sleepCtx(ctx, 15*time.Minute) continue } // Get current data current := data.Properties.Timeseries[0] currentData := current.Data.Instant.Details // Add temperature event err = dev.AddWeatherEvent(weather.TemperatureEvent{ TimelineHeader: weather.NewHeader( time.Now(), weather.EventTypeTemperature, time.Hour, ), Temperature: int16(round(currentData.AirTemperature * 100)), DewPoint: int16(round(currentData.DewPoint)), }) if err != nil { log.Error("Error adding temperature event").Err(err).Send() } // Add precipitation event err = dev.AddWeatherEvent(weather.PrecipitationEvent{ TimelineHeader: weather.NewHeader( time.Now(), weather.EventTypePrecipitation, time.Hour, ), Type: parseSymbol(current.Data.NextHour.Summary.SymbolCode), Amount: uint8(round(current.Data.NextHour.Details.PrecipitationAmount)), }) if err != nil { log.Error("Error adding precipitation event").Err(err).Send() } // Add wind event err = dev.AddWeatherEvent(weather.WindEvent{ TimelineHeader: weather.NewHeader( time.Now(), weather.EventTypeWind, time.Hour, ), SpeedMin: uint8(round(currentData.WindSpeed)), SpeedMax: uint8(round(currentData.WindSpeed)), DirectionMin: uint8(round(currentData.WindDirection)), DirectionMax: uint8(round(currentData.WindDirection)), }) if err != nil { log.Error("Error adding wind event").Err(err).Send() } // Add cloud event err = dev.AddWeatherEvent(weather.CloudsEvent{ TimelineHeader: weather.NewHeader( time.Now(), weather.EventTypeClouds, time.Hour, ), Amount: uint8(round(currentData.CloudAreaFraction)), }) if err != nil { log.Error("Error adding clouds event").Err(err).Send() } // Add humidity event err = dev.AddWeatherEvent(weather.HumidityEvent{ TimelineHeader: weather.NewHeader( time.Now(), weather.EventTypeHumidity, time.Hour, ), Humidity: uint8(round(currentData.RelativeHumidity)), }) if err != nil { log.Error("Error adding humidity event").Err(err).Send() } // Add pressure event err = dev.AddWeatherEvent(weather.PressureEvent{ TimelineHeader: weather.NewHeader( time.Now(), weather.EventTypePressure, time.Hour, ), Pressure: int16(round(currentData.AirPressure)), }) if err != nil { log.Error("Error adding pressure event").Err(err).Send() } // Reset timer to 1 hour timer.Stop() timer.Reset(time.Hour) // Wait for timer to fire or manual update signal select { case <-timer.C: case <-sendWeatherCh: case <-ctx.Done(): return } } }() return nil } // getLocation returns the latitude and longitude // given a location func getLocation(ctx context.Context, loc string) (lat, lon float64, err error) { // Create request URL and perform GET request reqURL := fmt.Sprintf("https://nominatim.openstreetmap.org/search.php?q=%s&format=jsonv2", url.QueryEscape(loc)) req, err := http.NewRequestWithContext(ctx, http.MethodGet, reqURL, nil) if err != nil { return } res, err := http.DefaultClient.Do(req) if err != nil { return } // Decode JSON from response into OSMData data := OSMData{} err = json.NewDecoder(res.Body).Decode(&data) if err != nil { return } // If no data points if len(data) == 0 { return } // Get first data point out := data[0] // Attempt to parse latitude lat, err = strconv.ParseFloat(out.Lat, 64) if err != nil { return } // Attempt to parse longitude lon, err = strconv.ParseFloat(out.Lon, 64) if err != nil { return } return } // getWeather gets weather data given a latitude and longitude func getWeather(ctx context.Context, lat, lon float64) (*METResponse, error) { // Create new GET request req, err := http.NewRequestWithContext( ctx, http.MethodGet, fmt.Sprintf( "https://api.met.no/weatherapi/locationforecast/2.0/complete?lat=%.2f&lon=%.2f", lat, lon, ), nil, ) if err != nil { return nil, err } // Set identifying user agent as per NMI requirements req.Header.Set("User-Agent", fmt.Sprintf("ITD/%s gitea.arsenm.dev/Arsen6331/itd", strings.TrimSpace(version))) // Perform request res, err := http.DefaultClient.Do(req) if err != nil { return nil, err } // Decode JSON from response to METResponse struct out := &METResponse{} err = json.NewDecoder(res.Body).Decode(out) if err != nil { return nil, err } return out, nil } // parseSymbol determines what type of precipitation a symbol code // codes for. func parseSymbol(symCode string) weather.PrecipitationType { switch { case strings.Contains(symCode, "lightrain"): return weather.PrecipitationTypeRain case strings.Contains(symCode, "rain"): return weather.PrecipitationTypeRain case strings.Contains(symCode, "snow"): return weather.PrecipitationTypeSnow case strings.Contains(symCode, "sleet"): return weather.PrecipitationTypeSleet case strings.Contains(symCode, "snow"): return weather.PrecipitationTypeSnow default: return weather.PrecipitationTypeNone } } // round rounds 32-bit floats to 32-bit integers func round(f float32) int32 { return int32(math.Round(float64(f))) }