From d5002c4412f6fffc3093f60ee6ae014c0dd9a0cd Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Se=C3=A1n=20C=20McCord?= <ulexus@gmail.com>
Date: Mon, 13 Jun 2022 20:44:02 -0400
Subject: [PATCH] add autotrigger command
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: Seán C McCord <ulexus@gmail.com>
---
 cmd/autotrigger/.gitignore |   1 +
 cmd/autotrigger/Dockerfile |   3 ++
 cmd/autotrigger/main.go    | 106 +++++++++++++++++++++++++++++++++++++
 3 files changed, 110 insertions(+)
 create mode 100644 cmd/autotrigger/.gitignore
 create mode 100644 cmd/autotrigger/Dockerfile
 create mode 100644 cmd/autotrigger/main.go

diff --git a/cmd/autotrigger/.gitignore b/cmd/autotrigger/.gitignore
new file mode 100644
index 0000000..b973ef8
--- /dev/null
+++ b/cmd/autotrigger/.gitignore
@@ -0,0 +1 @@
+/autotrigger
diff --git a/cmd/autotrigger/Dockerfile b/cmd/autotrigger/Dockerfile
new file mode 100644
index 0000000..09cb13c
--- /dev/null
+++ b/cmd/autotrigger/Dockerfile
@@ -0,0 +1,3 @@
+FROM gcr.io/distroless/static
+COPY autotrigger /go/bin/app
+ENTRYPOINT ["/go/bin/app"]
diff --git a/cmd/autotrigger/main.go b/cmd/autotrigger/main.go
new file mode 100644
index 0000000..a3deb11
--- /dev/null
+++ b/cmd/autotrigger/main.go
@@ -0,0 +1,106 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"flag"
+	"fmt"
+	"log"
+	"net/http"
+	"os"
+	"os/signal"
+	"time"
+
+	"github.com/CyCoreSystems/audimance/agenda"
+)
+
+var cueName string
+var interval time.Duration
+var baseURL string
+
+func init() {
+	flag.DurationVar(&interval, "r", 3*time.Minute, "repeat interval")
+	flag.StringVar(&cueName, "c", "", "name of cue to be triggered")
+	flag.StringVar(&baseURL, "u", "http://localhost:3000", "base URL")
+}
+
+func main() {
+	flag.Parse()
+
+	ctx, cancel := signal.NotifyContext(context.Background(), os.Interrupt, os.Kill)
+	defer cancel()
+
+	if cueName == "" {
+		log.Fatalln("cue name must be set")
+	}
+
+	cueID, err := findCue(cueName)
+	if err != nil {
+		log.Fatalln("failed to find cue: %s", err.Error())
+	}
+
+	if err := trigger(cueID); err != nil {
+		log.Printf("failed to trigger cue: %s", err.Error())
+	} else {
+		log.Printf("triggered cue %s (%s)", cueName, cueID)
+	}
+
+	for {
+		select {
+		case <-ctx.Done():
+			os.Exit(0)
+		case <-time.After(interval):
+			if err := trigger(cueID); err != nil {
+				log.Printf("failed to trigger cue: %s", err.Error())
+			} else {
+				log.Printf("triggered cue %s (%s)", cueName, cueID)
+			}
+		}
+	}
+}
+
+func findCue(cueName string) (string, error) {
+	resp, err := http.Get(fmt.Sprintf("%s/agenda.json", baseURL))
+	if err != nil {
+		return "", fmt.Errorf("failed to retrieve agenda: %w", err)
+	}
+
+	defer resp.Body.Close()
+
+	var a agenda.Agenda
+
+	if err := json.NewDecoder(resp.Body).Decode(&a); err != nil {
+		return "", fmt.Errorf("failed to decode agenda: %w", err)
+	}
+
+	for _, c := range a.Cues {
+		if c.Name == cueName {
+			return c.ID, nil
+		}
+
+		// Handle cases where the user passes the ID instaed of the name
+		if c.ID == cueName {
+			return c.ID, nil
+		}
+	}
+
+	return "", fmt.Errorf("cue %q not found in agenda", cueName)
+}
+
+func trigger(cueID string) error {
+	req, err := http.NewRequest(http.MethodPut, fmt.Sprintf("%s/cues/%s", baseURL, cueID), nil)
+	if err != nil {
+		return fmt.Errorf("failed to construct PUT request: %w", err)
+	}
+
+	resp, err := http.DefaultClient.Do(req)
+	if err != nil {
+		return fmt.Errorf("request failed: %w", err)
+	}
+
+	if resp.StatusCode != http.StatusOK {
+		return fmt.Errorf("non-200 OK response: %s", resp.Status)
+	}
+
+	return nil
+}