7 "github.com/Sirupsen/logrus"
8 "github.com/gin-gonic/gin"
9 "github.com/googollee/go-socket.io"
10 "github.com/iotbzh/xds-agent/lib/apiv1"
11 "github.com/iotbzh/xds-agent/lib/session"
12 "github.com/iotbzh/xds-agent/lib/xdsconfig"
16 type ServerService struct {
19 sIOServer *socketio.Server
20 webApp *gin.RouterGroup
22 sessions *session.Sessions
24 stop chan struct{} // signals intentional stop
27 const indexFilename = "index.html"
28 const cookieMaxAge = "3600"
30 // NewServer creates an instance of ServerService
31 func NewServer(conf *xdsconfig.Config, log *logrus.Logger) *ServerService {
33 // Setup logging for gin router
34 if log.Level == logrus.DebugLevel {
35 gin.SetMode(gin.DebugMode)
37 gin.SetMode(gin.ReleaseMode)
41 // - try to bind gin DefaultWriter & DefaultErrorWriter to logrus logger
42 // - try to fix pb about isTerminal=false when out is in VSC Debug Console
43 //gin.DefaultWriter = ??
44 //gin.DefaultErrorWriter = ??
49 svr := &ServerService{
57 stop: make(chan struct{}),
63 // Serve starts a new instance of the Web Server
64 func (s *ServerService) Serve() error {
68 s.router.Use(gin.Logger())
69 s.router.Use(gin.Recovery())
70 s.router.Use(s.middlewareXDSDetails())
71 s.router.Use(s.middlewareCORS())
74 s.sessions = session.NewClientSessions(s.router, s.log, cookieMaxAge)
77 s.api = apiv1.New(s.sessions, s.cfg, s.log, s.router)
80 s.sIOServer, err = socketio.NewServer(nil)
85 s.router.GET("/socket.io/", s.socketHandler)
86 s.router.POST("/socket.io/", s.socketHandler)
87 /* TODO: do we want to support ws://... ?
88 s.router.Handle("WS", "/socket.io/", s.socketHandler)
89 s.router.Handle("WSS", "/socket.io/", s.socketHandler)
92 // Serve in the background
93 serveError := make(chan error, 1)
95 fmt.Printf("Web Server running on localhost:%s ...\n", s.cfg.HTTPPort)
96 serveError <- http.ListenAndServe(":"+s.cfg.HTTPPort, s.router)
99 fmt.Printf("XDS agent running...\n")
101 // Wait for stop, restart or error signals
104 // Shutting down permanently
106 s.log.Infoln("shutting down (stop)")
107 case err = <-serveError:
108 // Error due to listen/serve failure
116 func (s *ServerService) Stop() {
120 // Add details in Header
121 func (s *ServerService) middlewareXDSDetails() gin.HandlerFunc {
122 return func(c *gin.Context) {
123 c.Header("XDS-Agent-Version", s.cfg.Version)
124 c.Header("XDS-API-Version", s.cfg.APIVersion)
130 func (s *ServerService) middlewareCORS() gin.HandlerFunc {
131 return func(c *gin.Context) {
133 if c.Request.Method == "OPTIONS" {
134 c.Header("Access-Control-Allow-Origin", "*")
135 c.Header("Access-Control-Allow-Headers", "Content-Type")
136 c.Header("Access-Control-Allow-Methods", "POST, DELETE, GET, PUT")
137 c.Header("Content-Type", "application/json")
138 c.Header("Access-Control-Max-Age", cookieMaxAge)
139 c.AbortWithStatus(204)
147 // socketHandler is the handler for the "main" websocket connection
148 func (s *ServerService) socketHandler(c *gin.Context) {
150 // Retrieve user session
151 sess := s.sessions.Get(c)
153 c.JSON(500, gin.H{"error": "Cannot retrieve session"})
157 s.sIOServer.On("connection", func(so socketio.Socket) {
158 s.log.Debugf("WS Connected (SID=%v)", so.Id())
159 s.sessions.UpdateIOSocket(sess.ID, &so)
161 so.On("disconnection", func() {
162 s.log.Debugf("WS disconnected (SID=%v)", so.Id())
163 s.sessions.UpdateIOSocket(sess.ID, nil)
167 s.sIOServer.On("error", func(so socketio.Socket, err error) {
168 s.log.Errorf("WS SID=%v Error : %v", so.Id(), err.Error())
171 s.sIOServer.ServeHTTP(c.Writer, c.Request)