Fix direct H.264 issues and optimize UI

feat: support custom mouse wheel speed
fix: prevent direct H.264 stream buffer overflow and replay
style: optimize virtual keyboard layout
style: optimize menu bar and settings styling

server/proto/vm.go

@@ -87,6 +87,10 @@ type GetOLEDRsp struct {
 	Sleep int  `json:"sleep"`
 }
 
+type GetGetHdmiStateRsp struct {
+	Enabled bool `json:"enabled"`
+}
+
 type GetSSHStateRsp struct {
 	Enabled bool `json:"enabled"`
 }

server/router/vm.go

@@ -35,8 +35,9 @@ func vmRouter(r *gin.Engine) {
 	api.GET("/vm/oled", service.GetOLED)  // get OLED configuration
 	api.POST("/vm/oled", service.SetOLED) // set OLED configuration
 
+	// Only supported by PCIe version
 	api.GET("/vm/hdmi", service.GetHdmiState)         // get HDMI state
-	api.POST("/vm/hdmi/reset", service.ResetHdmi)     // reset hdmi (pcie only)
+	api.POST("/vm/hdmi/reset", service.ResetHdmi)     // reset hdmi
 	api.POST("/vm/hdmi/enable", service.EnableHdmi)   // enable hdmi
 	api.POST("/vm/hdmi/disable", service.DisableHdmi) // disable hdmi
 

server/service/stream/direct/h264.go

@@ -21,9 +21,10 @@ type Frame struct {
 }
 
 var (
-	mutex    = sync.Mutex{}
-	wsMap    = make(map[*websocket.Conn]bool)
-	upgrader = websocket.Upgrader{
+	mutex     = sync.Mutex{}
+	wsMap     = make(map[*websocket.Conn]bool)
+	isSending = false
+	upgrader  = websocket.Upgrader{
 		CheckOrigin: func(r *http.Request) bool {
 			return true
 		},
@@ -47,7 +48,7 @@ func Connect(c *gin.Context) {
 
 	mutex.Lock()
 	wsMap[ws] = true
-	if len(wsMap) == 1 {
+	if len(wsMap) == 1 && !isSending {
 		go send()
 	}
 	mutex.Unlock()
@@ -62,6 +63,7 @@ func Connect(c *gin.Context) {
 }
 
 func send() {
+	isSending = true
 	screen := common.GetScreen()
 	common.CheckScreen()
 
@@ -76,6 +78,7 @@ func send() {
 
 	for range ticker.C {
 		if len(wsMap) == 0 {
+			isSending = false
 			return
 		}
 

server/service/vm/hdmi.go

@@ -1,15 +1,15 @@
 package vm
 
 import (
+	"time"
+
 	"NanoKVM-Server/common"
 	"NanoKVM-Server/proto"
-	"time"
 
 	"github.com/gin-gonic/gin"
 	log "github.com/sirupsen/logrus"
 )
 
-// ideally read state from the device
 var hdmiEnabled = true
 
 func (s *Service) ResetHdmi(c *gin.Context) {
@@ -53,15 +53,9 @@ func (s *Service) DisableHdmi(c *gin.Context) {
 func (s *Service) GetHdmiState(c *gin.Context) {
 	var rsp proto.Response
 
-	if hdmiEnabled {
-		rsp.OkRspWithData(c, &map[string]string{
-			"state": "enabled",
-		})
-	} else {
-		rsp.OkRspWithData(c, &map[string]string{
-			"state": "disabled",
-		})
-	}
+	rsp.OkRspWithData(c, &proto.GetGetHdmiStateRsp{
+		Enabled: hdmiEnabled,
+	})
 
 	log.Debug("get hdmi state")
 }