Inital commit: skeleton of runner logic

2022-10-16 14:57:18 -06:00
commit c21e7d5d6b
7 changed files with 1065 additions and 0 deletions
@@ -0,0 +1,85 @@
+package main
+
+import (
+	"context"
+	"fmt"
+	"git.ohea.xyz/cursorius/runner/config"
+	"github.com/op/go-logging"
+	"nhooyr.io/websocket"
+	"nhooyr.io/websocket/wsjson"
+	"os"
+	"time"
+)
+
+var log = logging.MustGetLogger("cursorius-server")
+
+type Register struct {
+	Secret string
+	Id     string
+	Tags   []string
+}
+
+type Job struct {
+	URL    *string
+	Folder *string
+}
+
+func main() {
+
+	var format = logging.MustStringFormatter(
+		`%{color}%{time:15:04:05.000} %{level:.4s}:%{color:reset} %{message}`,
+	)
+
+	backend := logging.NewLogBackend(os.Stderr, "", 0)
+	backendFormatter := logging.NewBackendFormatter(backend, format)
+	backendLeveled := logging.AddModuleLevel(backendFormatter)
+	backendLeveled.SetLevel(logging.DEBUG, "")
+
+	logging.SetBackend(backendLeveled)
+
+	log.Info("Starting cursorius-runner")
+
+	configData, isNew, err := config.GetConfig()
+	if err != nil {
+		log.Fatalf("Could not get configuration: %v", err)
+		os.Exit(1)
+	}
+	if isNew {
+		log.Infof("New config file created at %s, please update and relaunch", *configData.ConfigFile)
+		os.Exit(0)
+	}
+
+	registrationData := Register{
+		Secret: configData.Config.Secret,
+		Id:     configData.Config.Id,
+		Tags:   configData.Config.Tags,
+	}
+
+	connectString := fmt.Sprintf("ws://%v/runner", configData.Config.ServerUrl)
+
+	ctx, cancel := context.WithTimeout(context.Background(), time.Minute)
+	defer cancel()
+
+	c, _, err := websocket.Dial(ctx, connectString, nil)
+	if err != nil {
+		log.Fatalf("Could not connect to server at \"%v\", %v", connectString, err)
+		os.Exit(1)
+	}
+	defer c.Close(websocket.StatusInternalError, "the sky is falling")
+
+	err = wsjson.Write(ctx, c, registrationData)
+	if err != nil {
+		log.Fatalf("Could not send registration information to server: %v", err)
+		os.Exit(1)
+	}
+
+	for {
+		var job Job
+		err = wsjson.Read(ctx, c, &job)
+		if err != nil {
+			log.Fatalf("Could not receive from connection: %v", err)
+			continue
+		}
+		log.Infof("Received job from server: %v", *job.URL)
+	}
+}