diff --git a/internal/logging/logger.go b/internal/logging/logger.go
index f37c2dc..39156ec 100644
--- a/internal/logging/logger.go
+++ b/internal/logging/logger.go
@@ -35,7 +35,12 @@ func (w *logOutput) Write(p []byte) (n int, err error) {
defer w.mu.Unlock()
// TODO: write to file or syslog
-
+ if sseServer != nil {
+ // use a goroutine to avoid blocking the Write method
+ go func() {
+ sseServer.Message <- string(p)
+ }()
+ }
return len(p), nil
}
diff --git a/internal/logging/sse.go b/internal/logging/sse.go
new file mode 100644
index 0000000..a466432
--- /dev/null
+++ b/internal/logging/sse.go
@@ -0,0 +1,138 @@
+package logging
+
+import (
+ "embed"
+ "io"
+ "net/http"
+
+ "github.com/gin-gonic/gin"
+ "github.com/rs/zerolog"
+)
+
+//go:embed sse.html
+var sseHTML embed.FS
+
+type sseEvent struct {
+ Message chan string
+ NewClients chan chan string
+ ClosedClients chan chan string
+ TotalClients map[chan string]bool
+}
+
+// New event messages are broadcast to all registered client connection channels
+type sseClientChan chan string
+
+var (
+ sseServer *sseEvent
+ sseLogger *zerolog.Logger
+)
+
+func init() {
+ sseServer = newSseServer()
+ sseLogger = GetSubsystemLogger("sse")
+}
+
+// Initialize event and Start procnteessing requests
+func newSseServer() (event *sseEvent) {
+ event = &sseEvent{
+ Message: make(chan string),
+ NewClients: make(chan chan string),
+ ClosedClients: make(chan chan string),
+ TotalClients: make(map[chan string]bool),
+ }
+
+ go event.listen()
+
+ return
+}
+
+// It Listens all incoming requests from clients.
+// Handles addition and removal of clients and broadcast messages to clients.
+func (stream *sseEvent) listen() {
+ for {
+ select {
+ // Add new available client
+ case client := <-stream.NewClients:
+ stream.TotalClients[client] = true
+ sseLogger.Info().
+ Int("total_clients", len(stream.TotalClients)).
+ Msg("new client connected")
+
+ // Remove closed client
+ case client := <-stream.ClosedClients:
+ delete(stream.TotalClients, client)
+ close(client)
+ sseLogger.Info().Int("total_clients", len(stream.TotalClients)).Msg("client disconnected")
+
+ // Broadcast message to client
+ case eventMsg := <-stream.Message:
+ for clientMessageChan := range stream.TotalClients {
+ select {
+ case clientMessageChan <- eventMsg:
+ // Message sent successfully
+ default:
+ // Failed to send, dropping message
+ }
+ }
+ }
+ }
+}
+
+func (stream *sseEvent) serveHTTP() gin.HandlerFunc {
+ return func(c *gin.Context) {
+ clientChan := make(sseClientChan)
+ stream.NewClients <- clientChan
+
+ go func() {
+ <-c.Writer.CloseNotify()
+
+ for range clientChan {
+ }
+
+ stream.ClosedClients <- clientChan
+ }()
+
+ c.Set("clientChan", clientChan)
+ c.Next()
+ }
+}
+
+func sseHeadersMiddleware() gin.HandlerFunc {
+ return func(c *gin.Context) {
+ if c.Request.Method == "GET" && c.NegotiateFormat(gin.MIMEHTML) == gin.MIMEHTML {
+ c.FileFromFS("/sse.html", http.FS(sseHTML))
+ c.Status(http.StatusOK)
+ c.Abort()
+ return
+ }
+
+ c.Writer.Header().Set("Content-Type", "text/event-stream")
+ c.Writer.Header().Set("Cache-Control", "no-cache")
+ c.Writer.Header().Set("Connection", "keep-alive")
+ c.Writer.Header().Set("Transfer-Encoding", "chunked")
+ c.Writer.Header().Set("Access-Control-Allow-Origin", "*")
+ c.Next()
+ }
+}
+
+func AttachSSEHandler(router *gin.RouterGroup) {
+
+ router.StaticFS("/log-stream", http.FS(sseHTML))
+ router.GET("/log-stream", sseHeadersMiddleware(), sseServer.serveHTTP(), func(c *gin.Context) {
+ v, ok := c.Get("clientChan")
+ if !ok {
+ return
+ }
+ clientChan, ok := v.(sseClientChan)
+ if !ok {
+ return
+ }
+ c.Stream(func(w io.Writer) bool {
+ if msg, ok := <-clientChan; ok {
+ c.SSEvent("message", msg)
+ return true
+ }
+ return false
+ })
+ })
+}
diff --git a/internal/logging/sse.html b/internal/logging/sse.html
new file mode 100644
index 0000000..192b464
--- /dev/null
+++ b/internal/logging/sse.html
@@ -0,0 +1,319 @@
+
+
+
+
+
+ Server Sent Event
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/ui/public/sse.html b/ui/public/sse.html
new file mode 120000
index 0000000..0a8b4f3
--- /dev/null
+++ b/ui/public/sse.html
@@ -0,0 +1 @@
+../../internal/logging/sse.html
\ No newline at end of file
diff --git a/ui/vite.config.ts b/ui/vite.config.ts
index f6aae50..2c153e7 100644
--- a/ui/vite.config.ts
+++ b/ui/vite.config.ts
@@ -26,6 +26,7 @@ export default defineConfig(({ mode, command }) => {
"/auth": JETKVM_PROXY_URL,
"/storage": JETKVM_PROXY_URL,
"/cloud": JETKVM_PROXY_URL,
+ "/developer": JETKVM_PROXY_URL,
}
: undefined,
},
diff --git a/web.go b/web.go
index 31d3133..766eaf5 100644
--- a/web.go
+++ b/web.go
@@ -19,6 +19,7 @@ import (
gin_logger "github.com/gin-contrib/logger"
"github.com/gin-gonic/gin"
"github.com/google/uuid"
+ "github.com/jetkvm/kvm/internal/logging"
"github.com/pion/webrtc/v4"
"github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promhttp"
@@ -121,6 +122,8 @@ func setupRouter() *gin.Engine {
developerModeRouter.GET("/pprof/heap", gin.WrapH(pprof.Handler("heap")))
developerModeRouter.GET("/pprof/mutex", gin.WrapH(pprof.Handler("mutex")))
developerModeRouter.GET("/pprof/threadcreate", gin.WrapH(pprof.Handler("threadcreate")))
+
+ logging.AttachSSEHandler(developerModeRouter)
}
// Protected routes (allows both password and noPassword modes)