whishper/backend/api/server.go

package api

import (
	"os"

	"github.com/goccy/go-json"
	"github.com/gofiber/contrib/websocket"
	"github.com/gofiber/fiber/v2"
	"github.com/gofiber/fiber/v2/middleware/cors"
	"github.com/rs/zerolog/log"

	"codeberg.org/pluja/whishper/database"
	"codeberg.org/pluja/whishper/models"
)

type Server struct {
	ListenAddr         string
	Router             *fiber.App
	Db                 database.Db
	NewTranscriptionCh chan bool
	clients            []*websocket.Conn
}

func NewServer(listenAddr string, db database.Db) *Server {
	return &Server{
		ListenAddr: listenAddr,
		Router: fiber.New(fiber.Config{
			JSONEncoder:  json.Marshal,
			JSONDecoder:  json.Unmarshal,
			BodyLimit:    100000 * 1024 * 1024, // Increase body limit to 100000MB (100GB)
			ServerHeader: "Fiber",              // Optional, for easier debugging
		}),
		Db:                 db,
		clients:            make([]*websocket.Conn, 0),
		NewTranscriptionCh: make(chan bool, 100),
	}
}

func (s *Server) Run() {
	s.SetupWebsocket()
	s.SetupMiddleware()
	s.RegisterRoutes()
	s.Router.Listen(s.ListenAddr)
}

func (s *Server) SetupWebsocket() {
	s.Router.Get("/ws/transcriptions", websocket.New(func(c *websocket.Conn) {

		// Add this connection to the slice of clients
		s.clients = append(s.clients, c)

		for {
			_, msg, err := c.ReadMessage()
			if err != nil {
				// Check for normal close error (1000) or going away error (1001)
				if err.Error() != "websocket: close 1000 (normal)" &&
					err.Error() != "websocket: close 1001 (going away)" {
					log.Debug().Err(err).Msgf("Error reading message")
				}
				// Remove the client from the slice if it has disconnected
				s.clients = removeWsClient(s.clients, c)
				return
			}
			s.handleWebsocketMessage(c, msg)
		}
	}))
}

func (s *Server) BroadcastTranscription(t *models.Transcription) {
	// Convert the transcription to JSON.
	json, err := json.Marshal(&t)
	if err != nil {
		log.Error().Err(err).Msg("Error marshalling transcription to JSON:")
		return
	}
	for _, client := range s.clients {
		if err := client.WriteMessage(websocket.TextMessage, json); err != nil {
			log.Error().Err(err).Msg("Error broadcasting message:")
		}
	}
}

func (s *Server) SetupMiddleware() {
	s.Router.Use(cors.New())
}

func (s *Server) RegisterRoutes() {
	// Static routes
	s.Router.Static("/api/video", os.Getenv("UPLOAD_DIR"))

	// Register HTTP route for getting initial state.
	s.Router.Get("/api/transcriptions", func(c *fiber.Ctx) error {
		err := s.handleGetAllTranscriptions(c)
		if err != nil {
			log.Error().Err(err).Msg("Error handling POST /api/transcriptions")
		}
		return err
	})

	// Register HTTP route for getting initial state.
	s.Router.Get("/api/transcriptions/:id", func(c *fiber.Ctx) error {
		log.Debug().Msgf("GET /api/transcriptions/%v", c.Params("id"))
		err := s.handleGetTranscriptionById(c)
		if err != nil {
			log.Error().Err(err).Msg("Error handling GET /api/transcriptions/:id")
		}
		return err
	})

	// Register HTTP route for getting initial state.
	s.Router.Get("/api/translate/:id/:target", func(c *fiber.Ctx) error {
		log.Debug().Msgf("GET /api/translate/%v/%v", c.Params("id"), c.Params("target"))
		err := s.handleTranslate(c)
		if err != nil {
			log.Error().Err(err).Msg("Error handling GET /api/translate/:id/:source")
		}
		return err
	})

	// Register HTTP route for receiving the form data and creating new transcription job.
	s.Router.Post("/api/transcriptions", func(c *fiber.Ctx) error {
		log.Debug().Msg("POST /api/transcriptions")
		err := s.handlePostTranscription(c)
		if err != nil {
			log.Error().Err(err).Msg("Error handling POST /api/transcriptions")
		}
		return err
	})

	s.Router.Patch("/api/transcriptions", func(c *fiber.Ctx) error {
		//log.Debug().Msgf("PATCH /api/transcriptions/%v", c.Params("id"))
		err := s.handlePatchTranscription(c)
		if err != nil {
			log.Error().Err(err).Msg("Error handling PATCH /api/transcriptions")
		}
		return err
	})

	// Register HTTP route for receiving the form data and creating new transcription job.
	s.Router.Delete("/api/transcriptions/:id", func(c *fiber.Ctx) error {
		log.Debug().Msgf("DELETE /api/transcriptions/%v", c.Params("id"))
		err := s.handleDeleteTranscription(c)
		if err != nil {
			log.Error().Err(err).Msg("Error handling DELETE /api/transcriptions")
		}
		return err
	})
}

// Helper function to remove a WebSocket connection from the slice
func removeWsClient(s []*websocket.Conn, r *websocket.Conn) []*websocket.Conn {
	for i, v := range s {
		if v == r {
			return append(s[:i], s[i+1:]...)
		}
	}
	return s
}