mirror of
https://github.com/taigrr/wtf
synced 2025-01-18 04:03:14 -08:00
Refactor GCal calendar code to simplify things
This commit is contained in:
Chris Cummer
parent
ea930d02b7
commit
3f010f8a9f
219
gcal/display.go
Normal file
219
gcal/display.go
Normal file
@ -0,0 +1,219 @@
|
||||
package gcal
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/senorprogrammer/wtf/wtf"
|
||||
"google.golang.org/api/calendar/v3"
|
||||
)
|
||||
|
||||
func (widget *Widget) display() {
|
||||
if widget.events == nil || len(widget.events.Items) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
widget.mutex.Lock()
|
||||
defer widget.mutex.Unlock()
|
||||
|
||||
widget.View.SetText(widget.contentFrom(widget.events))
|
||||
}
|
||||
|
||||
func (widget *Widget) contentFrom(events *calendar.Events) string {
|
||||
if events == nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
var prevEvent *calendar.Event
|
||||
|
||||
str := ""
|
||||
|
||||
for _, event := range events.Items {
|
||||
timestamp := fmt.Sprintf("[%s]%s",
|
||||
widget.descriptionColor(event),
|
||||
widget.eventTimestamp(event))
|
||||
|
||||
title := fmt.Sprintf("[%s]%s",
|
||||
widget.titleColor(event),
|
||||
widget.eventSummary(event, widget.conflicts(event, events)))
|
||||
|
||||
lineOne := fmt.Sprintf(
|
||||
"%s %s %s %s %s[white]",
|
||||
widget.dayDivider(event, prevEvent),
|
||||
widget.responseIcon(event),
|
||||
timestamp,
|
||||
title,
|
||||
widget.timeUntil(event),
|
||||
)
|
||||
|
||||
str = str + fmt.Sprintf("%s%s\n\n",
|
||||
lineOne,
|
||||
widget.location(event), // prefixes newline if non-empty
|
||||
)
|
||||
|
||||
prevEvent = event
|
||||
}
|
||||
|
||||
return str
|
||||
}
|
||||
|
||||
func (widget *Widget) dayDivider(event, prevEvent *calendar.Event) string {
|
||||
var prevStartTime time.Time
|
||||
|
||||
if prevEvent != nil {
|
||||
prevStartTime, _ = time.Parse(time.RFC3339, prevEvent.Start.DateTime)
|
||||
}
|
||||
|
||||
currStartTime, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
||||
|
||||
if currStartTime.Day() != prevStartTime.Day() {
|
||||
_, _, width, _ := widget.View.GetInnerRect()
|
||||
|
||||
return fmt.Sprintf("[%s]", wtf.Config.UString("wtf.mods.gcal.colors.day", "forestgreen")) +
|
||||
wtf.CenterText(currStartTime.Format(wtf.FullDateFormat), width) +
|
||||
"\n"
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (widget *Widget) descriptionColor(event *calendar.Event) string {
|
||||
color := wtf.Config.UString("wtf.mods.gcal.colors.description", "white")
|
||||
|
||||
if widget.eventIsPast(event) {
|
||||
color = wtf.Config.UString("wtf.mods.gcal.colors.past", "gray")
|
||||
}
|
||||
|
||||
return color
|
||||
}
|
||||
|
||||
func (widget *Widget) eventSummary(event *calendar.Event, conflict bool) string {
|
||||
summary := event.Summary
|
||||
|
||||
if widget.eventIsNow(event) {
|
||||
summary = fmt.Sprintf(
|
||||
"%s %s",
|
||||
wtf.Config.UString("wtf.mods.gcal.currentIcon", "🔸"),
|
||||
event.Summary,
|
||||
)
|
||||
}
|
||||
|
||||
if conflict {
|
||||
return fmt.Sprintf("%s %s", wtf.Config.UString("wtf.mods.gcal.conflictIcon", "🚨"), summary)
|
||||
} else {
|
||||
return summary
|
||||
}
|
||||
}
|
||||
|
||||
func (widget *Widget) eventTimestamp(event *calendar.Event) string {
|
||||
if widget.eventIsAllDay(event) {
|
||||
startTime, _ := time.Parse("2006-01-02", event.Start.Date)
|
||||
return startTime.Format(wtf.FriendlyDateFormat)
|
||||
} else {
|
||||
startTime, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
||||
return startTime.Format(wtf.MinimumTimeFormat)
|
||||
}
|
||||
}
|
||||
|
||||
// timeUuntil returns the number of hours or days until the event
|
||||
// If the event is in the past, returns nil
|
||||
func (widget *Widget) timeUntil(event *calendar.Event) string {
|
||||
startTime, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
||||
duration := time.Until(startTime).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 := ""
|
||||
|
||||
color := "[lightblue]"
|
||||
if days > 0 {
|
||||
untilStr = fmt.Sprintf("%dd", days)
|
||||
} else if hours > 0 {
|
||||
untilStr = fmt.Sprintf("%dh", hours)
|
||||
} else {
|
||||
untilStr = fmt.Sprintf("%dm", mins)
|
||||
if mins < 30 {
|
||||
color = "[red]"
|
||||
}
|
||||
}
|
||||
|
||||
return color + untilStr + "[white]"
|
||||
}
|
||||
|
||||
func (widget *Widget) titleColor(event *calendar.Event) string {
|
||||
color := wtf.Config.UString("wtf.mods.gcal.colors.title", "white")
|
||||
|
||||
for _, untypedArr := range wtf.Config.UList("wtf.mods.gcal.colors.highlights") {
|
||||
highlightElements := wtf.ToStrs(untypedArr.([]interface{}))
|
||||
|
||||
match, _ := regexp.MatchString(
|
||||
strings.ToLower(highlightElements[0]),
|
||||
strings.ToLower(event.Summary),
|
||||
)
|
||||
|
||||
if match == true {
|
||||
color = highlightElements[1]
|
||||
}
|
||||
}
|
||||
|
||||
if widget.eventIsPast(event) {
|
||||
color = wtf.Config.UString("wtf.mods.gcal.colors.past", "gray")
|
||||
}
|
||||
|
||||
return color
|
||||
}
|
||||
|
||||
func (widget *Widget) location(event *calendar.Event) string {
|
||||
if wtf.Config.UBool("wtf.mods.gcal.displayLocation", true) == false {
|
||||
return ""
|
||||
}
|
||||
|
||||
if event.Location == "" {
|
||||
return ""
|
||||
}
|
||||
|
||||
return fmt.Sprintf(
|
||||
"\n [%s]%s",
|
||||
widget.descriptionColor(event),
|
||||
event.Location,
|
||||
)
|
||||
}
|
||||
|
||||
func (widget *Widget) responseIcon(event *calendar.Event) string {
|
||||
if false == wtf.Config.UBool("wtf.mods.gcal.displayResponseStatus", true) {
|
||||
return ""
|
||||
}
|
||||
|
||||
for _, attendee := range event.Attendees {
|
||||
if attendee.Email == wtf.Config.UString("wtf.mods.gcal.email") {
|
||||
icon := "[gray]"
|
||||
|
||||
switch attendee.ResponseStatus {
|
||||
case "accepted":
|
||||
return icon + "✔︎"
|
||||
case "declined":
|
||||
return icon + "✘"
|
||||
case "needsAction":
|
||||
return icon + "?"
|
||||
case "tentative":
|
||||
return icon + "~"
|
||||
default:
|
||||
return icon + " "
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return " "
|
||||
}
|
||||
220
gcal/widget.go
220
gcal/widget.go
@ -1,9 +1,6 @@
|
||||
package gcal
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"regexp"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
@ -48,16 +45,6 @@ func (widget *Widget) Disable() {
|
||||
|
||||
/* -------------------- Unexported Functions -------------------- */
|
||||
|
||||
func (widget *Widget) display() {
|
||||
if widget.events == nil || len(widget.events.Items) == 0 {
|
||||
return
|
||||
}
|
||||
|
||||
widget.mutex.Lock()
|
||||
defer widget.mutex.Unlock()
|
||||
widget.View.SetText(widget.contentFrom(widget.events))
|
||||
}
|
||||
|
||||
// conflicts returns TRUE if this event conflicts with another, FALSE if it does not
|
||||
func (widget *Widget) conflicts(event *calendar.Event, events *calendar.Events) bool {
|
||||
conflict := false
|
||||
@ -82,99 +69,8 @@ func (widget *Widget) conflicts(event *calendar.Event, events *calendar.Events)
|
||||
return conflict
|
||||
}
|
||||
|
||||
func (widget *Widget) contentFrom(events *calendar.Events) string {
|
||||
if events == nil {
|
||||
return ""
|
||||
}
|
||||
|
||||
var prevEvent *calendar.Event
|
||||
|
||||
str := ""
|
||||
for _, event := range events.Items {
|
||||
conflict := widget.conflicts(event, events)
|
||||
|
||||
dayDivider := widget.dayDivider(event, prevEvent)
|
||||
responseIcon := widget.responseIcon(event)
|
||||
timestamp := fmt.Sprintf("[%s]%s",
|
||||
widget.descriptionColor(event),
|
||||
widget.eventTimestamp(event))
|
||||
title := fmt.Sprintf("[%s]%s",
|
||||
widget.titleColor(event),
|
||||
widget.eventSummary(event, conflict))
|
||||
until := widget.until(event)
|
||||
|
||||
lineOne := fmt.Sprintf(
|
||||
"%s %s %s %s %s[white]",
|
||||
dayDivider,
|
||||
responseIcon,
|
||||
timestamp,
|
||||
title,
|
||||
until,
|
||||
)
|
||||
str = str + fmt.Sprintf("%s%s\n\n",
|
||||
lineOne,
|
||||
widget.location(event), // prefixes newline if non-empty
|
||||
)
|
||||
|
||||
prevEvent = event
|
||||
}
|
||||
|
||||
return str
|
||||
}
|
||||
|
||||
func (widget *Widget) dayDivider(event, prevEvent *calendar.Event) string {
|
||||
var prevStartTime time.Time
|
||||
if prevEvent != nil {
|
||||
prevStartTime, _ = time.Parse(time.RFC3339, prevEvent.Start.DateTime)
|
||||
}
|
||||
currStartTime, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
||||
|
||||
if currStartTime.Day() != prevStartTime.Day() {
|
||||
_, _, width, _ := widget.View.GetInnerRect()
|
||||
return fmt.Sprintf("[%s]", wtf.Config.UString("wtf.mods.gcal.colors.day", "forestgreen")) +
|
||||
wtf.CenterText(currStartTime.Format(wtf.FullDateFormat), width) +
|
||||
"\n"
|
||||
}
|
||||
|
||||
return ""
|
||||
}
|
||||
|
||||
func (widget *Widget) descriptionColor(event *calendar.Event) string {
|
||||
color := wtf.Config.UString("wtf.mods.gcal.colors.description", "white")
|
||||
|
||||
if widget.eventIsPast(event) {
|
||||
color = wtf.Config.UString("wtf.mods.gcal.colors.past", "gray")
|
||||
}
|
||||
|
||||
return color
|
||||
}
|
||||
|
||||
func (widget *Widget) eventSummary(event *calendar.Event, conflict bool) string {
|
||||
summary := event.Summary
|
||||
|
||||
if widget.eventIsNow(event) {
|
||||
summary = fmt.Sprintf(
|
||||
"%s %s",
|
||||
wtf.Config.UString("wtf.mods.gcal.currentIcon", "🔸"),
|
||||
event.Summary,
|
||||
)
|
||||
}
|
||||
|
||||
if conflict {
|
||||
return fmt.Sprintf("%s %s", wtf.Config.UString("wtf.mods.gcal.conflictIcon", "🚨"), summary)
|
||||
} else {
|
||||
return summary
|
||||
}
|
||||
}
|
||||
|
||||
func (widget *Widget) eventTimestamp(event *calendar.Event) string {
|
||||
if len(event.Start.Date) > 0 {
|
||||
startTime, _ := time.Parse("2006-01-02", event.Start.Date)
|
||||
return startTime.Format(wtf.FriendlyDateFormat)
|
||||
} else {
|
||||
startTime, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
||||
return startTime.Format(wtf.MinimumTimeFormat)
|
||||
}
|
||||
func (widget *Widget) eventIsAllDay(event *calendar.Event) bool {
|
||||
return len(event.Start.Date) > 0
|
||||
}
|
||||
|
||||
// eventIsNow returns true if the event is happening now, false if it not
|
||||
@ -186,116 +82,12 @@ func (widget *Widget) eventIsNow(event *calendar.Event) bool {
|
||||
}
|
||||
|
||||
func (widget *Widget) eventIsPast(event *calendar.Event) bool {
|
||||
ts, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
||||
return (widget.eventIsNow(event) == false) && ts.Before(time.Now())
|
||||
}
|
||||
|
||||
func (widget *Widget) titleColor(event *calendar.Event) string {
|
||||
color := wtf.Config.UString("wtf.mods.gcal.colors.title", "white")
|
||||
|
||||
for _, untypedArr := range wtf.Config.UList("wtf.mods.gcal.colors.highlights") {
|
||||
highlightElements := wtf.ToStrs(untypedArr.([]interface{}))
|
||||
|
||||
match, _ := regexp.MatchString(
|
||||
strings.ToLower(highlightElements[0]),
|
||||
strings.ToLower(event.Summary),
|
||||
)
|
||||
|
||||
if match == true {
|
||||
color = highlightElements[1]
|
||||
}
|
||||
}
|
||||
|
||||
if widget.eventIsPast(event) {
|
||||
color = wtf.Config.UString("wtf.mods.gcal.colors.past", "gray")
|
||||
}
|
||||
|
||||
return color
|
||||
}
|
||||
|
||||
func (widget *Widget) location(event *calendar.Event) string {
|
||||
if wtf.Config.UBool("wtf.mods.gcal.displayLocation", true) == false {
|
||||
return ""
|
||||
}
|
||||
|
||||
if event.Location == "" {
|
||||
return ""
|
||||
}
|
||||
|
||||
return fmt.Sprintf(
|
||||
"\n [%s]%s",
|
||||
widget.descriptionColor(event),
|
||||
event.Location,
|
||||
)
|
||||
}
|
||||
|
||||
func (widget *Widget) responseIcon(event *calendar.Event) string {
|
||||
if false == wtf.Config.UBool("wtf.mods.gcal.displayResponseStatus", true) {
|
||||
return ""
|
||||
}
|
||||
|
||||
response := ""
|
||||
|
||||
for _, attendee := range event.Attendees {
|
||||
if attendee.Email == wtf.Config.UString("wtf.mods.gcal.email") {
|
||||
response = attendee.ResponseStatus
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
icon := "[gray]"
|
||||
|
||||
switch response {
|
||||
case "accepted":
|
||||
icon = icon + "✔︎"
|
||||
case "declined":
|
||||
icon = icon + "✘"
|
||||
case "needsAction":
|
||||
icon = icon + "?"
|
||||
case "tentative":
|
||||
icon = icon + "~"
|
||||
default:
|
||||
icon = icon + " "
|
||||
}
|
||||
|
||||
return icon
|
||||
}
|
||||
|
||||
// until returns the number of hours or days until the event
|
||||
// If the event is in the past, returns nil
|
||||
func (widget *Widget) until(event *calendar.Event) string {
|
||||
startTime, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
||||
duration := time.Until(startTime)
|
||||
|
||||
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 := ""
|
||||
|
||||
var color = "[lightblue]"
|
||||
if days > 0 {
|
||||
untilStr = fmt.Sprintf("%dd", days)
|
||||
} else if hours > 0 {
|
||||
untilStr = fmt.Sprintf("%dh", hours)
|
||||
if widget.eventIsAllDay(event) {
|
||||
return false
|
||||
} else {
|
||||
untilStr = fmt.Sprintf("%dm", mins)
|
||||
if mins < 30 {
|
||||
color = "[red]"
|
||||
}
|
||||
ts, _ := time.Parse(time.RFC3339, event.Start.DateTime)
|
||||
return (widget.eventIsNow(event) == false) && ts.Before(time.Now())
|
||||
}
|
||||
|
||||
return color + untilStr + "[white]"
|
||||
}
|
||||
|
||||
func updateLoop(widget *Widget) {
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user