Genesis

internal/websocket/connect.go

@@ -0,0 +1,38 @@
+package websocket
+
+import (
+	"github.com/charmbracelet/log"
+	"github.com/gorilla/websocket"
+	"os"
+	"time"
+)
+
+func Connect(url string) *websocket.Conn {
+	log.Info("Connected 󰖟 ", "host", url)
+
+	conn, _, err := websocket.DefaultDialer.Dial(url, nil)
+	if err != nil {
+		log.Fatal("Dial error!", "reason", err)
+	}
+
+	return conn
+}
+
+func WaitForDisconnect(conn *websocket.Conn, interrupt chan os.Signal) {
+	for {
+		select {
+		case <-Done:
+			// session closed by remote
+			return
+		case <-interrupt:
+			// Cleanly close the connection by sending a close message and then
+			// waiting (with timeout) for the server to close the connection.
+			conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
+			select {
+			case <-Done:
+			case <-time.After(time.Second):
+			}
+			return
+		}
+	}
+}

internal/websocket/receive.go

@@ -0,0 +1,37 @@
+package websocket
+
+import (
+	"ChronoTomato/internal/notifications"
+	"encoding/json"
+	"fmt"
+	"git.smsvc.net/pomodoro/GoTomato/pkg/models"
+	"github.com/charmbracelet/log"
+	"github.com/gorilla/websocket"
+)
+
+var Done = make(chan struct{})
+
+func ProcessServerMessages(conn *websocket.Conn) {
+	var serverMessage models.ServerMessage
+
+	defer close(Done)
+
+	for {
+		_, message, err := conn.ReadMessage()
+		if err != nil {
+			fmt.Println()
+			log.Error("Read error!", "reason", err)
+			return
+		}
+
+		err = json.Unmarshal(message, &serverMessage)
+		if err != nil {
+			log.Error("Error unmarshalling!", "reason", err)
+			continue
+		}
+
+		notifications.DesktopNotifications(serverMessage)
+		notifications.TerminalOutput(serverMessage)
+	}
+
+}

internal/websocket/send.go

@@ -0,0 +1,43 @@
+package websocket
+
+import (
+	"ChronoTomato/config"
+	"encoding/json"
+	"git.smsvc.net/pomodoro/GoTomato/pkg/models"
+	"github.com/charmbracelet/log"
+	"github.com/gorilla/websocket"
+)
+
+func sendClientCommand(conn *websocket.Conn, msg models.ClientCommand) {
+	messageBytes, err := json.Marshal(msg)
+
+	if err != nil {
+		log.Error("Error marshalling!", "reason", err)
+		return
+	}
+
+	err = conn.WriteMessage(websocket.TextMessage, messageBytes)
+	if err != nil {
+		log.Error("Write error!", "reason", err)
+		return
+	}
+}
+
+func SendCmd(conn *websocket.Conn, cmd string, pwd string) {
+	message := models.ClientCommand{
+		Command:  cmd,
+		Password: pwd,
+	}
+
+	sendClientCommand(conn, message)
+}
+
+func Send_updateSettings(conn *websocket.Conn, pwd string) {
+	message := models.ClientCommand{
+		Command:          "updateSettings",
+		Password:         pwd,
+		PomodoroSettings: config.PomodoroSettings,
+	}
+
+	sendClientCommand(conn, message)
+}