mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
151 lines
3.2 KiB
Go
151 lines
3.2 KiB
Go
package gcal
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/gdamore/tcell"
|
|
"github.com/olebedev/config"
|
|
"github.com/rivo/tview"
|
|
"github.com/senorprogrammer/wtf/wtf"
|
|
"google.golang.org/api/calendar/v3"
|
|
)
|
|
|
|
var Config *config.Config
|
|
|
|
type Widget struct {
|
|
wtf.BaseWidget
|
|
|
|
View *tview.TextView
|
|
}
|
|
|
|
func NewWidget() *Widget {
|
|
widget := Widget{
|
|
BaseWidget: wtf.BaseWidget{
|
|
Name: "Calendar",
|
|
RefreshedAt: time.Now(),
|
|
RefreshInt: Config.UInt("wtf.gcal.refreshInterval", 300),
|
|
},
|
|
}
|
|
|
|
widget.addView()
|
|
|
|
return &widget
|
|
}
|
|
|
|
/* -------------------- Exported Functions -------------------- */
|
|
|
|
func (widget *Widget) Refresh() {
|
|
events, _ := Fetch()
|
|
|
|
widget.View.SetTitle(" 🍿 Calendar ")
|
|
widget.RefreshedAt = time.Now()
|
|
|
|
widget.View.Clear()
|
|
fmt.Fprintf(widget.View, "%s", widget.contentFrom(events))
|
|
}
|
|
|
|
/* -------------------- Unexported Functions -------------------- */
|
|
|
|
func (widget *Widget) addView() {
|
|
view := tview.NewTextView()
|
|
|
|
view.SetBorder(true)
|
|
view.SetBorderColor(tcell.ColorGrey)
|
|
view.SetDynamicColors(true)
|
|
view.SetTitle(widget.Name)
|
|
view.SetWrap(false)
|
|
|
|
widget.View = view
|
|
}
|
|
|
|
func (widget *Widget) contentFrom(events *calendar.Events) string {
|
|
if events == nil {
|
|
return ""
|
|
}
|
|
|
|
str := "\n"
|
|
|
|
for _, event := range events.Items {
|
|
startTime, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
|
timestamp := startTime.Format("Mon, Jan 2, 15:04")
|
|
until := widget.until(startTime)
|
|
|
|
summary := event.Summary
|
|
if widget.eventIsNow(event) {
|
|
summary = "🔥 " + summary
|
|
}
|
|
|
|
str = str + fmt.Sprintf(" [%s]%s[white]\n [%s]%s %s[white]\n\n", widget.titleColor(event), summary, widget.descriptionColor(event), timestamp, until)
|
|
}
|
|
|
|
return str
|
|
}
|
|
|
|
// eventIsNow returns true if the event is happening now, false if it not
|
|
func (widget *Widget) eventIsNow(event *calendar.Event) bool {
|
|
startTime, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
|
endTime, _ := time.Parse(time.RFC3339, event.End.DateTime)
|
|
|
|
return time.Now().After(startTime) && time.Now().Before(endTime)
|
|
}
|
|
|
|
func (widget *Widget) descriptionColor(event *calendar.Event) string {
|
|
ts, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
|
|
|
color := "white"
|
|
if (widget.eventIsNow(event) == false) && ts.Before(time.Now()) {
|
|
color = "grey"
|
|
}
|
|
|
|
return color
|
|
}
|
|
|
|
func (widget *Widget) titleColor(event *calendar.Event) string {
|
|
ts, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
|
|
|
color := "red"
|
|
if strings.Contains(event.Summary, "1on1") {
|
|
color = "green"
|
|
}
|
|
|
|
if (widget.eventIsNow(event) == false) && ts.Before(time.Now()) {
|
|
color = "grey"
|
|
}
|
|
|
|
return color
|
|
}
|
|
|
|
// until returns the number of hours or days until the event
|
|
// If the event is in the past, returns nil
|
|
func (widget *Widget) until(start time.Time) string {
|
|
duration := time.Until(start)
|
|
|
|
duration = duration.Round(time.Minute)
|
|
|
|
if duration < 0 {
|
|
return ""
|
|
}
|
|
|
|
days := duration / (24 * time.Hour)
|
|
duration -= days * (24 * time.Hour)
|
|
|
|
hours := duration / time.Hour
|
|
duration -= hours * time.Hour
|
|
|
|
mins := duration / time.Minute
|
|
|
|
untilStr := ""
|
|
|
|
if days > 0 {
|
|
untilStr = fmt.Sprintf("%dd", days)
|
|
} else if hours > 0 {
|
|
untilStr = fmt.Sprintf("%dh", hours)
|
|
} else {
|
|
untilStr = fmt.Sprintf("%dm", mins)
|
|
}
|
|
|
|
return "[grey]" + untilStr + "[white]"
|
|
}
|