nerve-centre.go

package main

import (
	"encoding/json"
	"errors"
	"flag"
	"strconv"
	"strings"
	"syscall"
	"time"
)

func main() {

	username := flag.String("username", "", "Nerve Centre username")
	password := flag.String("password", "", "Nerve Centre password")
	namespace := flag.String("namespace", "", "Nerve Centre namespace")
	webhookUrl := flag.String("webhook", "", "Slack webhook url")
	channel := flag.String("channel", "", "Slack channel override")
	flag.Parse()

	if *username == "" || *password == "" || *namespace == "" || *webhookUrl == "" {
		flag.Usage()
		syscall.Exit(1)
	}

	// Apply namespace
	nerveCentreBaseUrl = nerveCentreBaseUrl + *namespace
	usernameWithNamespace := *username + "@" + *namespace

	err := Login(usernameWithNamespace, *password)

	if err != nil {
		sendFailureToSlack(webhookUrl, Schedule{}, channel, err)
	}

	users := GetMembers()
	schedules := *GetSchedules()

	if len(*users) == 0 || len(schedules) == 0 {
		sendFailureToSlack(webhookUrl, Schedule{}, channel, errors.New("could not load users or schedules, check username and password"))
	}

	schedule := schedules[0]

	runTime := time.Now()
	planningTime := time.Now()
	planningEnd := time.Now()
	planning, err := GetPlanning(schedule, runTime)

	if err != nil {
		sendFailureToSlack(webhookUrl, schedule, channel, err)
	}

	today := planning.GetActiveSlot(runTime)
	var next *Slot

	var currentPlanningEnd time.Time
	var currentMembers []string

	if planning.HasMembers() {
		slot := planning.GetActiveSlot(runTime)
		currentPlanningEnd = slot.End
		currentMembers = slot.GetMembers(users)
	}

	foundOther := false

	for planning.HasMembers() {

		for _, slot := range planning.BaseTimeSlots {

			// Filter current active slot and older slots
			if runTime.After(slot.Start) || runTime.Equal(slot.Start) {
				continue
			}

			planningEnd = slot.End

			if !foundOther {
				members := slot.GetMembers(users)

				if !Equal(currentMembers, members) {
					foundOther = true
					next = &slot
				} else {
					currentPlanningEnd = slot.End
				}
			}
		}

		planningTime = planningTime.Add(24 * time.Hour)
		planning, err = GetPlanning(schedule, planningTime)
		if err != nil {
			sendFailureToSlack(webhookUrl, schedule, channel, err)
		}
	}

	todayMembersString := "<<geen>>"
	todayColor := "#ec0045"
	if len(currentMembers) > 0 {
		todayMembersString = strings.Join(currentMembers, ", ") + " tot " + currentPlanningEnd.Format("02-01-2006 15:04")
		todayColor = "#007a5a"
	}

	attachments := make([]Attachment, 0, 3)

	attachments = append(attachments, Attachment{
		Fallback: "Vandaag: " + todayMembersString,
		Color:    todayColor,
		Title:    "Vandaag",
		Text:     todayMembersString,
	})

	if foundOther && next != nil {
		nextMembers := next.GetMembers(users)
		nextMembersString := "<<geen>>"
		nextColor := "#ec0045"

		if len(currentMembers) > 0 {
			nextMembersString = strings.Join(nextMembers, ", ") + " op " + next.Start.Format("02-01-2006 om 15:04")
			nextColor = "#ffc917"
		}

		attachments = append(attachments, Attachment{
			Fallback: "Volgende: " + nextMembersString,
			Color:    nextColor,
			Title:    "Volgende",
			Text:     nextMembersString,
			Ts:       json.Number(strconv.FormatInt(next.Start.Unix(), 10)),
		})
	}

	if len(today.GetMembers(users)) > 0 {
		attachments = append(attachments, Attachment{
			Fallback: "Er is een rooster tot " + planningEnd.Format("02-01-2006 15:04"),
			Color:    "#ec0045",
			Title:    "Einde rooster",
			Text:     "Er is een rooster tot " + planningEnd.Format("02-01-2006 15:04"),
			Ts:       json.Number(strconv.FormatInt(planningEnd.Unix(), 10)),
		})
	}

	message := SlackPayload{
		Username:    "📞 Wachtdienst " + schedule.GroupName,
		Channel:     *channel,
		Text:        "Een overzicht van de de huidige wachtdiensten die zijn ingeregeld voor " + schedule.GroupName + " in Nerve Centre",
		Attachments: attachments,
	}

	err = SendSlack(*webhookUrl, &message)
	if err != nil {
		panic(err)
	}
}

func sendFailureToSlack(webhookUrl *string, schedule Schedule, channel *string, err error) {
	SendSlack(*webhookUrl, &SlackPayload{
		Username: "⚠️ Wachtdienst " + schedule.GroupName,
		Channel:  *channel,
		Text:     "Kon wachtdiensten niet ophalen uit Nerve Centre: " + err.Error(),
	})
	panic(err)
}