updated some logic and better error handling
This commit is contained in:
264
main.go
Normal file
264
main.go
Normal file
@@ -0,0 +1,264 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
log "github.com/sirupsen/logrus"
|
||||
|
||||
interfaces "vikingowl/poly-weather-script-go/interfaces"
|
||||
)
|
||||
|
||||
// Settings
|
||||
var (
|
||||
// optional
|
||||
city = ""
|
||||
location_lat = ""
|
||||
location_lon = ""
|
||||
// required
|
||||
apiKey = "" // get your api key on https://openweathermap.org/
|
||||
units = "metric"
|
||||
temperature_unit = "C"
|
||||
baseUrl = "https://api.openweathermap.org/data/2.5"
|
||||
|
||||
atmophere_icons_list = map[int]string{
|
||||
701: "", // Mist
|
||||
711: "", // Smoke
|
||||
721: "", // Haze
|
||||
731: "", // Dust (Sand / dust whirls)
|
||||
741: "", // Fog
|
||||
751: "", // Sand
|
||||
761: "", // Dust
|
||||
762: "", // Ash
|
||||
771: "", // Squalls
|
||||
781: "", // Tornado
|
||||
}
|
||||
)
|
||||
|
||||
func setColor(apiIcon string) (color string, icon string) {
|
||||
colorIcons := map[string]map[string]string{
|
||||
"#fdd835": { // yellow
|
||||
"01d": "",
|
||||
"01n": "",
|
||||
"02d": "",
|
||||
"02n": "",
|
||||
"04d": "",
|
||||
"04n": "",
|
||||
"11d": "",
|
||||
"11n": "",
|
||||
},
|
||||
"$foreground": { // foreground
|
||||
"03d": "",
|
||||
"03n": "",
|
||||
},
|
||||
"#42a5f5": { // blue
|
||||
"09d": "",
|
||||
"09n": "",
|
||||
"10d": "",
|
||||
"10n": "",
|
||||
},
|
||||
"#4dd0e1": { // cyan
|
||||
"13d": "ﰕ",
|
||||
"13n": "ﰕ",
|
||||
},
|
||||
"#9e9e9e": { // gray
|
||||
"50d": "",
|
||||
"50n": "",
|
||||
},
|
||||
}
|
||||
|
||||
for colorI, icons := range colorIcons {
|
||||
if iconI, f := icons[apiIcon]; f {
|
||||
color = colorI
|
||||
icon = iconI
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func getLocation() {
|
||||
res, err := http.Get("https://location.services.mozilla.com/v1/geolocate?key=geoclue")
|
||||
if err != nil {
|
||||
log.Warningf("[GL] unable to get location: %v", err)
|
||||
}
|
||||
|
||||
if res.StatusCode >= 200 && res.StatusCode < 300 {
|
||||
defer res.Body.Close()
|
||||
|
||||
body, err := ioutil.ReadAll(res.Body)
|
||||
if err != nil {
|
||||
log.Warningf("[GL] unable to read location response: %v", err)
|
||||
}
|
||||
|
||||
location := interfaces.GeoLocation{}
|
||||
if err := json.Unmarshal(body, &location); err != nil {
|
||||
log.Warningf("[GL] unable to marshal location data: %v", err)
|
||||
}
|
||||
|
||||
location_lat = fmt.Sprint(location.Location.Lat)
|
||||
location_lon = fmt.Sprint(location.Location.Lon)
|
||||
} else {
|
||||
// TODO: needs updating if errors occur
|
||||
log.Warningf("[GL] something went wrong while getting the location: %v", res.StatusCode)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func setUrls(apiKey string) (urls interfaces.Urls) {
|
||||
if city == "" {
|
||||
if location_lat == "" && location_lon == "" {
|
||||
getLocation()
|
||||
}
|
||||
|
||||
urls.Url = fmt.Sprintf("%s/weather?lat=%s&lon=%s&units=%s&appid=%s", baseUrl, location_lat, location_lon, units, apiKey)
|
||||
urls.Url_forecast = fmt.Sprintf("%s/forecast?lat=%s&lon=%s&units=%s&appid=%s", baseUrl, location_lat, location_lon, units, apiKey)
|
||||
} else {
|
||||
urls.Url = fmt.Sprintf("%s/weather?q=%s&units=%s&appid=%s", baseUrl, city, units, apiKey)
|
||||
urls.Url_forecast = fmt.Sprintf("%s/forecast?q=%s&units=%s&appid=%s", baseUrl, city, units, apiKey)
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func getWeather(url string) (weather interfaces.Weather) {
|
||||
res, err := http.Get(url)
|
||||
if err != nil {
|
||||
log.Warningf("[GW] unable to get weather: %v", err)
|
||||
}
|
||||
|
||||
if res.StatusCode >= 200 && res.StatusCode < 300 {
|
||||
defer res.Body.Close()
|
||||
|
||||
body, err := ioutil.ReadAll(res.Body)
|
||||
if err != nil {
|
||||
log.Warningf("[GW] unable to read weather: %v", err)
|
||||
}
|
||||
|
||||
if err := json.Unmarshal(body, &weather); err != nil {
|
||||
log.Warningf("[GW] unable to marshal weather data: %v", err)
|
||||
}
|
||||
} else {
|
||||
if res.StatusCode == 400 {
|
||||
log.Warning("[GW] bad request")
|
||||
} else if res.StatusCode == 401 {
|
||||
log.Warning("[GW] unauthorized - wrong api key")
|
||||
} else {
|
||||
log.Warningf("[GW] something went wrong while getting the weather: %v", res.StatusCode)
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func getForecast(url string) (forecast interfaces.Forecast) {
|
||||
res, err := http.Get(url)
|
||||
if err != nil {
|
||||
log.Warningf("[GF] unable to get forecast: %v", err)
|
||||
}
|
||||
|
||||
if res.StatusCode >= 200 && res.StatusCode < 300 {
|
||||
defer res.Body.Close()
|
||||
|
||||
body, err := ioutil.ReadAll(res.Body)
|
||||
if err != nil {
|
||||
log.Warningf("[GF] unable to read forecast: %v", err)
|
||||
}
|
||||
|
||||
if err := json.Unmarshal(body, &forecast); err != nil {
|
||||
log.Warningf("[GF] unable to marshal forecast data: %v", err)
|
||||
}
|
||||
} else {
|
||||
if res.StatusCode == 400 {
|
||||
log.Warning("[GF] bad request")
|
||||
} else if res.StatusCode == 401 {
|
||||
log.Warning("[GF] unauthorized - wrong api key")
|
||||
} else {
|
||||
log.Warningf("[GF] something went wrong while getting the forecast: %v", res.StatusCode)
|
||||
}
|
||||
}
|
||||
|
||||
return
|
||||
}
|
||||
|
||||
func main() {
|
||||
// try to get apiKey and location from environment
|
||||
// LOOKS LIKE THIS DOESN'T WORK IN POLYBAR
|
||||
// TODO: Needs fix
|
||||
// if apiKey == "" {
|
||||
// apiKey = os.Getenv("OPEN_WEATHER_MAP_API_KEY")
|
||||
// if apiKey == "" {
|
||||
// log.Warning("[main] you need an API KEY to use this script! Get one on https://openweathermap.org/")
|
||||
// os.Exit(1)
|
||||
// }
|
||||
// }
|
||||
// if location_lat == "" {
|
||||
// location_lat = fmt.Sprint(os.Getenv("LAT"))
|
||||
// }
|
||||
// if location_lon == "" {
|
||||
// location_lon = fmt.Sprint(os.Getenv("LNG"))
|
||||
// }
|
||||
// -----------------------------------------------
|
||||
|
||||
// api calls
|
||||
urls := setUrls(apiKey)
|
||||
weatherData := getWeather(urls.Url)
|
||||
forecastData := getForecast(urls.Url_forecast)
|
||||
|
||||
var (
|
||||
curr string
|
||||
forecast string
|
||||
atmosphere string
|
||||
atmosphereForecast string
|
||||
)
|
||||
|
||||
// process weather data
|
||||
if weatherData.Weather[0].ID != 0 {
|
||||
// Get info from data
|
||||
id := int(weatherData.Weather[0].ID)
|
||||
group := strings.Title(weatherData.Weather[0].Main)
|
||||
apiIcon := weatherData.Weather[0].Icon
|
||||
temp := fmt.Sprintf("%.1f", weatherData.Main.Temp)
|
||||
|
||||
color, icon := setColor(apiIcon)
|
||||
|
||||
// Load other icons for Atmosphere group
|
||||
if group == "Atmosphere" {
|
||||
atmosphere = "%{F#e57c46}" + atmophere_icons_list[id] + "%{F$foreground} " + temp + "°" + temperature_unit + "%{F-}"
|
||||
}
|
||||
|
||||
curr = "%{F" + color + "}" + icon + "%{F$foreground} " + temp + "°" + temperature_unit + "%{F-}"
|
||||
} else {
|
||||
forecast = "勒"
|
||||
}
|
||||
|
||||
// process forecast data
|
||||
if forecastData.List[0].Weather[0].ID != 0 {
|
||||
// Get info from data
|
||||
id := int(forecastData.List[0].Weather[0].ID)
|
||||
group := strings.Title(forecastData.List[0].Weather[0].Main)
|
||||
apiIcon := forecastData.List[0].Weather[0].Icon
|
||||
temp := fmt.Sprintf("%.1f", forecastData.List[0].Main.Temp)
|
||||
|
||||
color, icon := setColor(apiIcon)
|
||||
|
||||
// Load other icons for Atmosphere group
|
||||
if group == "Atmosphere" {
|
||||
atmosphereForecast = "%{F#e57c46}" + atmophere_icons_list[id] + "{F$foreground} " + temp + "°" + temperature_unit + "%{F-}"
|
||||
}
|
||||
|
||||
forecast = "%{F" + color + "}" + icon + "%{F$foreground} " + temp + "°" + temperature_unit + "%{F-}"
|
||||
} else {
|
||||
forecast = "勒"
|
||||
}
|
||||
|
||||
// print output
|
||||
if atmosphere == "" {
|
||||
fmt.Printf("%s %s", curr, forecast)
|
||||
} else {
|
||||
fmt.Printf("%s %s", atmosphere, atmosphereForecast)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user