LastMUD/internal/server/server.go

package server

import (
	"context"
	"net"
	"sync"
	"time"

	"code.haedhutner.dev/mvv/LastMUD/internal/game"
	"code.haedhutner.dev/mvv/LastMUD/internal/logging"
	"github.com/google/uuid"
)

type Server struct {
	ctx context.Context
	wg  *sync.WaitGroup

	listener *net.TCPListener

	connections map[uuid.UUID]*Connection

	lastmudgame *game.Game
}

func CreateServer(ctx context.Context, wg *sync.WaitGroup, port string) (srv *Server, err error) {

	logging.Info(" _              _   __  __ _   _ ____")
	logging.Info("| |    __ _ ___| |_|  \\/  | | | |  _ \\")
	logging.Info("| |   / _` / __| __| |\\/| | | | | | | |")
	logging.Info("| |__| (_| \\__ \\ |_| |  | | |_| | |_| |")
	logging.Info("|_____\\__,_|___/\\__|_|  |_|\\___/|____/")
	logging.Info("")

	addr, err := net.ResolveTCPAddr("tcp", port)

	if err != nil {
		logging.Error(err)
		return nil, err
	}

	ln, err := net.ListenTCP("tcp", addr)

	if err != nil {
		logging.Error(err)
		return nil, err
	}

	logging.Info("Starting server, listening on port ", port)

	srv = &Server{
		ctx:         ctx,
		wg:          wg,
		listener:    ln,
		connections: map[uuid.UUID]*Connection{},
	}

	srv.lastmudgame = game.CreateGame(ctx, srv.wg)

	srv.wg.Add(2)
	go srv.listen()
	go srv.consumeGameOutput()

	return
}

func (srv *Server) game() *game.Game {
	return srv.lastmudgame
}

func (srv *Server) listen() {
	defer srv.wg.Done()
	defer srv.shutdown()

	for {
		srv.listener.SetDeadline(time.Now().Add(1 * time.Second))

		if srv.shouldStop() {
			break
		}

		conn, err := srv.listener.Accept()

		if err != nil {
			continue
		}

		tcpConn, ok := conn.(*net.TCPConn)

		if !ok {
			logging.Warn("Attempted invalid connection from", conn.RemoteAddr())
			continue
		}

		c := CreateConnection(srv, tcpConn, srv.ctx, srv.wg)

		srv.connections[c.Id()] = c
	}
}

func (srv *Server) consumeGameOutput() {
	defer srv.wg.Done()

	for {
		if srv.shouldStop() {
			break
		}

		output := srv.lastmudgame.ConsumeNextOutput()

		if output == nil {
			continue
		}

		conn, ok := srv.connections[output.Id()]

		if ok && output.Contents() != nil {
			conn.Write(output.Contents())
		}

		if output.ShouldCloseConnection() {
			conn.CommandClose()
			delete(srv.connections, output.Id())
		}
	}
}

func (srv *Server) shutdown() {
	logging.Info("Stopping Server...")

	err := srv.listener.Close()

	if err != nil {
		logging.Error(err)
	}
}

func (srv *Server) shouldStop() bool {
	select {
	case <-srv.ctx.Done():
		return true
	default:
		return false
	}
}