Compare commits

..

No commits in common. "90ce4a514b1cf27744af9b9ed2847bab487bf301" and "64050cd1d2d30e3c16e73c7c7dd1a1c7df64f9aa" have entirely different histories.

2 changed files with 138 additions and 48 deletions

View file

@ -4,6 +4,7 @@ import (
"encoding/json" "encoding/json"
"log" "log"
"net/http" "net/http"
"sync"
"time" "time"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
@ -23,7 +24,14 @@ type BroadcastMessage struct {
TimeLeft int `json:"time_left"` TimeLeft int `json:"time_left"`
} }
type ClientCommand struct {
Command string `json:"command"`
}
var clients = make(map[*websocket.Conn]bool) var clients = make(map[*websocket.Conn]bool)
var timerRunning bool
var timerStopChannel = make(chan bool, 1)
var mu sync.Mutex // to synchronize access to shared state
// broadcastMessage sends the remaining time to all connected clients. // broadcastMessage sends the remaining time to all connected clients.
func broadcastMessage(message BroadcastMessage) { func broadcastMessage(message BroadcastMessage) {
@ -38,37 +46,54 @@ func broadcastMessage(message BroadcastMessage) {
} }
} }
// start a countdown and broadcast ever second // startTimer runs the countdown and broadcasts every second.
func startTimer(remaining_seconds int, mode string, session int) { func startTimer(remainingSeconds int, mode string, session int) bool {
for remaining_seconds > 0 { for remainingSeconds > 0 {
broadcastMessage(BroadcastMessage{ select {
Mode: mode, case <-timerStopChannel:
Session: session, return false // Stop the timer if a stop command is received
MaxSession: sessions, default:
TimeLeft: remaining_seconds, broadcastMessage(BroadcastMessage{
}) Mode: mode,
time.Sleep(time.Second) Session: session,
remaining_seconds-- MaxSession: sessions,
} TimeLeft: remainingSeconds,
// send final 0 timer })
broadcastMessage(BroadcastMessage{ time.Sleep(time.Second)
Mode: mode, remainingSeconds--
Session: 1,
MaxSession: 4,
TimeLeft: 0,
})
}
// iterate the Pomodoro work/break sessions
func runPomodoroTimer() {
for session := 1; session <= sessions; session++ {
startTimer(workDuration, "Work", session)
if session == sessions {
startTimer(longBreakDuration, "LongBreak", session)
} else {
startTimer(shortBreakDuration, "ShortBreak", session)
} }
} }
broadcastMessage(BroadcastMessage{
Mode: mode,
Session: session,
MaxSession: sessions,
TimeLeft: 0,
})
return true
}
// runPomodoroTimer iterates the Pomodoro work/break sessions.
func runPomodoroTimer() {
mu.Lock()
timerRunning = true
for session := 1; session <= sessions; session++ {
if !startTimer(workDuration, "Work", session) {
break
}
if session == sessions {
if !startTimer(longBreakDuration, "LongBreak", session) {
break
}
} else {
if !startTimer(shortBreakDuration, "ShortBreak", session) {
break
}
}
}
timerRunning = false
mu.Unlock()
} }
// upgrade HTTP requests to WebSocket connections. // upgrade HTTP requests to WebSocket connections.
@ -88,23 +113,40 @@ func handleConnections(w http.ResponseWriter, r *http.Request) {
// Register the new client // Register the new client
clients[ws] = true clients[ws] = true
// Clean up when the client disconnects // Listen for commands from this client
for { for {
_, _, err := ws.ReadMessage() _, message, err := ws.ReadMessage()
if err != nil { if err != nil {
log.Printf("Client disconnected: %v", err) log.Printf("Client disconnected: %v", err)
delete(clients, ws) delete(clients, ws)
break break
} }
// Handle incoming commands
var command ClientCommand
err = json.Unmarshal(message, &command)
if err != nil {
log.Printf("Error unmarshalling command: %v", err)
continue
}
// Process the commands
switch command.Command {
case "start":
if !timerRunning {
go runPomodoroTimer()
}
case "stop":
if timerRunning {
timerStopChannel <- true
}
}
} }
} }
func main() { func main() {
http.HandleFunc("/ws", handleConnections) http.HandleFunc("/ws", handleConnections)
// Start a goroutine that runs the Pomodoro timer and broadcasts updates.
go runPomodoroTimer()
log.Println("Pomodoro WebSocket server started on :8080") log.Println("Pomodoro WebSocket server started on :8080")
err := http.ListenAndServe(":8080", nil) err := http.ListenAndServe(":8080", nil)
if err != nil { if err != nil {

View file

@ -1,24 +1,72 @@
<!DOCTYPE html> <!DOCTYPE html>
<html lang="en"> <html lang="en">
<head> <head>
<meta charset="UTF-8"> <meta charset="UTF-8">
<meta name="viewport" content="width=device-width, initial-scale=1.0"> <meta name="viewport" content="width=device-width, initial-scale=1.0">
<title>Pomodoro Timer</title> <title>Pomodoro Timer</title>
<style>
button {
padding: 10px 20px;
margin: 5px;
font-size: 16px;
}
#timer {
font-size: 24px;
margin-top: 20px;
}
</style>
</head> </head>
<body> <body>
<h1>Pomodoro Timer</h1> <h1>Pomodoro Timer</h1>
<div id="timer">Connecting to server...</div> <div id="timer">Connecting to server...</div>
<script> <!-- Buttons to start and stop the timer -->
var ws = new WebSocket("ws://localhost:8080/ws"); <button id="startButton">Start</button>
<button id="stopButton">Stop</button>
ws.onmessage = function(event) { <script>
document.getElementById("timer").innerText = event.data; var ws = new WebSocket("ws://localhost:8080/ws");
};
ws.onclose = function() { ws.onopen = function () {
document.getElementById("timer").innerText = "Connection closed."; document.getElementById("timer").innerText = "Connected to server.";
}; };
</script>
// Handle incoming messages and update the timer display
ws.onmessage = function (event) {
var data = JSON.parse(event.data);
var mode = data.mode;
var session = data.session;
var maxSession = data.max_session;
var timeLeft = data.time_left;
document.getElementById("timer").innerText =
mode + " Session " + session + "/" + maxSession + ": " + formatTime(timeLeft);
};
ws.onclose = function () {
document.getElementById("timer").innerText = "Connection closed.";
};
// Format time in MM:SS
function formatTime(seconds) {
var minutes = Math.floor(seconds / 60);
var remainingSeconds = seconds % 60;
return minutes.toString().padStart(2, '0') + ":" + remainingSeconds.toString().padStart(2, '0');
}
// Send start command to the server
document.getElementById("startButton").addEventListener("click", function () {
ws.send(JSON.stringify({command: "start"}));
});
// Send stop command to the server
document.getElementById("stopButton").addEventListener("click", function () {
ws.send(JSON.stringify({command: "stop"}));
});
</script>
</body> </body>
</html> </html>