Refactor GCal calendar code to simplify things

gcal/display.go

@@ -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 " "
+}