2fas-server/internal/websocket/app.go

package websocket

import (
	"github.com/gin-gonic/gin"

	"github.com/twofas/2fas-server/internal/common/http"
	"github.com/twofas/2fas-server/internal/common/recovery"
	"github.com/twofas/2fas-server/internal/websocket/browser_extension"
	"github.com/twofas/2fas-server/internal/websocket/common"
)

type Server struct {
	router            *gin.Engine
	addr              string
	connectionHandler *common.ConnectionHandler
}

func NewServer(addr string) *Server {
	router := gin.New()

	router.Use(recovery.RecoveryMiddleware())
	router.Use(http.LoggingMiddleware())
	router.Use(http.RequestJsonLogger())

	connectionHandler := common.NewConnectionHandler()

	routesHandler := browser_extension.NewRoutesHandler(connectionHandler)

	browser_extension.GinRoutesHandler(routesHandler, router)

	return &Server{
		router:            router,
		addr:              addr,
		connectionHandler: connectionHandler,
	}
}

func (s *Server) RunWebsocketServer() {
	err := s.router.Run(s.addr)

	if err != nil {
		panic(err)
	}
}