mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
151 lines
3.1 KiB
Go
151 lines
3.1 KiB
Go
package weather
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
owm "github.com/briandowns/openweathermap"
|
|
"github.com/rivo/tview"
|
|
"github.com/senorprogrammer/wtf/wtf"
|
|
)
|
|
|
|
type Widget struct {
|
|
wtf.BaseWidget
|
|
View *tview.TextView
|
|
}
|
|
|
|
func NewWidget() *Widget {
|
|
widget := Widget{
|
|
BaseWidget: wtf.BaseWidget{
|
|
Name: "Weather",
|
|
RefreshedAt: time.Now(),
|
|
RefreshInterval: 600,
|
|
},
|
|
}
|
|
|
|
widget.addView()
|
|
go widget.refresher()
|
|
|
|
return &widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
func (widget *Widget) Refresh() {
|
|
data := Fetch()
|
|
|
|
widget.View.SetTitle(fmt.Sprintf(" %s Weather - %s ", icon(data), data.Name))
|
|
widget.RefreshedAt = time.Now()
|
|
|
|
widget.View.Clear()
|
|
fmt.Fprintf(widget.View, "%s", widget.contentFrom(data))
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) addView() {
|
|
view := tview.NewTextView()
|
|
|
|
view.SetBorder(true)
|
|
view.SetDynamicColors(true)
|
|
view.SetTitle(widget.Name)
|
|
|
|
widget.View = view
|
|
}
|
|
|
|
func centerText(str string, width int) string {
|
|
return fmt.Sprintf("%[1]*s", -width, fmt.Sprintf("%[1]*s", (width+len(str))/2, str))
|
|
}
|
|
|
|
func (widget *Widget) contentFrom(data *owm.CurrentWeatherData) string {
|
|
str := fmt.Sprintf("\n")
|
|
|
|
descs := []string{}
|
|
for _, weather := range data.Weather {
|
|
descs = append(descs, fmt.Sprintf(" %s", weather.Description))
|
|
}
|
|
|
|
str = str + strings.Join(descs, ",") + "\n\n"
|
|
|
|
str = str + fmt.Sprintf("%10s: %4.1f° C\n\n", "Current", data.Main.Temp)
|
|
str = str + fmt.Sprintf("%10s: %4.1f° C\n", "High", data.Main.TempMax)
|
|
str = str + fmt.Sprintf("%10s: %4.1f° C\n", "Low", data.Main.TempMin)
|
|
str = str + "\n\n\n\n"
|
|
str = str + centerText(fmt.Sprintf("Refreshed at %s", widget.refreshedAt()), 38)
|
|
|
|
return str
|
|
}
|
|
|
|
// icon returns an emoji for the current weather
|
|
// src: https://github.com/chubin/wttr.in/blob/master/share/translations/en.txt
|
|
func icon(data *owm.CurrentWeatherData) string {
|
|
var icon string
|
|
|
|
switch data.Weather[0].Description {
|
|
case "broken clouds":
|
|
icon = "☁️"
|
|
case "clear":
|
|
icon = "☀️"
|
|
case "clear sky":
|
|
icon = "☀️"
|
|
case "cloudy":
|
|
icon = "⛅️"
|
|
case "few clouds":
|
|
icon = "🌤"
|
|
case "fog":
|
|
icon = "🌫"
|
|
case "heavy rain":
|
|
icon = "💦"
|
|
case "heavy snow":
|
|
icon = "⛄️"
|
|
case "light intensity shower rain":
|
|
icon = "☔️"
|
|
case "light rain":
|
|
icon = "🌦"
|
|
case "light snow":
|
|
icon = "🌨"
|
|
case "mist":
|
|
icon = "🌬"
|
|
case "moderate rain":
|
|
icon = "🌧"
|
|
case "moderate snow":
|
|
icon = "🌨"
|
|
case "overcast":
|
|
icon = "🌥"
|
|
case "overcast clouds":
|
|
icon = "🌥"
|
|
case "partly cloudy":
|
|
icon = "🌤"
|
|
case "scattered clouds":
|
|
icon = "☁️"
|
|
case "snow":
|
|
icon = "❄️"
|
|
case "sunny":
|
|
icon = "☀️"
|
|
default:
|
|
icon = "💥"
|
|
}
|
|
|
|
return icon
|
|
}
|
|
|
|
func (widget *Widget) refresher() {
|
|
tick := time.NewTicker(time.Duration(widget.RefreshInterval) * time.Second)
|
|
quit := make(chan struct{})
|
|
|
|
for {
|
|
select {
|
|
case <-tick.C:
|
|
widget.Refresh()
|
|
case <-quit:
|
|
tick.Stop()
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (widget *Widget) refreshedAt() string {
|
|
return widget.RefreshedAt.Format("15:04:05")
|
|
}
|