GoTomato/internal/websocket/client_commands.go

package websocket

import (
	"encoding/json"
	"github.com/charmbracelet/log"

	"git.smsvc.net/pomodoro/GoTomato/internal/pomodoro"
	"git.smsvc.net/pomodoro/GoTomato/internal/shared"
	"git.smsvc.net/pomodoro/GoTomato/pkg/models"
)

func checkSettings(settings models.PomodoroConfig) bool {
	return settings.Work > 0 && settings.ShortBreak > 0 && settings.LongBreak > 0 && settings.Sessions > 0
}

// Listens for commands from a client and handles them
func handleClientCommands(c models.WebsocketClient) {
	ws := c.Conn
	for {
		var clientCommand models.ClientCommand

		_, message, err := ws.ReadMessage()
		if err != nil {
			log.Info("Client disconnected:", "msg", err, "host", ws.RemoteAddr(), "clients", len(Clients)-1)
			delete(Clients, ws.LocalAddr())
			break
		}

		// Handle incoming commands
		err = json.Unmarshal(message, &clientCommand)
		if err != nil {
			log.Error("Error unmarshalling command:", "msg", err)
			continue
		}

		// Process the command if pomodoro password matches
		if clientCommand.Password == shared.PomodoroPassword {
			switch clientCommand.Command {
			case "start":
				if !pomodoro.IsPomodoroOngoing() {
					go pomodoro.RunPomodoro() // Start the timer with the list of clients
				}
			case "stop":
				if pomodoro.IsPomodoroOngoing() {
					pomodoro.ResetPomodoro() // Reset Pomodoro
				}
			case "pause":
				if pomodoro.IsPomodoroOngoing() && !pomodoro.IsPomodoroPaused() {
					pomodoro.PausePomodoro() // Pause the timer
				}
			case "resume":
				if pomodoro.IsPomodoroOngoing() && pomodoro.IsPomodoroPaused() {
					pomodoro.ResumePomodoro() // Resume the timer
				}
			case "updateSettings":
				if !pomodoro.IsPomodoroOngoing() {
					if !checkSettings(clientCommand.Settings) {
						log.Warn("Ignoring invalid config:", "msg", clientCommand.Settings, "host", c.Conn.RemoteAddr())
						break
					}
					log.Info("Client send config", "config", clientCommand.Settings, "host", c.Conn.RemoteAddr())
					pomodoro.UpdateSettings(clientCommand.Settings)
				}
			}
		}

	}
}