ChronoTomato/internal/websocket/connect.go

package websocket

import (
	"github.com/charmbracelet/log"
	"github.com/gorilla/websocket"
	"time"

	ChronoTomato "git.smsvc.net/pomodoro/ChronoTomato/pkg/models"
)

type Client ChronoTomato.GoTomatoClient // New websocket client

// Connects to websocket
func Connect(url string) Client {
	log.Info("Connected 󰖟 ", "host", url)

	conn, _, err := websocket.DefaultDialer.Dial(url, nil)
	if err != nil {
		log.Fatal("Dial error!", "reason", err)
	}

	return Client{Conn: conn}
}

// Disconnects from websocket
func (c Client) Disconnect() {
	select {
	case <-Done:
		// session closed by remote
		return
	default:
		// Cleanly close the connection by sending a close message and then
		// waiting (with timeout) for the server to close the connection.
		c.Conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, ""))
		select {
		case <-Done:
		case <-time.After(time.Second):
		}
		return
	}
}