mirror of
https://github.com/twofas/2fas-server.git
synced 2024-12-12 04:00:15 +01:00
46 lines
1.0 KiB
Go
46 lines
1.0 KiB
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.RequestIdMiddleware())
|
|
router.Use(http.CorrelationIdMiddleware())
|
|
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)
|
|
}
|
|
}
|