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.ServerMessage) {
+func BroadcastMessage(clients map[*websocket.Conn]*models.Client, message models.BroadcastMessage) {
 	// Marshal the message into JSON format
 	jsonMessage, err := json.Marshal(message)
 	if err != nil {

internal/pomodoro/pomodoro.go

@@ -23,8 +23,8 @@ var pomodoroResumeChannel = make(chan bool, 1)
 
 var mu sync.Mutex // to synchronize access to shared state
 
-// RunPomodoro iterates the Pomodoro work/break sessions.
-func RunPomodoro(clients map[*websocket.Conn]*models.Client) {
+// RunPomodoroTimer iterates the Pomodoro work/break sessions.
+func RunPomodoroTimer(clients map[*websocket.Conn]*models.Client) {
 	mu.Lock()
 	pomodoroOngoing = true
 	pomodoroPaused = false
@@ -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.ServerMessage{
+	broadcast.BroadcastMessage(clients, models.BroadcastMessage{
 		Mode:       "none",
 		Session:    0,
 		MaxSession: 0,

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.ServerMessage{
+				broadcast.BroadcastMessage(clients, models.BroadcastMessage{
 					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.ServerMessage{
+	broadcast.BroadcastMessage(clients, models.BroadcastMessage{
 		Mode:       mode,
 		Session:    session,
 		MaxSession: PomodoroConfig.Sessions,

internal/websocket/client_commands.go

@@ -30,7 +30,7 @@ func handleClientCommands(ws *websocket.Conn) {
 		switch command.Command {
 		case "start":
 			if !pomodoro.IsPomodoroOngoing() {
-				go pomodoro.RunPomodoro(Clients) // Start the timer with the list of clients
+				go pomodoro.RunPomodoroTimer(Clients) // Start the timer with the list of clients
 			}
 		case "stop":
 			if pomodoro.IsPomodoroOngoing() {

pkg/models/broadcast.go

@@ -1,7 +1,7 @@
 package models
 
-// ServerMessage represents the data sent to the client via WebSocket.
-type ServerMessage struct {
+// BroadcastMessage represents the data sent to the client via WebSocket.
+type BroadcastMessage 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