diff --git a/internal/broadcast/broadcast.go b/internal/broadcast/broadcast.go index 229ae67..1be04ce 100644 --- a/internal/broadcast/broadcast.go +++ b/internal/broadcast/broadcast.go @@ -8,7 +8,7 @@ import ( ) // BroadcastMessage sends a message to all connected WebSocket clients. -func BroadcastMessage(clients map[*websocket.Conn]*models.Client, message models.BroadcastMessage) { +func BroadcastMessage(clients map[*websocket.Conn]*models.Client, message models.ServerMessage) { // Marshal the message into JSON format jsonMessage, err := json.Marshal(message) if err != nil { diff --git a/internal/pomodoro/pomodoro.go b/internal/pomodoro/pomodoro.go index c030acf..d5e88de 100644 --- a/internal/pomodoro/pomodoro.go +++ b/internal/pomodoro/pomodoro.go @@ -58,7 +58,7 @@ func ResetPomodoro(clients map[*websocket.Conn]*models.Client) { mu.Unlock() // Broadcast the reset message to all clients - broadcast.BroadcastMessage(clients, models.BroadcastMessage{ + broadcast.BroadcastMessage(clients, models.ServerMessage{ Mode: "none", Session: 0, MaxSession: 0, diff --git a/internal/pomodoro/timer.go b/internal/pomodoro/timer.go index e4d95dd..bca8116 100644 --- a/internal/pomodoro/timer.go +++ b/internal/pomodoro/timer.go @@ -20,7 +20,7 @@ func startTimer(clients map[*websocket.Conn]*models.Client, remainingSeconds int default: // Broadcast the current state to all clients if !IsPomodoroPaused() { - broadcast.BroadcastMessage(clients, models.BroadcastMessage{ + broadcast.BroadcastMessage(clients, models.ServerMessage{ Mode: mode, Session: session, MaxSession: PomodoroConfig.Sessions, @@ -33,7 +33,7 @@ func startTimer(clients map[*websocket.Conn]*models.Client, remainingSeconds int } // Final broadcast when time reaches zero - broadcast.BroadcastMessage(clients, models.BroadcastMessage{ + broadcast.BroadcastMessage(clients, models.ServerMessage{ Mode: mode, Session: session, MaxSession: PomodoroConfig.Sessions, diff --git a/pkg/models/broadcast.go b/pkg/models/server.go similarity index 74% rename from pkg/models/broadcast.go rename to pkg/models/server.go index 78463b9..8c68258 100644 --- a/pkg/models/broadcast.go +++ b/pkg/models/server.go @@ -1,7 +1,7 @@ package models -// BroadcastMessage represents the data sent to the client via WebSocket. -type BroadcastMessage struct { +// ServerMessage represents the data sent to the client via WebSocket. +type ServerMessage struct { Mode string `json:"mode"` // "Work", "ShortBreak", or "LongBreak" Session int `json:"session"` // Current session number MaxSession int `json:"max_session"` // Total number of sessions