GoTomato/cmd/server/main.go

package server

import (
	"flag"
	"fmt"
	"github.com/charmbracelet/log"
	"github.com/gorilla/handlers"
	"net/http"
	"os"

	"git.smsvc.net/pomodoro/GoTomato/internal/metadata"
	"git.smsvc.net/pomodoro/GoTomato/internal/shared"
	"git.smsvc.net/pomodoro/GoTomato/internal/websocket"
	"git.smsvc.net/pomodoro/GoTomato/pkg/models"
)

var (
	// define CLI flags
	listenAddress   = flag.String("listenAddress", shared.DefaultServerConfig.ListenAddress, "IP address to listen on")
	listenPort      = flag.Int("listenPort", shared.DefaultServerConfig.ListenPort, "Port to listen on")
	password        = flag.String("password", "", "Control password for pomodoro session (optional)")
	showVersionFlag = flag.Bool("version", false, "Output version")
)

// Start the pomodoro server
func Start() {
	flag.Parse()

	// show server and protocl version and exit
	if *showVersionFlag {
		fmt.Println("Server-Version:", metadata.GoTomatoVersion)
		fmt.Println("Protocol-Version:", metadata.ProtocolVersion)
		os.Exit(0)
	}

	// set server config
	serverConfig := models.ServerConfig{
		ListenAddress: *listenAddress,
		ListenPort:    *listenPort,
	}
	shared.PomodoroPassword = *password

	// define listen address for websocket
	listen := fmt.Sprintf("%s:%d", serverConfig.ListenAddress, serverConfig.ListenPort)

	// start connection handler and broadcast
	r := http.NewServeMux()
	r.HandleFunc("/ws", websocket.HandleConnection)
	go websocket.SendPermanentBroadCastMessage()

	log.Info("GoTomato started", "version", metadata.GoTomatoVersion)
	log.Info("Websocket listening", "address", listen)

	// start the listener
	err := http.ListenAndServe(listen, handlers.ProxyHeaders(r))
	if err != nil {
		log.Fatal("Error starting server:", "msg", err)
	}
}