Sebastian Mark
44a64bfce4
- add Gorilla handlers package for enhanced HTTP handling - refactor HTTP server to use a new ServeMux for routing - update ListenAndServe to utilize ProxyHeaders for better proxy support - add RealIP field to client model - use RealIP fild for connect/disconnect log output
68 lines
2 KiB
Go
68 lines
2 KiB
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 {
|
|
delete(Clients, ws.LocalAddr())
|
|
log.Info("Client disconnected:", "msg", err, "host", c.RealIP, "clients", len(Clients))
|
|
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)
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|
|
}
|