mirror of https://github.com/jetkvm/kvm.git
101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
package kvm
|
|
|
|
import (
|
|
"time"
|
|
|
|
"github.com/jetkvm/kvm/internal/usbgadget"
|
|
)
|
|
|
|
var gadget *usbgadget.UsbGadget
|
|
|
|
// initUsbGadget initializes the USB gadget.
|
|
// call it only after the config is loaded.
|
|
func initUsbGadget() {
|
|
gadget = usbgadget.NewUsbGadget(
|
|
"jetkvm",
|
|
config.UsbDevices,
|
|
config.UsbConfig,
|
|
usbLogger,
|
|
)
|
|
|
|
go func() {
|
|
for {
|
|
checkUSBState()
|
|
time.Sleep(500 * time.Millisecond)
|
|
}
|
|
}()
|
|
|
|
gadget.SetOnKeyboardStateChange(func(state usbgadget.KeyboardState) {
|
|
if currentSession != nil {
|
|
writeJSONRPCEvent("keyboardLedState", state, currentSession)
|
|
}
|
|
})
|
|
|
|
gadget.SetOnKeysDownChange(func(state usbgadget.KeysDownState) {
|
|
if currentSession != nil {
|
|
writeJSONRPCEvent("keysDownState", state, currentSession)
|
|
}
|
|
})
|
|
|
|
// open the keyboard hid file to listen for keyboard events
|
|
if err := gadget.OpenKeyboardHidFile(); err != nil {
|
|
usbLogger.Error().Err(err).Msg("failed to open keyboard hid file")
|
|
}
|
|
}
|
|
|
|
func rpcKeyboardReport(modifier byte, keys []byte) error {
|
|
return gadget.KeyboardReport(modifier, keys)
|
|
}
|
|
|
|
func rpcKeypressReport(key byte, press bool) (usbgadget.KeysDownState, error) {
|
|
return gadget.KeypressReport(key, press)
|
|
}
|
|
|
|
func rpcAbsMouseReport(x, y int, buttons uint8) error {
|
|
return gadget.AbsMouseReport(x, y, buttons)
|
|
}
|
|
|
|
func rpcRelMouseReport(dx, dy int8, buttons uint8) error {
|
|
return gadget.RelMouseReport(dx, dy, buttons)
|
|
}
|
|
|
|
func rpcWheelReport(wheelY int8) error {
|
|
return gadget.AbsMouseWheelReport(wheelY)
|
|
}
|
|
|
|
func rpcGetKeyboardLedState() (state usbgadget.KeyboardState) {
|
|
return gadget.GetKeyboardState()
|
|
}
|
|
|
|
func rpcGetKeysDownState() (state usbgadget.KeysDownState) {
|
|
return gadget.GetKeysDownState()
|
|
}
|
|
|
|
var usbState = "unknown"
|
|
|
|
func rpcGetUSBState() (state string) {
|
|
return gadget.GetUsbState()
|
|
}
|
|
|
|
func triggerUSBStateUpdate() {
|
|
go func() {
|
|
if currentSession == nil {
|
|
usbLogger.Info().Msg("No active RPC session, skipping USB state update")
|
|
return
|
|
}
|
|
writeJSONRPCEvent("usbState", usbState, currentSession)
|
|
}()
|
|
}
|
|
|
|
func checkUSBState() {
|
|
newState := gadget.GetUsbState()
|
|
if newState == usbState {
|
|
return
|
|
}
|
|
usbLogger.Info().Str("from", usbState).Str("to", newState).Msg("USB state changed")
|
|
usbState = newState
|
|
|
|
requestDisplayUpdate(true)
|
|
triggerUSBStateUpdate()
|
|
}
|