Migrate all modules to their own subfolder

Handles #375

modules/opsgenie/client.go

@@ -0,0 +1,76 @@
+package opsgenie
+
+import (
+	"encoding/json"
+	"fmt"
+	"net/http"
+	"os"
+
+	"github.com/wtfutil/wtf/wtf"
+)
+
+type OnCallResponse struct {
+	OnCallData OnCallData `json:"data"`
+	Message    string     `json:"message"`
+	RequestID  string     `json:"requestId"`
+	Took       float32    `json:"took"`
+}
+
+type OnCallData struct {
+	Recipients []string `json:"onCallRecipients"`
+	Parent     Parent   `json:"_parent"`
+}
+
+type Parent struct {
+	ID      string `json:"id"`
+	Name    string `json:"name"`
+	Enabled bool   `json:"enabled"`
+}
+
+/* -------------------- Exported Functions -------------------- */
+
+func Fetch(scheduleIdentifierType string, schedules []string) ([]*OnCallResponse, error) {
+	agregatedResponses := []*OnCallResponse{}
+	for _, sched := range schedules {
+		scheduleUrl := fmt.Sprintf("https://api.opsgenie.com/v2/schedules/%s/on-calls?scheduleIdentifierType=%s&flat=true", sched, scheduleIdentifierType)
+		response, err := opsGenieRequest(scheduleUrl, apiKey())
+		agregatedResponses = append(agregatedResponses, response)
+		if err != nil {
+			return nil, err
+		}
+	}
+	return agregatedResponses, nil
+}
+
+/* -------------------- Unexported Functions -------------------- */
+
+func apiKey() string {
+	return wtf.Config.UString(
+		"wtf.mods.opsgenie.apiKey",
+		os.Getenv("WTF_OPS_GENIE_API_KEY"),
+	)
+}
+
+func opsGenieRequest(url string, apiKey string) (*OnCallResponse, error) {
+	req, err := http.NewRequest("GET", url, nil)
+	if err != nil {
+		return nil, err
+	}
+
+	req.Header.Set("Authorization", fmt.Sprintf("GenieKey %s", apiKey))
+
+	client := &http.Client{}
+
+	resp, err := client.Do(req)
+	if err != nil {
+		return nil, err
+	}
+	defer resp.Body.Close()
+
+	response := &OnCallResponse{}
+	if err := json.NewDecoder(resp.Body).Decode(response); err != nil {
+		return nil, err
+	}
+
+	return response, nil
+}

modules/opsgenie/widget.go

@@ -0,0 +1,91 @@
+package opsgenie
+
+import (
+	"fmt"
+	"strings"
+
+	"github.com/rivo/tview"
+	"github.com/wtfutil/wtf/wtf"
+)
+
+type Widget struct {
+	wtf.TextWidget
+}
+
+func NewWidget(app *tview.Application) *Widget {
+	widget := Widget{
+		TextWidget: wtf.NewTextWidget(app, "OpsGenie", "opsgenie", false),
+	}
+
+	return &widget
+}
+
+/* -------------------- Exported Functions -------------------- */
+
+func (widget *Widget) Refresh() {
+	data, err := Fetch(
+		wtf.Config.UString("wtf.mods.opsgenie.scheduleIdentifierType"),
+		getSchedules(),
+	)
+	widget.View.SetTitle(widget.ContextualTitle(widget.Name))
+
+	var content string
+	if err != nil {
+		widget.View.SetWrap(true)
+		content = err.Error()
+	} else {
+		widget.View.SetWrap(false)
+		content = widget.contentFrom(data)
+	}
+
+	widget.View.SetText(content)
+}
+
+/* -------------------- Unexported Functions -------------------- */
+
+func getSchedules() []string {
+	// see if schedule is set to a single string
+	configPath := "wtf.mods.opsgenie.schedule"
+	singleSchedule, err := wtf.Config.String(configPath)
+	if err == nil {
+		return []string{singleSchedule}
+	}
+	// else, assume list
+	scheduleList := wtf.Config.UList(configPath)
+	var ret []string
+	for _, schedule := range scheduleList {
+		if str, ok := schedule.(string); ok {
+			ret = append(ret, str)
+		}
+	}
+	return ret
+}
+
+func (widget *Widget) contentFrom(onCallResponses []*OnCallResponse) string {
+	str := ""
+
+	displayEmpty := wtf.Config.UBool("wtf.mods.opsgenie.displayEmpty", true)
+
+	for _, data := range onCallResponses {
+		if (len(data.OnCallData.Recipients) == 0) && (displayEmpty == false) {
+			continue
+		}
+
+		var msg string
+		if len(data.OnCallData.Recipients) == 0 {
+			msg = " [gray]no one[white]\n\n"
+		} else {
+			msg = fmt.Sprintf(" %s\n\n", strings.Join(wtf.NamesFromEmails(data.OnCallData.Recipients), ", "))
+		}
+
+		str = str + widget.cleanScheduleName(data.OnCallData.Parent.Name)
+		str = str + msg
+	}
+
+	return str
+}
+
+func (widget *Widget) cleanScheduleName(schedule string) string {
+	cleanedName := strings.Replace(schedule, "_", " ", -1)
+	return fmt.Sprintf(" [green]%s[white]\n", cleanedName)
+}