Compare commits
2 commits
d0b1260f62
...
bdfd5c3b84
Author | SHA1 | Date | |
---|---|---|---|
bdfd5c3b84 | |||
94b6786c7c |
5 changed files with 17 additions and 16 deletions
|
@ -42,7 +42,7 @@ func RunPomodoro() {
|
||||||
|
|
||||||
// Work
|
// Work
|
||||||
shared.State.Mode = "Work"
|
shared.State.Mode = "Work"
|
||||||
go timer.Start(pomodoroConfig.Work)
|
timer.StartAsync(pomodoroConfig.Work)
|
||||||
if !waitForTimer(timer) {
|
if !waitForTimer(timer) {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
@ -50,13 +50,13 @@ func RunPomodoro() {
|
||||||
// Breaks
|
// Breaks
|
||||||
if session < pomodoroConfig.Sessions {
|
if session < pomodoroConfig.Sessions {
|
||||||
shared.State.Mode = "ShortBreak"
|
shared.State.Mode = "ShortBreak"
|
||||||
go timer.Start(pomodoroConfig.ShortBreak)
|
timer.StartAsync(pomodoroConfig.ShortBreak)
|
||||||
if !waitForTimer(timer) {
|
if !waitForTimer(timer) {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
} else { // last phase, prepare for finish
|
} else { // last phase, prepare for finish
|
||||||
shared.State.Mode = "LongBreak"
|
shared.State.Mode = "LongBreak"
|
||||||
go timer.Start(pomodoroConfig.LongBreak)
|
timer.StartAsync(pomodoroConfig.LongBreak)
|
||||||
if !waitForTimer(timer) {
|
if !waitForTimer(timer) {
|
||||||
break
|
break
|
||||||
}
|
}
|
||||||
|
|
|
@ -24,7 +24,12 @@ func (t Timer) Init() Timer {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// Start the timer
|
// Start the timer (goroutine)
|
||||||
|
func (t *Timer) StartAsync(duration int) {
|
||||||
|
go t.Start(duration)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Start the timer (blocking)
|
||||||
func (t *Timer) Start(duration int) {
|
func (t *Timer) Start(duration int) {
|
||||||
tick := time.NewTicker(time.Second)
|
tick := time.NewTicker(time.Second)
|
||||||
for timeLeft := duration; timeLeft > 0; {
|
for timeLeft := duration; timeLeft > 0; {
|
||||||
|
|
|
@ -3,7 +3,6 @@ package websocket
|
||||||
import (
|
import (
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gorilla/websocket"
|
|
||||||
|
|
||||||
"git.smsvc.net/pomodoro/GoTomato/internal/pomodoro"
|
"git.smsvc.net/pomodoro/GoTomato/internal/pomodoro"
|
||||||
"git.smsvc.net/pomodoro/GoTomato/internal/shared"
|
"git.smsvc.net/pomodoro/GoTomato/internal/shared"
|
||||||
|
@ -11,7 +10,8 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
// Listens for commands from a client and handles them
|
// Listens for commands from a client and handles them
|
||||||
func handleClientCommands(ws *websocket.Conn) {
|
func handleClientCommands(c models.Client) {
|
||||||
|
ws := c.Conn
|
||||||
for {
|
for {
|
||||||
var clientCommand models.ClientCommand
|
var clientCommand models.ClientCommand
|
||||||
|
|
||||||
|
|
|
@ -31,12 +31,13 @@ func HandleConnection(w http.ResponseWriter, r *http.Request) {
|
||||||
log.Info("Client connected", "host", ws.NetConn().RemoteAddr(), "clients", len(Clients)+1)
|
log.Info("Client connected", "host", ws.NetConn().RemoteAddr(), "clients", len(Clients)+1)
|
||||||
|
|
||||||
// Register the new client
|
// Register the new client
|
||||||
mu.Lock()
|
client := models.Client{
|
||||||
Clients[ws] = &models.Client{
|
Conn: ws,
|
||||||
Conn: ws, // Store the WebSocket connection
|
|
||||||
}
|
}
|
||||||
|
mu.Lock()
|
||||||
|
Clients[ws] = &client
|
||||||
mu.Unlock()
|
mu.Unlock()
|
||||||
|
|
||||||
// Listen for commands from the connected client
|
// Listen for commands from the connected client
|
||||||
handleClientCommands(ws)
|
handleClientCommands(client)
|
||||||
}
|
}
|
||||||
|
|
|
@ -3,7 +3,6 @@ package models
|
||||||
import (
|
import (
|
||||||
"github.com/charmbracelet/log"
|
"github.com/charmbracelet/log"
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
"sync"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Represents a command from the client (start/stop)
|
// Represents a command from the client (start/stop)
|
||||||
|
@ -15,16 +14,12 @@ type ClientCommand struct {
|
||||||
|
|
||||||
// Represents a single client
|
// Represents a single client
|
||||||
type Client struct {
|
type Client struct {
|
||||||
Conn *websocket.Conn // Websocket connection of the client
|
Conn *websocket.Conn
|
||||||
Mutex sync.Mutex // Mutex used to lock
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// Sends a message to the websocket.
|
// Sends a message to the websocket.
|
||||||
// Automatically locks and unlocks the client mutex, to ensure that only one goroutine can write at a time.
|
// Automatically locks and unlocks the client mutex, to ensure that only one goroutine can write at a time.
|
||||||
func (c *Client) SendMessage(messageType int, data []byte) error {
|
func (c *Client) SendMessage(messageType int, data []byte) error {
|
||||||
c.Mutex.Lock()
|
|
||||||
defer c.Mutex.Unlock()
|
|
||||||
|
|
||||||
err := c.Conn.WriteMessage(messageType, data)
|
err := c.Conn.WriteMessage(messageType, data)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error("Error writing to WebSocket:", "msg", err)
|
log.Error("Error writing to WebSocket:", "msg", err)
|
||||||
|
|
Loading…
Reference in a new issue