mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
7f3daaac59
Define help with keys This means that keys and help are automatically in sync This means that you can't define keys, but forget help This unfortunately also means that formatting may not be quite as good
133 lines
2.9 KiB
Go
133 lines
2.9 KiB
Go
package weather
|
|
|
|
import (
|
|
owm "github.com/briandowns/openweathermap"
|
|
"github.com/rivo/tview"
|
|
"github.com/wtfutil/wtf/wtf"
|
|
)
|
|
|
|
// Widget is the container for weather data.
|
|
type Widget struct {
|
|
wtf.KeyboardWidget
|
|
wtf.TextWidget
|
|
|
|
// APIKey string
|
|
Data []*owm.CurrentWeatherData
|
|
Idx int
|
|
|
|
settings *Settings
|
|
}
|
|
|
|
// NewWidget creates and returns a new instance of the weather Widget
|
|
func NewWidget(app *tview.Application, pages *tview.Pages, settings *Settings) *Widget {
|
|
widget := Widget{
|
|
KeyboardWidget: wtf.NewKeyboardWidget(app, pages, settings.common),
|
|
TextWidget: wtf.NewTextWidget(app, settings.common, true),
|
|
|
|
Idx: 0,
|
|
|
|
settings: settings,
|
|
}
|
|
|
|
widget.initializeKeyboardControls()
|
|
widget.View.SetInputCapture(widget.InputCapture)
|
|
|
|
widget.KeyboardWidget.SetView(widget.View)
|
|
|
|
return &widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
// Fetch retrieves OpenWeatherMap data from the OpenWeatherMap API.
|
|
// It takes a list of OpenWeatherMap city IDs.
|
|
// It returns a list of OpenWeatherMap CurrentWeatherData structs, one per valid city code.
|
|
func (widget *Widget) Fetch(cityIDs []int) []*owm.CurrentWeatherData {
|
|
data := []*owm.CurrentWeatherData{}
|
|
|
|
for _, cityID := range cityIDs {
|
|
result, err := widget.currentWeather(cityID)
|
|
if err == nil {
|
|
data = append(data, result)
|
|
}
|
|
}
|
|
|
|
return data
|
|
}
|
|
|
|
// Refresh fetches new data from the OpenWeatherMap API and loads the new data into the.
|
|
// widget's view for rendering
|
|
func (widget *Widget) Refresh() {
|
|
if widget.apiKeyValid() {
|
|
widget.Data = widget.Fetch(wtf.ToInts(widget.settings.cityIDs))
|
|
}
|
|
|
|
widget.display()
|
|
}
|
|
|
|
// Next displays data for the next city data in the list. If the current city is the last
|
|
// city, it wraps to the first city.
|
|
func (widget *Widget) Next() {
|
|
widget.Idx = widget.Idx + 1
|
|
if widget.Idx == len(widget.Data) {
|
|
widget.Idx = 0
|
|
}
|
|
|
|
widget.display()
|
|
}
|
|
|
|
// Prev displays data for the previous city in the list. If the previous city is the first
|
|
// city, it wraps to the last city.
|
|
func (widget *Widget) Prev() {
|
|
widget.Idx = widget.Idx - 1
|
|
if widget.Idx < 0 {
|
|
widget.Idx = len(widget.Data) - 1
|
|
}
|
|
|
|
widget.display()
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) apiKeyValid() bool {
|
|
if widget.settings.apiKey == "" {
|
|
return false
|
|
}
|
|
|
|
if len(widget.settings.apiKey) != 32 {
|
|
return false
|
|
}
|
|
|
|
return true
|
|
}
|
|
|
|
func (widget *Widget) currentData() *owm.CurrentWeatherData {
|
|
if len(widget.Data) == 0 {
|
|
return nil
|
|
}
|
|
|
|
if widget.Idx < 0 || widget.Idx >= len(widget.Data) {
|
|
return nil
|
|
}
|
|
|
|
return widget.Data[widget.Idx]
|
|
}
|
|
|
|
func (widget *Widget) currentWeather(cityCode int) (*owm.CurrentWeatherData, error) {
|
|
weather, err := owm.NewCurrent(
|
|
widget.settings.tempUnit,
|
|
widget.settings.language,
|
|
widget.settings.apiKey,
|
|
)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
err = weather.CurrentByID(cityCode)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return weather, nil
|
|
}
|